test(cli): batch 105 (25 files)

src/cli/install-codex/link-cached-plugin-agents.ts

@@ -0,0 +1,95 @@
+import { copyFile, lstat, mkdir, readdir, rm, symlink, writeFile } from "node:fs/promises"
+import { basename, join } from "node:path"
+
+const MANIFEST_FILE = ".installed-agents.json"
+
+export interface LinkedAgent {
+  readonly name: string
+  readonly path: string
+  readonly target: string
+}
+
+type LinkPlatform = NodeJS.Platform
+
+export async function linkCachedPluginAgents(input: {
+  readonly codexHome: string
+  readonly pluginRoot: string
+  readonly platform?: LinkPlatform
+}): Promise<readonly LinkedAgent[]> {
+  const platform = input.platform ?? process.platform
+  const bundledAgents = await discoverBundledAgents(input.pluginRoot)
+  if (bundledAgents.length === 0) {
+    await writeManifest(input.pluginRoot, [])
+    return []
+  }
+  const agentsDir = join(input.codexHome, "agents")
+  await mkdir(agentsDir, { recursive: true })
+  const linked: LinkedAgent[] = []
+  for (const agentPath of bundledAgents) {
+    const linkPath = join(agentsDir, basename(agentPath))
+    if (platform === "win32") {
+      await replaceWithCopy(linkPath, agentPath)
+    } else {
+      await replaceWithSymlink(linkPath, agentPath)
+    }
+    linked.push({ name: basename(agentPath), path: linkPath, target: agentPath })
+  }
+  await writeManifest(
+    input.pluginRoot,
+    linked.map((entry) => entry.path),
+  )
+  return linked
+}
+
+async function discoverBundledAgents(pluginRoot: string): Promise<readonly string[]> {
+  const componentsRoot = join(pluginRoot, "components")
+  if (!(await exists(componentsRoot))) return []
+  const componentEntries = await readdir(componentsRoot, { withFileTypes: true })
+  const agents: string[] = []
+  for (const entry of componentEntries) {
+    if (!entry.isDirectory()) continue
+    const agentsRoot = join(componentsRoot, entry.name, "agents")
+    if (!(await exists(agentsRoot))) continue
+    const agentEntries = await readdir(agentsRoot, { withFileTypes: true })
+    for (const file of agentEntries) {
+      if (!file.isFile() || !file.name.endsWith(".toml")) continue
+      agents.push(join(agentsRoot, file.name))
+    }
+  }
+  agents.sort()
+  return agents
+}
+
+async function replaceWithSymlink(linkPath: string, target: string): Promise<void> {
+  await prepareReplacement(linkPath)
+  await symlink(target, linkPath)
+}
+
+async function replaceWithCopy(linkPath: string, target: string): Promise<void> {
+  await prepareReplacement(linkPath)
+  await copyFile(target, linkPath)
+}
+
+async function prepareReplacement(linkPath: string): Promise<void> {
+  if (!(await exists(linkPath))) return
+  const entryStat = await lstat(linkPath)
+  if (entryStat.isDirectory() && !entryStat.isSymbolicLink()) {
+    throw new Error(`${linkPath} already exists and is a directory; refusing to replace`)
+  }
+  await rm(linkPath, { force: true })
+}
+
+async function writeManifest(pluginRoot: string, agentPaths: readonly string[]): Promise<void> {
+  const manifestPath = join(pluginRoot, MANIFEST_FILE)
+  const payload = { agents: [...agentPaths].sort() }
+  await writeFile(manifestPath, `${JSON.stringify(payload, null, "\t")}\n`)
+}
+
+async function exists(path: string): Promise<boolean> {
+  try {
+    await lstat(path)
+    return true
+  } catch {
+    return false
+  }
+}