automate the process of adding an API token to .env (#5)

.gitignore

@@ -1,2 +1,3 @@
 template/package-lock.json
-my-replicate-app
+my-replicate-app
+foo

index.mjs

@@ -5,15 +5,18 @@ import path, { dirname } from 'path'
 import { execSync } from 'child_process'
 import minimist from 'minimist'
 import JSON5 from 'json5'
+import readline from 'readline'
+import { promisify } from 'util'
+import open from 'open'
 
 // ESM shenanigans for dealing with local files
 import { fileURLToPath } from 'url'
 import { getModel, getModelInputs, getModelNameWithVersion } from './lib/models.js'
+const readlineSync = await import('readline-sync').then(module => module.default)
 const __filename = fileURLToPath(import.meta.url)
 const __dirname = dirname(__filename)
 
 const argv = minimist(process.argv.slice(2))
-
 const targetAppName = argv._[0] || 'my-replicate-app'
 const modelName = argv.model || 'stability-ai/sdxl'
 
@@ -35,6 +38,25 @@ const templateDir = path.join(__dirname, 'template')
 const targetDir = path.join(process.cwd(), targetAppName)
 fs.cpSync(templateDir, targetDir, { recursive: true })
 
+// Get env from existing REPLICATE_API_TOKEN, or prompt user to get it from the browser
+const envFile = path.join(targetDir, '.env')
+
+if (process.env.REPLICATE_API_TOKEN) {
+  fs.writeFileSync(envFile, `REPLICATE_API_TOKEN=${process.env.REPLICATE_API_TOKEN}`)
+  console.log('Adding API token to .env file')
+} else {
+  console.log('API token not found in environment.')
+  const rl = readline.createInterface({ input: process.stdin, output: process.stdout })
+  const question = promisify(rl.question).bind(rl)
+  const answer = await question('Open your browser to copy a Replicate API token? (Y/n) ')
+  if (answer.toLowerCase() === 'y' || answer === '') {
+    await open('https://replicate.com/account')
+    const token = readlineSync.question('Paste your API token here: ', { hideEchoBack: true })
+    fs.writeFileSync(envFile, `REPLICATE_API_TOKEN=${token}`)
+    console.log('API token written to .env file')
+  }
+}
+
 console.log('Setting package name...')
 execSync(`npm pkg set name=${targetAppName}`, { cwd: targetDir, stdio: 'ignore' })
 

package.json

@@ -21,6 +21,8 @@
   "dependencies": {
     "all-the-public-replicate-models": "^1.97.0",
     "json5": "^2.2.3",
-    "minimist": "^1.2.8"
+    "minimist": "^1.2.8",
+    "open": "^10.0.3",
+    "readline-sync": "^1.4.10"
   }
 }

template/.gitignore

@@ -0,0 +1,2 @@
+.env
+.npmrc

template/index.js

@@ -1,4 +1,14 @@
 import Replicate from 'replicate'
+import dotenv from 'dotenv'
+dotenv.config()
+
+if (!process.env.REPLICATE_API_TOKEN) {
+  console.error('REPLICATE_API_TOKEN not found in environment')
+  console.error('To copy a token, go to https://replicate.com/account')
+  console.error('Then add your token to a local .env file like this:')
+  console.error('echo "REPLICATE_API_TOKEN=your-token" > .env')
+  process.exit()
+}
 
 const replicate = new Replicate({ auth: process.env.REPLICATE_API_TOKEN })
 const model = '{{MODEL}}'

template/package.json

@@ -9,6 +9,7 @@
     "start": "node index.js"
   },
   "dependencies": {
+    "dotenv": "^16.4.1",
     "replicate": "*"
   }
 }