openAgent

1import { describe, expect } from "bun:test"
2import { $ } from "bun"
3import fs from "fs/promises"
4import path from "path"
5import { eq } from "drizzle-orm"
6import { Effect } from "effect"
7import { MoveSession } from "@opencode-ai/core/control-plane/move-session"
8import { Database } from "@opencode-ai/core/database/database"
9import { AppNodeBuilder } from "@opencode-ai/core/effect/app-node-builder"
10import { LayerNode } from "@opencode-ai/core/effect/layer-node"
11import { EventV2 } from "@opencode-ai/core/event"
12import { Project } from "@opencode-ai/core/project"
13import { ProjectTable } from "@opencode-ai/core/project/sql"
14import { ProjectDirectories } from "@opencode-ai/core/project/directories"
15import { AbsolutePath } from "@opencode-ai/core/schema"
16import { SessionV2 } from "@opencode-ai/core/session"
17import { SessionProjector } from "@opencode-ai/core/session/projector"
18import { SessionTable } from "@opencode-ai/core/session/sql"
19import { SessionStore } from "@opencode-ai/core/session/store"
20import { tmpdir } from "./fixture/tmpdir"
21import { testEffect } from "./lib/effect"
22 
23const it = testEffect(
24  AppNodeBuilder.build(
25    LayerNode.group([
26      MoveSession.node,
27      Database.node,
28      EventV2.node,
29      ProjectDirectories.node,
30      Project.node,
31      SessionProjector.node,
32      SessionStore.node,
33    ]),
34  ),
35)
36 
37function abs(input: string) {
38  return AbsolutePath.make(input)
39}
40 
41async function initRepo(directory: string) {
42  await $`git init`.cwd(directory).quiet()
43  await $`git config core.autocrlf false`.cwd(directory).quiet()
44  await $`git config core.fsmonitor false`.cwd(directory).quiet()
45  await $`git config commit.gpgsign false`.cwd(directory).quiet()
46  await $`git config user.email test@opencode.test`.cwd(directory).quiet()
47  await $`git config user.name Test`.cwd(directory).quiet()
48  await fs.writeFile(path.join(directory, "tracked.txt"), "initial\n")
49  await $`git add tracked.txt`.cwd(directory).quiet()
50  await $`git commit -m root`.cwd(directory).quiet()
51}
52 
53describe("MoveSession", () => {
54  it.live("moves session changes to another project directory", () =>
55    Effect.gen(function* () {
56      const root = yield* Effect.acquireRelease(
57        Effect.promise(() => tmpdir()),
58        (dir) => Effect.promise(() => dir[Symbol.asyncDispose]()),
59      )
60      yield* Effect.promise(() => initRepo(root.path))
61      const source = abs(yield* Effect.promise(() => fs.realpath(root.path)))
62      const destination = abs(`${root.path}-move-destination`)
63      yield* Effect.addFinalizer(() =>
64        Effect.promise(() => fs.rm(destination, { recursive: true, force: true })).pipe(Effect.ignore),
65      )
66      yield* Effect.promise(() => $`git worktree add --detach ${destination} HEAD`.cwd(root.path).quiet())
67      const moved = abs(yield* Effect.promise(() => fs.realpath(destination)))
68      yield* Effect.promise(() => fs.writeFile(path.join(source, "tracked.txt"), "changed\n"))
69      yield* Effect.promise(() => fs.writeFile(path.join(source, "untracked.txt"), "new\n"))
70 
71      const projectID = (yield* Project.Service.use((service) => service.resolve(source))).id
72      const sessionID = SessionV2.ID.make("ses_move")
73      const { db } = yield* Database.Service
74      yield* db
75        .insert(ProjectTable)
76        .values({ id: projectID, worktree: source, sandboxes: [], time_created: 1, time_updated: 1 })
77        .run()
78        .pipe(Effect.orDie)
79      yield* db
80        .insert(SessionTable)
81        .values({
82          id: sessionID,
83          project_id: projectID,
84          slug: "move",
85          directory: source,
86          title: "move",
87          version: "test",
88          time_created: 1,
89          time_updated: 1,
90        })
91        .run()
92        .pipe(Effect.orDie)
93 
94      yield* MoveSession.Service.use((service) =>
95        service.moveSession({ sessionID, destination: { directory: moved }, moveChanges: true }),
96      )
97 
98      expect(yield* Effect.promise(() => fs.readFile(path.join(moved, "tracked.txt"), "utf8"))).toBe("changed\n")
99      expect(yield* Effect.promise(() => fs.readFile(path.join(moved, "untracked.txt"), "utf8"))).toBe("new\n")
100      expect(yield* Effect.promise(() => fs.readFile(path.join(source, "tracked.txt"), "utf8"))).toBe("initial\n")
101      expect(yield* Effect.promise(() => Bun.file(path.join(source, "untracked.txt")).exists())).toBe(false)
102      expect(
103        yield* db
104          .select({ directory: SessionTable.directory, path: SessionTable.path })
105          .from(SessionTable)
106          .where(eq(SessionTable.id, sessionID))
107          .get(),
108      ).toEqual({ directory: moved, path: "" })
109    }),
110  )
111 
112  it.live("moves within a checkout without transferring existing changes", () =>
113    Effect.gen(function* () {
114      const root = yield* Effect.acquireRelease(
115        Effect.promise(() => tmpdir()),
116        (dir) => Effect.promise(() => dir[Symbol.asyncDispose]()),
117      )
118      yield* Effect.promise(() => initRepo(root.path))
119      const source = abs(yield* Effect.promise(() => fs.realpath(root.path)))
120      const destination = abs(path.join(source, "packages"))
121      yield* Effect.promise(() => fs.mkdir(destination))
122      yield* Effect.promise(() => fs.writeFile(path.join(source, "tracked.txt"), "changed\n"))
123      yield* Effect.promise(() => fs.writeFile(path.join(source, "untracked.txt"), "new\n"))
124 
125      const projectID = (yield* Project.Service.use((service) => service.resolve(source))).id
126      const sessionID = SessionV2.ID.make("ses_move_nested")
127      const { db } = yield* Database.Service
128      yield* db
129        .insert(ProjectTable)
130        .values({ id: projectID, worktree: source, sandboxes: [], time_created: 1, time_updated: 1 })
131        .run()
132        .pipe(Effect.orDie)
133      yield* db
134        .insert(SessionTable)
135        .values({
136          id: sessionID,
137          project_id: projectID,
138          slug: "move-nested",
139          directory: source,
140          title: "move nested",
141          version: "test",
142          time_created: 1,
143          time_updated: 1,
144        })
145        .run()
146        .pipe(Effect.orDie)
147 
148      yield* MoveSession.Service.use((service) =>
149        service.moveSession({ sessionID, destination: { directory: destination }, moveChanges: true }),
150      )
151 
152      expect(yield* Effect.promise(() => fs.readFile(path.join(source, "tracked.txt"), "utf8"))).toBe("changed\n")
153      expect(yield* Effect.promise(() => fs.readFile(path.join(source, "untracked.txt"), "utf8"))).toBe("new\n")
154      expect(
155        yield* db
156          .select({ directory: SessionTable.directory, path: SessionTable.path })
157          .from(SessionTable)
158          .where(eq(SessionTable.id, sessionID))
159          .get(),
160      ).toEqual({ directory: destination, path: "packages" })
161    }),
162  )
163 
164  it.live("moves nested session changes without cleaning unrelated files", () =>
165    Effect.gen(function* () {
166      const root = yield* Effect.acquireRelease(
167        Effect.promise(() => tmpdir()),
168        (dir) => Effect.promise(() => dir[Symbol.asyncDispose]()),
169      )
170      yield* Effect.promise(() => initRepo(root.path))
171      const source = abs(yield* Effect.promise(() => fs.realpath(root.path)))
172      const sourceDirectory = abs(path.join(source, "packages"))
173      yield* Effect.promise(() => fs.mkdir(sourceDirectory))
174      yield* Effect.promise(() => fs.writeFile(path.join(sourceDirectory, "tracked.txt"), "initial\n"))
175      yield* Effect.promise(() => fs.writeFile(path.join(sourceDirectory, "staged.txt"), "initial\n"))
176      yield* Effect.promise(() => $`git add packages/tracked.txt packages/staged.txt`.cwd(source).quiet())
177      yield* Effect.promise(() => $`git commit -m packages`.cwd(source).quiet())
178      const destination = abs(`${root.path}-move-nested-destination`)
179      yield* Effect.addFinalizer(() =>
180        Effect.promise(() => fs.rm(destination, { recursive: true, force: true })).pipe(Effect.ignore),
181      )
182      yield* Effect.promise(() => $`git worktree add --detach ${destination} HEAD`.cwd(source).quiet())
183      const moved = abs(path.join(yield* Effect.promise(() => fs.realpath(destination)), "packages"))
184      yield* Effect.promise(() => fs.writeFile(path.join(sourceDirectory, "tracked.txt"), "changed\n"))
185      yield* Effect.promise(() => fs.writeFile(path.join(sourceDirectory, "staged.txt"), "staged\n"))
186      yield* Effect.promise(() => $`git add packages/staged.txt`.cwd(source).quiet())
187      yield* Effect.promise(() => fs.writeFile(path.join(sourceDirectory, "untracked.txt"), "new\n"))
188      yield* Effect.promise(() => fs.writeFile(path.join(source, "tracked.txt"), "unrelated\n"))
189      yield* Effect.promise(() => fs.writeFile(path.join(source, "untracked.txt"), "unrelated\n"))
190 
191      const projectID = (yield* Project.Service.use((service) => service.resolve(source))).id
192      const sessionID = SessionV2.ID.make("ses_move_nested_checkout")
193      const { db } = yield* Database.Service
194      yield* db
195        .insert(ProjectTable)
196        .values({ id: projectID, worktree: source, sandboxes: [], time_created: 1, time_updated: 1 })
197        .run()
198        .pipe(Effect.orDie)
199      yield* db
200        .insert(SessionTable)
201        .values({
202          id: sessionID,
203          project_id: projectID,
204          slug: "move-nested-checkout",
205          directory: sourceDirectory,
206          title: "move nested checkout",
207          version: "test",
208          time_created: 1,
209          time_updated: 1,
210        })
211        .run()
212        .pipe(Effect.orDie)
213 
214      yield* MoveSession.Service.use((service) =>
215        service.moveSession({ sessionID, destination: { directory: moved }, moveChanges: true }),
216      )
217 
218      expect(yield* Effect.promise(() => fs.readFile(path.join(moved, "tracked.txt"), "utf8"))).toBe("changed\n")
219      expect(yield* Effect.promise(() => fs.readFile(path.join(moved, "staged.txt"), "utf8"))).toBe("staged\n")
220      expect(yield* Effect.promise(() => fs.readFile(path.join(moved, "untracked.txt"), "utf8"))).toBe("new\n")
221      expect(yield* Effect.promise(() => fs.readFile(path.join(sourceDirectory, "tracked.txt"), "utf8"))).toBe(
222        "initial\n",
223      )
224      expect(yield* Effect.promise(() => Bun.file(path.join(sourceDirectory, "untracked.txt")).exists())).toBe(false)
225      expect(yield* Effect.promise(() => fs.readFile(path.join(sourceDirectory, "staged.txt"), "utf8"))).toBe(
226        "staged\n",
227      )
228      expect(yield* Effect.promise(() => $`git status --porcelain -- packages/staged.txt`.cwd(source).text())).toBe(
229        "M  packages/staged.txt\n",
230      )
231      expect(yield* Effect.promise(() => fs.readFile(path.join(source, "tracked.txt"), "utf8"))).toBe("unrelated\n")
232      expect(yield* Effect.promise(() => fs.readFile(path.join(source, "untracked.txt"), "utf8"))).toBe("unrelated\n")
233    }),
234  )
235})
236