openAgent

1export * as PtyTicket from "./ticket"
2 
3import { WorkspaceV2 } from "../workspace"
4import { PtyTicket } from "@opencode-ai/schema/pty-ticket"
5import { PtyID } from "./schema"
6import { Cache, Context, Duration, Effect, Layer } from "effect"
7import { makeGlobalNode } from "../effect/app-node"
8 
9const DEFAULT_TTL = Duration.seconds(60)
10const CAPACITY = 10_000
11 
12export const ConnectToken = PtyTicket.ConnectToken
13 
14export type Scope = {
15  readonly ptyID: PtyID
16  readonly directory?: string
17  readonly workspaceID?: WorkspaceV2.ID
18}
19 
20export interface Interface {
21  issue(input: Scope): Effect.Effect<typeof ConnectToken.Type>
22  consume(input: Scope & { readonly ticket: string }): Effect.Effect<boolean>
23}
24 
25export class Service extends Context.Service<Service, Interface>()("@opencode/PtyTicket") {}
26 
27function matches(record: Scope, input: Scope) {
28  return (
29    record.ptyID === input.ptyID && record.directory === input.directory && record.workspaceID === input.workspaceID
30  )
31}
32 
33// Tickets are inserted via Cache.set and removed atomically via invalidateWhen. The lookup is
34// never invoked; it dies if it ever is, which would signal a misuse of the Service interface.
35const noLookup = () => Effect.die("PtyTicket cache must be used via set/invalidateWhen, never get")
36 
37// Visible for tests so the TTL can be shortened. Production uses `layer` with the default TTL.
38export const make = (ttl: Duration.Input = DEFAULT_TTL) =>
39  Effect.gen(function* () {
40    const cache = yield* Cache.make<string, Scope>({ capacity: CAPACITY, lookup: noLookup, timeToLive: ttl })
41    const expiresIn = Math.max(1, Math.round(Duration.toSeconds(Duration.fromInputUnsafe(ttl))))
42    return Service.of({
43      issue: Effect.fn("PtyTicket.issue")(function* (input) {
44        const ticket = crypto.randomUUID()
45        yield* Cache.set(cache, ticket, input)
46        return { ticket, expires_in: expiresIn }
47      }),
48      consume: Effect.fn("PtyTicket.consume")(function* (input) {
49        return yield* Cache.invalidateWhen(cache, input.ticket, (stored) => matches(stored, input))
50      }),
51    })
52  })
53 
54const layer = Layer.effect(Service, make())
55 
56export const node = makeGlobalNode({ service: Service, layer: layer, deps: [] })
57