1 files changed, 80 insertions, 0 deletions
diff --git a/bin/update/create_full_mar.py b/bin/update/create_full_mar.py
new file mode 100755
index 0000000000..b4f53c48f1
--- /dev/null
+++ b/bin/update/create_full_mar.py
@@ -0,0 +1,80 @@
+#!/usr/bin/env python3
+
+import sys
+import glob
+import os
+import re
+import subprocess
+import json
+import argparse
+
+from tools import uncompress_file_to_dir, get_file_info, make_complete_mar_name
+from signing import sign_mar_file
+from path import UpdaterPath, convert_to_unix, convert_to_native
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('product_name')
+    parser.add_argument('workdir')
+    parser.add_argument('filename_prefix')
+    parser.add_argument('certificate_path')
+    parser.add_argument('certificate_name')
+    parser.add_argument('base_url')
+    parser.add_argument('version')
+    args = parser.parse_args()
+
+    certificate_path = args.certificate_path
+    certificate_name = args.certificate_name
+    base_url = args.base_url
+    filename_prefix = args.filename_prefix
+    workdir = args.workdir
+    product_name = args.product_name
+    version = args.version
+
+    update_path = UpdaterPath(workdir)
+    update_path.ensure_dir_exist()
+
+    target_dir = update_path.get_update_dir()
+    temp_dir = update_path.get_current_build_dir()
+
+    tar_file_glob = os.path.join(update_path.get_workdir(), "installation", product_name, "archive", "install", "*", f'{product_name}_*_archive*')
+    tar_files = glob.glob(tar_file_glob)
+    if len(tar_files) != 1:
+        raise Exception(f'`{tar_file_glob}` does not match exactly one file')
+    tar_file = tar_files[0]
+
+    uncompress_dir = uncompress_file_to_dir(tar_file, temp_dir)
+
+    metadatafile = os.path.join(
+        update_path.get_workdir(), 'installation', product_name, 'archive', 'install', 'metadata')
+    ifsfile = os.path.join(update_path.get_mar_dir(), 'ifs')
+    with open(metadatafile) as meta, open(ifsfile, 'w') as ifs:
+        for l in meta:
+            m = re.fullmatch('(skip|cond) (.*)', l.rstrip())
+            if m and m.group(2).startswith(f'{product_name}/'):
+                path = m.group(2)[len(f'{product_name}/'):]
+                if m.group(1) == 'skip':
+                    os.remove(os.path.join(uncompress_dir, path))
+                else:
+                    ifs.write(f'"{path}" "{path}"\n')
+
+    mar_file = make_complete_mar_name(target_dir, filename_prefix)
+    path = os.path.join(
+        workdir, 'UnpackedTarball/onlineupdate/tools/update-packaging/make_full_update.sh')
+    os.putenv('MOZ_PRODUCT_VERSION', version)
+    os.putenv('MAR_CHANNEL_ID', 'LOOnlineUpdater')
+    subprocess.call([
+        path, convert_to_native(mar_file), convert_to_native(uncompress_dir),
+        convert_to_native(ifsfile)])
+
+    sign_mar_file(target_dir, certificate_path, certificate_name, mar_file, filename_prefix)
+
+    file_info = {'complete': get_file_info(mar_file, base_url)}
+
+    with open(os.path.join(target_dir, 'complete_info.json'), "w") as complete_info_file:
+        json.dump(file_info, complete_info_file, indent=4)
+
+
+if __name__ == '__main__':
+    main()