feat: token handling

Author: ericc-ch

docs/api/completions.txt

@@ -0,0 +1,9 @@
+interface Tokens {
+  GITHUB_TOKEN: string | undefined
+  COPILOT_TOKEN: string | undefined
+}
+
+export const TOKENS: Tokens = {
+  GITHUB_TOKEN: undefined,
+  COPILOT_TOKEN: undefined,
+}

docs/api/models.txt

@@ -1,16 +1,21 @@
-import { chatCompletions } from "./services/copilot-vscode/chat-completions/service"
+import consola from "consola"
 
-const gen = chatCompletions({
-  messages: [
-    {
-      content:
-        "Hey how do write a factorial function in typescript in functional programming style",
-      role: "user",
-    },
-  ],
-  model: "gpt-4o-mini",
-})
+import { TOKENS } from "./lib/tokens"
+import { getCopilotToken } from "./services/copilot-vscode/get-token/copilot-token"
+import { getGitHubToken } from "./services/copilot-vscode/get-token/github-token"
 
-for await (const chunk of gen) {
-  console.log(chunk)
-}
+const githubToken = await getGitHubToken()
+TOKENS.GITHUB_TOKEN = githubToken
+
+const { token: copilotToken, refresh_in } = await getCopilotToken()
+TOKENS.COPILOT_TOKEN = copilotToken
+
+// refresh_in is in minutes
+// we're refreshing 100 minutes early
+const refreshInterval = (refresh_in - 100) * 60 * 1000
+
+setInterval(async () => {
+  consola.start("Refreshing copilot token")
+  const { token: copilotToken } = await getCopilotToken()
+  TOKENS.COPILOT_TOKEN = copilotToken
+}, refreshInterval)

src/lib/tokens.ts

@@ -1,18 +1,28 @@
 import { ofetch } from "ofetch"
 
-import { getToken } from "./get-token/service"
+import { TOKENS } from "~/lib/tokens"
 
-const result = await getToken()
+export const copilotVSCode = ofetch.create({
+  baseURL: "https://api.individual.githubcopilot.com",
+  headers: {
+    "copilot-integration-id": "vscode-chat",
+  },
+
+  onRequest({ options }) {
+    options.headers.set("authorization", `Bearer ${TOKENS.COPILOT_TOKEN}`)
+  },
+})
 
-export const COPILOT_VSCODE_BASE_URL =
-  "https://api.individual.githubcopilot.com"
-export const COPILOT_VSCODE_TOKEN = result.token
-export const COPILOT_VSCODE_HEADERS = {
-  authorization: `Bearer ${COPILOT_VSCODE_TOKEN}`,
-  "copilot-integration-id": "vscode-chat",
-}
+export const github = ofetch.create({
+  baseURL: "https://api.github.com",
+  headers: {
+    // "editor-plugin-version": "copilot-chat/0.23.2",
+    // "editor-version": "vscode/1.96.2",
+    // "user-agent": "GitHubCopilotChat/0.23.2",
+    // "x-github-api-version": "2024-12-15",
+  },
 
-export const copilotVSCode = ofetch.create({
-  baseURL: COPILOT_VSCODE_BASE_URL,
-  headers: COPILOT_VSCODE_HEADERS,
+  onRequest({ options }) {
+    options.headers.set("authorization", `token ${TOKENS.GITHUB_TOKEN}`)
+  },
 })

src/main.ts

@@ -0,0 +1,8 @@
+import type { GetCopilotTokenResponse } from "./types"
+
+import { github } from "../api-instance"
+
+export const getCopilotToken = async () =>
+  github<GetCopilotTokenResponse>("/copilot_internal/v2/token", {
+    method: "GET",
+  })

src/services/copilot-vscode/api-instance.ts

@@ -1,26 +1,7 @@
-import consola from "consola"
 import { execa } from "execa"
 
-import { PATHS } from "~/lib/paths"
-
-import type { GetTokenResponse } from "./types"
-
-const TEN_MINUTES = 10 * 60 * 1000
-
 // @ts-expect-error TypeScript can't analyze timeout
-export async function getToken(): Promise<GetTokenResponse> {
-  try {
-    const cachedToken = await readCachedToken()
-
-    if (Date.now() - cachedToken.expires_at > ONE_DAY) {
-      return cachedToken
-    }
-  } catch (e) {
-    if (!(e instanceof Error)) throw e
-    if (e.message === "No such file or directory")
-      consola.info(`No cached token found in ${PATHS.PATH_TOKEN_CACHE}`)
-  }
-
+export async function getGitHubToken(): Promise<string> {
   // Kill any existing vscode processes
   // otherwise, no token call will be made
   await killVSCodeProcesses()
@@ -34,20 +15,24 @@ export async function getToken(): Promise<GetTokenResponse> {
 
   for await (const line of mitmdump.stdout) {
     if (typeof line !== "string") continue
-    if (!line.includes("tid=")) continue
-
-    consola.debug(`Found token output line: ${line}`)
+    if (!line.includes("authorization: token")) continue
+
+    // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+    const token = line
+      .split("\n")
+      .map((line) => line.trim())
+      .filter(Boolean)
+      .find((line) => line.includes("authorization: token"))!
+      .split("authorization: token ")
+      .at(1)!
+      .trim()
 
     clearTimeout(timeout)
 
     await killVSCodeProcesses()
     mitmdump.kill()
 
-    const parsed = JSON.parse(line) as GetTokenResponse
-    parsed.expires_at = Date.now() + t
-
-    await writeCachedToken(line)
-    return JSON.parse(line) as GetTokenResponse
+    return token
   }
 }
 
@@ -73,11 +58,3 @@ const createVSCodeProcess = () =>
   ])
 
 const killVSCodeProcesses = () => execa({ reject: false })("pkill", ["code"])
-
-const readCachedToken = async () => {
-  const content = await Bun.file(PATHS.PATH_TOKEN_CACHE).text()
-  return JSON.parse(content) as GetTokenResponse
-}
-
-const writeCachedToken = async (token: string) =>
-  Bun.write(PATHS.PATH_TOKEN_CACHE, token)

src/services/copilot-vscode/get-token/copilot-token.ts

@@ -1,4 +1,4 @@
-export interface GetTokenResponse {
+export interface GetCopilotTokenResponse {
   annotations_enabled: boolean
   chat_enabled: boolean
   chat_jetbrains_enabled: boolean