refactor: reusable utility for getting api keys (#54)

* chore

* up

* Update utils.ts

Author: grabbou

packages/cali/src/cli.ts

@@ -2,8 +2,6 @@
 
 import 'dotenv/config'
 
-import { execSync } from 'node:child_process'
-
 import { createOpenAI } from '@ai-sdk/openai'
 import { confirm, outro, select, spinner, text } from '@clack/prompts'
 import { CoreMessage, generateText } from 'ai'
@@ -14,6 +12,7 @@ import { retro } from 'gradient-string'
 import { z } from 'zod'
 
 import { reactNativePrompt } from './prompt.js'
+import { getApiKey } from './utils.js'
 
 const MessageSchema = z.union([
   z.object({ type: z.literal('select'), content: z.string(), options: z.array(z.string()) }),
@@ -48,56 +47,12 @@ console.log(
   `)
 )
 
-const OPENAI_API_KEY =
-  process.env.OPENAI_API_KEY ||
-  (await (async () => {
-    const apiKey = await text({
-      message: dedent`
-        ${chalk.bold('Please provide your OpenAI API key.')}
-
-        To skip this message, set ${chalk.bold('OPENAI_API_KEY')} env variable, and run again. 
-        
-        You can do it in three ways:
-        - by creating an ${chalk.bold('.env.local')} file (make sure to ${chalk.bold('.gitignore')} it)
-          ${chalk.gray(`\`\`\`
-            OPENAI_API_KEY=<your-key>
-            \`\`\`
-          `)}
-        - by passing it inline:
-          ${chalk.gray(`\`\`\`
-            OPENAI_API_KEY=<your-key> npx cali
-            \`\`\`
-          `)}
-        - by setting it as an env variable in your shell (e.g. in ~/.zshrc or ~/.bashrc):
-          ${chalk.gray(`\`\`\`
-            export OPENAI_API_KEY=<your-key>
-            \`\`\`
-          `)},
-        `,
-      validate: (value) => (value.length > 0 ? undefined : 'Please provide a valid answer.'),
-    })
-
-    if (typeof apiKey === 'symbol') {
-      outro(chalk.gray('Bye!'))
-      process.exit(0)
-    }
-
-    const save = await confirm({
-      message: 'Do you want to save it for future runs in `.env.local`?',
-    })
-
-    if (save) {
-      execSync(`echo "OPENAI_API_KEY=${apiKey}" >> .env.local`)
-      execSync(`echo ".env.local" >> .gitignore`)
-    }
-
-    return apiKey
-  })())
+console.log()
 
 const AI_MODEL = process.env.AI_MODEL || 'gpt-4o'
 
 const openai = createOpenAI({
-  apiKey: OPENAI_API_KEY,
+  apiKey: await getApiKey('OpenAI', 'OPENAI_API_K2EY'),
 })
 
 async function startSession(): Promise<CoreMessage[]> {

packages/cali/src/utils.ts

@@ -0,0 +1,60 @@
+import { execSync } from 'node:child_process'
+
+import { confirm, text } from '@clack/prompts'
+import chalk from 'chalk'
+import dedent from 'dedent'
+
+/**
+ * Get API key from environment variables or prompt user for it.
+ */
+export async function getApiKey(name: string, key: string) {
+  if (key in process.env) {
+    return process.env[key]
+  }
+  return (async () => {
+    let apiKey: string | symbol
+    do {
+      apiKey = await text({
+        message: dedent`
+          ${chalk.bold(`Please provide your ${name} API key.`)}
+
+          To skip this message, set ${chalk.bold(key)} env variable, and run again. 
+          
+          You can do it in three ways:
+          - by creating an ${chalk.bold('.env.local')} file (make sure to ${chalk.bold('.gitignore')} it)
+            ${chalk.gray(`\`\`\`
+              ${key}=<your-key>
+              \`\`\`
+            `)}
+          - by passing it inline:
+            ${chalk.gray(`\`\`\`
+              ${key}=<your-key> npx cali
+              \`\`\`
+            `)}
+          - by setting it as an env variable in your shell (e.g. in ~/.zshrc or ~/.bashrc):
+            ${chalk.gray(`\`\`\`
+              export ${key}=<your-key>
+              \`\`\`
+            `)},
+          `,
+        validate: (value) => (value.length > 0 ? undefined : `Please provide a valid ${key}.`),
+      })
+    } while (typeof apiKey === 'undefined')
+
+    if (typeof apiKey === 'symbol') {
+      outro(chalk.gray('Bye!'))
+      process.exit(0)
+    }
+
+    const save = await confirm({
+      message: `Do you want to save it for future runs in .env.local?`,
+    })
+
+    if (save) {
+      execSync(`echo "${key}=${apiKey}" >> .env.local`)
+      execSync(`echo ".env.local" >> .gitignore`)
+    }
+
+    return apiKey
+  })()
+}