1 files changed, 67 insertions, 0 deletions
diff --git a/python/mozbuild/mozbuild/action/make_dmg.py b/python/mozbuild/mozbuild/action/make_dmg.py
new file mode 100644
index 0000000000..6dc19450fb
--- /dev/null
+++ b/python/mozbuild/mozbuild/action/make_dmg.py
@@ -0,0 +1,67 @@
+# 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 argparse
+import platform
+import sys
+from pathlib import Path
+
+from mozpack import dmg
+
+from mozbuild.bootstrap import bootstrap_toolchain
+from mozbuild.repackaging.application_ini import get_application_ini_value
+
+is_linux = platform.system() == "Linux"
+
+
+def main(args):
+    parser = argparse.ArgumentParser(
+        description="Explode a DMG into its relevant files"
+    )
+
+    parser.add_argument("--dsstore", help="DSStore file from")
+    parser.add_argument("--background", help="Background file from")
+    parser.add_argument("--icon", help="Icon file from")
+    parser.add_argument("--volume-name", help="Disk image volume name")
+
+    parser.add_argument("inpath", metavar="PATH_IN", help="Location of files to pack")
+    parser.add_argument("dmgfile", metavar="DMG_OUT", help="DMG File to create")
+
+    options = parser.parse_args(args)
+
+    extra_files = []
+    if options.dsstore:
+        extra_files.append((options.dsstore, ".DS_Store"))
+    if options.background:
+        extra_files.append((options.background, ".background/background.png"))
+    if options.icon:
+        extra_files.append((options.icon, ".VolumeIcon.icns"))
+
+    if options.volume_name:
+        volume_name = options.volume_name
+    else:
+        volume_name = get_application_ini_value(
+            options.inpath, "App", "CodeName", fallback="Name"
+        )
+
+    # Resolve required tools
+    dmg_tool = bootstrap_toolchain("dmg/dmg")
+    hfs_tool = bootstrap_toolchain("dmg/hfsplus")
+    mkfshfs_tool = bootstrap_toolchain("hfsplus/newfs_hfs")
+
+    dmg.create_dmg(
+        source_directory=Path(options.inpath),
+        output_dmg=Path(options.dmgfile),
+        volume_name=volume_name,
+        extra_files=extra_files,
+        dmg_tool=dmg_tool,
+        hfs_tool=hfs_tool,
+        mkfshfs_tool=mkfshfs_tool,
+    )
+
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main(sys.argv[1:]))