Feat: add pkg support in orasclient (#7)

.github/workflows/build-meson-tst.yml

@@ -37,7 +37,7 @@ jobs:
         source venv/bin/activate
         cd blint-db/
         python blint_db/cli.py --clean-start -f -Z1
-        python ./.oras/orasclient.py
+        python ./.oras/orasclient.py -p meson-tst
       env:
         GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         GITHUB_USERNAME: ${{ github.actor }}

.github/workflows/build-meson.yml

@@ -37,7 +37,7 @@ jobs:
         source venv/bin/activate
         cd blint-db/
         python blint_db/cli.py --clean-start -Z1
-        python ./.oras/orasclient.py
+        python ./.oras/orasclient.py -p meson
       env:
         GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         GITHUB_USERNAME: ${{ github.actor }}

.github/workflows/build-vcpkg-tst.yml

@@ -37,7 +37,7 @@ jobs:
         source venv/bin/activate
         cd blint-db/
         python3 blint-db/blint_db/cli.py --clean-start -f -Z2
-        python3 ./.oras/orasclient.py
+        python3 ./.oras/orasclient.py -p vcpkg-tst
       env:
         GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         GITHUB_USERNAME: ${{ github.actor }}

.github/workflows/build-vcpkg.yml

@@ -37,7 +37,7 @@ jobs:
         source venv/bin/activate
         cd blint-db/
         python3 blint-db/blint_db/cli.py --clean-start -Z2
-        python3 ./.oras/orasclient.py
+        python3 ./.oras/orasclient.py -p vcpkg
       env:
         GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
         GITHUB_USERNAME: ${{ github.actor }}

.oras/orasclient.py

@@ -1,19 +1,61 @@
 import oras.client
 import os
+import argparse
 
 client = oras.client.OrasClient()
 
 token = os.getenv("GITHUB_TOKEN", "")
 username = os.getenv("GITHUB_USERNAME", "")
 
-
 client.login(password=token, username=username)
 
-client.push(
-    target="ghcr.io/appthreat/blintdb-meson:v0.1",
-    config_path="./.oras/config.json",
-    annotation_file="./.oras/annotations.json",
-    files=[
-        "./blint.db:application/vnd.appthreat.blintdb.layer.v1+tar",
-    ],
-)
+parser = argparse.ArgumentParser(
+    prog="orasclient_blintdb", description="Helps pushing blint.db into a container and uploading to ghcr.io"
+)
+parser.add_argument(
+    "-p",
+    "--pkg-manager",
+    dest="pkg",
+    help="Path to the CDXGEN bom file (NOT IMPLEMENTED)",
+)
+args = vars(parser.parse_args())
+
+if pkg := args.get("pkg", None):
+    # not using fstring here to make sure value is correct
+    # otherwise wrong file may be uploaded
+    if pkg=="vcpkg":
+        client.push(
+            target="ghcr.io/appthreat/blintdb-vcpkg:v0.1",
+            config_path="./.oras/config.json",
+            annotation_file="./.oras/annotations.json",
+            files=[
+                "./blint.db:application/vnd.appthreat.blintdb.layer.v1+tar",
+            ],
+        )       
+    if pkg=="meson":
+        client.push(
+            target="ghcr.io/appthreat/blintdb-meson:v0.1",
+            config_path="./.oras/config.json",
+            annotation_file="./.oras/annotations.json",
+            files=[
+                "./blint.db:application/vnd.appthreat.blintdb.layer.v1+tar",
+            ],
+        )
+    if pkg=="vcpkg-tst":
+        client.push(
+            target="ghcr.io/appthreat/blintdb-vcpkg-tst:v0.1",
+            config_path="./.oras/config.json",
+            annotation_file="./.oras/annotations.json",
+            files=[
+                "./blint.db:application/vnd.appthreat.blintdb.layer.v1+tar",
+            ],
+        )       
+    if pkg=="meson-tst":
+        client.push(
+            target="ghcr.io/appthreat/blintdb-meson-tst:v0.1",
+            config_path="./.oras/config.json",
+            annotation_file="./.oras/annotations.json",
+            files=[
+                "./blint.db:application/vnd.appthreat.blintdb.layer.v1+tar",
+            ],
+        )