feat(core): add browser authority and tools

packages/core/src/browser-host.ts

@@ -0,0 +1,264 @@
+export * as BrowserHost from "./browser-host"
+
+import { Browser } from "@opencode-ai/schema/browser"
+import { Session } from "@opencode-ai/schema/session"
+import { makeGlobalNode } from "@opencode-ai/util/effect/app-node"
+import { Context, Deferred, Effect, Layer, Option, Schema, Scope, Stream, SynchronizedRef } from "effect"
+import { Bus } from "./bus"
+import { SessionEvent } from "./session/event"
+import { SessionStore } from "./session/store"
+
+export class RegistrationError extends Schema.TaggedErrorClass<RegistrationError>()("BrowserHost.RegistrationError", {
+  reason: Schema.Literals(["unknown_session", "already_registered", "stale_registration", "stale_lease"]),
+  message: Schema.String,
+}) {}
+
+export class RequestError extends Schema.TaggedErrorClass<RequestError>()("BrowserHost.RequestError", {
+  code: Browser.ErrorCode,
+  message: Schema.String,
+}) {}
+
+export interface Peer {
+  readonly open: Effect.Effect<void, RequestError>
+  readonly request: (
+    command: Browser.Command,
+    leaseID: Browser.LeaseID,
+  ) => Effect.Effect<Browser.Result, RequestError>
+}
+
+export interface Controller {
+  readonly attach: (leaseID: Browser.LeaseID, state: Browser.State) => Effect.Effect<void, RegistrationError>
+  readonly state: (leaseID: Browser.LeaseID, state: Browser.State) => Effect.Effect<void, RegistrationError>
+  readonly detach: (leaseID: Browser.LeaseID) => Effect.Effect<void, RegistrationError>
+}
+
+export interface Available {
+  readonly type: "available"
+  readonly open: Effect.Effect<void, RequestError>
+}
+
+export interface Attached {
+  readonly type: "attached"
+  readonly state: Browser.State
+  readonly revoked: Effect.Effect<void>
+  readonly request: (command: Browser.Command) => Effect.Effect<Browser.Result, RequestError>
+}
+
+export type Capability = Available | Attached
+
+export interface Interface {
+  readonly register: (
+    sessionID: Session.ID,
+    peer: Peer,
+  ) => Effect.Effect<Controller, RegistrationError, Scope.Scope>
+  readonly get: (sessionID: Session.ID) => Effect.Effect<Option.Option<Capability>>
+}
+
+export class Service extends Context.Service<Service, Interface>()("@opencode/BrowserHost") {}
+
+type Attachment = {
+  readonly token: object
+  readonly leaseID: Browser.LeaseID
+  readonly state: Browser.State
+  readonly revoked: Deferred.Deferred<void>
+}
+
+type Registration = {
+  readonly token: object
+  readonly peer: Peer
+  readonly attached: Deferred.Deferred<void>
+  readonly attachment?: Attachment
+}
+
+type State = ReadonlyMap<Session.ID, Registration>
+
+export function make(
+  sessionExists: (sessionID: Session.ID) => Effect.Effect<boolean>,
+  deleted: Stream.Stream<Session.ID> = Stream.never,
+) {
+  return Effect.gen(function* () {
+    const registrations = yield* SynchronizedRef.make<State>(new Map())
+
+    const remove = Effect.fn("BrowserHost.remove")(function* (sessionID: Session.ID, token?: object) {
+      const attachment = yield* SynchronizedRef.modify(registrations, (current): readonly [Attachment | undefined, State] => {
+        const registration = current.get(sessionID)
+        if (!registration || (token && registration.token !== token)) return [undefined, current]
+        const next = new Map(current)
+        next.delete(sessionID)
+        return [registration.attachment, next]
+      })
+      if (attachment) Deferred.doneUnsafe(attachment.revoked, Effect.void)
+    })
+
+    const register: Interface["register"] = Effect.fn("BrowserHost.register")(function* (sessionID, peer) {
+      if (!(yield* sessionExists(sessionID))) {
+        return yield* new RegistrationError({
+          reason: "unknown_session",
+          message: "The browser Session does not exist.",
+        })
+      }
+
+      const token = {}
+      yield* SynchronizedRef.modifyEffect(
+        registrations,
+        Effect.fnUntraced(function* (current) {
+          if (current.has(sessionID)) {
+            return yield* new RegistrationError({
+              reason: "already_registered",
+              message: "The browser Session is already registered.",
+            })
+          }
+          return [undefined, new Map(current).set(sessionID, { token, peer, attached: Deferred.makeUnsafe<void>() })] as const
+        }),
+      )
+      yield* Effect.addFinalizer(() => remove(sessionID, token))
+
+      const attach: Controller["attach"] = Effect.fn("BrowserHost.attach")(function* (leaseID, state) {
+        const previous = yield* SynchronizedRef.modifyEffect(
+        registrations,
+        Effect.fnUntraced(function* (current) {
+            const registration = current.get(sessionID)
+            if (registration?.token !== token) {
+              return yield* new RegistrationError({
+                reason: "stale_registration",
+                message: "The browser registration is no longer active.",
+              })
+            }
+            const attachment = { token: {}, leaseID, state, revoked: Deferred.makeUnsafe<void>() }
+            return [
+              registration.attachment,
+              new Map(current).set(sessionID, { ...registration, attachment }),
+            ] as const
+          }),
+        )
+        if (previous) Deferred.doneUnsafe(previous.revoked, Effect.void)
+        const current = (yield* SynchronizedRef.get(registrations)).get(sessionID)
+        if (current) Deferred.doneUnsafe(current.attached, Effect.void)
+      })
+
+      const update: Controller["state"] = Effect.fn("BrowserHost.state")(function* (leaseID, state) {
+        yield* SynchronizedRef.updateEffect(
+          registrations,
+          Effect.fnUntraced(function* (current) {
+            const registration = current.get(sessionID)
+            if (registration?.token !== token) {
+              return yield* new RegistrationError({
+                reason: "stale_registration",
+                message: "The browser registration is no longer active.",
+              })
+            }
+            const attachment = registration.attachment
+            if (attachment?.leaseID !== leaseID) {
+              return yield* new RegistrationError({
+                reason: "stale_lease",
+                message: "The browser attachment lease is no longer active.",
+              })
+            }
+            return new Map(current).set(sessionID, {
+              ...registration,
+              attachment: { ...attachment, state },
+            })
+          }),
+        )
+      })
+
+      const detach: Controller["detach"] = Effect.fn("BrowserHost.detach")(function* (leaseID) {
+        const attachment = yield* SynchronizedRef.modifyEffect(
+          registrations,
+          Effect.fnUntraced(function* (current) {
+            const registration = current.get(sessionID)
+            if (registration?.token !== token) {
+              return yield* new RegistrationError({
+                reason: "stale_registration",
+                message: "The browser registration is no longer active.",
+              })
+            }
+            const attachment = registration.attachment
+            if (attachment?.leaseID !== leaseID) {
+              return yield* new RegistrationError({
+                reason: "stale_lease",
+                message: "The browser attachment lease is no longer active.",
+              })
+            }
+            return [
+              attachment,
+              new Map(current).set(sessionID, { token, peer, attached: Deferred.makeUnsafe<void>() }),
+            ] as const
+          }),
+        )
+        Deferred.doneUnsafe(attachment.revoked, Effect.void)
+      })
+
+      return { attach, state: update, detach }
+    })
+
+    const get: Interface["get"] = Effect.fn("BrowserHost.get")(function* (sessionID) {
+      if (!(yield* sessionExists(sessionID))) {
+        yield* remove(sessionID)
+        return Option.none()
+      }
+      const registration = (yield* SynchronizedRef.get(registrations)).get(sessionID)
+      if (!registration) return Option.none()
+      if (!registration.attachment) {
+        return Option.some({
+          type: "available" as const,
+          open: Effect.gen(function* () {
+            const current = (yield* SynchronizedRef.get(registrations)).get(sessionID)
+            if (current?.token !== registration.token || current.attachment) return yield* unavailable()
+            yield* registration.peer.open
+            return yield* Deferred.await(registration.attached).pipe(
+              Effect.timeoutOrElse({
+                duration: "30 seconds",
+                orElse: () => Effect.fail(new RequestError({ code: "timeout", message: "Browser pane did not open." })),
+              }),
+            )
+          }),
+        })
+      }
+
+      const attachment = registration.attachment
+      return Option.some({
+        type: "attached" as const,
+        state: attachment.state,
+        revoked: Deferred.await(attachment.revoked),
+        request: (command) =>
+          Effect.gen(function* () {
+            const current = (yield* SynchronizedRef.get(registrations)).get(sessionID)
+            if (current?.token !== registration.token || current.attachment?.token !== attachment.token) {
+              return yield* unavailable()
+            }
+            const result = yield* registration.peer
+              .request(command, attachment.leaseID)
+              .pipe(Effect.raceFirst(Deferred.await(attachment.revoked).pipe(Effect.andThen(unavailable()))))
+            if (result.type === command.type) return result
+            return yield* new RequestError({ code: "protocol", message: "Browser response does not match its command." })
+          }),
+      })
+    })
+
+    yield* Stream.runForEach(deleted, (sessionID) => remove(sessionID)).pipe(Effect.forkScoped)
+    return Service.of({ register, get })
+  })
+}
+
+function unavailable() {
+  return new RequestError({ code: "not_attached", message: "The browser attachment is no longer available." })
+}
+
+export const layer = Layer.effect(
+  Service,
+  Effect.gen(function* () {
+    const sessions = yield* SessionStore.Service
+    const bus = yield* Bus.Service
+    return yield* make(
+      (sessionID) => sessions.get(sessionID).pipe(Effect.map((session) => session !== undefined)),
+      bus.subscribe(SessionEvent.Deleted).pipe(Stream.map((event) => event.data.sessionID)),
+    )
+  }),
+)
+
+export const node = makeGlobalNode({
+  service: Service,
+  layer,
+  deps: [SessionStore.node, Bus.node],
+})