effectify Skill service internals (#19364)
parent
21023337fa
commit
decb5e68ee
|
|
@ -63,16 +63,23 @@ export namespace Skill {
|
||||||
readonly available: (agent?: Agent.Info) => Effect.Effect<Info[]>
|
readonly available: (agent?: Agent.Info) => Effect.Effect<Info[]>
|
||||||
}
|
}
|
||||||
|
|
||||||
const add = async (state: State, match: string) => {
|
const add = Effect.fnUntraced(function* (state: State, match: string, bus: Bus.Interface) {
|
||||||
const md = await ConfigMarkdown.parse(match).catch(async (err) => {
|
const md = yield* Effect.tryPromise({
|
||||||
const message = ConfigMarkdown.FrontmatterError.isInstance(err)
|
try: () => ConfigMarkdown.parse(match),
|
||||||
? err.data.message
|
catch: (err) => err,
|
||||||
: `Failed to parse skill ${match}`
|
}).pipe(
|
||||||
const { Session } = await import("@/session")
|
Effect.catch(
|
||||||
Bus.publish(Session.Event.Error, { error: new NamedError.Unknown({ message }).toObject() })
|
Effect.fnUntraced(function* (err) {
|
||||||
log.error("failed to load skill", { skill: match, err })
|
const message = ConfigMarkdown.FrontmatterError.isInstance(err)
|
||||||
return undefined
|
? err.data.message
|
||||||
})
|
: `Failed to parse skill ${match}`
|
||||||
|
const { Session } = yield* Effect.promise(() => import("@/session"))
|
||||||
|
yield* bus.publish(Session.Event.Error, { error: new NamedError.Unknown({ message }).toObject() })
|
||||||
|
log.error("failed to load skill", { skill: match, err })
|
||||||
|
return undefined
|
||||||
|
}),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
if (!md) return
|
if (!md) return
|
||||||
|
|
||||||
|
|
@ -94,80 +101,115 @@ export namespace Skill {
|
||||||
location: match,
|
location: match,
|
||||||
content: md.content,
|
content: md.content,
|
||||||
}
|
}
|
||||||
}
|
})
|
||||||
|
|
||||||
const scan = async (state: State, root: string, pattern: string, opts?: { dot?: boolean; scope?: string }) => {
|
const scan = Effect.fnUntraced(function* (
|
||||||
return Glob.scan(pattern, {
|
state: State,
|
||||||
cwd: root,
|
bus: Bus.Interface,
|
||||||
absolute: true,
|
root: string,
|
||||||
include: "file",
|
pattern: string,
|
||||||
symlink: true,
|
opts?: { dot?: boolean; scope?: string },
|
||||||
dot: opts?.dot,
|
) {
|
||||||
})
|
const matches = yield* Effect.tryPromise({
|
||||||
.then((matches) => Promise.all(matches.map((match) => add(state, match))))
|
try: () =>
|
||||||
.catch((error) => {
|
Glob.scan(pattern, {
|
||||||
if (!opts?.scope) throw error
|
cwd: root,
|
||||||
|
absolute: true,
|
||||||
|
include: "file",
|
||||||
|
symlink: true,
|
||||||
|
dot: opts?.dot,
|
||||||
|
}),
|
||||||
|
catch: (error) => error,
|
||||||
|
}).pipe(
|
||||||
|
Effect.catch((error) => {
|
||||||
|
if (!opts?.scope) return Effect.die(error)
|
||||||
log.error(`failed to scan ${opts.scope} skills`, { dir: root, error })
|
log.error(`failed to scan ${opts.scope} skills`, { dir: root, error })
|
||||||
})
|
return Effect.succeed([] as string[])
|
||||||
}
|
}),
|
||||||
|
)
|
||||||
|
|
||||||
async function loadSkills(state: State, discovery: Discovery.Interface, directory: string, worktree: string) {
|
yield* Effect.forEach(matches, (match) => add(state, match, bus), {
|
||||||
|
concurrency: "unbounded",
|
||||||
|
discard: true,
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
const loadSkills = Effect.fnUntraced(function* (
|
||||||
|
state: State,
|
||||||
|
config: Config.Interface,
|
||||||
|
discovery: Discovery.Interface,
|
||||||
|
bus: Bus.Interface,
|
||||||
|
directory: string,
|
||||||
|
worktree: string,
|
||||||
|
) {
|
||||||
if (!Flag.OPENCODE_DISABLE_EXTERNAL_SKILLS) {
|
if (!Flag.OPENCODE_DISABLE_EXTERNAL_SKILLS) {
|
||||||
for (const dir of EXTERNAL_DIRS) {
|
for (const dir of EXTERNAL_DIRS) {
|
||||||
const root = path.join(Global.Path.home, dir)
|
const root = path.join(Global.Path.home, dir)
|
||||||
if (!(await Filesystem.isDir(root))) continue
|
const isDir = yield* Effect.promise(() => Filesystem.isDir(root))
|
||||||
await scan(state, root, EXTERNAL_SKILL_PATTERN, { dot: true, scope: "global" })
|
if (!isDir) continue
|
||||||
|
yield* scan(state, bus, root, EXTERNAL_SKILL_PATTERN, { dot: true, scope: "global" })
|
||||||
}
|
}
|
||||||
|
|
||||||
for await (const root of Filesystem.up({
|
const upDirs = yield* Effect.promise(async () => {
|
||||||
targets: EXTERNAL_DIRS,
|
const dirs: string[] = []
|
||||||
start: directory,
|
for await (const root of Filesystem.up({
|
||||||
stop: worktree,
|
targets: EXTERNAL_DIRS,
|
||||||
})) {
|
start: directory,
|
||||||
await scan(state, root, EXTERNAL_SKILL_PATTERN, { dot: true, scope: "project" })
|
stop: worktree,
|
||||||
|
})) {
|
||||||
|
dirs.push(root)
|
||||||
|
}
|
||||||
|
return dirs
|
||||||
|
})
|
||||||
|
|
||||||
|
for (const root of upDirs) {
|
||||||
|
yield* scan(state, bus, root, EXTERNAL_SKILL_PATTERN, { dot: true, scope: "project" })
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for (const dir of await Config.directories()) {
|
const configDirs = yield* config.directories()
|
||||||
await scan(state, dir, OPENCODE_SKILL_PATTERN)
|
for (const dir of configDirs) {
|
||||||
|
yield* scan(state, bus, dir, OPENCODE_SKILL_PATTERN)
|
||||||
}
|
}
|
||||||
|
|
||||||
const cfg = await Config.get()
|
const cfg = yield* config.get()
|
||||||
for (const item of cfg.skills?.paths ?? []) {
|
for (const item of cfg.skills?.paths ?? []) {
|
||||||
const expanded = item.startsWith("~/") ? path.join(os.homedir(), item.slice(2)) : item
|
const expanded = item.startsWith("~/") ? path.join(os.homedir(), item.slice(2)) : item
|
||||||
const dir = path.isAbsolute(expanded) ? expanded : path.join(directory, expanded)
|
const dir = path.isAbsolute(expanded) ? expanded : path.join(directory, expanded)
|
||||||
if (!(await Filesystem.isDir(dir))) {
|
const isDir = yield* Effect.promise(() => Filesystem.isDir(dir))
|
||||||
|
if (!isDir) {
|
||||||
log.warn("skill path not found", { path: dir })
|
log.warn("skill path not found", { path: dir })
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
await scan(state, dir, SKILL_PATTERN)
|
yield* scan(state, bus, dir, SKILL_PATTERN)
|
||||||
}
|
}
|
||||||
|
|
||||||
for (const url of cfg.skills?.urls ?? []) {
|
for (const url of cfg.skills?.urls ?? []) {
|
||||||
for (const dir of await Effect.runPromise(discovery.pull(url))) {
|
const pulledDirs = yield* discovery.pull(url)
|
||||||
|
for (const dir of pulledDirs) {
|
||||||
state.dirs.add(dir)
|
state.dirs.add(dir)
|
||||||
await scan(state, dir, SKILL_PATTERN)
|
yield* scan(state, bus, dir, SKILL_PATTERN)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
log.info("init", { count: Object.keys(state.skills).length })
|
log.info("init", { count: Object.keys(state.skills).length })
|
||||||
}
|
})
|
||||||
|
|
||||||
export class Service extends ServiceMap.Service<Service, Interface>()("@opencode/Skill") {}
|
export class Service extends ServiceMap.Service<Service, Interface>()("@opencode/Skill") {}
|
||||||
|
|
||||||
export const layer: Layer.Layer<Service, never, Discovery.Service> = Layer.effect(
|
export const layer: Layer.Layer<Service, never, Discovery.Service | Config.Service | Bus.Service> = Layer.effect(
|
||||||
Service,
|
Service,
|
||||||
Effect.gen(function* () {
|
Effect.gen(function* () {
|
||||||
const discovery = yield* Discovery.Service
|
const discovery = yield* Discovery.Service
|
||||||
|
const config = yield* Config.Service
|
||||||
|
const bus = yield* Bus.Service
|
||||||
const state = yield* InstanceState.make(
|
const state = yield* InstanceState.make(
|
||||||
Effect.fn("Skill.state")((ctx) =>
|
Effect.fn("Skill.state")(function* (ctx) {
|
||||||
Effect.gen(function* () {
|
const s: State = { skills: {}, dirs: new Set() }
|
||||||
const s: State = { skills: {}, dirs: new Set() }
|
yield* loadSkills(s, config, discovery, bus, ctx.directory, ctx.worktree)
|
||||||
yield* Effect.promise(() => loadSkills(s, discovery, ctx.directory, ctx.worktree))
|
return s
|
||||||
return s
|
}),
|
||||||
}),
|
|
||||||
),
|
|
||||||
)
|
)
|
||||||
|
|
||||||
const get = Effect.fn("Skill.get")(function* (name: string) {
|
const get = Effect.fn("Skill.get")(function* (name: string) {
|
||||||
|
|
@ -196,7 +238,11 @@ export namespace Skill {
|
||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
|
|
||||||
export const defaultLayer: Layer.Layer<Service> = layer.pipe(Layer.provide(Discovery.defaultLayer))
|
export const defaultLayer: Layer.Layer<Service> = layer.pipe(
|
||||||
|
Layer.provide(Discovery.defaultLayer),
|
||||||
|
Layer.provide(Config.defaultLayer),
|
||||||
|
Layer.provide(Bus.layer),
|
||||||
|
)
|
||||||
|
|
||||||
export function fmt(list: Info[], opts: { verbose: boolean }) {
|
export function fmt(list: Info[], opts: { verbose: boolean }) {
|
||||||
if (list.length === 0) return "No skills are currently available."
|
if (list.length === 0) return "No skills are currently available."
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue