1 files changed, 89 insertions, 0 deletions
diff --git a/python/mozbuild/mozbuild/action/exe_7z_archive.py b/python/mozbuild/mozbuild/action/exe_7z_archive.py
new file mode 100644
index 0000000000..b0d35be2bf
--- /dev/null
+++ b/python/mozbuild/mozbuild/action/exe_7z_archive.py
@@ -0,0 +1,89 @@
+# 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, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+import os
+import shutil
+import subprocess
+import sys
+import tempfile
+
+import buildconfig
+import mozpack.path as mozpath
+
+from mozbuild.base import BuildEnvironmentNotFoundException
+
+
+def archive_exe(pkg_dir, tagfile, sfx_package, package, use_upx):
+    tmpdir = tempfile.mkdtemp(prefix="tmp")
+    try:
+        if pkg_dir:
+            shutil.move(pkg_dir, "core")
+
+        if use_upx:
+            final_sfx = mozpath.join(tmpdir, "7zSD.sfx")
+            upx = buildconfig.substs.get("UPX", "upx")
+            wine = buildconfig.substs.get("WINE")
+            if wine and upx.lower().endswith(".exe"):
+                cmd = [wine, upx]
+            else:
+                cmd = [upx]
+            subprocess.check_call(
+                cmd
+                + [
+                    "--best",
+                    "-o",
+                    final_sfx,
+                    sfx_package,
+                ]
+            )
+        else:
+            final_sfx = sfx_package
+
+        try:
+            sevenz = buildconfig.config.substs["7Z"]
+        except BuildEnvironmentNotFoundException:
+            # configure hasn't been run, just use the default
+            sevenz = "7z"
+        subprocess.check_call(
+            [
+                sevenz,
+                "a",
+                "-r",
+                "-t7z",
+                mozpath.join(tmpdir, "app.7z"),
+                "-mx",
+                "-m0=BCJ2",
+                "-m1=LZMA:d25",
+                "-m2=LZMA:d19",
+                "-m3=LZMA:d19",
+                "-mb0:1",
+                "-mb0s1:2",
+                "-mb0s2:3",
+            ]
+        )
+
+        with open(package, "wb") as o:
+            for i in [final_sfx, tagfile, mozpath.join(tmpdir, "app.7z")]:
+                shutil.copyfileobj(open(i, "rb"), o)
+        os.chmod(package, 0o0755)
+    finally:
+        if pkg_dir:
+            shutil.move("core", pkg_dir)
+        shutil.rmtree(tmpdir)
+
+
+def main(args):
+    if len(args) != 4:
+        print(
+            "Usage: exe_7z_archive.py <pkg_dir> <tagfile> <sfx_package> <package> <use_upx>",
+            file=sys.stderr,
+        )
+        return 1
+    else:
+        archive_exe(args[0], args[1], args[2], args[3], args[4])
+        return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main(sys.argv[1:]))