feat: Add initialization module and refactor main.ts

Author: ericc-ch

src/lib/initialization/index.ts

@@ -0,0 +1,68 @@
+import consola from "consola"
+import fs from "node:fs"
+
+import { PATHS } from "../../config/paths"
+import { TOKENS } from "../../config/tokens"
+import { getModels } from "../../services/copilot/get-models/service"
+import { getCopilotToken } from "../../services/copilot/get-token/copilot-token"
+import { getGitHubToken } from "../../services/copilot/get-token/github-token"
+import { CACHE } from "../cache"
+
+async function initializeGithubToken() {
+  const EIGHT_HOURS = 8 * 60 * 60 * 1000
+  const cachedGithubToken = await CACHE.get("github-token")
+
+  // If exists and at most 8 hours old
+  if (
+    cachedGithubToken &&
+    Date.now() - cachedGithubToken.createdAt < EIGHT_HOURS
+  ) {
+    return cachedGithubToken.value
+  }
+
+  const newToken = await getGitHubToken()
+  await CACHE.set("github-token", newToken)
+  return newToken
+}
+
+async function initializeCopilotToken() {
+  const { token, refresh_in } = await getCopilotToken()
+  TOKENS.COPILOT_TOKEN = token
+
+  // refresh_in is in seconds
+  // we're refreshing 10 minutes (600 seconds) early
+  const refreshInterval = (refresh_in - 600) * 1000
+
+  setInterval(async () => {
+    consola.start("Refreshing copilot token")
+    const { token: newToken } = await getCopilotToken()
+    TOKENS.COPILOT_TOKEN = newToken
+  }, refreshInterval)
+
+  return token
+}
+
+async function initializeCache() {
+  if (!fs.existsSync(PATHS.PATH_CACHE_FILE)) {
+    fs.mkdirSync(PATHS.DIR_CACHE, { recursive: true })
+    await CACHE._write({})
+  }
+}
+
+async function logAvailableModels() {
+  const models = await getModels()
+  consola.info(
+    `Available models: \n${models.data.map((model) => `- ${model.id}`).join("\n")}`,
+  )
+}
+
+export async function initialize() {
+  await initializeCache()
+
+  // Initialize tokens
+  TOKENS.GITHUB_TOKEN = await initializeGithubToken()
+  await initializeCopilotToken()
+
+  // Log available models
+  await logAvailableModels()
+}

src/main.ts

@@ -1,74 +1,7 @@
 import type { Serve } from "bun"
 
-import consola from "consola"
-import fs from "node:fs"
-
-import { PATHS } from "./config/paths"
-import { TOKENS } from "./config/tokens"
-import { CACHE } from "./lib/cache"
+import { initialize } from "./lib/initialization"
 import { server } from "./server"
-import { getModels } from "./services/copilot/get-models/service"
-import { getCopilotToken } from "./services/copilot/get-token/copilot-token"
-import { getGitHubToken } from "./services/copilot/get-token/github-token"
-
-async function initializeGithubToken() {
-  const EIGHT_HOURS = 8 * 60 * 60 * 1000
-  const cachedGithubToken = await CACHE.get("github-token")
-
-  // If exists and at most 8 hours old
-  if (
-    cachedGithubToken &&
-    Date.now() - cachedGithubToken.createdAt < EIGHT_HOURS
-  ) {
-    return cachedGithubToken.value
-  }
-
-  const newToken = await getGitHubToken()
-  await CACHE.set("github-token", newToken)
-  return newToken
-}
-
-async function initializeCopilotToken() {
-  const { token, refresh_in } = await getCopilotToken()
-  TOKENS.COPILOT_TOKEN = token
-
-  // refresh_in is in seconds
-  // we're refreshing 10 minutes (600 seconds) early
-  const refreshInterval = (refresh_in - 600) * 1000
-
-  setInterval(async () => {
-    consola.start("Refreshing copilot token")
-    const { token: newToken } = await getCopilotToken()
-    TOKENS.COPILOT_TOKEN = newToken
-  }, refreshInterval)
-
-  return token
-}
-
-async function initializeCache() {
-  if (!fs.existsSync(PATHS.PATH_CACHE_FILE)) {
-    fs.mkdirSync(PATHS.DIR_CACHE, { recursive: true })
-    await CACHE._write({})
-  }
-}
-
-async function logAvailableModels() {
-  const models = await getModels()
-  consola.info(
-    `Available models: \n${models.data.map((model) => `- ${model.id}`).join("\n")}`,
-  )
-}
-
-async function initialize() {
-  await initializeCache()
-
-  // Initialize tokens
-  TOKENS.GITHUB_TOKEN = await initializeGithubToken()
-  await initializeCopilotToken()
-
-  // Log available models
-  await logAvailableModels()
-}
 
 // Initialize the application
 await initialize()