claude oauth support

This commit is contained in:
Dax Raad
2025-06-05 11:50:54 -04:00
parent b3555cda30
commit 35b03e4cb3
12 changed files with 457 additions and 195 deletions

View File

@@ -0,0 +1,20 @@
import { Server } from "../../server/server"
import fs from "fs/promises"
import path from "path"
import type { CommandModule } from "yargs"
export const GenerateCommand = {
command: "generate",
describe: "Generate OpenAPI and event specs",
handler: async () => {
const specs = await Server.openapi()
const dir = "gen"
await fs.rmdir(dir, { recursive: true }).catch(() => {})
await fs.mkdir(dir, { recursive: true })
await Bun.write(
path.join(dir, "openapi.json"),
JSON.stringify(specs, null, 2),
)
},
} satisfies CommandModule

View File

@@ -0,0 +1,22 @@
import { AuthAnthropic } from "../../auth/anthropic"
import { UI } from "../ui"
// Example: https://claude.ai/oauth/authorize?code=true&client_id=9d1c250a-e61b-44d9-88ed-5944d1962f5e&response_type=code&redirect_uri=https%3A%2F%2Fconsole.anthropic.com%2Foauth%2Fcode%2Fcallback&scope=org%3Acreate_api_key+user%3Aprofile+user%3Ainference&code_challenge=MdFtFgFap23AWDSN0oa3-eaKjQRFE4CaEhXx8M9fHZg&code_challenge_method=S256&state=rKLtaDzm88GSwekyEqdi0wXX-YqIr13tSzYymSzpvfs
import { generatePKCE } from "@openauthjs/openauth/pkce"
export const LoginAnthropicCommand = {
command: "anthropic",
describe: "Login to Anthropic",
handler: async () => {
const { url, verifier } = await AuthAnthropic.authorize()
UI.print("Login to Anthropic")
UI.print("Open the following URL in your browser:")
UI.print(url)
UI.print("")
const code = await UI.input("Paste the authorization code here: ")
await AuthAnthropic.exchange(code, verifier)
},
}

View File

@@ -0,0 +1,140 @@
import type { Argv } from "yargs"
import { App } from "../../app/app"
import { version } from "bun"
import { Bus } from "../../bus"
import { Provider } from "../../provider/provider"
import { Session } from "../../session"
import { Share } from "../../share/share"
import { Message } from "../../session/message"
export const RunCommand = {
command: "run [message..]",
describe: "Run OpenCode with a message",
builder: (yargs: Argv) => {
return yargs
.positional("message", {
describe: "Message to send",
type: "string",
array: true,
default: [],
})
.option("session", {
describe: "Session ID to continue",
type: "string",
})
},
handler: async (args: { message: string[]; session?: string }) => {
const message = args.message.join(" ")
await App.provide({ cwd: process.cwd(), version }, async () => {
await Share.init()
const session = args.session
? await Session.get(args.session)
: await Session.create()
const styles = {
TEXT_HIGHLIGHT: "\x1b[96m",
TEXT_HIGHLIGHT_BOLD: "\x1b[96m\x1b[1m",
TEXT_DIM: "\x1b[90m",
TEXT_DIM_BOLD: "\x1b[90m\x1b[1m",
TEXT_NORMAL: "\x1b[0m",
TEXT_NORMAL_BOLD: "\x1b[1m",
TEXT_WARNING: "\x1b[93m",
TEXT_WARNING_BOLD: "\x1b[93m\x1b[1m",
TEXT_DANGER: "\x1b[91m",
TEXT_DANGER_BOLD: "\x1b[91m\x1b[1m",
TEXT_SUCCESS: "\x1b[92m",
TEXT_SUCCESS_BOLD: "\x1b[92m\x1b[1m",
TEXT_INFO: "\x1b[94m",
TEXT_INFO_BOLD: "\x1b[94m\x1b[1m",
}
let isEmpty = false
function stderr(...message: string[]) {
isEmpty = true
Bun.stderr.write(message.join(" "))
Bun.stderr.write("\n")
}
function empty() {
stderr("" + styles.TEXT_NORMAL)
isEmpty = true
}
stderr(styles.TEXT_HIGHLIGHT_BOLD + "◍ OpenCode", version)
empty()
stderr(styles.TEXT_NORMAL_BOLD + "> ", message)
empty()
stderr(
styles.TEXT_INFO_BOLD +
"~ https://dev.opencode.ai/s?id=" +
session.id.slice(-8),
)
empty()
function printEvent(color: string, type: string, title: string) {
stderr(
color + `|`,
styles.TEXT_NORMAL + styles.TEXT_DIM + ` ${type.padEnd(7, " ")}`,
"",
styles.TEXT_NORMAL + title,
)
}
Bus.subscribe(Message.Event.PartUpdated, async (message) => {
const part = message.properties.part
if (
part.type === "tool-invocation" &&
part.toolInvocation.state === "result"
) {
if (part.toolInvocation.toolName === "opencode_todowrite") return
const messages = await Session.messages(session.id)
const metadata =
messages[messages.length - 1].metadata.tool[
part.toolInvocation.toolCallId
]
const args = part.toolInvocation.args as any
const tool = part.toolInvocation.toolName
if (tool === "opencode_edit")
printEvent(styles.TEXT_SUCCESS_BOLD, "Edit", args.filePath)
if (tool === "opencode_bash")
printEvent(styles.TEXT_WARNING_BOLD, "Execute", args.command)
if (tool === "opencode_read")
printEvent(styles.TEXT_INFO_BOLD, "Read", args.filePath)
if (tool === "opencode_write")
printEvent(styles.TEXT_SUCCESS_BOLD, "Create", args.filePath)
if (tool === "opencode_glob")
printEvent(
styles.TEXT_INFO_BOLD,
"Glob",
args.pattern + (args.path ? " in " + args.path : ""),
)
}
if (part.type === "text") {
if (part.text.includes("\n")) {
empty()
stderr(part.text)
empty()
return
}
printEvent(styles.TEXT_NORMAL_BOLD, "Text", part.text)
}
})
const { providerID, modelID } = await Provider.defaultModel()
const result = await Session.chat({
sessionID: session.id,
providerID,
modelID,
parts: [
{
type: "text",
text: message,
},
],
})
empty()
})
},
}