refactor: use native fetch for github device flow

Author: ericc-ch

src/lib/constants.ts

@@ -1,5 +1,4 @@
 // VSCode client ID
-const GITHUB_CLIENT_ID = "01ab8ac9400c4e429b23"
 const GITHUB_OAUTH_SCOPES = [
   "read:org",
   "read:user",
@@ -9,7 +8,7 @@ const GITHUB_OAUTH_SCOPES = [
 ].join(" ")
 
 export const ENV = {
-  GITHUB_CLIENT_ID,
+  GITHUB_CLIENT_ID: "01ab8ac9400c4e429b23",
   GITHUB_OAUTH_SCOPES,
 }
 
@@ -29,3 +28,13 @@ export const GITHUB_API_CONFIG = {
 export const GITHUB_WEB_API_CONFIG = {
   baseURL: "https://github.com",
 } as const
+
+export const GITHUB_BASE_URL = "https://github.com"
+export const GITHUB_CLIENT_ID = "01ab8ac9400c4e429b23"
+export const GITHUB_APP_SCOPES = [
+  "read:org",
+  "read:user",
+  "repo",
+  "user:email",
+  "workflow",
+].join(" ")

src/lib/initialization.ts

@@ -62,8 +62,7 @@ async function logModelInformation(): Promise<void> {
 
 async function initializeGithubToken() {
   consola.start("Getting GitHub device code")
-  const tokenResponse = await getGitHubToken()
-  return tokenResponse.access_token
+  return await getGitHubToken()
 }
 
 async function logUser() {

src/lib/sleep.ts

@@ -0,0 +1,4 @@
+export const sleep = (ms: number) =>
+  new Promise((resolve) => {
+    setTimeout(resolve, ms)
+  })

src/lib/state.ts

@@ -0,0 +1,9 @@
+import type { GetModelsResponse } from "~/services/copilot/get-models/types"
+
+interface State {
+  githubToken?: string
+  copilotToken?: string
+  models?: GetModelsResponse
+}
+
+export const state: State = {}

src/services/github/get-token/service.ts

@@ -1,7 +1,7 @@
+import { sleep } from "bun"
 import consola from "consola"
 
-import { ENV } from "~/lib/constants"
-import { _github } from "~/services/api-instance"
+import { GITHUB_CLIENT_ID, GITHUB_BASE_URL } from "~/lib/constants"
 
 interface DeviceCodeResponse {
   device_code: string
@@ -18,42 +18,52 @@ interface AccessTokenResponse {
 }
 
 export async function getGitHubToken() {
-  const response = await _github<DeviceCodeResponse>("/login/device/code", {
+  const response = await fetch(`${GITHUB_BASE_URL}/login/device/code`, {
     method: "POST",
-    body: {
-      client_id: ENV.GITHUB_CLIENT_ID,
-      scope: ENV.GITHUB_OAUTH_SCOPES,
-    },
+    body: JSON.stringify({
+      client_id: GITHUB_CLIENT_ID,
+    }),
   })
 
-  consola.info(
-    `Please enter the code "${response.user_code}" in ${response.verification_uri}`,
-  )
+  if (!response.ok) {
+    throw new Error("Failed to get device code", {
+      cause: await response.json(),
+    })
+  }
+
+  const { user_code, verification_uri, device_code, interval } =
+    (await response.json()) as DeviceCodeResponse
+
+  consola.info(`Please enter the code "${user_code}" in ${verification_uri}`)
 
   while (true) {
-    const pollResponse = await _github<AccessTokenResponse>(
-      "/login/oauth/access_token",
+    const response = await fetch(
+      `${GITHUB_BASE_URL}/login/oauth/access_token`,
       {
         method: "POST",
-        body: {
-          client_id: ENV.GITHUB_CLIENT_ID,
-          device_code: response.device_code,
+        body: JSON.stringify({
+          client_id: GITHUB_CLIENT_ID,
+          device_code,
           grant_type: "urn:ietf:params:oauth:grant-type:device_code",
-        },
+        }),
       },
     )
 
-    if (pollResponse.access_token) {
-      consola.info(
-        `Got token ${pollResponse.access_token.replaceAll(/./g, "*")}`,
-      )
-      return pollResponse
+    // Interval is in seconds, we need to multiply by 1000 to get milliseconds
+    // I'm also adding another second, just to be safe
+    const sleepDuration = (interval + 1) * 1000
+
+    if (!response.ok) {
+      await sleep(sleepDuration)
+      continue
+    }
+
+    const { access_token } = (await response.json()) as AccessTokenResponse
+
+    if (access_token) {
+      return access_token
     } else {
-      // Interval is in seconds, we need to multiply by 1000 to get milliseconds
-      // I'm also adding another second, just to be safe
-      await new Promise((resolve) =>
-        setTimeout(resolve, (response.interval + 1) * 1000),
-      )
+      await sleep(sleepDuration)
     }
   }
 }