|
@@ -1,5 +1,3 @@
|
|
|
-import { AuthAnthropic } from "../../auth/anthropic"
|
|
|
|
|
-import { AuthCopilot } from "../../auth/copilot"
|
|
|
|
|
import { Auth } from "../../auth"
|
|
import { Auth } from "../../auth"
|
|
|
import { cmd } from "./cmd"
|
|
import { cmd } from "./cmd"
|
|
|
import * as prompts from "@clack/prompts"
|
|
import * as prompts from "@clack/prompts"
|
|
@@ -10,6 +8,8 @@ import { map, pipe, sortBy, values } from "remeda"
|
|
|
import path from "path"
|
|
import path from "path"
|
|
|
import os from "os"
|
|
import os from "os"
|
|
|
import { Global } from "../../global"
|
|
import { Global } from "../../global"
|
|
|
|
|
+import { Plugin } from "../../plugin"
|
|
|
|
|
+import { App } from "../../app/app"
|
|
|
|
|
|
|
|
export const AuthCommand = cmd({
|
|
export const AuthCommand = cmd({
|
|
|
command: "auth",
|
|
command: "auth",
|
|
@@ -75,242 +75,179 @@ export const AuthLoginCommand = cmd({
|
|
|
type: "string",
|
|
type: "string",
|
|
|
}),
|
|
}),
|
|
|
async handler(args) {
|
|
async handler(args) {
|
|
|
- UI.empty()
|
|
|
|
|
- prompts.intro("Add credential")
|
|
|
|
|
- if (args.url) {
|
|
|
|
|
- const wellknown = await fetch(`${args.url}/.well-known/opencode`).then((x) => x.json())
|
|
|
|
|
- prompts.log.info(`Running \`${wellknown.auth.command.join(" ")}\``)
|
|
|
|
|
- const proc = Bun.spawn({
|
|
|
|
|
- cmd: wellknown.auth.command,
|
|
|
|
|
- stdout: "pipe",
|
|
|
|
|
- })
|
|
|
|
|
- const exit = await proc.exited
|
|
|
|
|
- if (exit !== 0) {
|
|
|
|
|
- prompts.log.error("Failed")
|
|
|
|
|
|
|
+ await App.provide({ cwd: process.cwd() }, async () => {
|
|
|
|
|
+ UI.empty()
|
|
|
|
|
+ prompts.intro("Add credential")
|
|
|
|
|
+ if (args.url) {
|
|
|
|
|
+ const wellknown = await fetch(`${args.url}/.well-known/opencode`).then((x) => x.json())
|
|
|
|
|
+ prompts.log.info(`Running \`${wellknown.auth.command.join(" ")}\``)
|
|
|
|
|
+ const proc = Bun.spawn({
|
|
|
|
|
+ cmd: wellknown.auth.command,
|
|
|
|
|
+ stdout: "pipe",
|
|
|
|
|
+ })
|
|
|
|
|
+ const exit = await proc.exited
|
|
|
|
|
+ if (exit !== 0) {
|
|
|
|
|
+ prompts.log.error("Failed")
|
|
|
|
|
+ prompts.outro("Done")
|
|
|
|
|
+ return
|
|
|
|
|
+ }
|
|
|
|
|
+ const token = await new Response(proc.stdout).text()
|
|
|
|
|
+ await Auth.set(args.url, {
|
|
|
|
|
+ type: "wellknown",
|
|
|
|
|
+ key: wellknown.auth.env,
|
|
|
|
|
+ token: token.trim(),
|
|
|
|
|
+ })
|
|
|
|
|
+ prompts.log.success("Logged into " + args.url)
|
|
|
prompts.outro("Done")
|
|
prompts.outro("Done")
|
|
|
return
|
|
return
|
|
|
}
|
|
}
|
|
|
- const token = await new Response(proc.stdout).text()
|
|
|
|
|
- await Auth.set(args.url, {
|
|
|
|
|
- type: "wellknown",
|
|
|
|
|
- key: wellknown.auth.env,
|
|
|
|
|
- token: token.trim(),
|
|
|
|
|
- })
|
|
|
|
|
- prompts.log.success("Logged into " + args.url)
|
|
|
|
|
- prompts.outro("Done")
|
|
|
|
|
- return
|
|
|
|
|
- }
|
|
|
|
|
- await ModelsDev.refresh().catch(() => {})
|
|
|
|
|
- const providers = await ModelsDev.get()
|
|
|
|
|
- const priority: Record<string, number> = {
|
|
|
|
|
- anthropic: 0,
|
|
|
|
|
- "github-copilot": 1,
|
|
|
|
|
- openai: 2,
|
|
|
|
|
- google: 3,
|
|
|
|
|
- openrouter: 4,
|
|
|
|
|
- vercel: 5,
|
|
|
|
|
- }
|
|
|
|
|
- let provider = await prompts.autocomplete({
|
|
|
|
|
- message: "Select provider",
|
|
|
|
|
- maxItems: 8,
|
|
|
|
|
- options: [
|
|
|
|
|
- ...pipe(
|
|
|
|
|
- providers,
|
|
|
|
|
- values(),
|
|
|
|
|
- sortBy(
|
|
|
|
|
- (x) => priority[x.id] ?? 99,
|
|
|
|
|
- (x) => x.name ?? x.id,
|
|
|
|
|
- ),
|
|
|
|
|
- map((x) => ({
|
|
|
|
|
- label: x.name,
|
|
|
|
|
- value: x.id,
|
|
|
|
|
- hint: priority[x.id] === 0 ? "recommended" : undefined,
|
|
|
|
|
- })),
|
|
|
|
|
- ),
|
|
|
|
|
- {
|
|
|
|
|
- value: "other",
|
|
|
|
|
- label: "Other",
|
|
|
|
|
- },
|
|
|
|
|
- ],
|
|
|
|
|
- })
|
|
|
|
|
-
|
|
|
|
|
- if (prompts.isCancel(provider)) throw new UI.CancelledError()
|
|
|
|
|
-
|
|
|
|
|
- if (provider === "other") {
|
|
|
|
|
- provider = await prompts.text({
|
|
|
|
|
- message: "Enter provider id",
|
|
|
|
|
- validate: (x) => (x && x.match(/^[0-9a-z-]+$/) ? undefined : "a-z, 0-9 and hyphens only"),
|
|
|
|
|
- })
|
|
|
|
|
- if (prompts.isCancel(provider)) throw new UI.CancelledError()
|
|
|
|
|
- provider = provider.replace(/^@ai-sdk\//, "")
|
|
|
|
|
- if (prompts.isCancel(provider)) throw new UI.CancelledError()
|
|
|
|
|
- prompts.log.warn(
|
|
|
|
|
- `This only stores a credential for ${provider} - you will need configure it in opencode.json, check the docs for examples.`,
|
|
|
|
|
- )
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- if (provider === "amazon-bedrock") {
|
|
|
|
|
- prompts.log.info(
|
|
|
|
|
- "Amazon bedrock can be configured with standard AWS environment variables like AWS_BEARER_TOKEN_BEDROCK, AWS_PROFILE or AWS_ACCESS_KEY_ID",
|
|
|
|
|
- )
|
|
|
|
|
- prompts.outro("Done")
|
|
|
|
|
- return
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- if (provider === "anthropic") {
|
|
|
|
|
- const method = await prompts.select({
|
|
|
|
|
- message: "Login method",
|
|
|
|
|
|
|
+ await ModelsDev.refresh().catch(() => {})
|
|
|
|
|
+ const providers = await ModelsDev.get()
|
|
|
|
|
+ const priority: Record<string, number> = {
|
|
|
|
|
+ anthropic: 0,
|
|
|
|
|
+ "github-copilot": 1,
|
|
|
|
|
+ openai: 2,
|
|
|
|
|
+ google: 3,
|
|
|
|
|
+ openrouter: 4,
|
|
|
|
|
+ vercel: 5,
|
|
|
|
|
+ }
|
|
|
|
|
+ let provider = await prompts.autocomplete({
|
|
|
|
|
+ message: "Select provider",
|
|
|
|
|
+ maxItems: 8,
|
|
|
options: [
|
|
options: [
|
|
|
|
|
+ ...pipe(
|
|
|
|
|
+ providers,
|
|
|
|
|
+ values(),
|
|
|
|
|
+ sortBy(
|
|
|
|
|
+ (x) => priority[x.id] ?? 99,
|
|
|
|
|
+ (x) => x.name ?? x.id,
|
|
|
|
|
+ ),
|
|
|
|
|
+ map((x) => ({
|
|
|
|
|
+ label: x.name,
|
|
|
|
|
+ value: x.id,
|
|
|
|
|
+ hint: priority[x.id] === 0 ? "recommended" : undefined,
|
|
|
|
|
+ })),
|
|
|
|
|
+ ),
|
|
|
{
|
|
{
|
|
|
- label: "Claude Pro/Max",
|
|
|
|
|
- value: "max",
|
|
|
|
|
- },
|
|
|
|
|
- {
|
|
|
|
|
- label: "Create API Key",
|
|
|
|
|
- value: "console",
|
|
|
|
|
- },
|
|
|
|
|
- {
|
|
|
|
|
- label: "Manually enter API Key",
|
|
|
|
|
- value: "api",
|
|
|
|
|
|
|
+ value: "other",
|
|
|
|
|
+ label: "Other",
|
|
|
},
|
|
},
|
|
|
],
|
|
],
|
|
|
})
|
|
})
|
|
|
- if (prompts.isCancel(method)) throw new UI.CancelledError()
|
|
|
|
|
-
|
|
|
|
|
- if (method === "max") {
|
|
|
|
|
- // some weird bug where program exits without this
|
|
|
|
|
- await new Promise((resolve) => setTimeout(resolve, 10))
|
|
|
|
|
- const { url, verifier } = await AuthAnthropic.authorize("max")
|
|
|
|
|
- prompts.note("Trying to open browser...")
|
|
|
|
|
- try {
|
|
|
|
|
- await open(url)
|
|
|
|
|
- } catch (e) {
|
|
|
|
|
- prompts.log.error(
|
|
|
|
|
- "Failed to open browser perhaps you are running without a display or X server, please open the following URL in your browser:",
|
|
|
|
|
- )
|
|
|
|
|
- }
|
|
|
|
|
- prompts.log.info(url)
|
|
|
|
|
|
|
|
|
|
- const code = await prompts.text({
|
|
|
|
|
- message: "Paste the authorization code here: ",
|
|
|
|
|
- validate: (x) => (x && x.length > 0 ? undefined : "Required"),
|
|
|
|
|
- })
|
|
|
|
|
- if (prompts.isCancel(code)) throw new UI.CancelledError()
|
|
|
|
|
|
|
+ if (prompts.isCancel(provider)) throw new UI.CancelledError()
|
|
|
|
|
|
|
|
- try {
|
|
|
|
|
- const credentials = await AuthAnthropic.exchange(code, verifier)
|
|
|
|
|
- await Auth.set("anthropic", {
|
|
|
|
|
- type: "oauth",
|
|
|
|
|
- refresh: credentials.refresh,
|
|
|
|
|
- access: credentials.access,
|
|
|
|
|
- expires: credentials.expires,
|
|
|
|
|
|
|
+ const plugin = await Plugin.list().then((x) => x.find((x) => x.auth?.provider === provider))
|
|
|
|
|
+ if (plugin && plugin.auth) {
|
|
|
|
|
+ let index = 0
|
|
|
|
|
+ if (plugin.auth.methods.length > 1) {
|
|
|
|
|
+ const method = await prompts.select({
|
|
|
|
|
+ message: "Login method",
|
|
|
|
|
+ options: [
|
|
|
|
|
+ ...plugin.auth.methods.map((x, index) => ({
|
|
|
|
|
+ label: x.label,
|
|
|
|
|
+ value: index.toString(),
|
|
|
|
|
+ })),
|
|
|
|
|
+ ],
|
|
|
})
|
|
})
|
|
|
- prompts.log.success("Login successful")
|
|
|
|
|
- } catch {
|
|
|
|
|
- prompts.log.error("Invalid code")
|
|
|
|
|
|
|
+ if (prompts.isCancel(method)) throw new UI.CancelledError()
|
|
|
|
|
+ index = parseInt(method)
|
|
|
}
|
|
}
|
|
|
- prompts.outro("Done")
|
|
|
|
|
- return
|
|
|
|
|
- }
|
|
|
|
|
|
|
+ const method = plugin.auth.methods[index]
|
|
|
|
|
+ if (method.type === "oauth") {
|
|
|
|
|
+ await new Promise((resolve) => setTimeout(resolve, 10))
|
|
|
|
|
+ const authorize = await method.authorize()
|
|
|
|
|
|
|
|
- if (method === "console") {
|
|
|
|
|
- // some weird bug where program exits without this
|
|
|
|
|
- await new Promise((resolve) => setTimeout(resolve, 10))
|
|
|
|
|
- const { url, verifier } = await AuthAnthropic.authorize("console")
|
|
|
|
|
- prompts.note("Trying to open browser...")
|
|
|
|
|
- try {
|
|
|
|
|
- await open(url)
|
|
|
|
|
- } catch (e) {
|
|
|
|
|
- prompts.log.error(
|
|
|
|
|
- "Failed to open browser perhaps you are running without a display or X server, please open the following URL in your browser:",
|
|
|
|
|
- )
|
|
|
|
|
- }
|
|
|
|
|
- prompts.log.info(url)
|
|
|
|
|
-
|
|
|
|
|
- const code = await prompts.text({
|
|
|
|
|
- message: "Paste the authorization code here: ",
|
|
|
|
|
- validate: (x) => (x && x.length > 0 ? undefined : "Required"),
|
|
|
|
|
- })
|
|
|
|
|
- if (prompts.isCancel(code)) throw new UI.CancelledError()
|
|
|
|
|
|
|
+ if (authorize.url) {
|
|
|
|
|
+ try {
|
|
|
|
|
+ await open(authorize.url)
|
|
|
|
|
+ } catch (e) {}
|
|
|
|
|
+ prompts.log.info("Go to: " + authorize.url)
|
|
|
|
|
+ }
|
|
|
|
|
|
|
|
- try {
|
|
|
|
|
- const credentials = await AuthAnthropic.exchange(code, verifier)
|
|
|
|
|
- const accessToken = credentials.access
|
|
|
|
|
- const response = await fetch("https://api.anthropic.com/api/oauth/claude_cli/create_api_key", {
|
|
|
|
|
- method: "POST",
|
|
|
|
|
- headers: {
|
|
|
|
|
- Authorization: `Bearer ${accessToken}`,
|
|
|
|
|
- "Content-Type": "application/x-www-form-urlencoded",
|
|
|
|
|
- Accept: "application/json, text/plain, */*",
|
|
|
|
|
- },
|
|
|
|
|
- })
|
|
|
|
|
- if (!response.ok) {
|
|
|
|
|
- throw new Error("Failed to create API key")
|
|
|
|
|
|
|
+ if (authorize.method === "auto") {
|
|
|
|
|
+ if (authorize.instructions) {
|
|
|
|
|
+ prompts.log.info(authorize.instructions)
|
|
|
|
|
+ }
|
|
|
|
|
+ const spinner = prompts.spinner()
|
|
|
|
|
+ spinner.start("Waiting for authorization...")
|
|
|
|
|
+ const result = await authorize.callback()
|
|
|
|
|
+ if (result.type === "failed") {
|
|
|
|
|
+ spinner.stop("Failed to authorize", 1)
|
|
|
|
|
+ }
|
|
|
|
|
+ if (result.type === "success") {
|
|
|
|
|
+ await Auth.set(provider, {
|
|
|
|
|
+ type: "oauth",
|
|
|
|
|
+ refresh: result.refresh,
|
|
|
|
|
+ access: result.access,
|
|
|
|
|
+ expires: result.expires,
|
|
|
|
|
+ })
|
|
|
|
|
+ spinner.stop("Login successful")
|
|
|
|
|
+ }
|
|
|
}
|
|
}
|
|
|
- const json = await response.json()
|
|
|
|
|
- await Auth.set("anthropic", {
|
|
|
|
|
- type: "api",
|
|
|
|
|
- key: json.raw_key,
|
|
|
|
|
- })
|
|
|
|
|
|
|
|
|
|
- prompts.log.success("Login successful - API key created and saved")
|
|
|
|
|
- } catch (error) {
|
|
|
|
|
- prompts.log.error("Invalid code or failed to create API key")
|
|
|
|
|
|
|
+ if (authorize.method === "code") {
|
|
|
|
|
+ const code = await prompts.text({
|
|
|
|
|
+ message: "Paste the authorization code here: ",
|
|
|
|
|
+ validate: (x) => (x && x.length > 0 ? undefined : "Required"),
|
|
|
|
|
+ })
|
|
|
|
|
+ if (prompts.isCancel(code)) throw new UI.CancelledError()
|
|
|
|
|
+ const result = await authorize.callback(code)
|
|
|
|
|
+ if (result.type === "failed") {
|
|
|
|
|
+ prompts.log.error("Failed to authorize")
|
|
|
|
|
+ }
|
|
|
|
|
+ if (result.type === "success") {
|
|
|
|
|
+ await Auth.set(provider, {
|
|
|
|
|
+ type: "oauth",
|
|
|
|
|
+ refresh: result.refresh,
|
|
|
|
|
+ access: result.access,
|
|
|
|
|
+ expires: result.expires,
|
|
|
|
|
+ })
|
|
|
|
|
+ prompts.log.success("Login successful")
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+ prompts.outro("Done")
|
|
|
|
|
+ return
|
|
|
}
|
|
}
|
|
|
- prompts.outro("Done")
|
|
|
|
|
- return
|
|
|
|
|
}
|
|
}
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- const copilot = await AuthCopilot()
|
|
|
|
|
- if (provider === "github-copilot" && copilot) {
|
|
|
|
|
- await new Promise((resolve) => setTimeout(resolve, 10))
|
|
|
|
|
- const deviceInfo = await copilot.authorize()
|
|
|
|
|
|
|
|
|
|
- prompts.note(`Please visit: ${deviceInfo.verification}\nEnter code: ${deviceInfo.user}`)
|
|
|
|
|
-
|
|
|
|
|
- const spinner = prompts.spinner()
|
|
|
|
|
- spinner.start("Waiting for authorization...")
|
|
|
|
|
|
|
+ if (provider === "other") {
|
|
|
|
|
+ provider = await prompts.text({
|
|
|
|
|
+ message: "Enter provider id",
|
|
|
|
|
+ validate: (x) => (x && x.match(/^[0-9a-z-]+$/) ? undefined : "a-z, 0-9 and hyphens only"),
|
|
|
|
|
+ })
|
|
|
|
|
+ if (prompts.isCancel(provider)) throw new UI.CancelledError()
|
|
|
|
|
+ provider = provider.replace(/^@ai-sdk\//, "")
|
|
|
|
|
+ if (prompts.isCancel(provider)) throw new UI.CancelledError()
|
|
|
|
|
+ prompts.log.warn(
|
|
|
|
|
+ `This only stores a credential for ${provider} - you will need configure it in opencode.json, check the docs for examples.`,
|
|
|
|
|
+ )
|
|
|
|
|
+ }
|
|
|
|
|
|
|
|
- while (true) {
|
|
|
|
|
- await new Promise((resolve) => setTimeout(resolve, deviceInfo.interval * 1000))
|
|
|
|
|
- const response = await copilot.poll(deviceInfo.device)
|
|
|
|
|
- if (response.status === "pending") continue
|
|
|
|
|
- if (response.status === "success") {
|
|
|
|
|
- await Auth.set("github-copilot", {
|
|
|
|
|
- type: "oauth",
|
|
|
|
|
- refresh: response.refresh,
|
|
|
|
|
- access: response.access,
|
|
|
|
|
- expires: response.expires,
|
|
|
|
|
- })
|
|
|
|
|
- spinner.stop("Login successful")
|
|
|
|
|
- break
|
|
|
|
|
- }
|
|
|
|
|
- if (response.status === "failed") {
|
|
|
|
|
- spinner.stop("Failed to authorize", 1)
|
|
|
|
|
- break
|
|
|
|
|
- }
|
|
|
|
|
|
|
+ if (provider === "amazon-bedrock") {
|
|
|
|
|
+ prompts.log.info(
|
|
|
|
|
+ "Amazon bedrock can be configured with standard AWS environment variables like AWS_BEARER_TOKEN_BEDROCK, AWS_PROFILE or AWS_ACCESS_KEY_ID",
|
|
|
|
|
+ )
|
|
|
|
|
+ prompts.outro("Done")
|
|
|
|
|
+ return
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- prompts.outro("Done")
|
|
|
|
|
- return
|
|
|
|
|
- }
|
|
|
|
|
|
|
+ if (provider === "vercel") {
|
|
|
|
|
+ prompts.log.info("You can create an api key in the dashboard")
|
|
|
|
|
+ }
|
|
|
|
|
|
|
|
- if (provider === "vercel") {
|
|
|
|
|
- prompts.log.info("You can create an api key in the dashboard")
|
|
|
|
|
- }
|
|
|
|
|
|
|
+ const key = await prompts.password({
|
|
|
|
|
+ message: "Enter your API key",
|
|
|
|
|
+ validate: (x) => (x && x.length > 0 ? undefined : "Required"),
|
|
|
|
|
+ })
|
|
|
|
|
+ if (prompts.isCancel(key)) throw new UI.CancelledError()
|
|
|
|
|
+ await Auth.set(provider, {
|
|
|
|
|
+ type: "api",
|
|
|
|
|
+ key,
|
|
|
|
|
+ })
|
|
|
|
|
|
|
|
- const key = await prompts.password({
|
|
|
|
|
- message: "Enter your API key",
|
|
|
|
|
- validate: (x) => (x && x.length > 0 ? undefined : "Required"),
|
|
|
|
|
- })
|
|
|
|
|
- if (prompts.isCancel(key)) throw new UI.CancelledError()
|
|
|
|
|
- await Auth.set(provider, {
|
|
|
|
|
- type: "api",
|
|
|
|
|
- key,
|
|
|
|
|
|
|
+ prompts.outro("Done")
|
|
|
})
|
|
})
|
|
|
-
|
|
|
|
|
- prompts.outro("Done")
|
|
|
|
|
},
|
|
},
|
|
|
})
|
|
})
|
|
|
|
|
|