openAgent

1import { expect } from "bun:test"
2import { Effect, Layer, Context } from "effect"
3import { InstanceRef } from "../../src/effect/instance-ref"
4import { makeRuntime } from "../../src/effect/run-service"
5import { ProjectV2 } from "@opencode-ai/core/project"
6import { it } from "../lib/effect"
7 
8class Shared extends Context.Service<Shared, { readonly id: number }>()("@test/Shared") {}
9const testDirectory = "/tmp/opencode-test"
10 
11it.live("makeRuntime shares dependent layers through the shared memo map", () =>
12  Effect.gen(function* () {
13    let n = 0
14 
15    const shared = Layer.effect(
16      Shared,
17      Effect.sync(() => {
18        n += 1
19        return Shared.of({ id: n })
20      }),
21    )
22 
23    class One extends Context.Service<One, { readonly get: () => Effect.Effect<number> }>()("@test/One") {}
24    const one = Layer.effect(
25      One,
26      Effect.gen(function* () {
27        const svc = yield* Shared
28        return One.of({
29          get: Effect.fn("One.get")(() => Effect.succeed(svc.id)),
30        })
31      }),
32    ).pipe(Layer.provide(shared))
33 
34    class Two extends Context.Service<Two, { readonly get: () => Effect.Effect<number> }>()("@test/Two") {}
35    const two = Layer.effect(
36      Two,
37      Effect.gen(function* () {
38        const svc = yield* Shared
39        return Two.of({
40          get: Effect.fn("Two.get")(() => Effect.succeed(svc.id)),
41        })
42      }),
43    ).pipe(Layer.provide(shared))
44 
45    const { runPromise: runOne } = makeRuntime(One, one)
46    const { runPromise: runTwo } = makeRuntime(Two, two)
47 
48    expect(yield* Effect.promise(() => runOne((svc) => svc.get()))).toBe(1)
49    expect(yield* Effect.promise(() => runTwo((svc) => svc.get()))).toBe(1)
50    expect(n).toBe(1)
51  }),
52)
53 
54it.live("makeRuntime inherits InstanceRef from the current fiber", () =>
55  Effect.gen(function* () {
56    class NeedsInstance extends Context.Service<
57      NeedsInstance,
58      { readonly directory: () => Effect.Effect<string | undefined> }
59    >()("@test/NeedsInstance") {}
60 
61    const runtime = makeRuntime(
62      NeedsInstance,
63      Layer.succeed(
64        NeedsInstance,
65        NeedsInstance.of({
66          directory: () =>
67            Effect.gen(function* () {
68              return (yield* InstanceRef)?.directory
69            }),
70        }),
71      ),
72    )
73 
74    const actual = yield* Effect.promise(() => runtime.runPromise((svc) => svc.directory()))
75 
76    expect(actual).toBe(testDirectory)
77  }).pipe(
78    Effect.provideService(InstanceRef, {
79      directory: testDirectory,
80      worktree: testDirectory,
81      project: {
82        id: ProjectV2.ID.global,
83        worktree: testDirectory,
84        time: { created: 0, updated: 0 },
85        sandboxes: [],
86      },
87    }),
88  ),
89)
90