location-layer.ts 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. import { Layer, LayerMap } from "effect"
  2. import { Location } from "./location"
  3. import { Policy } from "./policy"
  4. import { Config } from "./config"
  5. import { PluginV2 } from "./plugin"
  6. import { Catalog } from "./catalog"
  7. import { CommandV2 } from "./command"
  8. import { AgentV2 } from "./agent"
  9. import { PluginBoot } from "./plugin/boot"
  10. import { Project } from "./project"
  11. import { EventV2 } from "./event"
  12. import { Auth } from "./auth"
  13. import { Npm } from "./npm"
  14. import { ModelsDev } from "./models-dev"
  15. import { FSUtil } from "./fs-util"
  16. import { Global } from "./global"
  17. import { Database } from "./database/database"
  18. import { PermissionV2 } from "./permission"
  19. import { PermissionSaved } from "./permission/saved"
  20. import { FileSystem } from "./filesystem"
  21. import { Watcher } from "./filesystem/watcher"
  22. import { LocationMutation } from "./location-mutation"
  23. import { LocationSearch } from "./location-search"
  24. import { FileMutation } from "./file-mutation"
  25. import { ProjectReference } from "./project-reference"
  26. import { RepositoryCache } from "./repository-cache"
  27. import { Pty } from "./pty"
  28. import { SkillV2 } from "./skill"
  29. import { BuiltInTools } from "./tool/builtins"
  30. import { ToolRegistry } from "./tool-registry"
  31. import { ToolOutputStore } from "./tool-output-store"
  32. import { AppProcess } from "./process"
  33. import { Ripgrep } from "./ripgrep"
  34. import { SessionStore } from "./session/store"
  35. import { SessionTodo } from "./session/todo"
  36. import { QuestionV2 } from "./question"
  37. import { LLMClient } from "@opencode-ai/llm"
  38. import { RequestExecutor } from "@opencode-ai/llm/route"
  39. import * as SessionRunnerLLM from "./session/runner/llm"
  40. import { SessionRunnerModel } from "./session/runner/model"
  41. import { SessionRunCoordinator } from "./session/run-coordinator"
  42. import { FetchHttpClient } from "effect/unstable/http"
  43. export class LocationServiceMap extends LayerMap.Service<LocationServiceMap>()("@opencode/example/LocationServiceMap", {
  44. lookup: (ref: Location.Ref) => {
  45. const location = Location.layer(ref)
  46. const permissionsAndTools = ToolRegistry.layer.pipe(Layer.provideMerge(PermissionV2.locationLayer))
  47. const services = Layer.mergeAll(
  48. location,
  49. Policy.locationLayer,
  50. Config.locationLayer,
  51. ProjectReference.locationLayer,
  52. PluginV2.locationLayer,
  53. Catalog.locationLayer,
  54. CommandV2.locationLayer,
  55. AgentV2.locationLayer,
  56. PluginBoot.locationLayer,
  57. FileSystem.locationLayer,
  58. Watcher.locationLayer,
  59. Pty.locationLayer,
  60. SkillV2.locationLayer,
  61. permissionsAndTools,
  62. LocationMutation.locationLayer.pipe(Layer.orDie),
  63. ).pipe(Layer.provideMerge(location))
  64. const commits = FileMutation.locationLayer.pipe(Layer.provide(services))
  65. const searches = LocationSearch.layer.pipe(Layer.provide(Ripgrep.layer), Layer.provide(services))
  66. const resources = ToolOutputStore.layer.pipe(Layer.provide(services))
  67. const todos = SessionTodo.layer.pipe(Layer.provide(services))
  68. const questions = QuestionV2.locationLayer.pipe(Layer.provide(services))
  69. const builtInTools = BuiltInTools.locationLayer.pipe(
  70. Layer.provide(services),
  71. Layer.provide(commits),
  72. Layer.provide(searches),
  73. Layer.provide(resources),
  74. Layer.provide(todos),
  75. Layer.provide(questions),
  76. )
  77. const model = SessionRunnerModel.locationLayer.pipe(Layer.provide(services))
  78. const runner = SessionRunnerLLM.defaultLayer.pipe(Layer.provide(services), Layer.provide(model))
  79. const coordinator = SessionRunCoordinator.layer.pipe(Layer.provide(runner))
  80. return Layer.mergeAll(
  81. services,
  82. commits,
  83. searches,
  84. resources,
  85. todos,
  86. questions,
  87. model,
  88. runner,
  89. coordinator,
  90. builtInTools,
  91. ).pipe(Layer.fresh)
  92. },
  93. idleTimeToLive: "60 minutes",
  94. dependencies: [
  95. Project.defaultLayer,
  96. EventV2.defaultLayer,
  97. Auth.defaultLayer,
  98. Npm.defaultLayer,
  99. ModelsDev.defaultLayer,
  100. FSUtil.defaultLayer,
  101. AppProcess.defaultLayer,
  102. Global.defaultLayer,
  103. Database.defaultLayer,
  104. SessionStore.layer.pipe(Layer.provide(Database.defaultLayer)),
  105. PermissionSaved.defaultLayer,
  106. RepositoryCache.defaultLayer,
  107. LLMClient.layer.pipe(Layer.provide(RequestExecutor.defaultLayer)),
  108. FetchHttpClient.layer,
  109. ToolOutputStore.defaultCleanupLayer,
  110. ],
  111. }) {}