335 lines
12 KiB
TypeScript
335 lines
12 KiB
TypeScript
import { createServer } from "node:http"
|
|
import type { IntegrationOAuthMethodRegistration } from "@opencode-ai/plugin/v2/effect/integration"
|
|
import { define } from "@opencode-ai/plugin/v2/effect/plugin"
|
|
import { Deferred, Effect, Option, Schema, Semaphore, Stream } from "effect"
|
|
import { App } from "../../app"
|
|
import { Credential } from "../../credential"
|
|
import { EventV2 } from "../../event"
|
|
import { Integration } from "../../integration"
|
|
import { ModelV2 } from "../../model"
|
|
import { OauthCallbackPage } from "../../oauth/page"
|
|
import { ProviderV2 } from "../../provider"
|
|
import type { PluginInternal } from "../internal"
|
|
import { OpenAICodex } from "./openai-codex"
|
|
|
|
const clientID = "app_EMoamEEZ73f0CkXaXp7hrann"
|
|
const issuer = "https://auth.openai.com"
|
|
const callbackPort = 1455
|
|
const pollingSafetyMargin = 3000
|
|
const browserMethodID = Integration.MethodID.make("chatgpt-browser")
|
|
const headlessMethodID = Integration.MethodID.make("chatgpt-headless")
|
|
|
|
type Pkce = {
|
|
verifier: string
|
|
challenge: string
|
|
}
|
|
|
|
type TokenResponse = {
|
|
id_token: string
|
|
access_token: string
|
|
refresh_token: string
|
|
expires_in?: number
|
|
}
|
|
|
|
const Claims = Schema.fromJsonString(
|
|
Schema.Struct({
|
|
chatgpt_account_id: Schema.optional(Schema.String),
|
|
organizations: Schema.optional(Schema.Array(Schema.Struct({ id: Schema.String }))),
|
|
"https://api.openai.com/auth": Schema.optional(
|
|
Schema.Struct({ chatgpt_account_id: Schema.optional(Schema.String) }),
|
|
),
|
|
}),
|
|
)
|
|
const decodeClaims = Schema.decodeUnknownOption(Claims)
|
|
|
|
const browser = (app: App.Info) => ({
|
|
integrationID: Integration.ID.make("openai"),
|
|
method: {
|
|
id: browserMethodID,
|
|
type: "oauth",
|
|
label: "ChatGPT Pro/Plus (browser)",
|
|
},
|
|
authorize: () =>
|
|
Effect.gen(function* () {
|
|
const pkce = yield* Effect.promise(generatePKCE)
|
|
const state = base64UrlEncode(crypto.getRandomValues(new Uint8Array(32)).buffer)
|
|
const code = yield* Deferred.make<string, Error>()
|
|
const redirect = `http://localhost:${callbackPort}/auth/callback`
|
|
const server = createServer((request, response) => {
|
|
const url = new URL(request.url ?? "/", `http://localhost:${callbackPort}`)
|
|
if (url.pathname !== "/auth/callback") {
|
|
response.writeHead(404).end("Not found")
|
|
return
|
|
}
|
|
const error = url.searchParams.get("error_description") ?? url.searchParams.get("error")
|
|
const value = url.searchParams.get("code")
|
|
if (error) {
|
|
Effect.runFork(Deferred.fail(code, new Error(error)))
|
|
response
|
|
.writeHead(400, { "Content-Type": "text/html" })
|
|
.end(OauthCallbackPage.error(error, { provider: "ChatGPT" }))
|
|
return
|
|
}
|
|
if (!value || url.searchParams.get("state") !== state) {
|
|
const message = value ? "Invalid OAuth state" : "Missing authorization code"
|
|
Effect.runFork(Deferred.fail(code, new Error(message)))
|
|
response
|
|
.writeHead(400, { "Content-Type": "text/html" })
|
|
.end(OauthCallbackPage.error(message, { provider: "ChatGPT" }))
|
|
return
|
|
}
|
|
Effect.runFork(Deferred.succeed(code, value))
|
|
response.writeHead(200, { "Content-Type": "text/html" }).end(OauthCallbackPage.success({ provider: "ChatGPT" }))
|
|
})
|
|
yield* Effect.callback<void, Error>((resume) => {
|
|
server.once("error", (error) => resume(Effect.fail(error)))
|
|
server.listen(callbackPort, "localhost", () => resume(Effect.void))
|
|
})
|
|
yield* Effect.addFinalizer(() => Effect.sync(() => server.close()))
|
|
return {
|
|
mode: "auto" as const,
|
|
url: authorizeURL(redirect, pkce, state),
|
|
instructions: "Complete authorization in your browser. This window will close automatically.",
|
|
callback: Deferred.await(code).pipe(
|
|
Effect.flatMap((value) => exchange(value, redirect, pkce, app)),
|
|
Effect.map((tokens) => credential(browserMethodID, tokens)),
|
|
),
|
|
}
|
|
}),
|
|
refresh: (value) => refresh(browserMethodID, value, app),
|
|
}) satisfies IntegrationOAuthMethodRegistration
|
|
|
|
const headless = (app: App.Info) => ({
|
|
integrationID: Integration.ID.make("openai"),
|
|
method: {
|
|
id: headlessMethodID,
|
|
type: "oauth",
|
|
label: "ChatGPT Pro/Plus (headless)",
|
|
},
|
|
authorize: () =>
|
|
Effect.gen(function* () {
|
|
const device = yield* request<{ device_auth_id: string; user_code: string; interval: string }>(
|
|
`${issuer}/api/accounts/deviceauth/usercode`,
|
|
{
|
|
method: "POST",
|
|
headers: headers("application/json", app),
|
|
body: JSON.stringify({ client_id: clientID }),
|
|
},
|
|
)
|
|
const interval = Math.max(Number.parseInt(device.interval) || 5, 1) * 1000
|
|
return {
|
|
mode: "auto" as const,
|
|
url: `${issuer}/codex/device`,
|
|
instructions: `Enter code: ${device.user_code}`,
|
|
callback: Effect.gen(function* () {
|
|
while (true) {
|
|
const response = yield* Effect.tryPromise({
|
|
try: (signal) =>
|
|
fetch(`${issuer}/api/accounts/deviceauth/token`, {
|
|
method: "POST",
|
|
headers: headers("application/json", app),
|
|
body: JSON.stringify({ device_auth_id: device.device_auth_id, user_code: device.user_code }),
|
|
signal,
|
|
}),
|
|
catch: (cause) => cause,
|
|
})
|
|
if (response.ok) {
|
|
const data = (yield* Effect.promise(() => response.json())) as {
|
|
authorization_code: string
|
|
code_verifier: string
|
|
}
|
|
return credential(
|
|
headlessMethodID,
|
|
yield* exchange(
|
|
data.authorization_code,
|
|
`${issuer}/deviceauth/callback`,
|
|
{ verifier: data.code_verifier, challenge: "" },
|
|
app,
|
|
),
|
|
)
|
|
}
|
|
if (response.status !== 403 && response.status !== 404) {
|
|
return yield* Effect.fail(new Error(`Device authorization failed: ${response.status}`))
|
|
}
|
|
yield* Effect.sleep(interval + pollingSafetyMargin)
|
|
}
|
|
}),
|
|
}
|
|
}),
|
|
refresh: (value) => refresh(headlessMethodID, value, app),
|
|
}) satisfies IntegrationOAuthMethodRegistration
|
|
|
|
export const OpenAIPlugin = define({
|
|
id: "opencode.provider.openai",
|
|
effect: Effect.fn(function* (ctx) {
|
|
const events = yield* EventV2.Service
|
|
const loading = Semaphore.makeUnsafe(1)
|
|
let chatgpt = false
|
|
|
|
const load = Effect.fn("OpenAIPlugin.load")(function* () {
|
|
const connection = yield* ctx.integration.connection.active("openai")
|
|
const credential = connection
|
|
? yield* ctx.integration.connection.resolve(connection).pipe(Effect.catch(() => Effect.succeed(undefined)))
|
|
: undefined
|
|
chatgpt = OpenAICodex.isChatGPT(credential)
|
|
})
|
|
|
|
yield* ctx.integration.transform((draft) => {
|
|
draft.method.update(browser(ctx.app))
|
|
draft.method.update(headless(ctx.app))
|
|
})
|
|
yield* load()
|
|
yield* ctx.catalog.transform((evt) => {
|
|
for (const item of evt.provider.list()) {
|
|
if (!ProviderV2.isAISDK(item.provider.package)) continue
|
|
if (ProviderV2.packageName(item.provider.package) !== "@ai-sdk/openai") continue
|
|
if (!item.models.has(ModelV2.ID.make("gpt-5-chat-latest"))) continue
|
|
evt.model.update(item.provider.id, ModelV2.ID.make("gpt-5-chat-latest"), (model) => {
|
|
// OpenAIPlugin sends OpenAI models through Responses; this alias is a
|
|
// chat-completions-only model, so hide it only from OpenAI's catalog.
|
|
model.enabled = false
|
|
})
|
|
}
|
|
if (!chatgpt) return
|
|
const item = evt.provider.get(ProviderV2.ID.openai)
|
|
if (!item) return
|
|
for (const model of item.models.values()) {
|
|
// ChatGPT-plan tokens only authorize codex-eligible models, and the
|
|
// subscription covers usage, so hide the rest and zero the cost.
|
|
evt.model.update(item.provider.id, model.id, (draft) => {
|
|
if (Schema.is(Schema.Struct({ mode: Schema.Literal("pro") }))(draft.body?.reasoning)) {
|
|
draft.enabled = false
|
|
return
|
|
}
|
|
if (!OpenAICodex.eligible(draft.modelID ?? draft.id)) {
|
|
draft.enabled = false
|
|
return
|
|
}
|
|
draft.cost = []
|
|
})
|
|
}
|
|
})
|
|
|
|
const refresh = () => loading.withPermit(load().pipe(Effect.andThen(ctx.catalog.reload())))
|
|
yield* events.subscribe(Integration.Event.ConnectionUpdated).pipe(
|
|
Stream.filter((event) => event.data.integrationID === Integration.ID.make("openai")),
|
|
Stream.runForEach(refresh),
|
|
Effect.forkScoped({ startImmediately: true }),
|
|
)
|
|
yield* ctx.aisdk.hook(
|
|
"sdk",
|
|
Effect.fn(function* (evt) {
|
|
if (evt.package !== "@ai-sdk/openai") return
|
|
const mod = yield* Effect.promise(() => import("@ai-sdk/openai"))
|
|
evt.sdk = mod.createOpenAI(evt.options)
|
|
}),
|
|
)
|
|
yield* ctx.aisdk.hook(
|
|
"language",
|
|
Effect.fn(function* (evt) {
|
|
if (evt.model.providerID !== ProviderV2.ID.openai) return
|
|
evt.language = evt.sdk.responses(evt.model.modelID ?? evt.model.id)
|
|
}),
|
|
)
|
|
}),
|
|
} satisfies PluginInternal.InternalPlugin)
|
|
|
|
function headers(contentType: string, app: App.Info) {
|
|
return { "Content-Type": contentType, "User-Agent": App.useragent(app) }
|
|
}
|
|
|
|
function exchange(code: string, redirect: string, pkce: Pkce, app: App.Info) {
|
|
return request<TokenResponse>(`${issuer}/oauth/token`, {
|
|
method: "POST",
|
|
headers: headers("application/x-www-form-urlencoded", app),
|
|
body: new URLSearchParams({
|
|
grant_type: "authorization_code",
|
|
code,
|
|
redirect_uri: redirect,
|
|
client_id: clientID,
|
|
code_verifier: pkce.verifier,
|
|
}).toString(),
|
|
})
|
|
}
|
|
|
|
function refresh(methodID: Integration.MethodID, value: Pick<Credential.OAuth, "refresh" | "metadata">, app: App.Info) {
|
|
return request<TokenResponse>(`${issuer}/oauth/token`, {
|
|
method: "POST",
|
|
headers: headers("application/x-www-form-urlencoded", app),
|
|
body: new URLSearchParams({
|
|
grant_type: "refresh_token",
|
|
refresh_token: value.refresh,
|
|
client_id: clientID,
|
|
}).toString(),
|
|
}).pipe(
|
|
Effect.map((tokens) => {
|
|
const next = credential(methodID, tokens)
|
|
return Credential.OAuth.make({ ...next, metadata: next.metadata ?? value.metadata })
|
|
}),
|
|
)
|
|
}
|
|
|
|
function request<A>(url: string, init: RequestInit) {
|
|
return Effect.tryPromise({
|
|
try: async (signal) => {
|
|
const response = await fetch(url, { ...init, signal })
|
|
if (!response.ok) throw new Error(`Request failed: ${response.status}`)
|
|
return response.json() as Promise<A>
|
|
},
|
|
catch: (cause) => cause,
|
|
})
|
|
}
|
|
|
|
function credential(methodID: Integration.MethodID, tokens: TokenResponse) {
|
|
const accountID = extractAccountID(tokens)
|
|
return Credential.OAuth.make({
|
|
type: "oauth",
|
|
methodID,
|
|
refresh: tokens.refresh_token,
|
|
access: tokens.access_token,
|
|
expires: Date.now() + (tokens.expires_in ?? 3600) * 1000,
|
|
metadata: accountID ? { accountID } : undefined,
|
|
})
|
|
}
|
|
|
|
async function generatePKCE(): Promise<Pkce> {
|
|
const chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-._~"
|
|
const verifier = Array.from(crypto.getRandomValues(new Uint8Array(43)), (byte) => chars[byte % chars.length]).join("")
|
|
const challenge = base64UrlEncode(await crypto.subtle.digest("SHA-256", new TextEncoder().encode(verifier)))
|
|
return { verifier, challenge }
|
|
}
|
|
|
|
function base64UrlEncode(buffer: ArrayBuffer) {
|
|
return Buffer.from(buffer).toString("base64url")
|
|
}
|
|
|
|
function authorizeURL(redirect: string, pkce: Pkce, state: string) {
|
|
return `${issuer}/oauth/authorize?${new URLSearchParams({
|
|
response_type: "code",
|
|
client_id: clientID,
|
|
redirect_uri: redirect,
|
|
scope: "openid profile email offline_access",
|
|
code_challenge: pkce.challenge,
|
|
code_challenge_method: "S256",
|
|
id_token_add_organizations: "true",
|
|
codex_cli_simplified_flow: "true",
|
|
state,
|
|
originator: "opencode",
|
|
})}`
|
|
}
|
|
|
|
function extractAccountID(tokens: TokenResponse) {
|
|
return claim(tokens.id_token) ?? claim(tokens.access_token)
|
|
}
|
|
|
|
function claim(token: string) {
|
|
const part = token.split(".")[1]
|
|
if (!part) return
|
|
const claims = Option.getOrUndefined(decodeClaims(Buffer.from(part, "base64url").toString()))
|
|
if (!claims) return
|
|
return (
|
|
claims.chatgpt_account_id ??
|
|
claims["https://api.openai.com/auth"]?.chatgpt_account_id ??
|
|
claims.organizations?.[0]?.id
|
|
)
|
|
}
|