feat: Add configuration and logging functionality to the app

Author: ericc-ch

src/config/config.ts

@@ -0,0 +1,4 @@
+export const CONFIG = {
+  EMULATE_STREAMING: false,
+  LOGGING_ENABLED: false,
+}

src/config/env.ts

@@ -11,5 +11,4 @@ const GITHUB_OAUTH_SCOPES = [
 export const ENV = {
   GITHUB_CLIENT_ID,
   GITHUB_OAUTH_SCOPES,
-  EMULATE_STREAMING: false,
 }

src/config/paths.ts

@@ -4,8 +4,12 @@ import path from "pathe"
 const APP_DIR = path.join(os.homedir(), ".local", "share", "copilot-api")
 
 const CACHE_PATH = path.join(APP_DIR, "cache.json")
+const LOG_PATH = path.join(APP_DIR, "logs")
+const LOG_FILE = path.join(LOG_PATH, "app.log")
 
 export const PATHS = {
   APP_DIR,
   CACHE_PATH,
+  LOG_PATH,
+  LOG_FILE,
 }

src/lib/cli.ts

@@ -18,6 +18,11 @@ const args = {
     default: "4141",
     description: "Port to listen on",
   },
+  logs: {
+    type: "boolean",
+    default: false,
+    description: "Write logs to the app directory",
+  },
 } satisfies ArgsDef
 
 export async function parseCli() {

src/lib/initialization.ts

@@ -2,17 +2,18 @@ import consola from "consola"
 import fs from "node:fs"
 import { FetchError } from "ofetch"
 
-import { ENV } from "~/config/env"
+import { CONFIG } from "~/config/config"
+import { PATHS } from "~/config/paths"
 import { getGitHubUser } from "~/services/github/get-user/service"
 
 import type { parseCli } from "./cli"
 
-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/github/get-token/service"
 import { CACHE } from "./cache"
+import { initializeLogger } from "./logger"
 
 interface InitStep {
   name: string
@@ -23,7 +24,7 @@ const initSteps: Array<InitStep> = [
   {
     name: "Emulation check",
     run: () => {
-      if (ENV.EMULATE_STREAMING) {
+      if (CONFIG.EMULATE_STREAMING) {
         consola.box("Streaming emulation is enabled.")
       }
     },
@@ -108,7 +109,11 @@ async function logUser() {
 export async function initializeApp(
   options: Awaited<ReturnType<typeof parseCli>>,
 ) {
-  ENV.EMULATE_STREAMING = options["emulate-streaming"]
+  CONFIG.EMULATE_STREAMING = options["emulate-streaming"]
+  CONFIG.LOGGING_ENABLED = options.logs
+
+  // Initialize logger if enabled
+  initializeLogger()
 
   await initialize()
 

src/lib/logger.ts

@@ -0,0 +1,17 @@
+import fs from "node:fs"
+
+import { CONFIG } from "~/config/config"
+import { PATHS } from "~/config/paths"
+
+export function initializeLogger() {
+  if (!CONFIG.LOGGING_ENABLED) return
+
+  fs.mkdirSync(PATHS.LOG_PATH, { recursive: true })
+}
+
+export function logToFile(type: string, message: string) {
+  if (!CONFIG.LOGGING_ENABLED) return
+
+  const timestamp = new Date().toISOString()
+  fs.appendFileSync(PATHS.LOG_FILE, `${timestamp} ${type}: ${message}\n`)
+}

src/routes/chat-completions/route.ts

@@ -2,7 +2,7 @@ import consola from "consola"
 import { Hono } from "hono"
 import { FetchError } from "ofetch"
 
-import { ENV } from "~/config/env"
+import { CONFIG } from "~/config/config"
 
 import { handler } from "./handler"
 import { handlerStreaming } from "./handler-streaming"
@@ -11,7 +11,7 @@ export const completionRoutes = new Hono()
 
 completionRoutes.post("/chat/completions", async (c) => {
   try {
-    if (ENV.EMULATE_STREAMING) {
+    if (CONFIG.EMULATE_STREAMING) {
       return await handler(c)
     }