openAgent

1import fs from "fs/promises"
2import path from "path"
3import { describe, expect } from "bun:test"
4import { Effect, Layer } from "effect"
5import { AgentV2 } from "@opencode-ai/core/agent"
6import { AppNodeBuilder } from "@opencode-ai/core/effect/app-node-builder"
7import { LayerNode } from "@opencode-ai/core/effect/layer-node"
8import { FSUtil } from "@opencode-ai/core/fs-util"
9import { AbsolutePath } from "@opencode-ai/core/schema"
10import { SkillV2 } from "@opencode-ai/core/skill"
11import { SkillDiscovery } from "@opencode-ai/core/skill/discovery"
12import { tmpdir } from "./fixture/tmpdir"
13import { testEffect } from "./lib/effect"
14 
15const urls = new Map<string, AbsolutePath[]>()
16let pulls = 0
17const discovery = Layer.succeed(
18  SkillDiscovery.Service,
19  SkillDiscovery.Service.of({
20    pull: (url) => {
21      pulls++
22      return Effect.succeed(urls.get(url) ?? [])
23    },
24  }),
25)
26const it = testEffect(
27  AppNodeBuilder.build(LayerNode.group([SkillV2.node, AgentV2.node]), [[SkillDiscovery.node, discovery]]),
28)
29 
30function write(directory: string, name: string, description: string) {
31  return fs.writeFile(
32    path.join(directory, name, "SKILL.md"),
33    `---
34name: ${name}
35description: ${description}
36---
37# ${name}`,
38  )
39}
40 
41describe("SkillV2", () => {
42  it.live("registers sources and resolves later source precedence", () =>
43    Effect.acquireRelease(
44      Effect.promise(() => tmpdir()),
45      (tmp) => Effect.promise(() => tmp[Symbol.asyncDispose]()),
46    ).pipe(
47      Effect.flatMap((tmp) =>
48        Effect.gen(function* () {
49          const first = path.join(tmp.path, "first")
50          const second = path.join(tmp.path, "second")
51          yield* Effect.promise(async () => {
52            await fs.mkdir(path.join(first, "review"), { recursive: true })
53            await fs.mkdir(path.join(second, "review"), { recursive: true })
54            await write(first, "review", "First")
55            await write(second, "review", "Second")
56            await fs.writeFile(path.join(first, "foo.md"), "---\nslash: true\n---\n# foo")
57          })
58 
59          const skill = yield* SkillV2.Service
60          yield* skill.transform((editor) => {
61            editor.source({ type: "directory", path: AbsolutePath.make(first) })
62            editor.source({ type: "directory", path: AbsolutePath.make(first) })
63            editor.source({ type: "directory", path: AbsolutePath.make(second) })
64            expect(editor.list()).toEqual([
65              { type: "directory", path: AbsolutePath.make(first) },
66              { type: "directory", path: AbsolutePath.make(second) },
67            ])
68          })
69 
70          expect(yield* skill.sources()).toEqual([
71            { type: "directory", path: AbsolutePath.make(first) },
72            { type: "directory", path: AbsolutePath.make(second) },
73          ])
74          expect(yield* skill.list()).toEqual([
75            SkillV2.Info.make({
76              name: "foo",
77              slash: true,
78              location: AbsolutePath.make(path.join(first, "foo.md")),
79              content: "# foo",
80            }),
81            {
82              name: "review",
83              description: "Second",
84              location: AbsolutePath.make(path.join(second, "review", "SKILL.md")),
85              content: "# review",
86            },
87          ])
88        }),
89      ),
90    ),
91  )
92 
93  it.live("loads URL sources and filters skills for agents", () =>
94    Effect.acquireRelease(
95      Effect.promise(() => tmpdir()),
96      (tmp) => Effect.promise(() => tmp[Symbol.asyncDispose]()),
97    ).pipe(
98      Effect.flatMap((tmp) =>
99        Effect.gen(function* () {
100          yield* Effect.promise(async () => {
101            await fs.mkdir(path.join(tmp.path, "deploy"), { recursive: true })
102            await write(tmp.path, "deploy", "Deploy production")
103          })
104          pulls = 0
105          urls.set("https://example.test/skills/", [AbsolutePath.make(tmp.path)])
106 
107          const agents = yield* AgentV2.Service
108          yield* agents.transform((editor) =>
109            editor.update(AgentV2.ID.make("reviewer"), (agent) => {
110              agent.permissions.push({ action: "skill", resource: "deploy", effect: "deny" })
111            }),
112          )
113 
114          const skill = yield* SkillV2.Service
115          yield* skill.transform((editor) => editor.source({ type: "url", url: "https://example.test/skills/" }))
116 
117          expect((yield* skill.list()).map((item) => item.name)).toEqual(["deploy"])
118          expect((yield* skill.list()).map((item) => item.name)).toEqual(["deploy"])
119          expect(pulls).toBe(1)
120          expect(SkillV2.available(yield* skill.list(), (yield* agents.get(AgentV2.ID.make("reviewer")))!)).toEqual([])
121        }),
122      ),
123    ),
124  )
125})
126