auth.ts

#!/usr/bin/env node

import { defineCommand } from "citty"
import consola from "consola"

import { parseBackendId } from "./lib/backend"
import { authenticateOpenAIOAuth } from "./lib/openai-oauth"
import { PATHS, ensurePaths } from "./lib/paths"
import { state } from "./lib/state"
import { setupGitHubToken } from "./lib/token"

interface RunAuthOptions {
  backend: string
  verbose: boolean
  showToken: boolean
}

export async function runAuth(options: RunAuthOptions): Promise<void> {
  if (options.verbose) {
    consola.level = 5
    consola.info("Verbose logging enabled")
  }

  state.backend = parseBackendId(options.backend)
  state.showToken = options.showToken

  await ensurePaths()
  if (state.backend === "openai-oauth") {
    await authenticateOpenAIOAuth()
    consola.success("OpenAI OAuth token written to", PATHS.OPENAI_OAUTH_PATH)
    return
  }

  await setupGitHubToken({ force: true })
  consola.success("GitHub token written to", PATHS.GITHUB_TOKEN_PATH)
}

export const auth = defineCommand({
  meta: {
    name: "auth",
    description: "Run GitHub auth flow without running the server",
  },
  args: {
    backend: {
      type: "string",
      default: "copilot",
      description: "Auth backend to use (copilot, openai-oauth)",
    },
    verbose: {
      alias: "v",
      type: "boolean",
      default: false,
      description: "Enable verbose logging",
    },
    "show-token": {
      type: "boolean",
      default: false,
      description: "Show GitHub token on auth",
    },
  },
  run({ args }) {
    return runAuth({
      backend: args.backend,
      verbose: args.verbose,
      showToken: args["show-token"],
    })
  },
})