openAgent

1import { beforeEach, describe, expect } from "bun:test"
2import { Effect, Exit, Layer, Option } from "effect"
3import { HttpClient, HttpClientRequest, HttpClientResponse } from "effect/unstable/http"
4import { LayerNode } from "@opencode-ai/core/effect/layer-node"
5import { httpClient } from "@opencode-ai/core/effect/app-node-platform"
6import { CrossSpawnSpawner } from "@opencode-ai/core/cross-spawn-spawner"
7import { SessionProjector } from "@opencode-ai/core/session/projector"
8 
9import { AccessToken, AccountID, OrgID, RefreshToken } from "../../src/account/schema"
10import { AccountRepo } from "../../src/account/repo"
11import { EventV2Bridge } from "../../src/event-v2-bridge"
12import { Session } from "@/session/session"
13import type { SessionID } from "../../src/session/schema"
14import { ShareNext } from "@/share/share-next"
15import { SessionShareTable } from "@opencode-ai/core/share/sql"
16import { Database } from "@opencode-ai/core/database/database"
17import { eq } from "drizzle-orm"
18import { provideTmpdirInstance } from "../fixture/fixture"
19import { resetDatabase } from "../fixture/db"
20import { pollWithTimeout, testEffect } from "../lib/effect"
21 
22const env = LayerNode.compile(LayerNode.group([CrossSpawnSpawner.node]))
23const it = testEffect(env)
24 
25const json = (req: Parameters<typeof HttpClientResponse.fromWeb>[0], body: unknown, status = 200) =>
26  HttpClientResponse.fromWeb(
27    req,
28    new Response(JSON.stringify(body), {
29      status,
30      headers: { "content-type": "application/json" },
31    }),
32  )
33 
34const none = HttpClient.make(() => Effect.die("unexpected http call"))
35 
36function requestLayer(client: HttpClient.HttpClient) {
37  const replacement = [httpClient, Layer.succeed(HttpClient.HttpClient, client)] as const
38  return LayerNode.compile(LayerNode.group([ShareNext.node, AccountRepo.node]), [replacement])
39}
40 
41function integrationLayer(client: HttpClient.HttpClient) {
42  const replacement = [httpClient, Layer.succeed(HttpClient.HttpClient, client)] as const
43  return LayerNode.compile(
44    LayerNode.group([
45      ShareNext.node,
46      EventV2Bridge.node,
47      Session.node,
48      SessionProjector.node,
49      AccountRepo.node,
50      Database.node,
51    ]),
52    [replacement],
53  )
54}
55 
56const share = (id: SessionID) =>
57  Effect.gen(function* () {
58    const { db } = yield* Database.Service
59    return yield* db
60      .select()
61      .from(SessionShareTable)
62      .where(eq(SessionShareTable.session_id, id))
63      .get()
64      .pipe(Effect.orDie)
65  })
66 
67const seed = (url: string, org?: string) =>
68  AccountRepo.Service.use((repo) =>
69    repo.persistAccount({
70      id: AccountID.make("account-1"),
71      email: "user@example.com",
72      url,
73      accessToken: AccessToken.make("st_test_token"),
74      refreshToken: RefreshToken.make("rt_test_token"),
75      expiry: Date.now() + 10 * 60_000,
76      orgID: org ? Option.some(OrgID.make(org)) : Option.none(),
77    }),
78  )
79 
80beforeEach(async () => {
81  await resetDatabase()
82})
83 
84describe("ShareNext", () => {
85  it.live("request uses legacy share API without active org account", () =>
86    provideTmpdirInstance(
87      () =>
88        ShareNext.Service.use((svc) =>
89          Effect.gen(function* () {
90            const req = yield* svc.request()
91 
92            expect(req.api.create).toBe("/api/share")
93            expect(req.api.sync("shr_123")).toBe("/api/share/shr_123/sync")
94            expect(req.api.remove("shr_123")).toBe("/api/share/shr_123")
95            expect(req.api.data("shr_123")).toBe("/api/share/shr_123/data")
96            expect(req.baseUrl).toBe("https://legacy-share.example.com")
97            expect(req.headers).toEqual({})
98          }),
99        ).pipe(Effect.provide(requestLayer(none))),
100      { config: { enterprise: { url: "https://legacy-share.example.com" } } },
101    ),
102  )
103 
104  it.live("request uses default URL when no enterprise config", () =>
105    provideTmpdirInstance(() =>
106      ShareNext.Service.use((svc) =>
107        Effect.gen(function* () {
108          const req = yield* svc.request()
109 
110          expect(req.baseUrl).toBe("https://opncd.ai")
111          expect(req.api.create).toBe("/api/share")
112          expect(req.headers).toEqual({})
113        }),
114      ).pipe(Effect.provide(requestLayer(none))),
115    ),
116  )
117 
118  it.live("request uses org share API with auth headers when account is active", () =>
119    provideTmpdirInstance(() =>
120      Effect.gen(function* () {
121        yield* seed("https://control.example.com", "org-1")
122 
123        const req = yield* ShareNext.use.request()
124 
125        expect(req.api.create).toBe("/api/shares")
126        expect(req.api.sync("shr_123")).toBe("/api/shares/shr_123/sync")
127        expect(req.api.remove("shr_123")).toBe("/api/shares/shr_123")
128        expect(req.api.data("shr_123")).toBe("/api/shares/shr_123/data")
129        expect(req.baseUrl).toBe("https://control.example.com")
130        expect(req.headers).toEqual({
131          authorization: "Bearer st_test_token",
132          "x-org-id": "org-1",
133        })
134      }).pipe(Effect.provide(requestLayer(none))),
135    ),
136  )
137 
138  it.live("create posts share, persists it, and returns the result", () =>
139    provideTmpdirInstance(
140      () => {
141        const createRequests: HttpClientRequest.HttpClientRequest[] = []
142        const client = HttpClient.make((req) => {
143          if (req.url.endsWith("/api/share")) {
144            createRequests.push(req)
145            return Effect.succeed(
146              json(req, {
147                id: "shr_abc",
148                url: "https://legacy-share.example.com/share/abc",
149                secret: "sec_123",
150              }),
151            )
152          }
153          return Effect.succeed(json(req, { ok: true }))
154        })
155        return Effect.gen(function* () {
156          const session = yield* (yield* Session.Service).create({ title: "test" })
157 
158          const result = yield* (yield* ShareNext.Service).create(session.id)
159 
160          expect(result.id).toBe("shr_abc")
161          expect(result.url).toBe("https://legacy-share.example.com/share/abc")
162          expect(result.secret).toBe("sec_123")
163 
164          const row = yield* share(session.id)
165          expect(row?.id).toBe("shr_abc")
166          expect(row?.url).toBe("https://legacy-share.example.com/share/abc")
167          expect(row?.secret).toBe("sec_123")
168 
169          expect(createRequests).toHaveLength(1)
170          expect(createRequests[0].method).toBe("POST")
171          expect(createRequests[0].url).toBe("https://legacy-share.example.com/api/share")
172        }).pipe(Effect.provide(integrationLayer(client)))
173      },
174      { config: { enterprise: { url: "https://legacy-share.example.com" } } },
175    ),
176  )
177 
178  it.live("remove deletes the persisted share and calls the delete endpoint", () =>
179    provideTmpdirInstance(
180      () => {
181        const seen: HttpClientRequest.HttpClientRequest[] = []
182        const client = HttpClient.make((req) => {
183          seen.push(req)
184          if (req.method === "POST") {
185            return Effect.succeed(
186              json(req, {
187                id: "shr_abc",
188                url: "https://legacy-share.example.com/share/abc",
189                secret: "sec_123",
190              }),
191            )
192          }
193          return Effect.succeed(HttpClientResponse.fromWeb(req, new Response(null, { status: 200 })))
194        })
195        return Effect.gen(function* () {
196          const session = yield* (yield* Session.Service).create({ title: "test" })
197          const service = yield* ShareNext.Service
198 
199          yield* service.create(session.id)
200          yield* service.remove(session.id)
201 
202          expect(yield* share(session.id)).toBeUndefined()
203          expect(seen.map((req) => [req.method, req.url])).toEqual([
204            ["POST", "https://legacy-share.example.com/api/share"],
205            ["DELETE", "https://legacy-share.example.com/api/share/shr_abc"],
206          ])
207        }).pipe(Effect.provide(integrationLayer(client)))
208      },
209      { config: { enterprise: { url: "https://legacy-share.example.com" } } },
210    ),
211  )
212 
213  it.live("create fails on a non-ok response and does not persist a share", () =>
214    provideTmpdirInstance(() => {
215      const client = HttpClient.make((req) => Effect.succeed(json(req, { error: "bad" }, 500)))
216      return Effect.gen(function* () {
217        const session = yield* (yield* Session.Service).create({ title: "test" })
218 
219        const exit = yield* ShareNext.Service.use((svc) => Effect.exit(svc.create(session.id)))
220 
221        expect(Exit.isFailure(exit)).toBe(true)
222        expect(yield* share(session.id)).toBeUndefined()
223      }).pipe(Effect.provide(integrationLayer(client)))
224    }),
225  )
226 
227  it.live("ShareNext coalesces rapid diff events into one delayed sync with latest data", () =>
228    provideTmpdirInstance(
229      () => {
230        const seen: Array<{ url: string; body: string }> = []
231        const client = HttpClient.make((req) => {
232          if (req.url.endsWith("/sync") && req.body._tag === "Uint8Array") {
233            seen.push({ url: req.url, body: new TextDecoder().decode(req.body.body) })
234          }
235          return Effect.succeed(json(req, { ok: true }))
236        })
237 
238        return Effect.gen(function* () {
239          const events = yield* EventV2Bridge.Service
240          const share = yield* ShareNext.Service
241          const session = yield* Session.Service
242 
243          const info = yield* session.create({ title: "first" })
244          yield* share.init()
245          yield* Effect.sleep(50)
246          const { db } = yield* Database.Service
247          yield* db
248            .insert(SessionShareTable)
249            .values({
250              session_id: info.id,
251              id: "shr_abc",
252              url: "https://legacy-share.example.com/share/abc",
253              secret: "sec_123",
254            })
255            .run()
256            .pipe(Effect.orDie)
257 
258          yield* events.publish(Session.Event.Diff, {
259            sessionID: info.id,
260            diff: [
261              {
262                file: "a.ts",
263                patch:
264                  "Index: a.ts\n===================================================================\n--- a.ts\t\n+++ a.ts\t\n@@ -1,1 +1,1 @@\n-one\n\\ No newline at end of file\n+two\n\\ No newline at end of file\n",
265                additions: 1,
266                deletions: 1,
267                status: "modified",
268              },
269            ],
270          })
271          yield* events.publish(Session.Event.Diff, {
272            sessionID: info.id,
273            diff: [
274              {
275                file: "b.ts",
276                patch:
277                  "Index: b.ts\n===================================================================\n--- b.ts\t\n+++ b.ts\t\n@@ -1,1 +1,1 @@\n-old\n\\ No newline at end of file\n+new\n\\ No newline at end of file\n",
278                additions: 2,
279                deletions: 0,
280                status: "modified",
281              },
282            ],
283          })
284          yield* pollWithTimeout(
285            Effect.sync(() => (seen.length === 1 ? true : undefined)),
286            "timed out waiting for share sync",
287            "5 seconds",
288          )
289 
290          expect(seen).toHaveLength(1)
291          expect(seen[0].url).toBe("https://legacy-share.example.com/api/share/shr_abc/sync")
292 
293          const body = JSON.parse(seen[0].body) as {
294            secret: string
295            data: Array<{
296              type: string
297              data: Array<{
298                file: string
299                patch: string
300                additions: number
301                deletions: number
302                status?: string
303              }>
304            }>
305          }
306          expect(body.secret).toBe("sec_123")
307          expect(body.data).toHaveLength(1)
308          expect(body.data[0].type).toBe("session_diff")
309          expect(body.data[0].data).toEqual([
310            {
311              file: "b.ts",
312              patch:
313                "Index: b.ts\n===================================================================\n--- b.ts\t\n+++ b.ts\t\n@@ -1,1 +1,1 @@\n-old\n\\ No newline at end of file\n+new\n\\ No newline at end of file\n",
314              additions: 2,
315              deletions: 0,
316              status: "modified",
317            },
318          ])
319        }).pipe(Effect.provide(integrationLayer(client)))
320      },
321      { config: { enterprise: { url: "https://legacy-share.example.com" } } },
322    ),
323  )
324})
325