openAgent

1export * as Reference from "./reference"
2 
3import { makeLocationNode } from "./effect/app-node"
4import { Context, Effect, Layer, Scope, Types } from "effect"
5import { Reference } from "@opencode-ai/schema/reference"
6import { Global } from "./global"
7import { EventV2 } from "./event"
8import { Repository } from "./repository"
9import { RepositoryCache } from "./repository-cache"
10import { AbsolutePath } from "./schema"
11import { State } from "./state"
12 
13export const LocalSource = Reference.LocalSource
14export type LocalSource = Reference.LocalSource
15 
16export const GitSource = Reference.GitSource
17export type GitSource = Reference.GitSource
18 
19export const Source = Reference.Source
20export type Source = Reference.Source
21 
22export const Event = Reference.Event
23 
24export const Info = Reference.Info
25export type Info = Reference.Info
26 
27type Data = {
28  sources: Map<string, Types.DeepMutable<Source>>
29}
30 
31type Draft = {
32  add(name: string, source: Source): void
33  remove(name: string): void
34  list(): readonly [string, Source][]
35}
36 
37export interface Interface extends State.Transformable<Draft> {
38  readonly list: () => Effect.Effect<Info[]>
39}
40 
41export class Service extends Context.Service<Service, Interface>()("@opencode/v2/Reference") {}
42 
43const layer = Layer.effect(
44  Service,
45  Effect.gen(function* () {
46    const global = yield* Global.Service
47    const events = yield* EventV2.Service
48    const cache = yield* RepositoryCache.Service
49    const scope = yield* Scope.Scope
50    const materialized = new Map<string, Info>()
51    const state = State.create<Data, Draft>({
52      initial: () => ({ sources: new Map() }),
53      draft: (draft) => ({
54        add: (name, source) => draft.sources.set(name, source as Types.DeepMutable<Source>),
55        remove: (name) => draft.sources.delete(name),
56        list: () => Array.from(draft.sources.entries()) as [string, Source][],
57      }),
58      finalize: (draft) =>
59        Effect.gen(function* () {
60          materialized.clear()
61          for (const [name, source] of draft.list()) {
62            if (source.type === "local") {
63              materialized.set(
64                name,
65                new Info({
66                  name,
67                  path: source.path,
68                  ...(source.description === undefined ? {} : { description: source.description }),
69                  ...(source.hidden === undefined ? {} : { hidden: source.hidden }),
70                  source,
71                }),
72              )
73              continue
74            }
75            const repository = Repository.parse(source.repository)
76            if (!repository || !Repository.isRemote(repository)) continue
77            if (source.branch) {
78              try {
79                Repository.validateBranch(source.branch)
80              } catch {
81                continue
82              }
83            }
84            materialized.set(
85              name,
86              new Info({
87                name,
88                path: AbsolutePath.make(Repository.cachePath(global.repos, repository, source.branch)),
89                ...(source.description === undefined ? {} : { description: source.description }),
90                ...(source.hidden === undefined ? {} : { hidden: source.hidden }),
91                source,
92              }),
93            )
94            yield* cache.ensure({ reference: repository, branch: source.branch, refresh: true }).pipe(
95              Effect.catchCause((cause) =>
96                Effect.logWarning("failed to materialize reference", {
97                  name,
98                  repository: source.repository,
99                  cause,
100                }),
101              ),
102              Effect.forkIn(scope),
103            )
104          }
105          yield* events.publish(Event.Updated, {})
106        }),
107    })
108 
109    return Service.of({
110      transform: state.transform,
111      reload: state.reload,
112      list: Effect.fn("Reference.list")(function* () {
113        return Array.from(materialized.values())
114      }),
115    })
116  }),
117)
118 
119export const locationLayer = layer
120 
121export const node = makeLocationNode({
122  service: Service,
123  layer,
124  deps: [Global.node, EventV2.node, RepositoryCache.node],
125})
126