Voice Input

/**
 * Voice TRPC router
 * Provides voice-to-text transcription using OpenAI Whisper API
 *
 * For authenticated users (with subscription): uses 21st.dev backend
 * For open-source users: requires OPENAI_API_KEY in environment
 */

function getOpenAIApiKey(): string | null {
  // First check user-configured key (highest priority)
  if (userConfiguredOpenAIKey && userConfiguredOpenAIKey.startsWith("sk-")) {
    return userConfiguredOpenAIKey
  }

  // Check Vite env vars (works with .env.local files)
  const viteKey = (import.meta.env as Record<string, string | undefined>)
    .MAIN_VITE_OPENAI_API_KEY
  if (viteKey) {
    return viteKey
  }

  // Check process.env
  if (process.env.OPENAI_API_KEY) {
    return process.env.OPENAI_API_KEY
  }

  // Try to get from shell environment
  const shell = process.env.SHELL || "/bin/zsh"
  const result = execSync(`${shell} -ilc 'echo $OPENAI_API_KEY'`, {
    encoding: "utf8",
    timeout: 5000,
  })

  return result.trim() || null
}

function cleanTranscribedText(text: string): string {
  return (
    text
      // Remove zero-width and invisible characters
      .replace(/[\u200B-\u200D\u2060\uFEFF\u00AD]/g, "")
      // Normalize unicode whitespace to regular space
      .replace(/[\u00A0\u1680\u2000-\u200A\u202F\u205F\u3000]/g, " ")
      // Replace all types of newlines and line breaks with space
      .replace(/[\r\n\u2028\u2029]+/g, " ")
      // Replace tabs with space
      .replace(/\t+/g, " ")
      // Collapse multiple spaces into one
      .replace(/ +/g, " ")
      // Trim leading/trailing whitespace
      .trim()
  )
}

isAvailable: publicProcedure.query(async () => {
  const hasLocalKey = !!getOpenAIApiKey()

  // Local API key always works
  if (hasLocalKey) {
    return {
      available: true,
      method: "local" as const,
      reason: undefined,
    }
  }

  // Check if user has paid subscription
  const hasPaid = await hasPaidSubscription()
  if (hasPaid) {
    return {
      available: true,
      method: "backend" as const,
      reason: undefined,
    }
  }

  // Check if authenticated but free plan
  const isAuthenticated = authManager?.isAuthenticated() ?? false
  if (isAuthenticated) {
    return {
      available: false,
      method: null,
      reason: "Voice input requires a paid subscription or OpenAI API key",
    }
  }

  return {
    available: false,
    method: null,
    reason: "Add your OpenAI API key in Settings > Models, or sign in with a paid subscription",
  }
})

transcribe: publicProcedure
  .input(
    z.object({
      audio: z.string(), // base64 encoded audio
      format: z.enum(["webm", "wav", "mp3", "m4a", "ogg"]).default("webm"),
      language: z.string().optional(), // ISO 639-1 code
    })
  )
  .mutation(async ({ input }) => {
    const audioBuffer = Buffer.from(input.audio, "base64")

    // If local OPENAI_API_KEY exists, use it directly
    const hasLocalKey = !!getOpenAIApiKey()
    if (hasLocalKey) {
      const text = await transcribeWithWhisper(
        audioBuffer,
        input.format,
        input.language
      )
      return { text }
    }

    // Otherwise, try backend if user is authenticated
    const isAuthenticated = authManager?.isAuthenticated() ?? false
    if (isAuthenticated) {
      const text = await transcribeViaBackend(
        audioBuffer,
        input.format,
        input.language
      )
      return { text }
    }

    throw new Error(
      "Voice input requires signing in or setting OPENAI_API_KEY"
    )
  })