From d3a052b9c126f79fce5bc25f45d968ed6ff4fa5c Mon Sep 17 00:00:00 2001
From: Adrien Ferrand <ferrand.ad@gmail.com>
Date: Fri, 10 Jul 2020 10:28:17 +0200
Subject: [PATCH] Convert to a python logic

---
 tools/snap/build_remote.py | 86 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)
 create mode 100644 tools/snap/build_remote.py

diff --git a/tools/snap/build_remote.py b/tools/snap/build_remote.py
new file mode 100644
index 000000000..2089d3d93
--- /dev/null
+++ b/tools/snap/build_remote.py
@@ -0,0 +1,86 @@
+#!/usr/bin/env python3
+import argparse
+import glob
+import multiprocessing
+import re
+import subprocess
+from os.path import join, realpath, dirname, basename
+
+
+CERTBOT_DIR = dirname(dirname(dirname(realpath(__file__))))
+PLUGINS = [basename(path) for path in glob.glob(join(CERTBOT_DIR, 'certbot-dns-*'))]
+
+
+def _build_snap(target, archs):
+    if target == 'certbot':
+        workspace = CERTBOT_DIR
+    else:
+        workspace = join(CERTBOT_DIR, target)
+
+    try:
+        subprocess.check_call([
+            'snapcraft', 'remote-build', '--launchpad-accept-public-upload',
+             '--build-on', ','.join(archs)
+        ], universal_newlines=True)
+    except subprocess.CalledProcessError as e:
+        # Will be handled after
+        pass
+
+    status = {}
+
+    for arch in archs:
+        with open(join(workspace, '{0}_{1}.txt'.format(target, arch))) as file_h:
+            build_output = file_h.read()
+
+        if not re.match(r'Snapped {0}_.*_{1}\.snap'.format(target, arch), build_output):
+            status[arch] = build_output
+        else:
+            status[arch] = None
+
+    return {target: status}
+
+
+def _dump_result(targets, archs, results):
+    for target in targets:
+        for arch in archs:
+            build_output = results[target][arch]
+            if build_output:
+                print('Output for failed build target={0} arch={1}'.format(target, arch))
+                print('-------------------------------------------')
+                print(build_output)
+                print('-------------------------------------------')
+
+    print('Summary')
+    print('=======')
+    for target in targets:
+        print('Builds for target={0}: {1}'.format(
+            target,
+            ', '.join([
+                'arch={0} ({1})'.format(arch, 'success' if not results[target][arch] else 'failure')
+                for arch in archs
+            ])
+        ))
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('targets', nargs='+', choices=['certbot', *PLUGINS],
+                        help='the list of snaps to build')
+    parser.add_argument('--archs', nargs='+', choices=['amd64', 'arm64', 'armhf'], default='amd64',
+                        help='the architectures for which snaps are built')
+    args = parser.parse_args()
+
+    archs = set(args.archs)
+    targets = set(args.targets)
+
+    pool = multiprocessing.Pool(processes=len(targets))
+    async_results = [pool.apply_async(_dump_result, (target, archs)) for target in targets]
+
+    results = {}
+
+    for async_result in async_results:
+        results.update(async_result.get())
+
+
+if __name__ == '__main__':
+    main()