ADO Pipeline to update a onefuzz instance with the latest release (#208)

contrib/deploy-onefuzz-via-azure-devops/get_latest_version.py

@@ -0,0 +1,71 @@
+#!/usr/bin/env python
+#
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+import argparse
+import os
+import requests
+
+BASE_URL = "https://api.github.com/repos/microsoft/onefuzz"
+
+
+class Onefuzz:
+    def get_latest_version(self):
+        latest_releasee = requests.get(f"{BASE_URL}/releases/latest").json()
+        return (latest_releasee["id"], latest_releasee["name"])
+
+    def list_assets(self, release_id):
+        assets = requests.get(f"{BASE_URL}/releases/{release_id}/assets").json()
+        artifacts = []
+        for asset in assets:
+            artifacts.append({"id": asset["id"], "name": asset["name"]})
+        return artifacts
+
+    def download_artifact(self, path, asset_id, asset_name):
+        headers = {"Accept": "application/octet-stream"}
+        asset = requests.get(
+            f"{BASE_URL}/releases/assets/{asset_id}", headers=headers
+        )
+        with open(os.path.join(path, asset_name), "wb") as artifact:
+            artifact.write(asset.content)
+
+    def download_artifacts(self, path, artifacts):
+        for artifact in artifacts:
+            self.download_artifact(path, artifact["id"], artifact["name"])
+
+    def onefuzz_release_artifacts(self, path):
+        release_id, _ = self.get_latest_version()
+        artifacts = self.list_assets(release_id)
+        self.download_artifacts(path, artifacts)
+
+
+def main():
+    def dir_path(path):
+        path = os.path.abspath(path)
+        if not os.path.isdir(path):
+            os.makedirs(path)
+        return path
+
+    parser = argparse.ArgumentParser(description="Download artifacts")
+    parser.add_argument(
+        "-path",
+        type=dir_path,
+        help="Path to download binaries",
+    )
+    parser.add_argument(
+        "-version",
+        action="store_true",
+        help="Get Onefuzz latest version",
+    )
+
+    args = parser.parse_args()
+    if args.path:
+        Onefuzz().onefuzz_release_artifacts(args.path)
+
+    if args.version:
+        print(Onefuzz().get_latest_version()[1])
+
+
+if __name__ == "__main__":
+    main()