From b2d6f8ef9caadf468469e9abbdc1c4dbae77e239 Mon Sep 17 00:00:00 2001
From: a4z <harald.achitz@gmail.com>
Date: Fri, 22 Dec 2023 14:57:38 +0100
Subject: [PATCH] More debug output

---
 .github/workflows/build.yml |  2 +-
 get-generator.py            | 15 +++++++++++----
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 274ad6a..69a6072 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -52,7 +52,7 @@ jobs:
         path: "requirements.txt"
     - name: Download Generator
       run: |
-        python get-generator.py $GENERATOR_BUILID
+        python get-generator.py $GENERATOR_BUILID ${{ secrets.GENERATOR_DL_TOKEN }}
       env:
         GENERATOR_BUILID: ${{ env.GENERATOR_BUILID }}
         GENERATOR_DL_TOKEN: ${{ secrets.GENERATOR_DL_TOKEN }}
diff --git a/get-generator.py b/get-generator.py
index 5d7c8ae..72f0b68 100644
--- a/get-generator.py
+++ b/get-generator.py
@@ -27,6 +27,17 @@
     10. Run the generator
 """
 
+dl_arg = sys.argv[1]
+if not dl_arg:
+    sys.exit('No argument given')
+
+
+key_arg = sys.argv[2]
+
+if key_arg:
+    print(f'Using key: {key_arg}')
+    os.environ['GENERATOR_DL_TOKEN'] = key_arg
+
 # Get the token from the environment variable
 TOKEN_ENV_VAR='GENERATOR_DL_TOKEN'
 TOKEN = os.getenv(TOKEN_ENV_VAR)
@@ -148,10 +159,6 @@ def get_latest_pre_release(owner, repo):
             make_executable(dl_file)
             break
 
-# Get the argument, TODO, use argparse and add optional a folder argument
-dl_arg = sys.argv[1]
-if not dl_arg:
-    sys.exit('No argument given')
 
 owner = 'cross-language-cpp'
 repo = 'djinni-generator'