location-layer.ts 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  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 { SkillGuidance } from "./skill/guidance"
  30. import { BuiltInTools } from "./tool/builtins"
  31. import { Image } from "./image"
  32. import { ToolRegistry } from "./tool/registry"
  33. import { ApplicationTools } from "./tool/application-tools"
  34. import { ToolOutputStore } from "./tool-output-store"
  35. import { AppProcess } from "./process"
  36. import { Ripgrep } from "./ripgrep"
  37. import { SessionStore } from "./session/store"
  38. import { SessionTodo } from "./session/todo"
  39. import { QuestionV2 } from "./question"
  40. import { LLMClient } from "@opencode-ai/llm"
  41. import { RequestExecutor } from "@opencode-ai/llm/route"
  42. import * as SessionRunnerLLM from "./session/runner/llm"
  43. import { SessionRunnerModel } from "./session/runner/model"
  44. import { SystemContextBuiltIns } from "./system-context/builtins"
  45. import { FetchHttpClient } from "effect/unstable/http"
  46. export class LocationServiceMap extends LayerMap.Service<LocationServiceMap>()("@opencode/example/LocationServiceMap", {
  47. lookup: (ref: Location.Ref) => {
  48. const location = Location.layer(ref)
  49. const systemContext = SystemContextBuiltIns.locationLayer
  50. const base = Layer.mergeAll(
  51. location,
  52. Policy.locationLayer,
  53. Config.locationLayer,
  54. ProjectReference.locationLayer,
  55. PluginV2.locationLayer,
  56. Catalog.locationLayer,
  57. CommandV2.locationLayer,
  58. AgentV2.locationLayer,
  59. PluginBoot.locationLayer,
  60. FileSystem.locationLayer,
  61. Watcher.locationLayer,
  62. Pty.locationLayer,
  63. SkillV2.locationLayer,
  64. systemContext,
  65. LocationMutation.locationLayer.pipe(Layer.orDie),
  66. ).pipe(Layer.provideMerge(location))
  67. const resources = ToolOutputStore.layer.pipe(Layer.provide(base))
  68. const permissionsAndTools = ToolRegistry.layer.pipe(
  69. Layer.provideMerge(PermissionV2.locationLayer),
  70. Layer.provide(resources),
  71. Layer.provide(base),
  72. )
  73. const services = Layer.mergeAll(base, resources, permissionsAndTools)
  74. const image = Image.layer.pipe(Layer.provide(services))
  75. const mutation = FileMutation.locationLayer.pipe(Layer.provide(services))
  76. const searches = LocationSearch.layer.pipe(Layer.provide(Ripgrep.layer), Layer.provide(services))
  77. const skillGuidance = SkillGuidance.locationLayer.pipe(Layer.provide(services))
  78. const todos = SessionTodo.layer.pipe(Layer.provide(services))
  79. const questions = QuestionV2.locationLayer.pipe(Layer.provide(services))
  80. const builtInTools = BuiltInTools.locationLayer.pipe(
  81. Layer.provide(services),
  82. Layer.provide(mutation),
  83. Layer.provide(searches),
  84. Layer.provide(resources),
  85. Layer.provide(todos),
  86. Layer.provide(questions),
  87. Layer.provide(image),
  88. )
  89. const model = SessionRunnerModel.locationLayer.pipe(Layer.provide(services))
  90. const runner = SessionRunnerLLM.defaultLayer.pipe(
  91. Layer.provide(services),
  92. Layer.provide(model),
  93. Layer.provide(skillGuidance),
  94. )
  95. return Layer.mergeAll(
  96. services,
  97. image,
  98. mutation,
  99. searches,
  100. resources,
  101. todos,
  102. questions,
  103. model,
  104. runner,
  105. builtInTools,
  106. ).pipe(Layer.fresh)
  107. },
  108. idleTimeToLive: "60 minutes",
  109. dependencies: [
  110. Project.defaultLayer,
  111. EventV2.defaultLayer,
  112. Auth.defaultLayer,
  113. Npm.defaultLayer,
  114. ModelsDev.defaultLayer,
  115. FSUtil.defaultLayer,
  116. AppProcess.defaultLayer,
  117. Global.defaultLayer,
  118. Database.defaultLayer,
  119. SessionStore.layer.pipe(Layer.provide(Database.defaultLayer)),
  120. PermissionSaved.defaultLayer,
  121. RepositoryCache.defaultLayer,
  122. LLMClient.layer.pipe(Layer.provide(RequestExecutor.defaultLayer)),
  123. FetchHttpClient.layer,
  124. ToolOutputStore.defaultCleanupLayer,
  125. ApplicationTools.layer,
  126. ],
  127. }) {}