Pass vmine group ID between review app jobs

.gitlab-ci.yml

@@ -60,14 +60,21 @@ full_test:
 
 full_test_review:
   <<: *base_test
-  after_script: []
+  after_script:
+    - with-ssh-key ./test-driver cleanup --no-vagrant $BUILD_DIR
   variables:
     VM_IMAGE: "bullseye"
     CREATE_ENV_VARS: "-e config.float_debian_dist=bullseye -e inventory.group_vars.vagrant.ansible_python_interpreter=/usr/bin/python3"
-    FLOATUP_ARGS: "--ttl 6h"
+    FLOATUP_ARGS: "--state-file .vmine_group_review_$CI_MERGE_REQUEST_ID --ttl 6h --env deploy.env --dashboard-url https://vm.investici.org"
     TEST_DIR: "test/full.ref"
+  allow_failure: true
+  artifacts:
+    reports:
+      dotenv: deploy.env
+    paths: ['.vmine_group_review*']
   environment:
     name: review/$CI_COMMIT_REF_SLUG
+    url: $VMINE_GROUP_URL
     on_stop: stop_full_test_review
     auto_stop_in: "6 hours"
   rules:
@@ -77,8 +84,7 @@ stop_full_test_review:
   stage: test
   image: registry.git.autistici.org/ai3/docker/float-runner:master
   script:
-    - with-ssh-key ./test-driver cleanup --no-vagrant $BUILD_DIR
-    - with-ssh-key ./scripts/floatup.py ${LIBVIRT:+--ssh $LIBVIRT} down
+    - with-ssh-key ./scripts/floatup.py --state-file .vmine_group_review_$CI_MERGE_REQUEST_ID ${LIBVIRT:+--ssh $LIBVIRT} down
   allow_failure: true
   environment:
     name: review/$CI_COMMIT_REF_SLUG

scripts/floatup.py

@@ -5,12 +5,14 @@
 #
 
 import argparse
+import base64
 import json
 import os
 import re
 import shlex
 import subprocess
 import yaml
+import zlib
 
 
 # The Vagrant "insecure" SSH key that is used to log onto the VMs.
@@ -76,6 +78,15 @@ def do_request(url, ssh_gw, payload):
         raise
 
 
+def encode_dashboard_request(req):
+    # JSON data, in a raw zlib stream, base64-encoded.
+    hosts = sorted(req['hosts'], key=lambda x: x['name'])
+    data = json.dumps(hosts, separators=(',', ':')).encode('utf-8')
+    comp = zlib.compressobj(level=9, wbits=-9)
+    comp.compress(data)
+    return base64.urlsafe_b64encode(comp.flush()).decode('ascii')
+
+
 def install_ssh_key():
     # Install the SSH key as Vagrant would do, for compatibility.
     key_path = os.path.join(
@@ -116,6 +127,12 @@ def main():
     parser.add_argument(
         '--ttl', metavar='DURATION', default='1h',
         help='TTL for the virtual machines')
+    parser.add_argument(
+        '--env', metavar='FILE',
+        help='generate a dotenv file (for Gitlab CI)')
+    parser.add_argument(
+        '--dashboard-url', metavar='URL',
+        help='vmine dashboard base URL (for Gitlab CI)')
     parser.add_argument(
         'cmd',
         choices=['up', 'down'])
@@ -142,6 +159,14 @@ def main():
 
         install_ssh_key()
 
+        if args.env:
+            with open(args.env, 'w') as fd:
+                fd.write(f'VMINE_ID={group_id}\n')
+                if args.dashboard_url:
+                    base_url = args.dashboard_url.rstrip('/')
+                    payload = encode_dashboard_request(req)
+                    fd.write(f'VMINE_GROUP_URL={base_url}/dash/{payload}\n')
+
     elif args.cmd == 'down':
         try:
             with open(args.state_file) as fd: