location-layer.ts 4.6 KB

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