set origin header

saml-to · Dec 7, 2023 · 1607bf8 · 1607bf8
1 parent 23a3d6f
commit 1607bf8
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 2 deletions.
diff --git a/package.json b/package.json
@@ -143,4 +143,4 @@
     "qrcode-svg": "^1.1.0",
     "which": "^4.0.0"
   }
-}
+}
diff --git a/src/api.ts b/src/api.ts
@@ -8,7 +8,7 @@ import packageJson from "../package.json";
 import os from "os";
 import { Configuration } from "./config";
 
-type Headers = { "user-agent": string; "x-2fa-code"?: string };
+type Headers = { "user-agent": string; "x-2fa-code"?: string; origin?: string };
 
 const USER_AGENT = `${packageJson.name}/${
   packageJson.version
@@ -113,6 +113,12 @@ export class ApiFactory {
       headers["x-2fa-code"] = `${twoFactorCode}`;
     }
 
+    // Origin header for Codespaces Identity
+    const repository = this.configuration.github.repository;
+    if (repository) {
+      headers["origin"] = repository;
+    }
+
     const configuration = new IDPConfiguration({
       accessToken: accessToken || (await this.getAccessToken(twoFactorCode)),
       baseOptions: {
@@ -137,6 +143,11 @@ export class ApiFactory {
       headers["x-2fa-code"] = `${twoFactorCode}`;
     }
 
+    const repository = this.configuration.github.repository;
+    if (repository) {
+      headers["origin"] = repository;
+    }
+
     const configuration = new IDPConfiguration({
       accessToken: accessToken || (await this.getAccessToken(twoFactorCode)),
       baseOptions: {

diff --git a/src/config.ts b/src/config.ts
@@ -43,6 +43,12 @@ class GitHubConfiguration {
     const fromEnv = process.env.GITHUB_TOKEN;
     return fromConfig || fromEnv;
   }
+
+  get repository(): string | undefined {
+    // TODO: alternate read from /workspaces/.codespaces/shared/.env
+    const fromEnv = process.env.GITHUB_REPOSITORY;
+    return fromEnv;
+  }
 }
 
 class AssumeAwsProfileConfiguration {