Something went wrong. Try again.
opencode v2 mirror
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482import { Plugin } from "@opencode/plugin/effect"import type { IntegrationMethod } from "@opencode/plugin/effect/integration"import { Agent } from "@opencode/core/agent"import { Credential } from "@opencode/core/credential"import { Integration } from "@opencode/core/integration"import { Location } from "@opencode/core/location"import { Model } from "@opencode/core/model"import { Project } from "@opencode/core/project"import { Provider } from "@opencode/core/provider"import { AbsolutePath } from "@opencode/core/schema"import { WebSearch } from "@opencode/core/websearch"import { Effect, Stream } from "effect"
type Overrides = Partial<Omit<Plugin.Context, "options" | "session">> & { readonly session?: Partial<Plugin.Context["session"]>}export function host(overrides: Overrides = {}): Plugin.Context { return { app: overrides.app ?? { name: "test", version: "test", channel: "test" }, location: overrides.location ?? new Location.Info({ directory: AbsolutePath.make("/workspace"), project: { id: Project.ID.global, directory: AbsolutePath.make("/workspace"), canonical: AbsolutePath.make("/workspace"), }, }), options: {}, rpc: overrides.rpc ?? Object.assign( () => { throw new Error("unused rpc.client") }, { register: () => Effect.die("unused rpc.register") }, ), agent: overrides.agent ?? { get: () => Effect.die("unused agent.get"), list: () => Effect.die("unused agent.list"), transform: () => Effect.die("unused agent.transform"), reload: () => Effect.die("unused agent.reload"), }, aisdk: overrides.aisdk ?? { hook: () => Effect.die("unused aisdk.hook"), }, provider: overrides.provider ?? { list: () => Effect.die("unused provider.list"), get: () => Effect.die("unused provider.get"), transform: () => Effect.die("unused provider.transform"), reload: () => Effect.die("unused provider.reload"), }, model: overrides.model ?? { list: () => Effect.die("unused model.list"), default: () => Effect.die("unused model.default"), transform: () => Effect.die("unused model.transform"), reload: () => Effect.die("unused model.reload"), }, command: overrides.command ?? { list: () => Effect.die("unused command.list"), transform: () => Effect.die("unused command.transform"), reload: () => Effect.die("unused command.reload"), }, event: overrides.event ?? { subscribe: () => Stream.empty, }, experimental: overrides.experimental ?? { terminal: { read: () => Effect.die("unused experimental.terminal.read"), }, }, generate: overrides.generate ?? { text: () => Effect.die("unused generate.text"), }, integration: overrides.integration ?? { list: () => Effect.die("unused integration.list"), get: () => Effect.die("unused integration.get"), connect: { key: () => Effect.die("unused integration.connect.key"), }, oauth: { connect: () => Effect.die("unused integration.oauth.connect"), status: () => Effect.die("unused integration.oauth.status"), complete: () => Effect.die("unused integration.oauth.complete"), cancel: () => Effect.die("unused integration.oauth.cancel"), }, command: { connect: () => Effect.die("unused integration.command.connect"), status: () => Effect.die("unused integration.command.status"), cancel: () => Effect.die("unused integration.command.cancel"), }, transform: () => Effect.die("unused integration.transform"), reload: () => Effect.die("unused integration.reload"), connection: { active: () => Effect.die("unused integration.connection.active"), resolve: () => Effect.die("unused integration.connection.resolve"), }, }, mcp: overrides.mcp ?? { list: () => Effect.die("unused mcp.list"), transform: () => Effect.die("unused mcp.transform"), reload: () => Effect.die("unused mcp.reload"), }, permission: overrides.permission ?? { hook: () => Effect.die("unused permission.hook"), list: () => Effect.die("unused permission.list"), get: () => Effect.die("unused permission.get"), reply: () => Effect.die("unused permission.reply"), }, plugin: overrides.plugin ?? { list: () => Effect.die("unused plugin.list"), }, reference: overrides.reference ?? { list: () => Effect.die("unused reference.list"), transform: () => Effect.die("unused reference.transform"), reload: () => Effect.die("unused reference.reload"), }, skill: overrides.skill ?? { list: () => Effect.die("unused skill.list"), transform: () => Effect.die("unused skill.transform"), reload: () => Effect.die("unused skill.reload"), }, storage: overrides.storage ?? { get: () => Effect.die("unused storage.get"), set: () => Effect.die("unused storage.set"), remove: () => Effect.die("unused storage.remove"), scan: () => Effect.die("unused storage.scan"), }, shell: overrides.shell ?? { hook: () => Effect.die("unused shell.hook"), }, tool: overrides.tool ?? { transform: () => Effect.die("unused tool.transform"), reload: () => Effect.die("unused tool.reload"), list: () => Effect.die("unused tool.list"), hook: () => Effect.die("unused tool.hook"), }, vcs: overrides.vcs ?? { base: () => Effect.die("unused vcs.base"), get: () => Effect.die("unused vcs.get"), branch: { list: () => Effect.die("unused vcs.branch.list"), }, status: () => Effect.die("unused vcs.status"), diff: () => Effect.die("unused vcs.diff"), transform: () => Effect.die("unused vcs.transform"), reload: () => Effect.die("unused vcs.reload"), }, worktree: overrides.worktree ?? { list: () => Effect.die("unused worktree.list"), create: () => Effect.die("unused worktree.create"), remove: () => Effect.die("unused worktree.remove"), refresh: () => Effect.die("unused worktree.refresh"), transform: () => Effect.die("unused worktree.transform"), reload: () => Effect.die("unused worktree.reload"), }, websearch: overrides.websearch ?? { providers: () => Effect.die("unused websearch.providers"), query: () => Effect.die("unused websearch.query"), transform: () => Effect.die("unused websearch.transform"), reload: () => Effect.die("unused websearch.reload"), }, session: { hook: overrides.session?.hook ?? (() => Effect.die("unused session.hook")), create: overrides.session?.create ?? (() => Effect.die("unused session.create")), get: overrides.session?.get ?? (() => Effect.die("unused session.get")), switchAgent: overrides.session?.switchAgent ?? (() => Effect.die("unused session.switchAgent")), switchModel: overrides.session?.switchModel ?? (() => Effect.die("unused session.switchModel")), prompt: overrides.session?.prompt ?? (() => Effect.die("unused session.prompt")), generate: overrides.session?.generate ?? (() => Effect.die("unused session.generate")), command: overrides.session?.command ?? (() => Effect.die("unused session.command")), update: overrides.session?.update ?? (() => Effect.die("unused session.update")), move: overrides.session?.move ?? (() => Effect.die("unused session.move")), synthetic: overrides.session?.synthetic ?? (() => Effect.die("unused session.synthetic")), interrupt: overrides.session?.interrupt ?? (() => Effect.die("unused session.interrupt")), wait: overrides.session?.wait ?? (() => Effect.die("unused session.wait")), context: overrides.session?.context ?? (() => Effect.die("unused session.context")), }, }}
export function agentHost(agent: Agent.Interface): Plugin.Context["agent"] { return { get: (input) => agent.get(input.agentID).pipe( Effect.flatMap((value) => value ? Effect.succeed({ location: new Location.Info({ directory: AbsolutePath.make("/"), project: { id: Project.ID.make("test"), directory: AbsolutePath.make("/"), canonical: AbsolutePath.make("/"), }, }), data: agentInfo(value), }) : Effect.fail(new Error(`Agent not found: ${input.agentID}`)), ), ), list: () => Effect.die("unused agent.list"), reload: agent.reload, transform: (callback) => agent.transform((editor) => callback({ list: () => editor.list().map(agentInfo), get: (id) => { const value = editor.get(Agent.ID.make(id)) return value && agentInfo(value) }, default: (id) => editor.default(id === undefined ? undefined : Agent.ID.make(id)), update: (id, update) => editor.update(Agent.ID.make(id), (value) => { const current = agentInfo(value) update(current) Object.assign(value, current, { id: Agent.ID.make(current.id) }) }), remove: (id) => editor.remove(Agent.ID.make(id)), }), ), }}
export function providerHost(providers: Provider.Interface): Plugin.Context["provider"] { return { list: () => providers.available().pipe(Effect.map(located)), get: (input) => providers.get(Provider.ID.make(input.providerID)).pipe( Effect.flatMap((provider) => provider === undefined ? Effect.fail(new Error(`Provider not found: ${input.providerID}`)) : Effect.succeed(located(provider)), ), ), reload: providers.reload, transform: (callback) => providers.transform((editor) => callback({ list: editor.list, get: (id) => editor.get(Provider.ID.make(id)), add: editor.add, update: (id, update) => editor.update(Provider.ID.make(id), update), remove: (id) => editor.remove(Provider.ID.make(id)), models: { set: (id, models) => editor.models.set(Provider.ID.make(id), models), update: (providerID, modelID, update) => editor.models.update(Provider.ID.make(providerID), Model.ID.make(modelID), update), remove: (providerID, modelID) => editor.models.remove(Provider.ID.make(providerID), Model.ID.make(modelID)), }, }), ), }}
/** Empty catalog for hosts that only need provider lookups to fall back. */export const noProviders: Plugin.Context["provider"] = { list: () => Effect.succeed(located([])), get: () => Effect.die("unused provider.get"), transform: () => Effect.die("unused provider.transform"), reload: () => Effect.die("unused provider.reload"),}
export function modelHost(models: Model.Interface): Plugin.Context["model"] { return { list: () => models.available().pipe(Effect.map(located)), default: () => models.default().pipe(Effect.map(located)), reload: models.reload, transform: (callback) => models.transform((editor) => callback({ list: (id) => editor.list(id === undefined ? undefined : Provider.ID.make(id)), get: (providerID, modelID) => editor.get(Provider.ID.make(providerID), Model.ID.make(modelID)), update: (providerID, modelID, update) => editor.update(Provider.ID.make(providerID), Model.ID.make(modelID), update), remove: (providerID, modelID) => editor.remove(Provider.ID.make(providerID), Model.ID.make(modelID)), default: { get: editor.default.get, set: (providerID, modelID) => editor.default.set(Provider.ID.make(providerID), Model.ID.make(modelID)), }, provider: { list: editor.provider.list, get: (id) => editor.provider.get(Provider.ID.make(id)), }, }), ), }}
export function integrationHost(integration: Integration.Interface): Plugin.Context["integration"] { return { list: () => Effect.die("unused integration.list"), get: () => Effect.die("unused integration.get"), connect: { key: () => Effect.die("unused integration.connect.key"), }, oauth: { connect: () => Effect.die("unused integration.oauth.connect"), status: () => Effect.die("unused integration.oauth.status"), complete: () => Effect.die("unused integration.oauth.complete"), cancel: () => Effect.die("unused integration.oauth.cancel"), }, command: { connect: () => Effect.die("unused integration.command.connect"), status: () => Effect.die("unused integration.command.status"), cancel: () => Effect.die("unused integration.command.cancel"), }, reload: integration.reload, connection: { active: (id) => integration.connection.active(Integration.ID.make(id)), resolve: (connection) => integration.connection.resolve( connection.type === "credential" ? { ...connection, id: Credential.ID.make(connection.id) } : connection, ), }, transform: (callback) => integration.transform((editor) => callback({ list: () => editor.list().map((value) => ({ id: value.id, name: value.name })), get: (id) => { const value = editor.get(Integration.ID.make(id)) return value && { id: value.id, name: value.name } }, update: (id, update) => editor.update(Integration.ID.make(id), update), remove: (id) => editor.remove(Integration.ID.make(id)), method: { list: (id) => editor.method.list(Integration.ID.make(id)), update: (input) => { if ("authorize" in input) { const methodID = Integration.MethodID.make(input.method.id) const refresh = input.refresh editor.method.update({ integrationID: Integration.ID.make(input.integrationID), method: { ...input.method, id: methodID }, authorize: (answer) => input.authorize(answer).pipe( Effect.map((authorization) => { if (authorization.mode === "auto") { return { ...authorization, callback: authorization.callback.pipe( Effect.map((credential) => Credential.OAuth.make({ ...credential, methodID: Integration.MethodID.make(credential.methodID), }), ), ), } } return { ...authorization, callback: (code: string) => authorization.callback(code).pipe( Effect.map((credential) => Credential.OAuth.make({ ...credential, methodID: Integration.MethodID.make(credential.methodID), }), ), ), } }), ), ...(refresh ? { refresh: (value: Credential.OAuth) => refresh(value).pipe( Effect.map((next) => Credential.OAuth.make({ ...next, methodID: Integration.MethodID.make(next.methodID), }), ), ), } : {}), ...(input.label ? { label: input.label } : {}), }) return } if (input.method.type === "env") { editor.method.update({ integrationID: Integration.ID.make(input.integrationID), method: input.method, }) return } if (input.method.type === "command") { editor.method.update({ integrationID: Integration.ID.make(input.integrationID), method: { ...input.method, id: Integration.MethodID.make(input.method.id), }, }) return } editor.method.update({ integrationID: Integration.ID.make(input.integrationID), method: input.method, }) }, remove: (id, item) => editor.method.remove(Integration.ID.make(id), internalMethod(item)), }, }), ), }}
export function webSearchHost(websearch: WebSearch.Interface): Plugin.Context["websearch"] { const location = Location.Info.make({ directory: AbsolutePath.make("/tmp/websearch-test"), project: { id: Project.ID.make("websearch-test"), directory: AbsolutePath.make("/tmp/websearch-test"), canonical: AbsolutePath.make("/tmp/websearch-test"), }, }) return { providers: () => websearch.providers().pipe(Effect.map((data) => ({ location, data }))), query: (input) => websearch .query({ query: input.query, providerID: input.providerID && WebSearch.ID.make(input.providerID) }) .pipe(Effect.map((data) => ({ location, data }))), reload: websearch.reload, transform: (callback) => websearch.transform((editor) => { callback({ add: (definition) => editor.add({ id: WebSearch.ID.make(definition.id), name: definition.name, execute: definition.execute, }), default: { get: editor.default.get, set: (selection) => editor.default.set( selection === false || selection === "random" ? selection : WebSearch.ID.make(selection), ), }, }) }), }}
function internalMethod(value: IntegrationMethod): Integration.Method { if (value.type === "oauth" || value.type === "command") { return { ...value, id: Integration.MethodID.make(value.id) } } return value}
function agentInfo(value: Agent.Info) { return { ...value, model: value.model && { ...value.model }, request: { settings: { ...value.request.settings }, headers: { ...value.request.headers }, body: { ...value.request.body }, }, permissions: value.permissions.map((permission) => ({ ...permission })), }}
function located<A>(data: A) { return { location: new Location.Info({ directory: AbsolutePath.make("/"), project: { id: Project.ID.make("test"), directory: AbsolutePath.make("/"), canonical: AbsolutePath.make("/"), }, }), data, }}