Source code
Revision control
Copy as Markdown
Other Tools
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this file,
import os
import shutil
import subprocess
import sys
import tempfile
from xml.dom import minidom
import mozpack.path as mozpath
from mozbuild.dirutils import ensureParentDir
_MSI_ARCH = {
"x86": "x86",
"x86_64": "x64",
}
def update_wsx(wfile, pvalues):
parsed = minidom.parse(wfile)
# construct a dictinary for the pre-processing options
# iterate over that list and add them to the wsx xml doc
for k, v in pvalues.items():
entry = parsed.createProcessingInstruction("define", k + ' = "' + v + '"')
root = parsed.firstChild
parsed.insertBefore(entry, root)
# write out xml to new wfile
new_w_file = wfile + ".new"
with open(new_w_file, "w") as fh:
parsed.writexml(fh)
shutil.move(new_w_file, wfile)
return wfile
def repackage_msi(
topsrcdir, wsx, version, locale, arch, setupexe, candle, light, output
):
if sys.platform != "win32":
raise Exception("repackage msi only works on windows")
if not os.path.isdir(topsrcdir):
raise Exception("%s does not exist." % topsrcdir)
if not os.path.isfile(wsx):
raise Exception("%s does not exist." % wsx)
if version is None:
raise Exception("version name must be provided.")
if locale is None:
raise Exception("locale name must be provided.")
if arch is None or arch not in _MSI_ARCH.keys():
raise Exception(
"arch name must be provided and one of {}.".format(_MSI_ARCH.keys())
)
if not os.path.isfile(setupexe):
raise Exception("%s does not exist." % setupexe)
if candle is not None and not os.path.isfile(candle):
raise Exception("%s does not exist." % candle)
if light is not None and not os.path.isfile(light):
raise Exception("%s does not exist." % light)
embeddedVersion = "0.0.0.0"
# Version string cannot contain 'a' or 'b' when embedding in msi manifest.
if "a" not in version and "b" not in version:
if version.endswith("esr"):
parts = version[:-3].split(".")
else:
parts = version.split(".")
while len(parts) < 4:
parts.append("0")
embeddedVersion = ".".join(parts)
wsx = mozpath.realpath(wsx)
setupexe = mozpath.realpath(setupexe)
output = mozpath.realpath(output)
ensureParentDir(output)
if sys.platform == "win32":
tmpdir = tempfile.mkdtemp()
old_cwd = os.getcwd()
try:
wsx_file = os.path.split(wsx)[1]
shutil.copy(wsx, tmpdir)
temp_wsx_file = os.path.join(tmpdir, wsx_file)
temp_wsx_file = mozpath.realpath(temp_wsx_file)
pre_values = {
"Vendor": "Mozilla",
"BrandFullName": "Mozilla Firefox",
"Version": version,
"AB_CD": locale,
"Architecture": _MSI_ARCH[arch],
"ExeSourcePath": setupexe,
"EmbeddedVersionCode": embeddedVersion,
}
# update wsx file with inputs from
newfile = update_wsx(temp_wsx_file, pre_values)
wix_object_file = os.path.join(tmpdir, "installer.wixobj")
env = os.environ.copy()
if candle is None:
candle = "candle.exe"
cmd = [candle, "-out", wix_object_file, newfile]
subprocess.check_call(cmd, env=env)
wix_installer = wix_object_file.replace(".wixobj", ".msi")
if light is None:
light = "light.exe"
light_cmd = [
light,
"-cultures:neutral",
"-sw1076",
"-sw1079",
"-out",
wix_installer,
wix_object_file,
]
subprocess.check_call(light_cmd, env=env)
os.remove(wix_object_file)
# mv file to output dir
shutil.move(wix_installer, output)
finally:
os.chdir(old_cwd)
shutil.rmtree(tmpdir)