wix: add support for additional wxs files

As with my previous change for an --extra-prebuiild-script, I'm
assuming this is predominantly useful in an enterprise environment
and am only adding this to wix and not also to inno install scripts.

Differential Revision: https://phab.mercurial-scm.org/D6179

contrib/packaging/hgpackaging/wix.py

@@ -12,6 +12,7 b' import pathlib'
 import re
 import subprocess
 import tempfile
+import typing
 import xml.dom.minidom
 
 from .downloads import (
@@ -178,7 +179,8 b' def make_libraries_xml(wix_dir: pathlib.'
 
 def build_installer(source_dir: pathlib.Path, python_exe: pathlib.Path,
                     msi_name='mercurial', version=None, post_build_fn=None,
-                    extra_packages_script=None):
+                    extra_packages_script: typing.Optional[str]=None,
+                    extra_wxs:typing.Optional[typing.Dict[str,str]]=None):
     """Build a WiX MSI installer.
 
     ``source_dir`` is the path to the Mercurial source tree to use.
@@ -194,6 +196,7 b' def build_installer(source_dir: pathlib.'
     into the py2exe binary. It should stage packages into the virtualenv and
     print a null byte followed by a newline-separated list of packages that
     should be included in the exe.
+    ``extra_wxs`` is a dict of {wxs_name: working_dir_for_wxs_build}.
     """
     arch = 'x64' if r'\x64' in os.environ.get('LIB', '') else 'x86'
 
@@ -235,6 +238,9 b' def build_installer(source_dir: pathlib.'
         wxs_source_dir = source_dir / rel_path
         run_candle(wix_path, build_dir, wxs, wxs_source_dir, defines=defines)
 
+    for source, rel_path in sorted((extra_wxs or {}).items()):
+        run_candle(wix_path, build_dir, source, rel_path, defines=defines)
+
     # candle.exe doesn't like when we have an open handle on the file.
     # So use TemporaryDirectory() instead of NamedTemporaryFile().
     with tempfile.TemporaryDirectory() as td:
@@ -269,6 +275,11 b' def build_installer(source_dir: pathlib.'
         assert source.endswith('.wxs')
         args.append(str(build_dir / ('%s.wixobj' % source[:-4])))
 
+    for source, rel_path in sorted((extra_wxs or {}).items()):
+        assert source.endswith('.wxs')
+        source = os.path.basename(source)
+        args.append(str(build_dir / ('%s.wixobj' % source[:-4])))
+
     args.extend([
         str(build_dir / 'library.wixobj'),
         str(build_dir / 'mercurial.wixobj'),
@@ -286,7 +297,8 b' def build_installer(source_dir: pathlib.'
 def build_signed_installer(source_dir: pathlib.Path, python_exe: pathlib.Path,
                            name: str, version=None, subject_name=None,
                            cert_path=None, cert_password=None,
-                           timestamp_url=None, extra_packages_script=None):
+                           timestamp_url=None, extra_packages_script=None,
+                           extra_wxs=None):
     """Build an installer with signed executables."""
 
     post_build_fn = make_post_build_signing_fn(
@@ -299,7 +311,8 b' def build_signed_installer(source_dir: p'
     info = build_installer(source_dir, python_exe=python_exe,
                            msi_name=name.lower(), version=version,
                            post_build_fn=post_build_fn,
-                           extra_packages_script=extra_packages_script)
+                           extra_packages_script=extra_packages_script,
+                           extra_wxs=extra_wxs)
 
     description = '%s %s' % (name, version)
 

contrib/packaging/wix/build.py

@@ -37,6 +37,8 b" if __name__ == '__main__':"
     parser.add_argument('--extra-packages-script',
                         help=('Script to execute to include extra packages in '
                               'py2exe binary.'))
+    parser.add_argument('--extra-wxs',
+                        help='CSV of path_to_wxs_file=working_dir_for_wxs_file')
 
     args = parser.parse_args()
 
@@ -59,6 +61,9 b" if __name__ == '__main__':"
 
     if args.extra_packages_script:
         kwargs['extra_packages_script'] = args.extra_packages_script
+    if args.extra_wxs:
+        kwargs['extra_wxs'] = dict(
+            thing.split("=") for thing in args.extra_wxs.split(','))
 
     if args.sign_sn or args.sign_cert:
         fn = build_signed_installer