|
|
@@ -1,6 +1,6 @@
|
|
|
export * as PluginV2 from "./plugin"
|
|
|
|
|
|
-import { Context, Effect, Exit, Layer, Schema, Scope } from "effect"
|
|
|
+import { Context, Deferred, Effect, Exit, Layer, Schema, Scope } from "effect"
|
|
|
import type { Plugin } from "@opencode-ai/plugin/v2/effect"
|
|
|
import { AgentV2 } from "./agent"
|
|
|
import { AISDK } from "./aisdk"
|
|
|
@@ -29,6 +29,7 @@ export const Event = {
|
|
|
export interface Interface {
|
|
|
readonly add: (id: ID, effect: Plugin["effect"]) => Effect.Effect<void>
|
|
|
readonly remove: (id: ID) => Effect.Effect<void>
|
|
|
+ readonly wait: (id: ID) => Effect.Effect<void>
|
|
|
}
|
|
|
|
|
|
export class Service extends Context.Service<Service, Interface>()("@opencode/v2/Plugin") {}
|
|
|
@@ -41,13 +42,18 @@ export const layer = Layer.effect(
|
|
|
const scope = yield* Scope.make()
|
|
|
const active = new Map<ID, Scope.Closeable>()
|
|
|
const loading = new Set<ID>()
|
|
|
+ const waiters = new Map<ID, Set<Deferred.Deferred<void>>>()
|
|
|
+ const failures = new Map<ID, Exit.Exit<void, never>>()
|
|
|
let host: Parameters<Plugin["effect"]>[0]
|
|
|
|
|
|
const add = Effect.fn("Plugin.add")(function* (id: ID, effect: Plugin["effect"]) {
|
|
|
if (loading.has(id)) return yield* Effect.die(`Plugin load cycle detected for ${id}`)
|
|
|
|
|
|
yield* locks.withLock(id)(
|
|
|
- Effect.sync(() => loading.add(id)).pipe(
|
|
|
+ Effect.sync(() => {
|
|
|
+ loading.add(id)
|
|
|
+ failures.delete(id)
|
|
|
+ }).pipe(
|
|
|
Effect.andThen(
|
|
|
State.batch(
|
|
|
Effect.gen(function* () {
|
|
|
@@ -61,11 +67,22 @@ export const layer = Layer.effect(
|
|
|
Effect.withSpan("Plugin.load", { attributes: { "plugin.id": id } }),
|
|
|
Effect.onExit((exit) => (Exit.isFailure(exit) ? Scope.close(child, exit) : Effect.void)),
|
|
|
)
|
|
|
- active.set(id, child)
|
|
|
yield* events.publish(Event.Added, { id })
|
|
|
+ active.set(id, child)
|
|
|
+ yield* Effect.forEach(waiters.get(id) ?? [], (waiter) => Deferred.succeed(waiter, undefined), {
|
|
|
+ discard: true,
|
|
|
+ })
|
|
|
+ waiters.delete(id)
|
|
|
}),
|
|
|
),
|
|
|
),
|
|
|
+ Effect.onExit((exit) => {
|
|
|
+ if (Exit.isSuccess(exit)) return Effect.void
|
|
|
+ failures.set(id, exit)
|
|
|
+ return Effect.forEach(waiters.get(id) ?? [], (waiter) => Deferred.done(waiter, exit), {
|
|
|
+ discard: true,
|
|
|
+ }).pipe(Effect.ensuring(Effect.sync(() => waiters.delete(id))))
|
|
|
+ }),
|
|
|
Effect.ensuring(Effect.sync(() => loading.delete(id))),
|
|
|
),
|
|
|
)
|
|
|
@@ -79,12 +96,41 @@ export const layer = Layer.effect(
|
|
|
Effect.gen(function* () {
|
|
|
const current = active.get(id)
|
|
|
active.delete(id)
|
|
|
+ failures.delete(id)
|
|
|
if (current) yield* Scope.close(current, Exit.void).pipe(Effect.ignore)
|
|
|
}),
|
|
|
),
|
|
|
)
|
|
|
})
|
|
|
|
|
|
+ const wait = Effect.fn("Plugin.wait")(function* (id: ID) {
|
|
|
+ const waiter = yield* Deferred.make<void>()
|
|
|
+ const pending = yield* locks.withLock(id)(
|
|
|
+ Effect.sync(() => {
|
|
|
+ if (active.has(id)) return false
|
|
|
+ const failure = failures.get(id)
|
|
|
+ if (failure) return failure
|
|
|
+ const current = waiters.get(id) ?? new Set()
|
|
|
+ current.add(waiter)
|
|
|
+ waiters.set(id, current)
|
|
|
+ return true
|
|
|
+ }),
|
|
|
+ )
|
|
|
+ if (!pending) return
|
|
|
+ if (typeof pending !== "boolean") return yield* pending
|
|
|
+ yield* Deferred.await(waiter).pipe(
|
|
|
+ Effect.ensuring(
|
|
|
+ locks.withLock(id)(
|
|
|
+ Effect.sync(() => {
|
|
|
+ const current = waiters.get(id)
|
|
|
+ current?.delete(waiter)
|
|
|
+ if (current?.size === 0) waiters.delete(id)
|
|
|
+ }),
|
|
|
+ ),
|
|
|
+ ),
|
|
|
+ )
|
|
|
+ })
|
|
|
+
|
|
|
yield* Effect.addFinalizer((exit) =>
|
|
|
Effect.gen(function* () {
|
|
|
active.clear()
|
|
|
@@ -95,6 +141,7 @@ export const layer = Layer.effect(
|
|
|
const service = Service.of({
|
|
|
add,
|
|
|
remove,
|
|
|
+ wait,
|
|
|
})
|
|
|
host = yield* PluginHost.make(service)
|
|
|
return service
|