94 lines
2.8 KiB
TypeScript
94 lines
2.8 KiB
TypeScript
import { existsSync, readdirSync, readFileSync } from "fs"
|
|
import { join, basename } from "path"
|
|
import { parseFrontmatter } from "../../shared/frontmatter"
|
|
import { isMarkdownFile } from "../../shared/file-utils"
|
|
import { getClaudeConfigDir } from "../../shared"
|
|
import type { AgentScope, AgentFrontmatter, ClaudeCodeAgentConfig, LoadedAgent } from "./types"
|
|
import { mapClaudeModelToOpenCode } from "./claude-model-mapper"
|
|
|
|
function parseToolsConfig(toolsStr?: string): Record<string, boolean> | undefined {
|
|
if (!toolsStr) return undefined
|
|
|
|
const tools = toolsStr.split(",").map((t) => t.trim()).filter(Boolean)
|
|
if (tools.length === 0) return undefined
|
|
|
|
const result: Record<string, boolean> = {}
|
|
for (const tool of tools) {
|
|
result[tool.toLowerCase()] = true
|
|
}
|
|
return result
|
|
}
|
|
|
|
function loadAgentsFromDir(agentsDir: string, scope: AgentScope): LoadedAgent[] {
|
|
if (!existsSync(agentsDir)) {
|
|
return []
|
|
}
|
|
|
|
const entries = readdirSync(agentsDir, { withFileTypes: true })
|
|
const agents: LoadedAgent[] = []
|
|
|
|
for (const entry of entries) {
|
|
if (!isMarkdownFile(entry)) continue
|
|
|
|
const agentPath = join(agentsDir, entry.name)
|
|
const agentName = basename(entry.name, ".md")
|
|
|
|
try {
|
|
const content = readFileSync(agentPath, "utf-8")
|
|
const { data, body } = parseFrontmatter<AgentFrontmatter>(content)
|
|
|
|
const name = data.name || agentName
|
|
const originalDescription = data.description || ""
|
|
|
|
const formattedDescription = `(${scope}) ${originalDescription}`
|
|
|
|
const mappedModelOverride = mapClaudeModelToOpenCode(data.model)
|
|
|
|
const config: ClaudeCodeAgentConfig = {
|
|
description: formattedDescription,
|
|
mode: data.mode || "subagent",
|
|
prompt: body.trim(),
|
|
...(mappedModelOverride ? { model: mappedModelOverride } : {}),
|
|
}
|
|
|
|
const toolsConfig = parseToolsConfig(data.tools)
|
|
if (toolsConfig) {
|
|
config.tools = toolsConfig
|
|
}
|
|
|
|
agents.push({
|
|
name,
|
|
path: agentPath,
|
|
config,
|
|
scope,
|
|
})
|
|
} catch {
|
|
continue
|
|
}
|
|
}
|
|
|
|
return agents
|
|
}
|
|
|
|
export function loadUserAgents(): Record<string, ClaudeCodeAgentConfig> {
|
|
const userAgentsDir = join(getClaudeConfigDir(), "agents")
|
|
const agents = loadAgentsFromDir(userAgentsDir, "user")
|
|
|
|
const result: Record<string, ClaudeCodeAgentConfig> = {}
|
|
for (const agent of agents) {
|
|
result[agent.name] = agent.config
|
|
}
|
|
return result
|
|
}
|
|
|
|
export function loadProjectAgents(directory?: string): Record<string, ClaudeCodeAgentConfig> {
|
|
const projectAgentsDir = join(directory ?? process.cwd(), ".claude", "agents")
|
|
const agents = loadAgentsFromDir(projectAgentsDir, "project")
|
|
|
|
const result: Record<string, ClaudeCodeAgentConfig> = {}
|
|
for (const agent of agents) {
|
|
result[agent.name] = agent.config
|
|
}
|
|
return result
|
|
}
|