refactor: extract token retrieval into smaller functions

Author: ericc-ch

src/lib/paths.ts

@@ -1,5 +1,6 @@
+import fs from "node:fs/promises"
 import os from "node:os"
-import path from "pathe"
+import path from "node:path"
 
 const APP_DIR = path.join(os.homedir(), ".local", "share", "copilot-api")
 
@@ -9,3 +10,17 @@ export const PATHS = {
   APP_DIR,
   GITHUB_TOKEN_PATH,
 }
+
+export async function ensurePaths(): Promise<void> {
+  await fs.mkdir(PATHS.APP_DIR, { recursive: true })
+  await ensureFile(PATHS.GITHUB_TOKEN_PATH)
+}
+
+async function ensureFile(filePath: string): Promise<void> {
+  try {
+    await fs.access(filePath, fs.constants.W_OK)
+  } catch {
+    await fs.writeFile(filePath, "")
+    await fs.chmod(filePath, 0o600)
+  }
+}

src/services/github/get-device-code.ts

@@ -0,0 +1,26 @@
+import { GITHUB_BASE_URL, GITHUB_CLIENT_ID } from "~/lib/constants"
+
+export async function getDeviceCode(): Promise<DeviceCodeResponse> {
+  const response = await fetch(`${GITHUB_BASE_URL}/login/device/code`, {
+    method: "POST",
+    body: JSON.stringify({
+      client_id: GITHUB_CLIENT_ID,
+    }),
+  })
+
+  if (!response.ok) {
+    throw new Error("Failed to get device code", {
+      cause: await response.json(),
+    })
+  }
+
+  return (await response.json()) as DeviceCodeResponse
+}
+
+export interface DeviceCodeResponse {
+  device_code: string
+  user_code: string
+  verification_uri: string
+  expires_in: number
+  interval: number
+}

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

@@ -1,69 +1,14 @@
-import { sleep } from "bun"
 import consola from "consola"
 
-import { GITHUB_CLIENT_ID, GITHUB_BASE_URL } from "~/lib/constants"
-
-interface DeviceCodeResponse {
-  device_code: string
-  user_code: string
-  verification_uri: string
-  expires_in: number
-  interval: number
-}
-
-interface AccessTokenResponse {
-  access_token: string
-  token_type: string
-  scope: string
-}
+import { getDeviceCode } from "../get-device-code"
+import { pollAccessToken } from "../poll-access-token"
 
 export async function getGitHubToken() {
-  const response = await fetch(`${GITHUB_BASE_URL}/login/device/code`, {
-    method: "POST",
-    body: JSON.stringify({
-      client_id: GITHUB_CLIENT_ID,
-    }),
-  })
-
-  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 response = await fetch(
-      `${GITHUB_BASE_URL}/login/oauth/access_token`,
-      {
-        method: "POST",
-        body: JSON.stringify({
-          client_id: GITHUB_CLIENT_ID,
-          device_code,
-          grant_type: "urn:ietf:params:oauth:grant-type:device_code",
-        }),
-      },
-    )
-
-    // 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 response = await getDeviceCode()
 
-    const { access_token } = (await response.json()) as AccessTokenResponse
+  consola.info(
+    `Please enter the code "${response.user_code}" in ${response.verification_uri}`,
+  )
 
-    if (access_token) {
-      return access_token
-    } else {
-      await sleep(sleepDuration)
-    }
-  }
+  return await pollAccessToken(response)
 }

src/services/github/poll-access-token.ts

@@ -0,0 +1,45 @@
+import { GITHUB_BASE_URL, GITHUB_CLIENT_ID } from "~/lib/constants"
+import { sleep } from "~/lib/sleep"
+
+import type { DeviceCodeResponse } from "./get-device-code"
+
+export async function pollAccessToken(
+  deviceCode: DeviceCodeResponse,
+): Promise<string> {
+  while (true) {
+    const response = await fetch(
+      `${GITHUB_BASE_URL}/login/oauth/access_token`,
+      {
+        method: "POST",
+        body: JSON.stringify({
+          client_id: GITHUB_CLIENT_ID,
+          device_code: deviceCode.device_code,
+          grant_type: "urn:ietf:params:oauth:grant-type:device_code",
+        }),
+      },
+    )
+
+    // 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 = (deviceCode.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 {
+      await sleep(sleepDuration)
+    }
+  }
+}
+
+interface AccessTokenResponse {
+  access_token: string
+  token_type: string
+  scope: string
+}