scripts: add fetch releases.py to automatically generate/update remotes.json with new releases

scripts/fetch_releases.py

@@ -0,0 +1,306 @@
+import json
+import optparse
+import os
+import re
+import requests
+
+IGNORE_VERSIONS_BEFORE = '4.3'
+
+
+def version_number_and_type(git_hash, ap_source_subdir):
+    url = (
+        "https://raw.githubusercontent.com/ArduPilot/ardupilot/"
+        f"{git_hash}/{ap_source_subdir}/version.h"
+    )
+    response = requests.get(url=url)
+
+    if response.status_code != 200:
+        print(response.text)
+        print(url)
+        raise Exception(
+            "Couldn't fetch version.h from github server. "
+            f"Got status code {response.status_code}"
+        )
+
+    exp = re.compile(
+        r'#define FIRMWARE_VERSION (\d+),(\d+),(\d+),FIRMWARE_VERSION_TYPE_(\w+)'  # noqa
+    )
+    matches = re.findall(exp, response.text)
+    major, minor, patch, fw_type = matches[0]
+    fw_type = fw_type.lower()
+
+    if fw_type == 'official':
+        # to avoid any confusion, beta is also 'official' ;-)
+        fw_type = 'stable'
+    return f'{major}.{minor}.{patch}', fw_type
+
+
+def fetch_tags_from_github():
+    url = 'https://api.github.com/repos/ardupilot/ardupilot/git/refs/tags'
+    headers = {
+        'X-GitHub-Api-Version': '2022-11-28',
+        'Accept': 'application/vnd.github+json'
+    }
+    response = requests.get(url=url, headers=headers)
+    if response.status_code != 200:
+        print(response.text)
+        print(url)
+        raise Exception(
+            "Couldn't fetch tags from github server. "
+            f"Got status code {response.status_code}"
+        )
+
+    tags_objs = response.json()
+    return tags_objs
+
+
+def remove_duplicate_entries(releases):
+    temp = {}
+    for release in releases:
+        # if we have already seen a version with similar hash
+        # and we now see a beta release with same hash, we skip it
+        if temp.get(release['commit_reference']) and \
+           release['release_type'] == 'beta':
+            continue
+
+        temp[release['commit_reference']] = release
+
+    return list(temp.values())
+
+
+def construct_vehicle_versions_list(vehicle, ap_source_subdir,
+                                    fw_server_vehicle_sdir,
+                                    tag_filter_exps, tags):
+    ret = []
+    for tag_info in tags:
+        tag = tag_info['ref'].replace('refs/tags/', '')
+
+        matches = []
+        for exp in tag_filter_exps:
+            # the regexes capture two groups
+            # first group is the matched substring itself
+            # second group is the tag body (e.g., beta, stable, 4.5.1 etc)
+            matches.extend(re.findall(re.compile(exp), tag))
+
+        if matches:
+            matched_string, tag_body = matches[0]
+
+            if len(matched_string) < len(tag):
+                print(
+                    f"Partial match. Ignoring. Matched '{matched_string}' "
+                    f"in '{tag}'."
+                )
+                continue
+
+            try:
+                v_num, v_type = version_number_and_type(
+                    tag_info['object']['sha'],
+                    ap_source_subdir
+                )
+            except Exception as e:
+                print(f'Cannot determine version number for tag {tag}')
+                print(e)
+                continue
+
+            if v_num < IGNORE_VERSIONS_BEFORE:
+                print(f"{v_num} Version too old. Ignoring.")
+                continue
+
+            if re.search(r'\d+\.\d+\.\d+', tag_body):
+                # we do stable version tags in this format
+                # e.g. Rover-4.5.1, Copter-4.5.1, where Rover and Copter
+                # are prefixes and 4.5.1 is the tag body
+                # artifacts for these versions are stored in firmware
+                # server at stable-x.y.z subdirs
+                afcts_url = (
+                    f'https://firmware.ardupilot.org/{fw_server_vehicle_sdir}'
+                    f'/stable-{tag_body}'
+                )
+            else:
+                afcts_url = (
+                    f'https://firmware.ardupilot.org/{fw_server_vehicle_sdir}'
+                    f'/{tag_body}'
+                )
+
+            ret.append({
+                'release_type': v_type,
+                'version_number': v_num,
+                'ap_build_atrifacts_url': afcts_url,
+                'commit_reference': tag_info['object']['sha']
+            })
+
+    ret = remove_duplicate_entries(ret)
+
+    # entry for master
+    ret.append({
+        'release_type': 'latest',
+        'version_number': 'NA',
+        'ap_build_atrifacts_url': (
+            f'https://firmware.ardupilot.org/{fw_server_vehicle_sdir}/latest'
+        ),
+        'commit_reference': 'refs/heads/master'
+    })
+
+    return {
+        'name': vehicle,
+        'releases': ret
+    }
+
+
+# swiped from app.py
+def get_auth_token():
+    try:
+        # try to read the secret token from the file
+        with open(
+            os.path.join(basedir, 'secrets', 'reload_token'),
+            'r'
+        ) as file:
+            token = file.read().strip()
+            return token
+    except (FileNotFoundError, PermissionError):
+        # if the file does not exist, check the environment variable
+        return os.getenv('CBS_REMOTES_RELOAD_TOKEN')
+
+
+parser = optparse.OptionParser("fetch_releases.py")
+parser.add_option(
+    "", "--basedir", type="string",
+    default=os.path.abspath(
+        os.path.join(os.path.dirname(__file__), "..", "..", "base")
+    ),
+    help="base directory"
+)
+
+parser.add_option(
+    "", "--remotename", type="string",
+    default="ardupilot",
+    help="Remote name to write in json file"
+)
+
+parser.add_option(
+    "", "--appurl", type="string",
+    default="https://custom.ardupilot.org/refresh_remotes",
+    help="URL for triggering remotes.json refresh on custom build server app"
+)
+
+cmd_opts, cmd_args = parser.parse_args()
+basedir = os.path.abspath(cmd_opts.basedir)
+remotes_json_path = os.path.join(basedir, 'configs', 'remotes.json')
+
+tags = fetch_tags_from_github()
+vehicles = []
+
+vehicles.append(construct_vehicle_versions_list(
+    "Copter",
+    "ArduCopter",
+    "Copter",
+    [
+        "(ArduCopter-(beta-4.3|beta|stable))",
+        "(Copter-(\d+\.\d+\.\d+))"  # noqa
+    ],
+    tags
+))
+
+vehicles.append(construct_vehicle_versions_list(
+    "Plane",
+    "ArduPlane",
+    "Plane",
+    [
+        "(ArduPlane-(beta-4.3|beta|stable))",
+        "(Plane-(\d+\.\d+\.\d+))"  # noqa
+    ],
+    tags
+))
+
+vehicles.append(construct_vehicle_versions_list(
+    "Rover",
+    "Rover",
+    "Rover",
+    [
+        "(APMrover2-(beta-4.3|beta|stable))",
+        "(Rover-(\d+\.\d+\.\d+))"  # noqa
+    ],
+    tags
+))
+
+vehicles.append(construct_vehicle_versions_list(
+    "Sub",
+    "ArduSub",
+    "Sub",
+    [
+        "(ArduSub-(beta-4.3|beta|stable))",
+        "(Sub-(\d+\.\d+\.\d+))"  # noqa
+    ],
+    tags
+))
+
+vehicles.append(construct_vehicle_versions_list(
+    "Tracker",
+    "AntennaTracker",
+    "AntennaTracker",
+    [
+        "(AntennaTracker-(beta-4.3|beta|stable))",
+        "(Tracker-(\d+\.\d+\.\d+))"  # noqa
+    ],
+    tags
+))
+
+vehicles.append(construct_vehicle_versions_list(
+    "Blimp",
+    "Blimp",
+    "Blimp",
+    [
+        "(Blimp-(beta-4.3|beta|stable|\d+\.\d+\.\d+))"  # noqa
+    ],
+    tags
+))
+
+vehicles.append(construct_vehicle_versions_list(
+    "Heli",
+    "ArduCopter",
+    "Copter",
+    [
+        "(ArduCopter-(beta-4.3|beta|stable)-heli)"
+    ],
+    tags
+))
+
+remotes_json = {
+    "name": cmd_opts.remotename,
+    "url": "https://github.com/ardupilot/ardupilot.git",
+    "vehicles": vehicles
+}
+
+try:
+    with open(remotes_json_path, 'r') as f:
+        remotes = json.loads(f.read())
+
+    # remove existing remote entry from the list
+    temp = []
+    for remote in remotes:
+        if remote['name'] != cmd_opts.remotename:
+            temp.append(remote)
+    remotes = temp
+except Exception as e:
+    print(e)
+    print("Writing to empty file")
+    remotes = []
+
+with open(remotes_json_path, 'w') as f:
+    remotes.append(remotes_json)
+    f.write(json.dumps(remotes, indent=2))
+    print(f"Wrote {remotes_json_path}")
+
+# retrieve authentication token
+auth_token = get_auth_token()
+# hit reload endpoint on custom build server app
+if not auth_token:
+    print("Failed to retrieve authentication token. Aborting.")
+else:
+    headers = {'Content-Type': 'application/json'}
+    data = {'token': auth_token}
+    response = requests.post(url=cmd_opts.appurl, json=data, headers=headers)
+    if response.status_code == 200:
+        print("Successfully refreshed remotes.json on app")
+    else:
+        print("Failed to refresh remotes.json on app")