feat: Add CLI argument parsing and refactor main entry point

Author: ericc-ch

src/lib/cli.ts

@@ -0,0 +1,40 @@
+import { defineCommand, parseArgs, showUsage, type ArgsDef } from "citty"
+
+const args = {
+  help: {
+    alias: "h",
+    type: "boolean",
+    default: false,
+    description: "Show this help message",
+  },
+  stream: {
+    alias: "s",
+    type: "boolean",
+    default: false,
+    description: "Enable streaming response for chat completions",
+  },
+  port: {
+    alias: "p",
+    type: "string",
+    default: "4141",
+    description: "Port to listen on",
+  },
+} satisfies ArgsDef
+
+export interface CliOptions {
+  help: boolean
+  stream: boolean
+  port: string
+}
+
+export async function parseCli() {
+  const command = defineCommand({ args })
+  const options = parseArgs<typeof args>(Bun.argv, args)
+
+  if (options.help) {
+    await showUsage(command)
+    process.exit()
+  }
+
+  return options
+}

src/main.ts

@@ -1,42 +1,13 @@
 import type { Serve } from "bun"
 
-import { defineCommand, parseArgs, showUsage, type ArgsDef } from "citty"
-
+import { parseCli } from "./lib/cli"
 import { initialize } from "./lib/initialization"
 import { server } from "./server"
 
-const args: ArgsDef = {
-  help: {
-    alias: "h",
-    type: "boolean",
-    default: false,
-    description: "Show this help message",
-  },
-  stream: {
-    alias: "s",
-    type: "boolean",
-    default: false,
-    description: "Enable streaming response for chat completions",
-  },
-  port: {
-    alias: "p",
-    type: "string",
-    default: "4141",
-    description: "Port to listen on",
-  },
-}
-
-const command = defineCommand({ args })
-const options = parseArgs(Bun.argv, args)
-
-if (options.help) {
-  await showUsage(command)
-  process.exit()
-}
-
+const options = await parseCli()
 await initialize()
 
 export default {
   fetch: server.fetch,
-  port: parseInt(options.port as string),
+  port: parseInt(options.port, 10),
 } satisfies Serve