fix: Correct token retrieval in getRandomToken function (#204)

FajarKim · Dec 8, 2023 · 810015e · 810015e
1 parent 782e955
commit 810015e
Show file tree

Hide file tree

Showing 3 changed files with 66 additions and 0 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -4,6 +4,7 @@
   "description": "🙀 Generate your Stats GitHub Profile in SVG",
   "main": "api/index.ts",
   "dependencies": {
+    "@actions/core": "^1.10.1",
     "@resvg/resvg-js": "^2.6.0",
     "@types/escape-html": "^1.0.4",
     "@types/express": "^4.17.21",

diff --git a/src/getRandomToken.ts b/src/getRandomToken.ts
@@ -1,5 +1,6 @@
 // Importing necessary module
 import dotenv from "dotenv";
+import { getInput } from "@actions/core";
 
 // Loads environment variables from a .env file if present
 dotenv.config();
@@ -19,9 +20,19 @@ function getRandomToken(bearerHeader: boolean): string {
   const getRandomEnv: string =
     getGhEnv[Math.floor(Math.random() * getGhEnv.length)];
 
+  if (!getRandomEnv) {
+    // Use GitHub Actions core module to get the token
+    const getRandomEnv = getInput("github_token");
+
+    if (!getRandomEnv) {
+      throw new Error("Could not find github token");
+    }
+  }
+
   if (bearerHeader) {
     return `Bearer ${getRandomEnv}`;
   }
+
   return getRandomEnv;
 }