bonecode 1.0.0 → 1.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (575) hide show
  1. package/LICENSE +21 -0
  2. package/README.md +64 -50
  3. package/bone/output/agent/.dockerignore +7 -7
  4. package/bone/output/agent/.env.example +36 -36
  5. package/bone/output/agent/.github/workflows/ci.yaml +58 -58
  6. package/bone/output/agent/AgentDomain.bone.map +349 -349
  7. package/bone/output/agent/AgentDomain.postman_collection.json +957 -957
  8. package/bone/output/agent/Dockerfile +22 -22
  9. package/bone/output/agent/README.md +47 -47
  10. package/bone/output/agent/admin/index.html +739 -739
  11. package/bone/output/agent/docker-compose.yaml +22 -22
  12. package/bone/output/agent/k8s/deployment.yaml +75 -75
  13. package/bone/output/agent/migrations/agent.sql +36 -36
  14. package/bone/output/agent/migrations/agent_instance.sql +36 -36
  15. package/bone/output/agent/migrations/audit_log.sql +18 -18
  16. package/bone/output/agent/migrations/build_step.sql +34 -34
  17. package/bone/output/agent/migrations/event_outbox.sql +31 -31
  18. package/bone/output/agent/migrations/plan.sql +30 -30
  19. package/bone/output/agent/migrations/task.sql +30 -30
  20. package/bone/output/agent/migrations/tool_call.sql +33 -33
  21. package/bone/output/agent/openapi.yaml +1116 -1116
  22. package/bone/output/agent/package.json +35 -35
  23. package/bone/output/agent/schema.graphql +233 -233
  24. package/bone/output/agent/sdk/client.ts +231 -231
  25. package/bone/output/agent/src/algorithms.ts +2 -2
  26. package/bone/output/agent/src/audit.ts +44 -44
  27. package/bone/output/agent/src/auth.ts +57 -57
  28. package/bone/output/agent/src/cron.ts +12 -12
  29. package/bone/output/agent/src/db.ts +31 -31
  30. package/bone/output/agent/src/debug.ts +66 -66
  31. package/bone/output/agent/src/events.ts +243 -243
  32. package/bone/output/agent/src/extensions.ts +54 -54
  33. package/bone/output/agent/src/failure_rules.ts +322 -322
  34. package/bone/output/agent/src/flows.ts +168 -168
  35. package/bone/output/agent/src/health.ts +43 -43
  36. package/bone/output/agent/src/index.ts +99 -99
  37. package/bone/output/agent/src/logger.ts +69 -66
  38. package/bone/output/agent/src/metrics.ts +75 -75
  39. package/bone/output/agent/src/migrate.ts +351 -351
  40. package/bone/output/agent/src/migration_diff.ts +108 -108
  41. package/bone/output/agent/src/notify.ts +125 -125
  42. package/bone/output/agent/src/routes/plan.ts +91 -91
  43. package/bone/output/agent/src/routes/task.ts +105 -105
  44. package/bone/output/agent/src/routes/tool_call.ts +166 -166
  45. package/bone/output/agent/src/schemas.ts +384 -384
  46. package/bone/output/agent/src/state_machines/agent_instance.ts +24 -24
  47. package/bone/output/agent/src/state_machines/build_step.ts +22 -22
  48. package/bone/output/agent/src/state_machines/plan.ts +22 -22
  49. package/bone/output/agent/src/state_machines/task.ts +22 -22
  50. package/bone/output/agent/src/state_machines/tool_call.ts +22 -22
  51. package/bone/output/agent/src/tests.ts +361 -361
  52. package/bone/output/agent/src/websocket.ts +200 -200
  53. package/bone/output/agent/tsconfig.json +24 -24
  54. package/bone/output/rag/.dockerignore +7 -7
  55. package/bone/output/rag/.env.example +36 -36
  56. package/bone/output/rag/.github/workflows/ci.yaml +58 -58
  57. package/bone/output/rag/Dockerfile +22 -22
  58. package/bone/output/rag/RAGDomain.bone.map +286 -286
  59. package/bone/output/rag/RAGDomain.postman_collection.json +922 -922
  60. package/bone/output/rag/README.md +47 -47
  61. package/bone/output/rag/admin/index.html +817 -817
  62. package/bone/output/rag/docker-compose.yaml +22 -22
  63. package/bone/output/rag/k8s/deployment.yaml +75 -75
  64. package/bone/output/rag/migrations/audit_log.sql +18 -18
  65. package/bone/output/rag/migrations/code_chunk.sql +34 -34
  66. package/bone/output/rag/migrations/code_file.sql +33 -33
  67. package/bone/output/rag/migrations/event_outbox.sql +31 -31
  68. package/bone/output/rag/migrations/indexing_job.sql +33 -33
  69. package/bone/output/rag/migrations/knowledge_base.sql +35 -35
  70. package/bone/output/rag/migrations/memory_entry.sql +34 -34
  71. package/bone/output/rag/openapi.yaml +1097 -1097
  72. package/bone/output/rag/package.json +35 -35
  73. package/bone/output/rag/schema.graphql +245 -245
  74. package/bone/output/rag/sdk/client.ts +234 -234
  75. package/bone/output/rag/src/algorithms.ts +2 -2
  76. package/bone/output/rag/src/audit.ts +37 -37
  77. package/bone/output/rag/src/auth.ts +57 -57
  78. package/bone/output/rag/src/cron.ts +12 -12
  79. package/bone/output/rag/src/db.ts +31 -31
  80. package/bone/output/rag/src/debug.ts +66 -66
  81. package/bone/output/rag/src/events.ts +243 -243
  82. package/bone/output/rag/src/extensions.ts +350 -350
  83. package/bone/output/rag/src/failure_rules.ts +314 -314
  84. package/bone/output/rag/src/flows.ts +239 -239
  85. package/bone/output/rag/src/health.ts +43 -43
  86. package/bone/output/rag/src/index.ts +94 -94
  87. package/bone/output/rag/src/logger.ts +69 -66
  88. package/bone/output/rag/src/metrics.ts +75 -75
  89. package/bone/output/rag/src/migrate.ts +363 -363
  90. package/bone/output/rag/src/migration_diff.ts +108 -108
  91. package/bone/output/rag/src/notify.ts +99 -99
  92. package/bone/output/rag/src/routes/code_chunk.ts +75 -75
  93. package/bone/output/rag/src/routes/code_file.ts +101 -101
  94. package/bone/output/rag/src/routes/indexing_job.ts +87 -87
  95. package/bone/output/rag/src/routes/knowledge_base.ts +230 -230
  96. package/bone/output/rag/src/routes/memory_entry.ts +87 -87
  97. package/bone/output/rag/src/schemas.ts +394 -394
  98. package/bone/output/rag/src/state_machines/code_file.ts +23 -23
  99. package/bone/output/rag/src/state_machines/indexing_job.ts +22 -22
  100. package/bone/output/rag/src/state_machines/knowledge_base.ts +23 -23
  101. package/bone/output/rag/src/state_machines/memory_entry.ts +20 -20
  102. package/bone/output/rag/src/tests.ts +339 -339
  103. package/bone/output/rag/tsconfig.json +24 -24
  104. package/bone/output/session/.dockerignore +7 -7
  105. package/bone/output/session/.env.example +36 -36
  106. package/bone/output/session/.github/workflows/ci.yaml +58 -58
  107. package/bone/output/session/Dockerfile +22 -22
  108. package/bone/output/session/README.md +47 -47
  109. package/bone/output/session/SessionDomain.bone.map +349 -349
  110. package/bone/output/session/SessionDomain.postman_collection.json +957 -957
  111. package/bone/output/session/admin/index.html +666 -666
  112. package/bone/output/session/docker-compose.yaml +22 -22
  113. package/bone/output/session/k8s/deployment.yaml +75 -75
  114. package/bone/output/session/migrations/audit_log.sql +18 -18
  115. package/bone/output/session/migrations/event_outbox.sql +31 -31
  116. package/bone/output/session/migrations/message.sql +31 -31
  117. package/bone/output/session/migrations/part.sql +28 -28
  118. package/bone/output/session/migrations/permission.sql +28 -28
  119. package/bone/output/session/migrations/project.sql +28 -28
  120. package/bone/output/session/migrations/session.sql +38 -38
  121. package/bone/output/session/openapi.yaml +1101 -1101
  122. package/bone/output/session/package.json +35 -35
  123. package/bone/output/session/schema.graphql +222 -222
  124. package/bone/output/session/sdk/client.ts +225 -225
  125. package/bone/output/session/src/algorithms.ts +2 -2
  126. package/bone/output/session/src/audit.ts +44 -44
  127. package/bone/output/session/src/auth.ts +57 -57
  128. package/bone/output/session/src/cron.ts +12 -12
  129. package/bone/output/session/src/db.ts +31 -31
  130. package/bone/output/session/src/debug.ts +66 -66
  131. package/bone/output/session/src/events.ts +270 -270
  132. package/bone/output/session/src/extensions.ts +215 -215
  133. package/bone/output/session/src/failure_rules.ts +283 -283
  134. package/bone/output/session/src/flows.ts +168 -168
  135. package/bone/output/session/src/health.ts +43 -43
  136. package/bone/output/session/src/index.ts +99 -99
  137. package/bone/output/session/src/logger.ts +67 -66
  138. package/bone/output/session/src/metrics.ts +75 -75
  139. package/bone/output/session/src/migrate.ts +331 -331
  140. package/bone/output/session/src/migration_diff.ts +108 -108
  141. package/bone/output/session/src/notify.ts +112 -112
  142. package/bone/output/session/src/routes/message.ts +93 -93
  143. package/bone/output/session/src/routes/part.ts +79 -79
  144. package/bone/output/session/src/routes/permission.ts +79 -79
  145. package/bone/output/session/src/routes/project.ts +79 -79
  146. package/bone/output/session/src/routes/session.ts +294 -294
  147. package/bone/output/session/src/schemas.ts +357 -357
  148. package/bone/output/session/src/state_machines/session.ts +23 -23
  149. package/bone/output/session/src/tests.ts +325 -325
  150. package/bone/output/session/src/websocket.ts +223 -200
  151. package/bone/output/session/tsconfig.json +24 -24
  152. package/bone/output/workspace/.dockerignore +7 -7
  153. package/bone/output/workspace/.env.example +36 -36
  154. package/bone/output/workspace/.github/workflows/ci.yaml +58 -58
  155. package/bone/output/workspace/Dockerfile +22 -22
  156. package/bone/output/workspace/README.md +45 -45
  157. package/bone/output/workspace/WorkspaceDomain.bone.map +188 -188
  158. package/bone/output/workspace/WorkspaceDomain.postman_collection.json +620 -620
  159. package/bone/output/workspace/admin/index.html +484 -484
  160. package/bone/output/workspace/docker-compose.yaml +22 -22
  161. package/bone/output/workspace/k8s/deployment.yaml +75 -75
  162. package/bone/output/workspace/migrations/audit_log.sql +18 -18
  163. package/bone/output/workspace/migrations/codebase.sql +34 -34
  164. package/bone/output/workspace/migrations/event_outbox.sql +31 -31
  165. package/bone/output/workspace/migrations/snapshot.sql +32 -32
  166. package/bone/output/workspace/migrations/workspace.sql +33 -33
  167. package/bone/output/workspace/openapi.yaml +721 -721
  168. package/bone/output/workspace/package.json +35 -35
  169. package/bone/output/workspace/schema.graphql +153 -153
  170. package/bone/output/workspace/sdk/client.ts +155 -155
  171. package/bone/output/workspace/src/algorithms.ts +2 -2
  172. package/bone/output/workspace/src/audit.ts +37 -37
  173. package/bone/output/workspace/src/auth.ts +57 -57
  174. package/bone/output/workspace/src/cron.ts +12 -12
  175. package/bone/output/workspace/src/db.ts +31 -31
  176. package/bone/output/workspace/src/debug.ts +66 -66
  177. package/bone/output/workspace/src/events.ts +243 -243
  178. package/bone/output/workspace/src/extensions.ts +44 -44
  179. package/bone/output/workspace/src/failure_rules.ts +152 -152
  180. package/bone/output/workspace/src/health.ts +43 -43
  181. package/bone/output/workspace/src/index.ts +88 -88
  182. package/bone/output/workspace/src/logger.ts +69 -66
  183. package/bone/output/workspace/src/metrics.ts +75 -75
  184. package/bone/output/workspace/src/migrate.ts +219 -219
  185. package/bone/output/workspace/src/migration_diff.ts +108 -108
  186. package/bone/output/workspace/src/notify.ts +73 -73
  187. package/bone/output/workspace/src/routes/codebase.ts +87 -87
  188. package/bone/output/workspace/src/routes/snapshot.ts +127 -127
  189. package/bone/output/workspace/src/routes/workspace.ts +190 -190
  190. package/bone/output/workspace/src/schemas.ts +231 -231
  191. package/bone/output/workspace/src/state_machines/codebase.ts +21 -21
  192. package/bone/output/workspace/src/state_machines/snapshot.ts +20 -20
  193. package/bone/output/workspace/src/state_machines/workspace.ts +21 -21
  194. package/bone/output/workspace/src/tests.ts +248 -248
  195. package/bone/output/workspace/tsconfig.json +24 -24
  196. package/compat/opencode_adapter.ts +94 -17
  197. package/package.json +15 -2
  198. package/src/cli.ts +66 -107
  199. package/src/db_adapter.ts +354 -0
  200. package/src/engine/account/account.sql.ts +39 -39
  201. package/src/engine/account/account.ts +456 -456
  202. package/src/engine/account/repo.ts +166 -166
  203. package/src/engine/account/schema.ts +99 -99
  204. package/src/engine/account/url.ts +8 -8
  205. package/src/engine/acp/README.md +174 -174
  206. package/src/engine/acp/agent.ts +1968 -1968
  207. package/src/engine/acp/runtime.ts +22 -22
  208. package/src/engine/acp/session.ts +122 -122
  209. package/src/engine/acp/types.ts +24 -24
  210. package/src/engine/agent/agent.ts +463 -463
  211. package/src/engine/agent/generate.txt +75 -75
  212. package/src/engine/agent/prompt/compaction.txt +9 -9
  213. package/src/engine/agent/prompt/explore.txt +18 -18
  214. package/src/engine/agent/prompt/scout.txt +36 -36
  215. package/src/engine/agent/prompt/summary.txt +11 -11
  216. package/src/engine/agent/prompt/title.txt +44 -44
  217. package/src/engine/agent/subagent-permissions.ts +34 -34
  218. package/src/engine/auth/index.ts +96 -96
  219. package/src/engine/background/background/job.ts +200 -200
  220. package/src/engine/background/job.ts +200 -200
  221. package/src/engine/bus/bus-event.ts +45 -45
  222. package/src/engine/bus/global.ts +22 -22
  223. package/src/engine/bus/index.ts +203 -203
  224. package/src/engine/command/command/index.ts +181 -181
  225. package/src/engine/command/command/template/initialize.txt +66 -66
  226. package/src/engine/command/command/template/review.txt +101 -101
  227. package/src/engine/command/index.ts +181 -181
  228. package/src/engine/command/template/initialize.txt +66 -66
  229. package/src/engine/command/template/review.txt +101 -101
  230. package/src/engine/config/agent.ts +172 -172
  231. package/src/engine/config/attachment.ts +25 -25
  232. package/src/engine/config/command.ts +62 -62
  233. package/src/engine/config/config.ts +833 -833
  234. package/src/engine/config/console-state.ts +14 -14
  235. package/src/engine/config/entry-name.ts +16 -16
  236. package/src/engine/config/error.ts +23 -23
  237. package/src/engine/config/formatter.ts +13 -13
  238. package/src/engine/config/layout.ts +6 -6
  239. package/src/engine/config/lsp.ts +43 -43
  240. package/src/engine/config/managed.ts +71 -71
  241. package/src/engine/config/markdown.ts +96 -96
  242. package/src/engine/config/mcp.ts +56 -56
  243. package/src/engine/config/model-id.ts +5 -5
  244. package/src/engine/config/parse.ts +79 -79
  245. package/src/engine/config/paths.ts +45 -45
  246. package/src/engine/config/permission.ts +58 -58
  247. package/src/engine/config/plugin.ts +84 -84
  248. package/src/engine/config/provider.ts +111 -111
  249. package/src/engine/config/reference.ts +23 -23
  250. package/src/engine/config/server.ts +19 -19
  251. package/src/engine/config/skills.ts +14 -14
  252. package/src/engine/config/variable.ts +90 -90
  253. package/src/engine/control-plane/adapters/index.ts +41 -41
  254. package/src/engine/control-plane/adapters/worktree.ts +96 -96
  255. package/src/engine/control-plane/dev/README.md +19 -19
  256. package/src/engine/control-plane/dev/debug-workspace-plugin.ts +73 -73
  257. package/src/engine/control-plane/schema.ts +14 -14
  258. package/src/engine/control-plane/types.ts +59 -59
  259. package/src/engine/control-plane/util.ts +39 -39
  260. package/src/engine/control-plane/workspace-adapter-runtime.ts +51 -51
  261. package/src/engine/control-plane/workspace-context.ts +26 -26
  262. package/src/engine/control-plane/workspace.sql.ts +20 -20
  263. package/src/engine/control-plane/workspace.ts +1072 -1072
  264. package/src/engine/data-migration.ts +161 -161
  265. package/src/engine/effect/app-runtime.ts +143 -143
  266. package/src/engine/effect/bootstrap-runtime.ts +29 -29
  267. package/src/engine/effect/bridge.ts +84 -84
  268. package/src/engine/effect/config-service.ts +67 -67
  269. package/src/engine/effect/instance-ref.ts +11 -11
  270. package/src/engine/effect/instance-registry.ts +12 -12
  271. package/src/engine/effect/instance-state.ts +72 -72
  272. package/src/engine/effect/promise.ts +17 -17
  273. package/src/engine/effect/run-service.ts +47 -47
  274. package/src/engine/effect/runner.ts +217 -217
  275. package/src/engine/effect/runtime-flags.ts +74 -74
  276. package/src/engine/effect/service-use.ts +38 -38
  277. package/src/engine/env/index.ts +37 -37
  278. package/src/engine/event-v2-bridge.ts +89 -89
  279. package/src/engine/file/file/ignore.ts +81 -81
  280. package/src/engine/file/file/index.ts +651 -651
  281. package/src/engine/file/file/protected.ts +59 -59
  282. package/src/engine/file/file/ripgrep.ts +481 -481
  283. package/src/engine/file/file/watcher.ts +167 -167
  284. package/src/engine/file/ignore.ts +81 -81
  285. package/src/engine/file/index.ts +651 -651
  286. package/src/engine/file/protected.ts +59 -59
  287. package/src/engine/file/ripgrep.ts +481 -481
  288. package/src/engine/file/watcher.ts +167 -167
  289. package/src/engine/format/format/formatter.ts +404 -404
  290. package/src/engine/format/format/index.ts +209 -209
  291. package/src/engine/format/formatter.ts +404 -404
  292. package/src/engine/format/index.ts +209 -209
  293. package/src/engine/git/git/index.ts +347 -347
  294. package/src/engine/git/index.ts +347 -347
  295. package/src/engine/id/id.ts +80 -80
  296. package/src/engine/ide/index.ts +70 -70
  297. package/src/engine/image/image/image.ts +176 -176
  298. package/src/engine/image/image.ts +176 -176
  299. package/src/engine/index.ts +251 -251
  300. package/src/engine/installation/index.ts +327 -327
  301. package/src/engine/lsp/client.ts +707 -707
  302. package/src/engine/lsp/diagnostic.ts +29 -29
  303. package/src/engine/lsp/language.ts +121 -121
  304. package/src/engine/lsp/launch.ts +21 -21
  305. package/src/engine/lsp/lsp/client.ts +707 -707
  306. package/src/engine/lsp/lsp/diagnostic.ts +29 -29
  307. package/src/engine/lsp/lsp/language.ts +121 -121
  308. package/src/engine/lsp/lsp/launch.ts +21 -21
  309. package/src/engine/lsp/lsp/lsp.ts +507 -507
  310. package/src/engine/lsp/lsp/server.ts +2064 -2064
  311. package/src/engine/lsp/lsp.ts +507 -507
  312. package/src/engine/lsp/server.ts +2064 -2064
  313. package/src/engine/mcp/auth.ts +146 -146
  314. package/src/engine/mcp/index.ts +958 -958
  315. package/src/engine/mcp/mcp/auth.ts +146 -146
  316. package/src/engine/mcp/mcp/index.ts +958 -958
  317. package/src/engine/mcp/mcp/oauth-callback.ts +232 -232
  318. package/src/engine/mcp/mcp/oauth-provider.ts +214 -214
  319. package/src/engine/mcp/oauth-callback.ts +232 -232
  320. package/src/engine/mcp/oauth-provider.ts +214 -214
  321. package/src/engine/node.ts +6 -6
  322. package/src/engine/patch/index.ts +689 -689
  323. package/src/engine/patch/patch/index.ts +689 -689
  324. package/src/engine/permission/arity.ts +163 -163
  325. package/src/engine/permission/evaluate.ts +15 -15
  326. package/src/engine/permission/index.ts +306 -306
  327. package/src/engine/permission/permission/arity.ts +163 -163
  328. package/src/engine/permission/permission/evaluate.ts +15 -15
  329. package/src/engine/permission/permission/index.ts +306 -306
  330. package/src/engine/permission/permission/schema.ts +13 -13
  331. package/src/engine/permission/schema.ts +13 -13
  332. package/src/engine/plugin/azure.ts +26 -26
  333. package/src/engine/plugin/cloudflare.ts +76 -76
  334. package/src/engine/plugin/codex.ts +622 -622
  335. package/src/engine/plugin/digitalocean.ts +411 -411
  336. package/src/engine/plugin/github-copilot/copilot.ts +394 -394
  337. package/src/engine/plugin/github-copilot/models.ts +196 -196
  338. package/src/engine/plugin/index.ts +295 -295
  339. package/src/engine/plugin/install.ts +439 -439
  340. package/src/engine/plugin/loader.ts +216 -216
  341. package/src/engine/plugin/meta.ts +188 -188
  342. package/src/engine/plugin/shared.ts +323 -323
  343. package/src/engine/project/bootstrap-service.ts +9 -9
  344. package/src/engine/project/bootstrap.ts +75 -75
  345. package/src/engine/project/instance-context.ts +24 -24
  346. package/src/engine/project/instance-layer.ts +11 -11
  347. package/src/engine/project/instance-runtime.ts +16 -16
  348. package/src/engine/project/instance-store.ts +193 -193
  349. package/src/engine/project/project.sql.ts +17 -17
  350. package/src/engine/project/project.ts +537 -537
  351. package/src/engine/project/schema.ts +13 -13
  352. package/src/engine/project/vcs.ts +405 -405
  353. package/src/engine/provider/auth.ts +225 -225
  354. package/src/engine/provider/error.ts +204 -204
  355. package/src/engine/provider/model-status.ts +8 -8
  356. package/src/engine/provider/provider.ts +1843 -1843
  357. package/src/engine/provider/schema.ts +30 -30
  358. package/src/engine/provider/transform.ts +1376 -1376
  359. package/src/engine/pty/index.ts +365 -365
  360. package/src/engine/pty/input.ts +24 -24
  361. package/src/engine/pty/pty/index.ts +365 -365
  362. package/src/engine/pty/pty/input.ts +24 -24
  363. package/src/engine/pty/pty/pty.bun.ts +26 -26
  364. package/src/engine/pty/pty/pty.node.ts +27 -27
  365. package/src/engine/pty/pty/pty.ts +25 -25
  366. package/src/engine/pty/pty/schema.ts +14 -14
  367. package/src/engine/pty/pty/ticket.ts +68 -68
  368. package/src/engine/pty/pty.bun.ts +26 -26
  369. package/src/engine/pty/pty.node.ts +27 -27
  370. package/src/engine/pty/pty.ts +25 -25
  371. package/src/engine/pty/schema.ts +14 -14
  372. package/src/engine/pty/ticket.ts +68 -68
  373. package/src/engine/question/index.ts +213 -213
  374. package/src/engine/question/question/index.ts +213 -213
  375. package/src/engine/question/question/schema.ts +10 -10
  376. package/src/engine/question/schema.ts +10 -10
  377. package/src/engine/reference/reference/reference.ts +241 -241
  378. package/src/engine/reference/reference/repository-cache.ts +147 -147
  379. package/src/engine/reference/reference.ts +241 -241
  380. package/src/engine/reference/repository-cache.ts +147 -147
  381. package/src/engine/session/compaction.ts +651 -651
  382. package/src/engine/session/instruction.ts +238 -238
  383. package/src/engine/session/llm.ts +459 -459
  384. package/src/engine/session/message-error.ts +14 -14
  385. package/src/engine/session/message-v2.ts +1202 -1202
  386. package/src/engine/session/message.ts +146 -146
  387. package/src/engine/session/overflow.ts +32 -32
  388. package/src/engine/session/processor.ts +823 -823
  389. package/src/engine/session/prompt/anthropic.txt +105 -105
  390. package/src/engine/session/prompt/beast.txt +147 -147
  391. package/src/engine/session/prompt/build-switch.txt +5 -5
  392. package/src/engine/session/prompt/codex.txt +79 -79
  393. package/src/engine/session/prompt/copilot-gpt-5.txt +143 -143
  394. package/src/engine/session/prompt/default.txt +105 -105
  395. package/src/engine/session/prompt/gemini.txt +155 -155
  396. package/src/engine/session/prompt/gpt.txt +107 -107
  397. package/src/engine/session/prompt/kimi.txt +95 -95
  398. package/src/engine/session/prompt/max-steps.txt +15 -15
  399. package/src/engine/session/prompt/plan-reminder-anthropic.txt +67 -67
  400. package/src/engine/session/prompt/plan.txt +26 -26
  401. package/src/engine/session/prompt/trinity.txt +97 -97
  402. package/src/engine/session/prompt.ts +66 -9
  403. package/src/engine/session/retry.ts +200 -200
  404. package/src/engine/session/revert.ts +162 -162
  405. package/src/engine/session/run-state.ts +153 -153
  406. package/src/engine/session/schema.ts +26 -26
  407. package/src/engine/session/session.sql.ts +137 -137
  408. package/src/engine/session/session.ts +1011 -1011
  409. package/src/engine/session/status.ts +94 -94
  410. package/src/engine/session/summary.ts +164 -164
  411. package/src/engine/session/system.ts +84 -84
  412. package/src/engine/session/todo.ts +81 -81
  413. package/src/engine/share/session.ts +61 -61
  414. package/src/engine/share/share-next.ts +376 -376
  415. package/src/engine/share/share.sql.ts +13 -13
  416. package/src/engine/shell/shell/shell.ts +215 -215
  417. package/src/engine/shell/shell.ts +215 -215
  418. package/src/engine/skill/discovery.ts +116 -116
  419. package/src/engine/skill/index.ts +336 -336
  420. package/src/engine/skill/prompt/customize-opencode.md +377 -377
  421. package/src/engine/skill/skill/discovery.ts +116 -116
  422. package/src/engine/skill/skill/index.ts +336 -336
  423. package/src/engine/skill/skill/prompt/customize-opencode.md +377 -377
  424. package/src/engine/snapshot/index.ts +762 -762
  425. package/src/engine/snapshot/snapshot/index.ts +762 -762
  426. package/src/engine/sync/README.md +179 -179
  427. package/src/engine/sync/event.sql.ts +17 -17
  428. package/src/engine/sync/index.ts +410 -410
  429. package/src/engine/sync/schema.ts +11 -11
  430. package/src/engine/temporary.ts +33 -33
  431. package/src/engine/tool/apply_patch.ts +313 -313
  432. package/src/engine/tool/apply_patch.txt +33 -33
  433. package/src/engine/tool/edit.ts +711 -711
  434. package/src/engine/tool/edit.txt +10 -10
  435. package/src/engine/tool/external-directory.ts +49 -49
  436. package/src/engine/tool/glob.ts +103 -103
  437. package/src/engine/tool/glob.txt +6 -6
  438. package/src/engine/tool/grep.ts +156 -156
  439. package/src/engine/tool/grep.txt +8 -8
  440. package/src/engine/tool/invalid.ts +21 -21
  441. package/src/engine/tool/json-schema.ts +164 -164
  442. package/src/engine/tool/lsp.ts +113 -113
  443. package/src/engine/tool/lsp.txt +24 -24
  444. package/src/engine/tool/mcp-websearch.ts +96 -96
  445. package/src/engine/tool/plan-enter.txt +14 -14
  446. package/src/engine/tool/plan-exit.txt +13 -13
  447. package/src/engine/tool/plan.ts +78 -78
  448. package/src/engine/tool/question.ts +44 -44
  449. package/src/engine/tool/question.txt +10 -10
  450. package/src/engine/tool/read.ts +337 -337
  451. package/src/engine/tool/read.txt +14 -14
  452. package/src/engine/tool/registry.ts +472 -472
  453. package/src/engine/tool/repo_clone.ts +80 -80
  454. package/src/engine/tool/repo_clone.txt +5 -5
  455. package/src/engine/tool/repo_overview.ts +279 -279
  456. package/src/engine/tool/repo_overview.txt +4 -4
  457. package/src/engine/tool/schema.ts +14 -14
  458. package/src/engine/tool/shell/id.ts +19 -19
  459. package/src/engine/tool/shell/prompt.ts +295 -295
  460. package/src/engine/tool/shell/shell.txt +77 -77
  461. package/src/engine/tool/shell.ts +647 -647
  462. package/src/engine/tool/skill.ts +75 -75
  463. package/src/engine/tool/skill.txt +5 -5
  464. package/src/engine/tool/task.ts +337 -337
  465. package/src/engine/tool/task.txt +58 -58
  466. package/src/engine/tool/task_status.ts +179 -179
  467. package/src/engine/tool/task_status.txt +13 -13
  468. package/src/engine/tool/todo.ts +57 -57
  469. package/src/engine/tool/todowrite.txt +167 -167
  470. package/src/engine/tool/tool/apply_patch.ts +313 -313
  471. package/src/engine/tool/tool/apply_patch.txt +33 -33
  472. package/src/engine/tool/tool/edit.ts +711 -711
  473. package/src/engine/tool/tool/edit.txt +10 -10
  474. package/src/engine/tool/tool/external-directory.ts +49 -49
  475. package/src/engine/tool/tool/glob.ts +103 -103
  476. package/src/engine/tool/tool/glob.txt +6 -6
  477. package/src/engine/tool/tool/grep.ts +156 -156
  478. package/src/engine/tool/tool/grep.txt +8 -8
  479. package/src/engine/tool/tool/invalid.ts +21 -21
  480. package/src/engine/tool/tool/json-schema.ts +164 -164
  481. package/src/engine/tool/tool/lsp.ts +113 -113
  482. package/src/engine/tool/tool/lsp.txt +24 -24
  483. package/src/engine/tool/tool/mcp-websearch.ts +96 -96
  484. package/src/engine/tool/tool/plan-enter.txt +14 -14
  485. package/src/engine/tool/tool/plan-exit.txt +13 -13
  486. package/src/engine/tool/tool/plan.ts +78 -78
  487. package/src/engine/tool/tool/question.ts +44 -44
  488. package/src/engine/tool/tool/question.txt +10 -10
  489. package/src/engine/tool/tool/read.ts +337 -337
  490. package/src/engine/tool/tool/read.txt +14 -14
  491. package/src/engine/tool/tool/registry.ts +472 -472
  492. package/src/engine/tool/tool/repo_clone.ts +80 -80
  493. package/src/engine/tool/tool/repo_clone.txt +5 -5
  494. package/src/engine/tool/tool/repo_overview.ts +279 -279
  495. package/src/engine/tool/tool/repo_overview.txt +4 -4
  496. package/src/engine/tool/tool/schema.ts +14 -14
  497. package/src/engine/tool/tool/shell/id.ts +19 -19
  498. package/src/engine/tool/tool/shell/prompt.ts +295 -295
  499. package/src/engine/tool/tool/shell/shell.txt +77 -77
  500. package/src/engine/tool/tool/shell.ts +647 -647
  501. package/src/engine/tool/tool/skill.ts +75 -75
  502. package/src/engine/tool/tool/skill.txt +5 -5
  503. package/src/engine/tool/tool/task.ts +337 -337
  504. package/src/engine/tool/tool/task.txt +58 -58
  505. package/src/engine/tool/tool/task_status.ts +179 -179
  506. package/src/engine/tool/tool/task_status.txt +13 -13
  507. package/src/engine/tool/tool/todo.ts +57 -57
  508. package/src/engine/tool/tool/todowrite.txt +167 -167
  509. package/src/engine/tool/tool/tool.ts +164 -164
  510. package/src/engine/tool/tool/truncate.ts +160 -160
  511. package/src/engine/tool/tool/truncation-dir.ts +4 -4
  512. package/src/engine/tool/tool/webfetch.ts +192 -192
  513. package/src/engine/tool/tool/webfetch.txt +13 -13
  514. package/src/engine/tool/tool/websearch.ts +143 -143
  515. package/src/engine/tool/tool/websearch.txt +14 -14
  516. package/src/engine/tool/tool/write.ts +104 -104
  517. package/src/engine/tool/tool/write.txt +8 -8
  518. package/src/engine/tool/tool.ts +164 -164
  519. package/src/engine/tool/truncate.ts +160 -160
  520. package/src/engine/tool/truncation-dir.ts +4 -4
  521. package/src/engine/tool/webfetch.ts +192 -192
  522. package/src/engine/tool/webfetch.txt +13 -13
  523. package/src/engine/tool/websearch.ts +143 -143
  524. package/src/engine/tool/websearch.txt +14 -14
  525. package/src/engine/tool/write.ts +104 -104
  526. package/src/engine/tool/write.txt +8 -8
  527. package/src/engine/util/archive.ts +17 -17
  528. package/src/engine/util/bom.ts +31 -31
  529. package/src/engine/util/data-url.ts +9 -9
  530. package/src/engine/util/defer.ts +10 -10
  531. package/src/engine/util/effect-http-client.ts +11 -11
  532. package/src/engine/util/error.ts +88 -88
  533. package/src/engine/util/filesystem.ts +252 -252
  534. package/src/engine/util/format.ts +20 -20
  535. package/src/engine/util/iife.ts +3 -3
  536. package/src/engine/util/lazy.ts +20 -20
  537. package/src/engine/util/local-context.ts +25 -25
  538. package/src/engine/util/locale.ts +86 -86
  539. package/src/engine/util/media.ts +26 -26
  540. package/src/engine/util/process.ts +176 -176
  541. package/src/engine/util/queue.ts +32 -32
  542. package/src/engine/util/record.ts +3 -3
  543. package/src/engine/util/repository.ts +158 -158
  544. package/src/engine/util/rpc.ts +66 -66
  545. package/src/engine/util/signal.ts +12 -12
  546. package/src/engine/util/timeout.ts +13 -13
  547. package/src/engine/util/token.ts +7 -7
  548. package/src/engine/util/util/archive.ts +17 -17
  549. package/src/engine/util/util/bom.ts +31 -31
  550. package/src/engine/util/util/data-url.ts +9 -9
  551. package/src/engine/util/util/defer.ts +10 -10
  552. package/src/engine/util/util/effect-http-client.ts +11 -11
  553. package/src/engine/util/util/error.ts +88 -88
  554. package/src/engine/util/util/filesystem.ts +252 -252
  555. package/src/engine/util/util/format.ts +20 -20
  556. package/src/engine/util/util/iife.ts +3 -3
  557. package/src/engine/util/util/lazy.ts +20 -20
  558. package/src/engine/util/util/local-context.ts +25 -25
  559. package/src/engine/util/util/locale.ts +86 -86
  560. package/src/engine/util/util/media.ts +26 -26
  561. package/src/engine/util/util/process.ts +176 -176
  562. package/src/engine/util/util/queue.ts +32 -32
  563. package/src/engine/util/util/record.ts +3 -3
  564. package/src/engine/util/util/repository.ts +158 -158
  565. package/src/engine/util/util/rpc.ts +66 -66
  566. package/src/engine/util/util/signal.ts +12 -12
  567. package/src/engine/util/util/timeout.ts +13 -13
  568. package/src/engine/util/util/token.ts +7 -7
  569. package/src/engine/util/util/which.ts +14 -14
  570. package/src/engine/util/util/wildcard.ts +59 -59
  571. package/src/engine/util/which.ts +14 -14
  572. package/src/engine/util/wildcard.ts +59 -59
  573. package/src/engine/worktree/index.ts +621 -621
  574. package/src/server.ts +121 -158
  575. package/src/tui.ts +485 -502
@@ -1,1072 +1,1072 @@
1
- import { Context, Effect, FiberMap, Iterable, Layer, Schema, Stream } from "effect"
2
- import { FetchHttpClient, HttpBody, HttpClient, HttpClientError, HttpClientRequest } from "effect/unstable/http"
3
- import { Database } from "@/storage/db"
4
- import { asc } from "drizzle-orm"
5
- import { eq } from "drizzle-orm"
6
- import { inArray } from "drizzle-orm"
7
- import { Project } from "@/project/project"
8
- import { BusEvent } from "@/bus/bus-event"
9
- import { GlobalBus } from "@/bus/global"
10
- import { Auth } from "@/auth"
11
- import { SyncEvent } from "@/sync"
12
- import { EventSequenceTable, EventTable } from "@/sync/event.sql"
13
- import { AppFileSystem } from "@opencode-ai/core/filesystem"
14
- import * as Log from "@opencode-ai/core/util/log"
15
- import { RuntimeFlags } from "@/effect/runtime-flags"
16
- import { ProjectID } from "@/project/schema"
17
- import { Slug } from "@opencode-ai/core/util/slug"
18
- import { WorkspaceTable } from "./workspace.sql"
19
- import { getAdapter, registeredAdapters } from "./adapters"
20
- import { type Target, type WorkspaceInfo, WorkspaceInfo as WorkspaceInfoSchema } from "./types"
21
- import { WorkspaceID } from "./schema"
22
- import { Session } from "@/session/session"
23
- import { SessionPrompt } from "@/session/prompt"
24
- import { SessionTable } from "@/session/session.sql"
25
- import { SessionID } from "@/session/schema"
26
- import { NotFoundError } from "@/storage/storage"
27
- import { errorData } from "@/util/error"
28
- import { waitEvent } from "./util"
29
- import { WorkspaceRef } from "@/effect/instance-ref"
30
- import { Vcs } from "@/project/vcs"
31
- import { InstanceStore } from "@/project/instance-store"
32
- import { InstanceBootstrap } from "@/project/bootstrap"
33
- import { WorkspaceAdapterRuntime } from "./workspace-adapter-runtime"
34
-
35
- export const Info = Schema.Struct({
36
- ...WorkspaceInfoSchema.fields,
37
- timeUsed: Schema.Number,
38
- }).annotate({ identifier: "Workspace" })
39
- export type Info = WorkspaceInfo & { timeUsed: number }
40
-
41
- export const ConnectionStatus = Schema.Struct({
42
- workspaceID: WorkspaceID,
43
- status: Schema.Literals(["connected", "connecting", "disconnected", "error"]),
44
- })
45
- export type ConnectionStatus = Schema.Schema.Type<typeof ConnectionStatus>
46
-
47
- export const Event = {
48
- Ready: BusEvent.define(
49
- "workspace.ready",
50
- Schema.Struct({
51
- name: Schema.String,
52
- }),
53
- ),
54
- Failed: BusEvent.define(
55
- "workspace.failed",
56
- Schema.Struct({
57
- message: Schema.String,
58
- }),
59
- ),
60
- Status: BusEvent.define("workspace.status", ConnectionStatus),
61
- }
62
-
63
- function fromRow(row: typeof WorkspaceTable.$inferSelect): Info {
64
- return {
65
- id: row.id,
66
- type: row.type,
67
- branch: row.branch,
68
- name: row.name,
69
- directory: row.directory,
70
- extra: row.extra,
71
- projectID: row.project_id,
72
- timeUsed: row.time_used,
73
- }
74
- }
75
-
76
- const db = <T>(fn: (d: Parameters<typeof Database.use>[0] extends (trx: infer D) => any ? D : never) => T) =>
77
- Effect.sync(() => Database.use(fn))
78
-
79
- const log = Log.create({ service: "workspace-sync" })
80
-
81
- export const CreateInput = Schema.Struct({
82
- id: Schema.optional(WorkspaceID),
83
- type: Info.fields.type,
84
- branch: Info.fields.branch,
85
- projectID: ProjectID,
86
- extra: Schema.optional(Info.fields.extra),
87
- })
88
- export type CreateInput = Schema.Schema.Type<typeof CreateInput>
89
-
90
- export const SessionWarpInput = Schema.Struct({
91
- workspaceID: Schema.NullOr(WorkspaceID),
92
- sessionID: SessionID,
93
- copyChanges: Schema.optional(Schema.Boolean),
94
- })
95
- export type SessionWarpInput = Schema.Schema.Type<typeof SessionWarpInput>
96
-
97
- export class SyncHttpError extends Schema.TaggedErrorClass<SyncHttpError>()("WorkspaceSyncHttpError", {
98
- message: Schema.String,
99
- status: Schema.Number,
100
- body: Schema.optional(Schema.String),
101
- }) {}
102
-
103
- export class WorkspaceNotFoundError extends Schema.TaggedErrorClass<WorkspaceNotFoundError>()(
104
- "WorkspaceNotFoundError",
105
- {
106
- message: Schema.String,
107
- workspaceID: WorkspaceID,
108
- },
109
- ) {}
110
-
111
- export class SessionEventsNotFoundError extends Schema.TaggedErrorClass<SessionEventsNotFoundError>()(
112
- "WorkspaceSessionEventsNotFoundError",
113
- {
114
- message: Schema.String,
115
- sessionID: SessionID,
116
- },
117
- ) {}
118
-
119
- export class SessionWarpHttpError extends Schema.TaggedErrorClass<SessionWarpHttpError>()(
120
- "WorkspaceSessionWarpHttpError",
121
- {
122
- message: Schema.String,
123
- workspaceID: WorkspaceID,
124
- sessionID: SessionID,
125
- status: Schema.Number,
126
- body: Schema.String,
127
- },
128
- ) {}
129
-
130
- export class SyncTimeoutError extends Schema.TaggedErrorClass<SyncTimeoutError>()("WorkspaceSyncTimeoutError", {
131
- message: Schema.String,
132
- state: Schema.Record(Schema.String, Schema.Number),
133
- }) {}
134
-
135
- export class SyncAbortedError extends Schema.TaggedErrorClass<SyncAbortedError>()("WorkspaceSyncAbortedError", {
136
- message: Schema.String,
137
- cause: Schema.optional(Schema.Defect),
138
- }) {}
139
-
140
- type CreateError = Auth.AuthError
141
- type SessionWarpError =
142
- | WorkspaceNotFoundError
143
- | SessionEventsNotFoundError
144
- | SessionWarpHttpError
145
- | Vcs.PatchApplyError
146
- | HttpClientError.HttpClientError
147
- type WaitForSyncError = SyncTimeoutError | SyncAbortedError
148
- type SyncLoopError = SyncHttpError | HttpClientError.HttpClientError
149
-
150
- export interface Interface {
151
- readonly create: (input: CreateInput) => Effect.Effect<Info, CreateError>
152
- readonly sessionWarp: (input: SessionWarpInput) => Effect.Effect<void, SessionWarpError>
153
- readonly list: (project: Project.Info) => Effect.Effect<Info[]>
154
- readonly syncList: (project: Project.Info) => Effect.Effect<void>
155
- readonly get: (id: WorkspaceID) => Effect.Effect<Info | undefined>
156
- readonly remove: (id: WorkspaceID) => Effect.Effect<Info | undefined>
157
- readonly status: () => Effect.Effect<ConnectionStatus[]>
158
- readonly isSyncing: (workspaceID: WorkspaceID) => Effect.Effect<boolean>
159
- readonly waitForSync: (
160
- workspaceID: WorkspaceID,
161
- state: Record<string, number>,
162
- signal?: AbortSignal,
163
- ) => Effect.Effect<void, WaitForSyncError>
164
- readonly startWorkspaceSyncing: (projectID: ProjectID) => Effect.Effect<void>
165
- }
166
-
167
- export class Service extends Context.Service<Service, Interface>()("@opencode/Workspace") {}
168
-
169
- export const layer = Layer.effect(
170
- Service,
171
- Effect.gen(function* () {
172
- const auth = yield* Auth.Service
173
- const session = yield* Session.Service
174
- const prompt = yield* SessionPrompt.Service
175
- const http = yield* HttpClient.HttpClient
176
- const sync = yield* SyncEvent.Service
177
- const vcs = yield* Vcs.Service
178
- const flags = yield* RuntimeFlags.Service
179
- const fs = yield* AppFileSystem.Service
180
- const connections = new Map<WorkspaceID, ConnectionStatus>()
181
- const syncFibers = yield* FiberMap.make<WorkspaceID, void, SyncLoopError>()
182
-
183
- const setStatus = (id: WorkspaceID, status: ConnectionStatus["status"]) => {
184
- const prev = connections.get(id)
185
- if (prev?.status === status) return
186
- const next = { workspaceID: id, status }
187
- connections.set(id, next)
188
-
189
- GlobalBus.emit("event", {
190
- directory: "global",
191
- workspace: id,
192
- payload: {
193
- type: Event.Status.type,
194
- properties: next,
195
- },
196
- })
197
- }
198
-
199
- const connectSSE = Effect.fn("Workspace.connectSSE")(function* (
200
- url: URL | string,
201
- headers: HeadersInit | undefined,
202
- ) {
203
- const response = yield* http.execute(
204
- HttpClientRequest.get(route(url, "/global/event"), {
205
- headers: new Headers(headers),
206
- accept: "text/event-stream",
207
- }),
208
- )
209
- if (response.status < 200 || response.status >= 300) {
210
- return yield* new SyncHttpError({
211
- message: `Workspace sync HTTP failure: ${response.status}`,
212
- status: response.status,
213
- })
214
- }
215
- return response.stream
216
- })
217
-
218
- const parseSSE = Effect.fn("Workspace.parseSSE")(function* (
219
- stream: Stream.Stream<Uint8Array, unknown>,
220
- onEvent: (event: unknown) => Effect.Effect<void>,
221
- ) {
222
- yield* stream.pipe(
223
- Stream.decodeText(),
224
- Stream.splitLines,
225
- Stream.mapAccum(
226
- () => ({ data: [] as string[], id: undefined as string | undefined, retry: 1000 }),
227
- (state, line) => {
228
- if (line === "") {
229
- if (!state.data.length) return [state, []]
230
- return [{ ...state, data: [] }, [{ data: state.data.join("\n"), id: state.id, retry: state.retry }]]
231
- }
232
-
233
- const index = line.indexOf(":")
234
- const field = index === -1 ? line : line.slice(0, index)
235
- const value = index === -1 ? "" : line.slice(index + (line[index + 1] === " " ? 2 : 1))
236
-
237
- if (field === "data") return [{ ...state, data: [...state.data, value] }, []]
238
- if (field === "id") return [{ ...state, id: value }, []]
239
- if (field === "retry") {
240
- const retry = Number.parseInt(value, 10)
241
- return [Number.isNaN(retry) ? state : { ...state, retry }, []]
242
- }
243
- return [state, []]
244
- },
245
- {
246
- onHalt: (state) =>
247
- state.data.length ? [{ data: state.data.join("\n"), id: state.id, retry: state.retry }] : [],
248
- },
249
- ),
250
- Stream.map((event) => {
251
- try {
252
- return JSON.parse(event.data) as unknown
253
- } catch {
254
- return {
255
- type: "sse.message",
256
- properties: {
257
- data: event.data,
258
- id: event.id || undefined,
259
- retry: event.retry,
260
- },
261
- }
262
- }
263
- }),
264
- Stream.runForEach(onEvent),
265
- )
266
- })
267
-
268
- const runInWorkspace = <A, E, R>(input: {
269
- workspaceID?: WorkspaceID
270
- local: () => Effect.Effect<A, E, R>
271
- remote: (input: {
272
- workspace: Info
273
- target: Extract<Target, { type: "remote" }>
274
- }) => HttpClientRequest.HttpClientRequest
275
- fallback: A
276
- response?: "json" | "text"
277
- }) =>
278
- Effect.gen(function* () {
279
- if (!input.workspaceID) return yield* input.local()
280
-
281
- const workspace = yield* get(input.workspaceID)
282
- if (!workspace) return input.fallback
283
-
284
- const target = yield* WorkspaceAdapterRuntime.target(workspace)
285
-
286
- if (target.type === "local") {
287
- const store = yield* InstanceStore.Service
288
- return yield* store.provide({ directory: target.directory }, input.local())
289
- }
290
-
291
- const response = yield* http.execute(input.remote({ workspace, target })).pipe(
292
- Effect.catch((error) =>
293
- Effect.sync(() => {
294
- log.warn("workspace target request failed", {
295
- workspaceID: workspace.id,
296
- error: errorData(error),
297
- })
298
- }),
299
- ),
300
- )
301
- if (!response) return input.fallback
302
- if (response.status < 200 || response.status >= 300) {
303
- const body = yield* response.text.pipe(Effect.catch(() => Effect.succeed("")))
304
- log.warn("workspace target request failed", {
305
- workspaceID: workspace.id,
306
- status: response.status,
307
- body,
308
- })
309
- return input.fallback
310
- }
311
-
312
- const body = input.response === "text" ? response.text : response.json
313
- return yield* body.pipe(
314
- Effect.map((result) => result as A),
315
- Effect.catch((error) =>
316
- Effect.sync(() => {
317
- log.warn("workspace target response decode failed", {
318
- workspaceID: workspace.id,
319
- error: errorData(error),
320
- })
321
- return input.fallback
322
- }),
323
- ),
324
- )
325
- })
326
-
327
- const syncHistory = Effect.fn("Workspace.syncHistory")(function* (
328
- space: Info,
329
- url: URL | string,
330
- headers: HeadersInit | undefined,
331
- ) {
332
- const sessionIDs = yield* db((db) =>
333
- db
334
- .select({ id: SessionTable.id })
335
- .from(SessionTable)
336
- .where(eq(SessionTable.workspace_id, space.id))
337
- .all()
338
- .map((row) => row.id),
339
- )
340
- const state = sessionIDs.length
341
- ? Object.fromEntries(
342
- (yield* db((db) =>
343
- db.select().from(EventSequenceTable).where(inArray(EventSequenceTable.aggregate_id, sessionIDs)).all(),
344
- )).map((row) => [row.aggregate_id, row.seq]),
345
- )
346
- : {}
347
-
348
- log.info("syncing workspace history", {
349
- workspaceID: space.id,
350
- sessions: sessionIDs.length,
351
- known: Object.keys(state).length,
352
- })
353
-
354
- const response = yield* http.execute(
355
- HttpClientRequest.post(route(url, "/sync/history"), {
356
- headers: new Headers(headers),
357
- body: HttpBody.jsonUnsafe(state),
358
- }),
359
- )
360
-
361
- if (response.status < 200 || response.status >= 300) {
362
- const body = yield* response.text
363
- return yield* new SyncHttpError({
364
- message: `Workspace history HTTP failure: ${response.status} ${body}`,
365
- status: response.status,
366
- body,
367
- })
368
- }
369
-
370
- const events = (yield* response.json) as HistoryEvent[]
371
-
372
- log.info("workspace history synced", {
373
- workspaceID: space.id,
374
- events: events.length,
375
- })
376
-
377
- yield* Effect.forEach(
378
- events,
379
- (event) =>
380
- sync
381
- .replay(
382
- {
383
- id: event.id,
384
- aggregateID: event.aggregate_id,
385
- seq: event.seq,
386
- type: event.type,
387
- data: event.data,
388
- },
389
- { publish: true },
390
- )
391
- .pipe(Effect.provideService(WorkspaceRef, space.id)),
392
- { discard: true },
393
- )
394
- })
395
-
396
- const syncWorkspaceLoop = Effect.fn("Workspace.syncWorkspaceLoop")(function* (space: Info) {
397
- const target = yield* WorkspaceAdapterRuntime.target(space)
398
-
399
- if (target.type === "local") return
400
-
401
- let attempt = 0
402
-
403
- while (true) {
404
- log.info("connecting to global sync", { workspace: space.name })
405
- setStatus(space.id, "connecting")
406
-
407
- const stream = yield* connectSSE(target.url, target.headers).pipe(
408
- Effect.tap(() => syncHistory(space, target.url, target.headers)),
409
- Effect.catch((err) =>
410
- Effect.sync(() => {
411
- setStatus(space.id, "error")
412
- log.info("failed to connect to global sync", {
413
- workspace: space.name,
414
- err,
415
- })
416
- return null
417
- }),
418
- ),
419
- )
420
-
421
- if (stream) {
422
- attempt = 0
423
-
424
- log.info("global sync connected", { workspace: space.name })
425
- setStatus(space.id, "connected")
426
-
427
- yield* parseSSE(stream, (evt) =>
428
- Effect.gen(function* () {
429
- if (!evt || typeof evt !== "object" || !("payload" in evt)) return
430
- const payload = evt.payload as { type?: string; syncEvent?: SyncEvent.SerializedEvent }
431
- if (payload.type === "server.heartbeat") return
432
-
433
- if (payload.type === "sync" && payload.syncEvent) {
434
- const failed = yield* sync.replay(payload.syncEvent).pipe(
435
- Effect.as(false),
436
- Effect.catchCause((error) =>
437
- Effect.sync(() => {
438
- log.info("failed to replay global event", {
439
- workspaceID: space.id,
440
- error,
441
- })
442
- return true
443
- }),
444
- ),
445
- )
446
- if (failed) return
447
- }
448
-
449
- try {
450
- const event = evt as { directory?: string; project?: string; payload: unknown }
451
- GlobalBus.emit("event", {
452
- directory: event.directory,
453
- project: event.project,
454
- workspace: space.id,
455
- payload: event.payload,
456
- })
457
- } catch (error) {
458
- log.info("failed to replay global event", {
459
- workspaceID: space.id,
460
- error,
461
- })
462
- }
463
- }),
464
- )
465
-
466
- log.info("disconnected from global sync: " + space.id)
467
- setStatus(space.id, "disconnected")
468
- }
469
-
470
- // Back off reconnect attempts up to 2 minutes while the workspace
471
- // stays unavailable.
472
- yield* Effect.sleep(`${Math.min(120_000, 1_000 * 2 ** attempt)} millis`)
473
- attempt += 1
474
- }
475
- })
476
-
477
- const startSync = Effect.fn("Workspace.startSync")(function* (space: Info) {
478
- if (!flags.experimentalWorkspaces) return
479
-
480
- const target = yield* WorkspaceAdapterRuntime.target(space).pipe(
481
- Effect.catch((error) =>
482
- Effect.sync(() => {
483
- setStatus(space.id, "error")
484
- log.warn("workspace target failed", {
485
- workspaceID: space.id,
486
- error: errorData(error),
487
- })
488
- return null
489
- }),
490
- ),
491
- )
492
- if (!target) return
493
-
494
- if (target.type === "local") {
495
- setStatus(space.id, (yield* fs.existsSafe(target.directory)) ? "connected" : "error")
496
- return
497
- }
498
-
499
- const exists = yield* FiberMap.has(syncFibers, space.id)
500
- if (exists && connections.get(space.id)?.status !== "error") return
501
-
502
- setStatus(space.id, "disconnected")
503
-
504
- yield* FiberMap.run(
505
- syncFibers,
506
- space.id,
507
- // TODO: look into `tapError` to set the status but still
508
- // allow the fiber to fail and automatically get removed
509
- syncWorkspaceLoop(space).pipe(
510
- Effect.catch((error) =>
511
- Effect.sync(() => {
512
- setStatus(space.id, "error")
513
- log.warn("workspace listener failed", {
514
- workspaceID: space.id,
515
- error,
516
- })
517
- }),
518
- ),
519
- ),
520
- )
521
- })
522
-
523
- const stopSync = Effect.fn("Workspace.stopSync")(function* (id: WorkspaceID) {
524
- yield* FiberMap.remove(syncFibers, id)
525
- connections.delete(id)
526
- })
527
-
528
- const create = Effect.fn("Workspace.create")(function* (input: CreateInput) {
529
- const id = WorkspaceID.ascending(input.id)
530
- const adapter = getAdapter(input.projectID, input.type)
531
- const config = yield* WorkspaceAdapterRuntime.configure(adapter, {
532
- ...input,
533
- id,
534
- name: Slug.create(),
535
- directory: null,
536
- extra: input.extra ?? null,
537
- })
538
-
539
- const info: Info = {
540
- id,
541
- type: config.type,
542
- branch: config.branch ?? null,
543
- name: config.name ?? null,
544
- directory: config.directory ?? null,
545
- extra: config.extra ?? null,
546
- projectID: input.projectID,
547
- timeUsed: Date.now(),
548
- }
549
-
550
- yield* db((db) => {
551
- db.insert(WorkspaceTable)
552
- .values({
553
- id: info.id,
554
- type: info.type,
555
- branch: info.branch,
556
- name: info.name,
557
- directory: info.directory,
558
- extra: info.extra,
559
- project_id: info.projectID,
560
- time_used: info.timeUsed,
561
- })
562
- .run()
563
- })
564
-
565
- const env = {
566
- OPENCODE_AUTH_CONTENT: JSON.stringify(yield* auth.all()),
567
- OPENCODE_WORKSPACE_ID: config.id,
568
- OPENCODE_EXPERIMENTAL_WORKSPACES: "true",
569
- OTEL_EXPORTER_OTLP_HEADERS: process.env.OTEL_EXPORTER_OTLP_HEADERS,
570
- OTEL_EXPORTER_OTLP_ENDPOINT: process.env.OTEL_EXPORTER_OTLP_ENDPOINT,
571
- OTEL_RESOURCE_ATTRIBUTES: process.env.OTEL_RESOURCE_ATTRIBUTES,
572
- }
573
-
574
- yield* WorkspaceAdapterRuntime.create(adapter, config, env)
575
- yield* Effect.all(
576
- [
577
- waitEvent({
578
- timeout: TIMEOUT,
579
- fn(event) {
580
- if (event.workspace === info.id && event.payload.type === Event.Status.type) {
581
- const { status } = event.payload.properties
582
- return status === "error" || status === "connected"
583
- }
584
- return false
585
- },
586
- }),
587
- startSync(info),
588
- ],
589
- { concurrency: 2, discard: true },
590
- )
591
-
592
- return info
593
- })
594
-
595
- const sessionWarp = Effect.fn("Workspace.sessionWarp")(function* (input: SessionWarpInput) {
596
- return yield* Effect.gen(function* () {
597
- log.info("session warp requested", {
598
- workspaceID: input.workspaceID,
599
- sessionID: input.sessionID,
600
- })
601
-
602
- const current = yield* db((db) =>
603
- db
604
- .select({ workspaceID: SessionTable.workspace_id })
605
- .from(SessionTable)
606
- .where(eq(SessionTable.id, input.sessionID))
607
- .get(),
608
- )
609
-
610
- if (current?.workspaceID) {
611
- const previous = yield* get(current.workspaceID)
612
- if (previous) {
613
- const target = yield* WorkspaceAdapterRuntime.target(previous)
614
-
615
- if (target.type === "remote") {
616
- yield* syncHistory(previous, target.url, target.headers).pipe(
617
- Effect.catch((error) =>
618
- Effect.sync(() => {
619
- log.warn("session warp final source sync failed", {
620
- workspaceID: previous.id,
621
- sessionID: input.sessionID,
622
- error: errorData(error),
623
- })
624
- }),
625
- ),
626
- )
627
- } else {
628
- yield* prompt.cancel(input.sessionID)
629
- }
630
-
631
- // "claim" this session so any future events coming from
632
- // the old workspace are ignored
633
- yield* sync.claim(input.sessionID, input.workspaceID ?? previous.projectID)
634
- }
635
- }
636
-
637
- const sourcePatch =
638
- input.copyChanges && current?.workspaceID
639
- ? yield* runInWorkspace({
640
- workspaceID: current?.workspaceID ?? undefined,
641
- local: () => vcs.diffRaw(),
642
- remote: ({ target }) =>
643
- HttpClientRequest.get(route(target.url, "/vcs/diff/raw"), {
644
- headers: new Headers(target.headers),
645
- }),
646
- fallback: "",
647
- response: "text",
648
- }).pipe(Effect.provide(InstanceStore.defaultLayer.pipe(Layer.provide(InstanceBootstrap.defaultLayer))))
649
- : ""
650
-
651
- if (sourcePatch) {
652
- // Attempt to apply the file changes to the new workspace.
653
- // We intentionally do first so if it fails we don't warp
654
- // the session.
655
- yield* runInWorkspace({
656
- workspaceID: input.workspaceID ?? undefined,
657
- local: () => vcs.apply({ patch: sourcePatch }),
658
- remote: ({ target }) =>
659
- HttpClientRequest.post(route(target.url, "/vcs/apply"), {
660
- headers: new Headers(target.headers),
661
- body: HttpBody.jsonUnsafe({ patch: sourcePatch }),
662
- }),
663
- fallback: { applied: false },
664
- }).pipe(Effect.provide(InstanceStore.defaultLayer.pipe(Layer.provide(InstanceBootstrap.defaultLayer))))
665
- }
666
-
667
- if (input.workspaceID === null) {
668
- yield* sync.run(Session.Event.Updated, {
669
- sessionID: input.sessionID,
670
- info: {
671
- workspaceID: null,
672
- },
673
- })
674
-
675
- log.info("session warp complete", {
676
- workspaceID: input.workspaceID,
677
- sessionID: input.sessionID,
678
- target: "local",
679
- })
680
- return
681
- }
682
-
683
- const workspaceID = input.workspaceID
684
- const space = yield* get(workspaceID)
685
- if (!space)
686
- return yield* new WorkspaceNotFoundError({
687
- message: `Workspace not found: ${workspaceID}`,
688
- workspaceID,
689
- })
690
-
691
- const target = yield* WorkspaceAdapterRuntime.target(space)
692
-
693
- if (target.type === "local") {
694
- yield* sync.run(Session.Event.Updated, {
695
- sessionID: input.sessionID,
696
- info: {
697
- workspaceID: input.workspaceID,
698
- },
699
- })
700
-
701
- log.info("session warp complete", {
702
- workspaceID: input.workspaceID,
703
- sessionID: input.sessionID,
704
- target: target.directory,
705
- })
706
- return
707
- }
708
-
709
- const rows = yield* db((db) =>
710
- db
711
- .select({
712
- id: EventTable.id,
713
- aggregateID: EventTable.aggregate_id,
714
- seq: EventTable.seq,
715
- type: EventTable.type,
716
- data: EventTable.data,
717
- })
718
- .from(EventTable)
719
- .where(eq(EventTable.aggregate_id, input.sessionID))
720
- .orderBy(asc(EventTable.seq))
721
- .all(),
722
- )
723
- if (rows.length === 0)
724
- return yield* new SessionEventsNotFoundError({
725
- message: `No events found for session: ${input.sessionID}`,
726
- sessionID: input.sessionID,
727
- })
728
-
729
- const batches = Iterable.chunksOf(rows, 10)
730
- const total = Iterable.size(batches)
731
-
732
- log.info("session warp prepared", {
733
- workspaceID: input.workspaceID,
734
- sessionID: input.sessionID,
735
- target: String(route(target.url, "/sync/replay")),
736
- events: rows.length,
737
- batches: total,
738
- first: rows[0]?.seq,
739
- last: rows.at(-1)?.seq,
740
- })
741
-
742
- yield* Effect.forEach(
743
- batches,
744
- (events, i) =>
745
- Effect.gen(function* () {
746
- const response = yield* http.execute(
747
- HttpClientRequest.post(route(target.url, "/sync/replay"), {
748
- headers: new Headers(target.headers),
749
- body: HttpBody.jsonUnsafe({
750
- directory: space.directory ?? "",
751
- events,
752
- }),
753
- }),
754
- )
755
-
756
- if (response.status < 200 || response.status >= 300) {
757
- const body = yield* response.text
758
- log.error("session warp batch failed", {
759
- workspaceID: input.workspaceID,
760
- sessionID: input.sessionID,
761
- step: i + 1,
762
- total,
763
- status: response.status,
764
- body,
765
- })
766
- return yield* new SessionWarpHttpError({
767
- message: `Failed to warp session ${input.sessionID} into workspace ${workspaceID}: HTTP ${response.status} ${body}`,
768
- workspaceID,
769
- sessionID: input.sessionID,
770
- status: response.status,
771
- body,
772
- })
773
- }
774
-
775
- log.info("session warp batch posted", {
776
- workspaceID: input.workspaceID,
777
- sessionID: input.sessionID,
778
- step: i + 1,
779
- total,
780
- status: response.status,
781
- })
782
- }),
783
- { discard: true },
784
- )
785
-
786
- const response = yield* http.execute(
787
- HttpClientRequest.post(route(target.url, "/sync/steal"), {
788
- headers: new Headers(target.headers),
789
- body: HttpBody.jsonUnsafe({ sessionID: input.sessionID }),
790
- }),
791
- )
792
- if (response.status < 200 || response.status >= 300) {
793
- const body = yield* response.text
794
- log.error("session warp steal failed", {
795
- workspaceID: input.workspaceID,
796
- sessionID: input.sessionID,
797
- status: response.status,
798
- body,
799
- })
800
- return yield* new SessionWarpHttpError({
801
- message: `Failed to steal session ${input.sessionID} into workspace ${workspaceID}: HTTP ${response.status} ${body}`,
802
- workspaceID,
803
- sessionID: input.sessionID,
804
- status: response.status,
805
- body,
806
- })
807
- }
808
-
809
- log.info("session warp complete", {
810
- workspaceID: input.workspaceID,
811
- sessionID: input.sessionID,
812
- batches: total,
813
- })
814
- }).pipe(
815
- Effect.tapError((err) =>
816
- Effect.sync(() =>
817
- log.error("session warp failed", {
818
- workspaceID: input.workspaceID,
819
- sessionID: input.sessionID,
820
- error: errorData(err),
821
- }),
822
- ),
823
- ),
824
- )
825
- })
826
-
827
- const list = Effect.fn("Workspace.list")(function* (project: Project.Info) {
828
- return yield* db((db) =>
829
- db
830
- .select()
831
- .from(WorkspaceTable)
832
- .where(eq(WorkspaceTable.project_id, project.id))
833
- .all()
834
- .map(fromRow)
835
- .sort((a, b) => a.id.localeCompare(b.id)),
836
- )
837
- })
838
-
839
- const syncList = Effect.fn("Workspace.syncList")(function* (project: Project.Info) {
840
- const names = new Set((yield* list(project)).map((workspace) => workspace.name))
841
- const discovered = yield* Effect.forEach(
842
- registeredAdapters(project.id),
843
- ([type, adapter]) =>
844
- WorkspaceAdapterRuntime.list(adapter).pipe(
845
- Effect.catchCause((error) =>
846
- Effect.sync(() => {
847
- log.warn("workspace adapter list failed", { type, error })
848
- return []
849
- }),
850
- ),
851
- ),
852
- { concurrency: "unbounded" },
853
- ).pipe(Effect.map((items) => items.flat()))
854
-
855
- yield* Effect.forEach(
856
- discovered,
857
- (item) =>
858
- Effect.gen(function* () {
859
- if (names.has(item.name)) return
860
- names.add(item.name)
861
-
862
- const info: Info = {
863
- id: WorkspaceID.ascending(),
864
- type: item.type,
865
- branch: item.branch,
866
- name: item.name,
867
- directory: item.directory,
868
- extra: item.extra,
869
- projectID: item.projectID,
870
- timeUsed: Date.now(),
871
- }
872
-
873
- yield* db((db) => {
874
- db.insert(WorkspaceTable)
875
- .values({
876
- id: info.id,
877
- type: info.type,
878
- branch: info.branch,
879
- name: info.name,
880
- directory: info.directory,
881
- extra: info.extra,
882
- project_id: info.projectID,
883
- time_used: info.timeUsed,
884
- })
885
- .run()
886
- })
887
-
888
- yield* startSync(info)
889
- }),
890
- { concurrency: 1 },
891
- )
892
- })
893
-
894
- const get = Effect.fn("Workspace.get")(function* (id: WorkspaceID) {
895
- const row = yield* db((db) => db.select().from(WorkspaceTable).where(eq(WorkspaceTable.id, id)).get())
896
- if (!row) return
897
- return fromRow(row)
898
- })
899
-
900
- const remove = Effect.fn("Workspace.remove")(function* (id: WorkspaceID) {
901
- const sessions = yield* db((db) =>
902
- db
903
- .select({ id: SessionTable.id, parentID: SessionTable.parent_id })
904
- .from(SessionTable)
905
- .where(eq(SessionTable.workspace_id, id))
906
- .all(),
907
- )
908
- const sessionIDs = new Set(sessions.map((sessionInfo) => sessionInfo.id))
909
- yield* Effect.forEach(
910
- sessions.filter((sessionInfo) => !sessionInfo.parentID || !sessionIDs.has(sessionInfo.parentID)),
911
- (sessionInfo) =>
912
- session.remove(sessionInfo.id).pipe(Effect.catchIf(NotFoundError.isInstance, () => Effect.void)),
913
- { discard: true },
914
- )
915
-
916
- const row = yield* db((db) => db.select().from(WorkspaceTable).where(eq(WorkspaceTable.id, id)).get())
917
- if (!row) return
918
-
919
- yield* stopSync(id)
920
-
921
- const info = fromRow(row)
922
- yield* Effect.catchCause(
923
- Effect.gen(function* () {
924
- yield* WorkspaceAdapterRuntime.remove(info)
925
- }),
926
- () =>
927
- Effect.sync(() => {
928
- log.error("adapter not available when removing workspace", { type: row.type })
929
- }),
930
- )
931
-
932
- yield* db((db) => db.delete(WorkspaceTable).where(eq(WorkspaceTable.id, id)).run())
933
- return info
934
- })
935
-
936
- const status = Effect.fn("Workspace.status")(function* () {
937
- return [...connections.values()]
938
- })
939
-
940
- const isSyncing = Effect.fn("Workspace.isSyncing")(function* (workspaceID: WorkspaceID) {
941
- const exists = yield* FiberMap.has(syncFibers, workspaceID)
942
- return exists && connections.get(workspaceID)?.status !== "error"
943
- })
944
-
945
- const waitForSync = Effect.fn("Workspace.waitForSync")(function* (
946
- workspaceID: WorkspaceID,
947
- state: Record<string, number>,
948
- signal?: AbortSignal,
949
- ) {
950
- if (synced(state)) return
951
-
952
- yield* Effect.catch(
953
- waitEvent({
954
- timeout: TIMEOUT,
955
- signal,
956
- fn(event) {
957
- if (event.workspace !== workspaceID && event.payload.type !== "sync") {
958
- return false
959
- }
960
- return synced(state)
961
- },
962
- }),
963
- (): Effect.Effect<never, WaitForSyncError> =>
964
- signal?.aborted
965
- ? Effect.fail(
966
- new SyncAbortedError({
967
- message: signal.reason instanceof Error ? signal.reason.message : "Request aborted",
968
- cause: signal.reason,
969
- }),
970
- )
971
- : Effect.fail(
972
- new SyncTimeoutError({
973
- message: `Timed out waiting for sync fence: ${JSON.stringify(state)}`,
974
- state,
975
- }),
976
- ),
977
- )
978
- })
979
-
980
- const startWorkspaceSyncing = Effect.fn("Workspace.startWorkspaceSyncing")(function* (projectID: ProjectID) {
981
- const rows = yield* db((db) =>
982
- db
983
- .selectDistinct({ workspace: WorkspaceTable })
984
- .from(WorkspaceTable)
985
- .where(eq(WorkspaceTable.project_id, projectID))
986
- .all(),
987
- )
988
-
989
- for (const { workspace } of rows) {
990
- yield* startSync(fromRow(workspace)).pipe(
991
- Effect.catch((error) =>
992
- Effect.sync(() => {
993
- setStatus(workspace.id, "error")
994
- log.warn("workspace sync failed to start", {
995
- workspaceID: workspace.id,
996
- error,
997
- })
998
- }),
999
- ),
1000
- Effect.forkDetach,
1001
- )
1002
- }
1003
- })
1004
-
1005
- return Service.of({
1006
- create,
1007
- sessionWarp,
1008
- list,
1009
- syncList,
1010
- get,
1011
- remove,
1012
- status,
1013
- isSyncing,
1014
- waitForSync,
1015
- startWorkspaceSyncing,
1016
- })
1017
- }),
1018
- )
1019
-
1020
- export const defaultLayer = layer.pipe(
1021
- Layer.provide(Auth.defaultLayer),
1022
- Layer.provide(Session.defaultLayer),
1023
- Layer.provide(SyncEvent.defaultLayer),
1024
- Layer.provide(SessionPrompt.defaultLayer),
1025
- Layer.provide(Project.defaultLayer),
1026
- Layer.provide(Vcs.defaultLayer),
1027
- Layer.provide(AppFileSystem.defaultLayer),
1028
- Layer.provide(FetchHttpClient.layer),
1029
- Layer.provide(RuntimeFlags.defaultLayer),
1030
- )
1031
-
1032
- const TIMEOUT = 5000
1033
-
1034
- type HistoryEvent = {
1035
- id: string
1036
- aggregate_id: string
1037
- seq: number
1038
- type: string
1039
- data: Record<string, unknown>
1040
- }
1041
-
1042
- function synced(state: Record<string, number>) {
1043
- const ids = Object.keys(state)
1044
- if (ids.length === 0) return true
1045
-
1046
- const done = Object.fromEntries(
1047
- Database.use((db) =>
1048
- db
1049
- .select({
1050
- id: EventSequenceTable.aggregate_id,
1051
- seq: EventSequenceTable.seq,
1052
- })
1053
- .from(EventSequenceTable)
1054
- .where(inArray(EventSequenceTable.aggregate_id, ids))
1055
- .all(),
1056
- ).map((row) => [row.id, row.seq]),
1057
- ) as Record<string, number>
1058
-
1059
- return ids.every((id) => {
1060
- return (done[id] ?? -1) >= state[id]
1061
- })
1062
- }
1063
-
1064
- function route(url: string | URL, path: string) {
1065
- const next = new URL(url)
1066
- next.pathname = `${next.pathname.replace(/\/$/, "")}${path}`
1067
- next.search = ""
1068
- next.hash = ""
1069
- return next
1070
- }
1071
-
1072
- export * as Workspace from "./workspace"
1
+ import { Context, Effect, FiberMap, Iterable, Layer, Schema, Stream } from "effect"
2
+ import { FetchHttpClient, HttpBody, HttpClient, HttpClientError, HttpClientRequest } from "effect/unstable/http"
3
+ import { Database } from "@/storage/db"
4
+ import { asc } from "drizzle-orm"
5
+ import { eq } from "drizzle-orm"
6
+ import { inArray } from "drizzle-orm"
7
+ import { Project } from "@/project/project"
8
+ import { BusEvent } from "@/bus/bus-event"
9
+ import { GlobalBus } from "@/bus/global"
10
+ import { Auth } from "@/auth"
11
+ import { SyncEvent } from "@/sync"
12
+ import { EventSequenceTable, EventTable } from "@/sync/event.sql"
13
+ import { AppFileSystem } from "@opencode-ai/core/filesystem"
14
+ import * as Log from "@opencode-ai/core/util/log"
15
+ import { RuntimeFlags } from "@/effect/runtime-flags"
16
+ import { ProjectID } from "@/project/schema"
17
+ import { Slug } from "@opencode-ai/core/util/slug"
18
+ import { WorkspaceTable } from "./workspace.sql"
19
+ import { getAdapter, registeredAdapters } from "./adapters"
20
+ import { type Target, type WorkspaceInfo, WorkspaceInfo as WorkspaceInfoSchema } from "./types"
21
+ import { WorkspaceID } from "./schema"
22
+ import { Session } from "@/session/session"
23
+ import { SessionPrompt } from "@/session/prompt"
24
+ import { SessionTable } from "@/session/session.sql"
25
+ import { SessionID } from "@/session/schema"
26
+ import { NotFoundError } from "@/storage/storage"
27
+ import { errorData } from "@/util/error"
28
+ import { waitEvent } from "./util"
29
+ import { WorkspaceRef } from "@/effect/instance-ref"
30
+ import { Vcs } from "@/project/vcs"
31
+ import { InstanceStore } from "@/project/instance-store"
32
+ import { InstanceBootstrap } from "@/project/bootstrap"
33
+ import { WorkspaceAdapterRuntime } from "./workspace-adapter-runtime"
34
+
35
+ export const Info = Schema.Struct({
36
+ ...WorkspaceInfoSchema.fields,
37
+ timeUsed: Schema.Number,
38
+ }).annotate({ identifier: "Workspace" })
39
+ export type Info = WorkspaceInfo & { timeUsed: number }
40
+
41
+ export const ConnectionStatus = Schema.Struct({
42
+ workspaceID: WorkspaceID,
43
+ status: Schema.Literals(["connected", "connecting", "disconnected", "error"]),
44
+ })
45
+ export type ConnectionStatus = Schema.Schema.Type<typeof ConnectionStatus>
46
+
47
+ export const Event = {
48
+ Ready: BusEvent.define(
49
+ "workspace.ready",
50
+ Schema.Struct({
51
+ name: Schema.String,
52
+ }),
53
+ ),
54
+ Failed: BusEvent.define(
55
+ "workspace.failed",
56
+ Schema.Struct({
57
+ message: Schema.String,
58
+ }),
59
+ ),
60
+ Status: BusEvent.define("workspace.status", ConnectionStatus),
61
+ }
62
+
63
+ function fromRow(row: typeof WorkspaceTable.$inferSelect): Info {
64
+ return {
65
+ id: row.id,
66
+ type: row.type,
67
+ branch: row.branch,
68
+ name: row.name,
69
+ directory: row.directory,
70
+ extra: row.extra,
71
+ projectID: row.project_id,
72
+ timeUsed: row.time_used,
73
+ }
74
+ }
75
+
76
+ const db = <T>(fn: (d: Parameters<typeof Database.use>[0] extends (trx: infer D) => any ? D : never) => T) =>
77
+ Effect.sync(() => Database.use(fn))
78
+
79
+ const log = Log.create({ service: "workspace-sync" })
80
+
81
+ export const CreateInput = Schema.Struct({
82
+ id: Schema.optional(WorkspaceID),
83
+ type: Info.fields.type,
84
+ branch: Info.fields.branch,
85
+ projectID: ProjectID,
86
+ extra: Schema.optional(Info.fields.extra),
87
+ })
88
+ export type CreateInput = Schema.Schema.Type<typeof CreateInput>
89
+
90
+ export const SessionWarpInput = Schema.Struct({
91
+ workspaceID: Schema.NullOr(WorkspaceID),
92
+ sessionID: SessionID,
93
+ copyChanges: Schema.optional(Schema.Boolean),
94
+ })
95
+ export type SessionWarpInput = Schema.Schema.Type<typeof SessionWarpInput>
96
+
97
+ export class SyncHttpError extends Schema.TaggedErrorClass<SyncHttpError>()("WorkspaceSyncHttpError", {
98
+ message: Schema.String,
99
+ status: Schema.Number,
100
+ body: Schema.optional(Schema.String),
101
+ }) {}
102
+
103
+ export class WorkspaceNotFoundError extends Schema.TaggedErrorClass<WorkspaceNotFoundError>()(
104
+ "WorkspaceNotFoundError",
105
+ {
106
+ message: Schema.String,
107
+ workspaceID: WorkspaceID,
108
+ },
109
+ ) {}
110
+
111
+ export class SessionEventsNotFoundError extends Schema.TaggedErrorClass<SessionEventsNotFoundError>()(
112
+ "WorkspaceSessionEventsNotFoundError",
113
+ {
114
+ message: Schema.String,
115
+ sessionID: SessionID,
116
+ },
117
+ ) {}
118
+
119
+ export class SessionWarpHttpError extends Schema.TaggedErrorClass<SessionWarpHttpError>()(
120
+ "WorkspaceSessionWarpHttpError",
121
+ {
122
+ message: Schema.String,
123
+ workspaceID: WorkspaceID,
124
+ sessionID: SessionID,
125
+ status: Schema.Number,
126
+ body: Schema.String,
127
+ },
128
+ ) {}
129
+
130
+ export class SyncTimeoutError extends Schema.TaggedErrorClass<SyncTimeoutError>()("WorkspaceSyncTimeoutError", {
131
+ message: Schema.String,
132
+ state: Schema.Record(Schema.String, Schema.Number),
133
+ }) {}
134
+
135
+ export class SyncAbortedError extends Schema.TaggedErrorClass<SyncAbortedError>()("WorkspaceSyncAbortedError", {
136
+ message: Schema.String,
137
+ cause: Schema.optional(Schema.Defect),
138
+ }) {}
139
+
140
+ type CreateError = Auth.AuthError
141
+ type SessionWarpError =
142
+ | WorkspaceNotFoundError
143
+ | SessionEventsNotFoundError
144
+ | SessionWarpHttpError
145
+ | Vcs.PatchApplyError
146
+ | HttpClientError.HttpClientError
147
+ type WaitForSyncError = SyncTimeoutError | SyncAbortedError
148
+ type SyncLoopError = SyncHttpError | HttpClientError.HttpClientError
149
+
150
+ export interface Interface {
151
+ readonly create: (input: CreateInput) => Effect.Effect<Info, CreateError>
152
+ readonly sessionWarp: (input: SessionWarpInput) => Effect.Effect<void, SessionWarpError>
153
+ readonly list: (project: Project.Info) => Effect.Effect<Info[]>
154
+ readonly syncList: (project: Project.Info) => Effect.Effect<void>
155
+ readonly get: (id: WorkspaceID) => Effect.Effect<Info | undefined>
156
+ readonly remove: (id: WorkspaceID) => Effect.Effect<Info | undefined>
157
+ readonly status: () => Effect.Effect<ConnectionStatus[]>
158
+ readonly isSyncing: (workspaceID: WorkspaceID) => Effect.Effect<boolean>
159
+ readonly waitForSync: (
160
+ workspaceID: WorkspaceID,
161
+ state: Record<string, number>,
162
+ signal?: AbortSignal,
163
+ ) => Effect.Effect<void, WaitForSyncError>
164
+ readonly startWorkspaceSyncing: (projectID: ProjectID) => Effect.Effect<void>
165
+ }
166
+
167
+ export class Service extends Context.Service<Service, Interface>()("@opencode/Workspace") {}
168
+
169
+ export const layer = Layer.effect(
170
+ Service,
171
+ Effect.gen(function* () {
172
+ const auth = yield* Auth.Service
173
+ const session = yield* Session.Service
174
+ const prompt = yield* SessionPrompt.Service
175
+ const http = yield* HttpClient.HttpClient
176
+ const sync = yield* SyncEvent.Service
177
+ const vcs = yield* Vcs.Service
178
+ const flags = yield* RuntimeFlags.Service
179
+ const fs = yield* AppFileSystem.Service
180
+ const connections = new Map<WorkspaceID, ConnectionStatus>()
181
+ const syncFibers = yield* FiberMap.make<WorkspaceID, void, SyncLoopError>()
182
+
183
+ const setStatus = (id: WorkspaceID, status: ConnectionStatus["status"]) => {
184
+ const prev = connections.get(id)
185
+ if (prev?.status === status) return
186
+ const next = { workspaceID: id, status }
187
+ connections.set(id, next)
188
+
189
+ GlobalBus.emit("event", {
190
+ directory: "global",
191
+ workspace: id,
192
+ payload: {
193
+ type: Event.Status.type,
194
+ properties: next,
195
+ },
196
+ })
197
+ }
198
+
199
+ const connectSSE = Effect.fn("Workspace.connectSSE")(function* (
200
+ url: URL | string,
201
+ headers: HeadersInit | undefined,
202
+ ) {
203
+ const response = yield* http.execute(
204
+ HttpClientRequest.get(route(url, "/global/event"), {
205
+ headers: new Headers(headers),
206
+ accept: "text/event-stream",
207
+ }),
208
+ )
209
+ if (response.status < 200 || response.status >= 300) {
210
+ return yield* new SyncHttpError({
211
+ message: `Workspace sync HTTP failure: ${response.status}`,
212
+ status: response.status,
213
+ })
214
+ }
215
+ return response.stream
216
+ })
217
+
218
+ const parseSSE = Effect.fn("Workspace.parseSSE")(function* (
219
+ stream: Stream.Stream<Uint8Array, unknown>,
220
+ onEvent: (event: unknown) => Effect.Effect<void>,
221
+ ) {
222
+ yield* stream.pipe(
223
+ Stream.decodeText(),
224
+ Stream.splitLines,
225
+ Stream.mapAccum(
226
+ () => ({ data: [] as string[], id: undefined as string | undefined, retry: 1000 }),
227
+ (state, line) => {
228
+ if (line === "") {
229
+ if (!state.data.length) return [state, []]
230
+ return [{ ...state, data: [] }, [{ data: state.data.join("\n"), id: state.id, retry: state.retry }]]
231
+ }
232
+
233
+ const index = line.indexOf(":")
234
+ const field = index === -1 ? line : line.slice(0, index)
235
+ const value = index === -1 ? "" : line.slice(index + (line[index + 1] === " " ? 2 : 1))
236
+
237
+ if (field === "data") return [{ ...state, data: [...state.data, value] }, []]
238
+ if (field === "id") return [{ ...state, id: value }, []]
239
+ if (field === "retry") {
240
+ const retry = Number.parseInt(value, 10)
241
+ return [Number.isNaN(retry) ? state : { ...state, retry }, []]
242
+ }
243
+ return [state, []]
244
+ },
245
+ {
246
+ onHalt: (state) =>
247
+ state.data.length ? [{ data: state.data.join("\n"), id: state.id, retry: state.retry }] : [],
248
+ },
249
+ ),
250
+ Stream.map((event) => {
251
+ try {
252
+ return JSON.parse(event.data) as unknown
253
+ } catch {
254
+ return {
255
+ type: "sse.message",
256
+ properties: {
257
+ data: event.data,
258
+ id: event.id || undefined,
259
+ retry: event.retry,
260
+ },
261
+ }
262
+ }
263
+ }),
264
+ Stream.runForEach(onEvent),
265
+ )
266
+ })
267
+
268
+ const runInWorkspace = <A, E, R>(input: {
269
+ workspaceID?: WorkspaceID
270
+ local: () => Effect.Effect<A, E, R>
271
+ remote: (input: {
272
+ workspace: Info
273
+ target: Extract<Target, { type: "remote" }>
274
+ }) => HttpClientRequest.HttpClientRequest
275
+ fallback: A
276
+ response?: "json" | "text"
277
+ }) =>
278
+ Effect.gen(function* () {
279
+ if (!input.workspaceID) return yield* input.local()
280
+
281
+ const workspace = yield* get(input.workspaceID)
282
+ if (!workspace) return input.fallback
283
+
284
+ const target = yield* WorkspaceAdapterRuntime.target(workspace)
285
+
286
+ if (target.type === "local") {
287
+ const store = yield* InstanceStore.Service
288
+ return yield* store.provide({ directory: target.directory }, input.local())
289
+ }
290
+
291
+ const response = yield* http.execute(input.remote({ workspace, target })).pipe(
292
+ Effect.catch((error) =>
293
+ Effect.sync(() => {
294
+ log.warn("workspace target request failed", {
295
+ workspaceID: workspace.id,
296
+ error: errorData(error),
297
+ })
298
+ }),
299
+ ),
300
+ )
301
+ if (!response) return input.fallback
302
+ if (response.status < 200 || response.status >= 300) {
303
+ const body = yield* response.text.pipe(Effect.catch(() => Effect.succeed("")))
304
+ log.warn("workspace target request failed", {
305
+ workspaceID: workspace.id,
306
+ status: response.status,
307
+ body,
308
+ })
309
+ return input.fallback
310
+ }
311
+
312
+ const body = input.response === "text" ? response.text : response.json
313
+ return yield* body.pipe(
314
+ Effect.map((result) => result as A),
315
+ Effect.catch((error) =>
316
+ Effect.sync(() => {
317
+ log.warn("workspace target response decode failed", {
318
+ workspaceID: workspace.id,
319
+ error: errorData(error),
320
+ })
321
+ return input.fallback
322
+ }),
323
+ ),
324
+ )
325
+ })
326
+
327
+ const syncHistory = Effect.fn("Workspace.syncHistory")(function* (
328
+ space: Info,
329
+ url: URL | string,
330
+ headers: HeadersInit | undefined,
331
+ ) {
332
+ const sessionIDs = yield* db((db) =>
333
+ db
334
+ .select({ id: SessionTable.id })
335
+ .from(SessionTable)
336
+ .where(eq(SessionTable.workspace_id, space.id))
337
+ .all()
338
+ .map((row) => row.id),
339
+ )
340
+ const state = sessionIDs.length
341
+ ? Object.fromEntries(
342
+ (yield* db((db) =>
343
+ db.select().from(EventSequenceTable).where(inArray(EventSequenceTable.aggregate_id, sessionIDs)).all(),
344
+ )).map((row) => [row.aggregate_id, row.seq]),
345
+ )
346
+ : {}
347
+
348
+ log.info("syncing workspace history", {
349
+ workspaceID: space.id,
350
+ sessions: sessionIDs.length,
351
+ known: Object.keys(state).length,
352
+ })
353
+
354
+ const response = yield* http.execute(
355
+ HttpClientRequest.post(route(url, "/sync/history"), {
356
+ headers: new Headers(headers),
357
+ body: HttpBody.jsonUnsafe(state),
358
+ }),
359
+ )
360
+
361
+ if (response.status < 200 || response.status >= 300) {
362
+ const body = yield* response.text
363
+ return yield* new SyncHttpError({
364
+ message: `Workspace history HTTP failure: ${response.status} ${body}`,
365
+ status: response.status,
366
+ body,
367
+ })
368
+ }
369
+
370
+ const events = (yield* response.json) as HistoryEvent[]
371
+
372
+ log.info("workspace history synced", {
373
+ workspaceID: space.id,
374
+ events: events.length,
375
+ })
376
+
377
+ yield* Effect.forEach(
378
+ events,
379
+ (event) =>
380
+ sync
381
+ .replay(
382
+ {
383
+ id: event.id,
384
+ aggregateID: event.aggregate_id,
385
+ seq: event.seq,
386
+ type: event.type,
387
+ data: event.data,
388
+ },
389
+ { publish: true },
390
+ )
391
+ .pipe(Effect.provideService(WorkspaceRef, space.id)),
392
+ { discard: true },
393
+ )
394
+ })
395
+
396
+ const syncWorkspaceLoop = Effect.fn("Workspace.syncWorkspaceLoop")(function* (space: Info) {
397
+ const target = yield* WorkspaceAdapterRuntime.target(space)
398
+
399
+ if (target.type === "local") return
400
+
401
+ let attempt = 0
402
+
403
+ while (true) {
404
+ log.info("connecting to global sync", { workspace: space.name })
405
+ setStatus(space.id, "connecting")
406
+
407
+ const stream = yield* connectSSE(target.url, target.headers).pipe(
408
+ Effect.tap(() => syncHistory(space, target.url, target.headers)),
409
+ Effect.catch((err) =>
410
+ Effect.sync(() => {
411
+ setStatus(space.id, "error")
412
+ log.info("failed to connect to global sync", {
413
+ workspace: space.name,
414
+ err,
415
+ })
416
+ return null
417
+ }),
418
+ ),
419
+ )
420
+
421
+ if (stream) {
422
+ attempt = 0
423
+
424
+ log.info("global sync connected", { workspace: space.name })
425
+ setStatus(space.id, "connected")
426
+
427
+ yield* parseSSE(stream, (evt) =>
428
+ Effect.gen(function* () {
429
+ if (!evt || typeof evt !== "object" || !("payload" in evt)) return
430
+ const payload = evt.payload as { type?: string; syncEvent?: SyncEvent.SerializedEvent }
431
+ if (payload.type === "server.heartbeat") return
432
+
433
+ if (payload.type === "sync" && payload.syncEvent) {
434
+ const failed = yield* sync.replay(payload.syncEvent).pipe(
435
+ Effect.as(false),
436
+ Effect.catchCause((error) =>
437
+ Effect.sync(() => {
438
+ log.info("failed to replay global event", {
439
+ workspaceID: space.id,
440
+ error,
441
+ })
442
+ return true
443
+ }),
444
+ ),
445
+ )
446
+ if (failed) return
447
+ }
448
+
449
+ try {
450
+ const event = evt as { directory?: string; project?: string; payload: unknown }
451
+ GlobalBus.emit("event", {
452
+ directory: event.directory,
453
+ project: event.project,
454
+ workspace: space.id,
455
+ payload: event.payload,
456
+ })
457
+ } catch (error) {
458
+ log.info("failed to replay global event", {
459
+ workspaceID: space.id,
460
+ error,
461
+ })
462
+ }
463
+ }),
464
+ )
465
+
466
+ log.info("disconnected from global sync: " + space.id)
467
+ setStatus(space.id, "disconnected")
468
+ }
469
+
470
+ // Back off reconnect attempts up to 2 minutes while the workspace
471
+ // stays unavailable.
472
+ yield* Effect.sleep(`${Math.min(120_000, 1_000 * 2 ** attempt)} millis`)
473
+ attempt += 1
474
+ }
475
+ })
476
+
477
+ const startSync = Effect.fn("Workspace.startSync")(function* (space: Info) {
478
+ if (!flags.experimentalWorkspaces) return
479
+
480
+ const target = yield* WorkspaceAdapterRuntime.target(space).pipe(
481
+ Effect.catch((error) =>
482
+ Effect.sync(() => {
483
+ setStatus(space.id, "error")
484
+ log.warn("workspace target failed", {
485
+ workspaceID: space.id,
486
+ error: errorData(error),
487
+ })
488
+ return null
489
+ }),
490
+ ),
491
+ )
492
+ if (!target) return
493
+
494
+ if (target.type === "local") {
495
+ setStatus(space.id, (yield* fs.existsSafe(target.directory)) ? "connected" : "error")
496
+ return
497
+ }
498
+
499
+ const exists = yield* FiberMap.has(syncFibers, space.id)
500
+ if (exists && connections.get(space.id)?.status !== "error") return
501
+
502
+ setStatus(space.id, "disconnected")
503
+
504
+ yield* FiberMap.run(
505
+ syncFibers,
506
+ space.id,
507
+ // TODO: look into `tapError` to set the status but still
508
+ // allow the fiber to fail and automatically get removed
509
+ syncWorkspaceLoop(space).pipe(
510
+ Effect.catch((error) =>
511
+ Effect.sync(() => {
512
+ setStatus(space.id, "error")
513
+ log.warn("workspace listener failed", {
514
+ workspaceID: space.id,
515
+ error,
516
+ })
517
+ }),
518
+ ),
519
+ ),
520
+ )
521
+ })
522
+
523
+ const stopSync = Effect.fn("Workspace.stopSync")(function* (id: WorkspaceID) {
524
+ yield* FiberMap.remove(syncFibers, id)
525
+ connections.delete(id)
526
+ })
527
+
528
+ const create = Effect.fn("Workspace.create")(function* (input: CreateInput) {
529
+ const id = WorkspaceID.ascending(input.id)
530
+ const adapter = getAdapter(input.projectID, input.type)
531
+ const config = yield* WorkspaceAdapterRuntime.configure(adapter, {
532
+ ...input,
533
+ id,
534
+ name: Slug.create(),
535
+ directory: null,
536
+ extra: input.extra ?? null,
537
+ })
538
+
539
+ const info: Info = {
540
+ id,
541
+ type: config.type,
542
+ branch: config.branch ?? null,
543
+ name: config.name ?? null,
544
+ directory: config.directory ?? null,
545
+ extra: config.extra ?? null,
546
+ projectID: input.projectID,
547
+ timeUsed: Date.now(),
548
+ }
549
+
550
+ yield* db((db) => {
551
+ db.insert(WorkspaceTable)
552
+ .values({
553
+ id: info.id,
554
+ type: info.type,
555
+ branch: info.branch,
556
+ name: info.name,
557
+ directory: info.directory,
558
+ extra: info.extra,
559
+ project_id: info.projectID,
560
+ time_used: info.timeUsed,
561
+ })
562
+ .run()
563
+ })
564
+
565
+ const env = {
566
+ OPENCODE_AUTH_CONTENT: JSON.stringify(yield* auth.all()),
567
+ OPENCODE_WORKSPACE_ID: config.id,
568
+ OPENCODE_EXPERIMENTAL_WORKSPACES: "true",
569
+ OTEL_EXPORTER_OTLP_HEADERS: process.env.OTEL_EXPORTER_OTLP_HEADERS,
570
+ OTEL_EXPORTER_OTLP_ENDPOINT: process.env.OTEL_EXPORTER_OTLP_ENDPOINT,
571
+ OTEL_RESOURCE_ATTRIBUTES: process.env.OTEL_RESOURCE_ATTRIBUTES,
572
+ }
573
+
574
+ yield* WorkspaceAdapterRuntime.create(adapter, config, env)
575
+ yield* Effect.all(
576
+ [
577
+ waitEvent({
578
+ timeout: TIMEOUT,
579
+ fn(event) {
580
+ if (event.workspace === info.id && event.payload.type === Event.Status.type) {
581
+ const { status } = event.payload.properties
582
+ return status === "error" || status === "connected"
583
+ }
584
+ return false
585
+ },
586
+ }),
587
+ startSync(info),
588
+ ],
589
+ { concurrency: 2, discard: true },
590
+ )
591
+
592
+ return info
593
+ })
594
+
595
+ const sessionWarp = Effect.fn("Workspace.sessionWarp")(function* (input: SessionWarpInput) {
596
+ return yield* Effect.gen(function* () {
597
+ log.info("session warp requested", {
598
+ workspaceID: input.workspaceID,
599
+ sessionID: input.sessionID,
600
+ })
601
+
602
+ const current = yield* db((db) =>
603
+ db
604
+ .select({ workspaceID: SessionTable.workspace_id })
605
+ .from(SessionTable)
606
+ .where(eq(SessionTable.id, input.sessionID))
607
+ .get(),
608
+ )
609
+
610
+ if (current?.workspaceID) {
611
+ const previous = yield* get(current.workspaceID)
612
+ if (previous) {
613
+ const target = yield* WorkspaceAdapterRuntime.target(previous)
614
+
615
+ if (target.type === "remote") {
616
+ yield* syncHistory(previous, target.url, target.headers).pipe(
617
+ Effect.catch((error) =>
618
+ Effect.sync(() => {
619
+ log.warn("session warp final source sync failed", {
620
+ workspaceID: previous.id,
621
+ sessionID: input.sessionID,
622
+ error: errorData(error),
623
+ })
624
+ }),
625
+ ),
626
+ )
627
+ } else {
628
+ yield* prompt.cancel(input.sessionID)
629
+ }
630
+
631
+ // "claim" this session so any future events coming from
632
+ // the old workspace are ignored
633
+ yield* sync.claim(input.sessionID, input.workspaceID ?? previous.projectID)
634
+ }
635
+ }
636
+
637
+ const sourcePatch =
638
+ input.copyChanges && current?.workspaceID
639
+ ? yield* runInWorkspace({
640
+ workspaceID: current?.workspaceID ?? undefined,
641
+ local: () => vcs.diffRaw(),
642
+ remote: ({ target }) =>
643
+ HttpClientRequest.get(route(target.url, "/vcs/diff/raw"), {
644
+ headers: new Headers(target.headers),
645
+ }),
646
+ fallback: "",
647
+ response: "text",
648
+ }).pipe(Effect.provide(InstanceStore.defaultLayer.pipe(Layer.provide(InstanceBootstrap.defaultLayer))))
649
+ : ""
650
+
651
+ if (sourcePatch) {
652
+ // Attempt to apply the file changes to the new workspace.
653
+ // We intentionally do first so if it fails we don't warp
654
+ // the session.
655
+ yield* runInWorkspace({
656
+ workspaceID: input.workspaceID ?? undefined,
657
+ local: () => vcs.apply({ patch: sourcePatch }),
658
+ remote: ({ target }) =>
659
+ HttpClientRequest.post(route(target.url, "/vcs/apply"), {
660
+ headers: new Headers(target.headers),
661
+ body: HttpBody.jsonUnsafe({ patch: sourcePatch }),
662
+ }),
663
+ fallback: { applied: false },
664
+ }).pipe(Effect.provide(InstanceStore.defaultLayer.pipe(Layer.provide(InstanceBootstrap.defaultLayer))))
665
+ }
666
+
667
+ if (input.workspaceID === null) {
668
+ yield* sync.run(Session.Event.Updated, {
669
+ sessionID: input.sessionID,
670
+ info: {
671
+ workspaceID: null,
672
+ },
673
+ })
674
+
675
+ log.info("session warp complete", {
676
+ workspaceID: input.workspaceID,
677
+ sessionID: input.sessionID,
678
+ target: "local",
679
+ })
680
+ return
681
+ }
682
+
683
+ const workspaceID = input.workspaceID
684
+ const space = yield* get(workspaceID)
685
+ if (!space)
686
+ return yield* new WorkspaceNotFoundError({
687
+ message: `Workspace not found: ${workspaceID}`,
688
+ workspaceID,
689
+ })
690
+
691
+ const target = yield* WorkspaceAdapterRuntime.target(space)
692
+
693
+ if (target.type === "local") {
694
+ yield* sync.run(Session.Event.Updated, {
695
+ sessionID: input.sessionID,
696
+ info: {
697
+ workspaceID: input.workspaceID,
698
+ },
699
+ })
700
+
701
+ log.info("session warp complete", {
702
+ workspaceID: input.workspaceID,
703
+ sessionID: input.sessionID,
704
+ target: target.directory,
705
+ })
706
+ return
707
+ }
708
+
709
+ const rows = yield* db((db) =>
710
+ db
711
+ .select({
712
+ id: EventTable.id,
713
+ aggregateID: EventTable.aggregate_id,
714
+ seq: EventTable.seq,
715
+ type: EventTable.type,
716
+ data: EventTable.data,
717
+ })
718
+ .from(EventTable)
719
+ .where(eq(EventTable.aggregate_id, input.sessionID))
720
+ .orderBy(asc(EventTable.seq))
721
+ .all(),
722
+ )
723
+ if (rows.length === 0)
724
+ return yield* new SessionEventsNotFoundError({
725
+ message: `No events found for session: ${input.sessionID}`,
726
+ sessionID: input.sessionID,
727
+ })
728
+
729
+ const batches = Iterable.chunksOf(rows, 10)
730
+ const total = Iterable.size(batches)
731
+
732
+ log.info("session warp prepared", {
733
+ workspaceID: input.workspaceID,
734
+ sessionID: input.sessionID,
735
+ target: String(route(target.url, "/sync/replay")),
736
+ events: rows.length,
737
+ batches: total,
738
+ first: rows[0]?.seq,
739
+ last: rows.at(-1)?.seq,
740
+ })
741
+
742
+ yield* Effect.forEach(
743
+ batches,
744
+ (events, i) =>
745
+ Effect.gen(function* () {
746
+ const response = yield* http.execute(
747
+ HttpClientRequest.post(route(target.url, "/sync/replay"), {
748
+ headers: new Headers(target.headers),
749
+ body: HttpBody.jsonUnsafe({
750
+ directory: space.directory ?? "",
751
+ events,
752
+ }),
753
+ }),
754
+ )
755
+
756
+ if (response.status < 200 || response.status >= 300) {
757
+ const body = yield* response.text
758
+ log.error("session warp batch failed", {
759
+ workspaceID: input.workspaceID,
760
+ sessionID: input.sessionID,
761
+ step: i + 1,
762
+ total,
763
+ status: response.status,
764
+ body,
765
+ })
766
+ return yield* new SessionWarpHttpError({
767
+ message: `Failed to warp session ${input.sessionID} into workspace ${workspaceID}: HTTP ${response.status} ${body}`,
768
+ workspaceID,
769
+ sessionID: input.sessionID,
770
+ status: response.status,
771
+ body,
772
+ })
773
+ }
774
+
775
+ log.info("session warp batch posted", {
776
+ workspaceID: input.workspaceID,
777
+ sessionID: input.sessionID,
778
+ step: i + 1,
779
+ total,
780
+ status: response.status,
781
+ })
782
+ }),
783
+ { discard: true },
784
+ )
785
+
786
+ const response = yield* http.execute(
787
+ HttpClientRequest.post(route(target.url, "/sync/steal"), {
788
+ headers: new Headers(target.headers),
789
+ body: HttpBody.jsonUnsafe({ sessionID: input.sessionID }),
790
+ }),
791
+ )
792
+ if (response.status < 200 || response.status >= 300) {
793
+ const body = yield* response.text
794
+ log.error("session warp steal failed", {
795
+ workspaceID: input.workspaceID,
796
+ sessionID: input.sessionID,
797
+ status: response.status,
798
+ body,
799
+ })
800
+ return yield* new SessionWarpHttpError({
801
+ message: `Failed to steal session ${input.sessionID} into workspace ${workspaceID}: HTTP ${response.status} ${body}`,
802
+ workspaceID,
803
+ sessionID: input.sessionID,
804
+ status: response.status,
805
+ body,
806
+ })
807
+ }
808
+
809
+ log.info("session warp complete", {
810
+ workspaceID: input.workspaceID,
811
+ sessionID: input.sessionID,
812
+ batches: total,
813
+ })
814
+ }).pipe(
815
+ Effect.tapError((err) =>
816
+ Effect.sync(() =>
817
+ log.error("session warp failed", {
818
+ workspaceID: input.workspaceID,
819
+ sessionID: input.sessionID,
820
+ error: errorData(err),
821
+ }),
822
+ ),
823
+ ),
824
+ )
825
+ })
826
+
827
+ const list = Effect.fn("Workspace.list")(function* (project: Project.Info) {
828
+ return yield* db((db) =>
829
+ db
830
+ .select()
831
+ .from(WorkspaceTable)
832
+ .where(eq(WorkspaceTable.project_id, project.id))
833
+ .all()
834
+ .map(fromRow)
835
+ .sort((a, b) => a.id.localeCompare(b.id)),
836
+ )
837
+ })
838
+
839
+ const syncList = Effect.fn("Workspace.syncList")(function* (project: Project.Info) {
840
+ const names = new Set((yield* list(project)).map((workspace) => workspace.name))
841
+ const discovered = yield* Effect.forEach(
842
+ registeredAdapters(project.id),
843
+ ([type, adapter]) =>
844
+ WorkspaceAdapterRuntime.list(adapter).pipe(
845
+ Effect.catchCause((error) =>
846
+ Effect.sync(() => {
847
+ log.warn("workspace adapter list failed", { type, error })
848
+ return []
849
+ }),
850
+ ),
851
+ ),
852
+ { concurrency: "unbounded" },
853
+ ).pipe(Effect.map((items) => items.flat()))
854
+
855
+ yield* Effect.forEach(
856
+ discovered,
857
+ (item) =>
858
+ Effect.gen(function* () {
859
+ if (names.has(item.name)) return
860
+ names.add(item.name)
861
+
862
+ const info: Info = {
863
+ id: WorkspaceID.ascending(),
864
+ type: item.type,
865
+ branch: item.branch,
866
+ name: item.name,
867
+ directory: item.directory,
868
+ extra: item.extra,
869
+ projectID: item.projectID,
870
+ timeUsed: Date.now(),
871
+ }
872
+
873
+ yield* db((db) => {
874
+ db.insert(WorkspaceTable)
875
+ .values({
876
+ id: info.id,
877
+ type: info.type,
878
+ branch: info.branch,
879
+ name: info.name,
880
+ directory: info.directory,
881
+ extra: info.extra,
882
+ project_id: info.projectID,
883
+ time_used: info.timeUsed,
884
+ })
885
+ .run()
886
+ })
887
+
888
+ yield* startSync(info)
889
+ }),
890
+ { concurrency: 1 },
891
+ )
892
+ })
893
+
894
+ const get = Effect.fn("Workspace.get")(function* (id: WorkspaceID) {
895
+ const row = yield* db((db) => db.select().from(WorkspaceTable).where(eq(WorkspaceTable.id, id)).get())
896
+ if (!row) return
897
+ return fromRow(row)
898
+ })
899
+
900
+ const remove = Effect.fn("Workspace.remove")(function* (id: WorkspaceID) {
901
+ const sessions = yield* db((db) =>
902
+ db
903
+ .select({ id: SessionTable.id, parentID: SessionTable.parent_id })
904
+ .from(SessionTable)
905
+ .where(eq(SessionTable.workspace_id, id))
906
+ .all(),
907
+ )
908
+ const sessionIDs = new Set(sessions.map((sessionInfo) => sessionInfo.id))
909
+ yield* Effect.forEach(
910
+ sessions.filter((sessionInfo) => !sessionInfo.parentID || !sessionIDs.has(sessionInfo.parentID)),
911
+ (sessionInfo) =>
912
+ session.remove(sessionInfo.id).pipe(Effect.catchIf(NotFoundError.isInstance, () => Effect.void)),
913
+ { discard: true },
914
+ )
915
+
916
+ const row = yield* db((db) => db.select().from(WorkspaceTable).where(eq(WorkspaceTable.id, id)).get())
917
+ if (!row) return
918
+
919
+ yield* stopSync(id)
920
+
921
+ const info = fromRow(row)
922
+ yield* Effect.catchCause(
923
+ Effect.gen(function* () {
924
+ yield* WorkspaceAdapterRuntime.remove(info)
925
+ }),
926
+ () =>
927
+ Effect.sync(() => {
928
+ log.error("adapter not available when removing workspace", { type: row.type })
929
+ }),
930
+ )
931
+
932
+ yield* db((db) => db.delete(WorkspaceTable).where(eq(WorkspaceTable.id, id)).run())
933
+ return info
934
+ })
935
+
936
+ const status = Effect.fn("Workspace.status")(function* () {
937
+ return [...connections.values()]
938
+ })
939
+
940
+ const isSyncing = Effect.fn("Workspace.isSyncing")(function* (workspaceID: WorkspaceID) {
941
+ const exists = yield* FiberMap.has(syncFibers, workspaceID)
942
+ return exists && connections.get(workspaceID)?.status !== "error"
943
+ })
944
+
945
+ const waitForSync = Effect.fn("Workspace.waitForSync")(function* (
946
+ workspaceID: WorkspaceID,
947
+ state: Record<string, number>,
948
+ signal?: AbortSignal,
949
+ ) {
950
+ if (synced(state)) return
951
+
952
+ yield* Effect.catch(
953
+ waitEvent({
954
+ timeout: TIMEOUT,
955
+ signal,
956
+ fn(event) {
957
+ if (event.workspace !== workspaceID && event.payload.type !== "sync") {
958
+ return false
959
+ }
960
+ return synced(state)
961
+ },
962
+ }),
963
+ (): Effect.Effect<never, WaitForSyncError> =>
964
+ signal?.aborted
965
+ ? Effect.fail(
966
+ new SyncAbortedError({
967
+ message: signal.reason instanceof Error ? signal.reason.message : "Request aborted",
968
+ cause: signal.reason,
969
+ }),
970
+ )
971
+ : Effect.fail(
972
+ new SyncTimeoutError({
973
+ message: `Timed out waiting for sync fence: ${JSON.stringify(state)}`,
974
+ state,
975
+ }),
976
+ ),
977
+ )
978
+ })
979
+
980
+ const startWorkspaceSyncing = Effect.fn("Workspace.startWorkspaceSyncing")(function* (projectID: ProjectID) {
981
+ const rows = yield* db((db) =>
982
+ db
983
+ .selectDistinct({ workspace: WorkspaceTable })
984
+ .from(WorkspaceTable)
985
+ .where(eq(WorkspaceTable.project_id, projectID))
986
+ .all(),
987
+ )
988
+
989
+ for (const { workspace } of rows) {
990
+ yield* startSync(fromRow(workspace)).pipe(
991
+ Effect.catch((error) =>
992
+ Effect.sync(() => {
993
+ setStatus(workspace.id, "error")
994
+ log.warn("workspace sync failed to start", {
995
+ workspaceID: workspace.id,
996
+ error,
997
+ })
998
+ }),
999
+ ),
1000
+ Effect.forkDetach,
1001
+ )
1002
+ }
1003
+ })
1004
+
1005
+ return Service.of({
1006
+ create,
1007
+ sessionWarp,
1008
+ list,
1009
+ syncList,
1010
+ get,
1011
+ remove,
1012
+ status,
1013
+ isSyncing,
1014
+ waitForSync,
1015
+ startWorkspaceSyncing,
1016
+ })
1017
+ }),
1018
+ )
1019
+
1020
+ export const defaultLayer = layer.pipe(
1021
+ Layer.provide(Auth.defaultLayer),
1022
+ Layer.provide(Session.defaultLayer),
1023
+ Layer.provide(SyncEvent.defaultLayer),
1024
+ Layer.provide(SessionPrompt.defaultLayer),
1025
+ Layer.provide(Project.defaultLayer),
1026
+ Layer.provide(Vcs.defaultLayer),
1027
+ Layer.provide(AppFileSystem.defaultLayer),
1028
+ Layer.provide(FetchHttpClient.layer),
1029
+ Layer.provide(RuntimeFlags.defaultLayer),
1030
+ )
1031
+
1032
+ const TIMEOUT = 5000
1033
+
1034
+ type HistoryEvent = {
1035
+ id: string
1036
+ aggregate_id: string
1037
+ seq: number
1038
+ type: string
1039
+ data: Record<string, unknown>
1040
+ }
1041
+
1042
+ function synced(state: Record<string, number>) {
1043
+ const ids = Object.keys(state)
1044
+ if (ids.length === 0) return true
1045
+
1046
+ const done = Object.fromEntries(
1047
+ Database.use((db) =>
1048
+ db
1049
+ .select({
1050
+ id: EventSequenceTable.aggregate_id,
1051
+ seq: EventSequenceTable.seq,
1052
+ })
1053
+ .from(EventSequenceTable)
1054
+ .where(inArray(EventSequenceTable.aggregate_id, ids))
1055
+ .all(),
1056
+ ).map((row) => [row.id, row.seq]),
1057
+ ) as Record<string, number>
1058
+
1059
+ return ids.every((id) => {
1060
+ return (done[id] ?? -1) >= state[id]
1061
+ })
1062
+ }
1063
+
1064
+ function route(url: string | URL, path: string) {
1065
+ const next = new URL(url)
1066
+ next.pathname = `${next.pathname.replace(/\/$/, "")}${path}`
1067
+ next.search = ""
1068
+ next.hash = ""
1069
+ return next
1070
+ }
1071
+
1072
+ export * as Workspace from "./workspace"