Add login feature

Author: antonmedv

package.json

@@ -19,7 +19,9 @@
     "chalk": "^5.2.0",
     "indent-string": "^5.0.0",
     "inquirer": "^9.2.7",
-    "minimist": "^1.2.8"
+    "minimist": "^1.2.8",
+    "node-fetch": "^3.3.1",
+    "open": "^9.1.0"
   },
   "devDependencies": {
     "@types/inquirer": "^9.0.3",

src/api.ts

@@ -0,0 +1,69 @@
+import os from 'node:os'
+import {URL} from 'node:url'
+import {createHash} from 'node:crypto'
+import chalk from 'chalk'
+import fetch from 'node-fetch'
+import open from 'open'
+import {randomString, sleep} from './utils.js'
+import {confirm} from './prompt.js'
+import fs from 'node:fs'
+import {webpodLocalConfig} from './env.js'
+
+export async function login() {
+  let baseUrl = 'https://app.webpod.dev'
+  const localDev = createHash('sha1').update(process.env.WEBPOD_LOCAL_DEV ?? '').digest('hex')
+  if (localDev == 'd4f57dba817bd3ddc1101e3498754ed234f08094') {
+    baseUrl = 'http://0.0.0.0:8000'
+  }
+  const resp = await fetch(`${baseUrl}/api/user`, {
+    headers: {
+      Authorization: `Bearer ${process.env.WEBPOD_TOKEN}`,
+      Accept: 'application/json',
+    },
+  })
+  if (resp.status == 200) {
+    const {id} = await resp.json() as { id: number }
+    if (!Number.isFinite(id)) {
+      throw new Error('Invalid user id')
+    }
+    return
+  }
+
+  const randomCode = [randomString(), randomString(), randomString()].join('-')
+  const url = new URL(`${baseUrl}/api-token`)
+  url.searchParams.set('code', randomCode)
+  url.searchParams.set('hostname', os.hostname())
+  console.log(`Please visit ${chalk.underline(url.href)} to login.`)
+  if (await confirm(`Open in browser?`)) {
+    await open(url.href)
+  }
+
+  console.log(chalk.magenta('Waiting for login...'))
+  console.log('Alternatively, you can:')
+  console.log('-  Create API Token: https://app.webpod.dev/user/api-tokens')
+  console.log(`-  Set the ${chalk.bold('WEBPOD_TOKEN')} environment variable to the token.`)
+
+  while (true) {
+    await sleep(3000)
+
+    const resp = await fetch(`${baseUrl}/api-token/${randomCode}`, {
+      headers: {
+        Accept: 'application/json',
+      },
+    })
+    if (resp.status != 200) {
+      continue
+    }
+
+    const json = await resp.json() as { ok: number, token: string }
+    if (!json.ok) {
+      continue
+    }
+
+    fs.appendFileSync(webpodLocalConfig(), `\nglobal.WEBPOD_TOKEN = '${json.token}';\n`)
+    process.env.WEBPOD_TOKEN = json.token
+    console.log(chalk.green('Login successful!'))
+
+    return
+  }
+}

src/cli.ts

@@ -12,6 +12,8 @@ import {ask, confirm, skipPrompts} from './prompt.js'
 import fs from 'node:fs'
 import {handleError, StopError} from './error.js'
 import {detectFramework, setupFramework} from './framework.js'
+import {login} from './api.js'
+import {loadUserConfig} from './env.js'
 
 const {cyan, grey, green, bold} = chalk
 
@@ -54,6 +56,9 @@ await async function main() {
     process.exit(1)
   }
 
+  await loadUserConfig()
+  await login()
+
   let task, remoteUser, hostname, become
   if (argv._.length == 2) {
     task = argv._[0];

src/env.ts

@@ -0,0 +1,17 @@
+import path from 'node:path'
+import os from 'node:os'
+import fs from 'node:fs'
+
+export function webpodLocalConfig() {
+  return path.join(os.homedir(), '.webpod.js')
+}
+
+export async function loadUserConfig() {
+  if (fs.existsSync(webpodLocalConfig())) {
+    await import(webpodLocalConfig())
+    const token = (global as any).WEBPOD_TOKEN
+    if (token && !process.env.WEBPOD_TOKEN) {
+      process.env.WEBPOD_TOKEN = token
+    }
+  }
+}

src/utils.ts

@@ -116,3 +116,11 @@ export function secondsToHumanReadableFormat(seconds: number): string {
 export function humanPath(...parts: string[]): string {
   return path.resolve(path.join(...parts)).replace(os.homedir(), '~')
 }
+
+export function randomString(): string {
+  return Math.random().toString(36).slice(2)
+}
+
+export function sleep(ms: number): Promise<void> {
+  return new Promise(resolve => setTimeout(resolve, ms))
+}