feat(mcp): agentmark-mcp install — auto-wire detected AI clients

src/mcp/cli.ts

@@ -1,35 +1,210 @@
 #!/usr/bin/env node
 
 /**
- * `agentmark-mcp` CLI — the bin entry referenced by package.json.
+ * `agentmark-mcp` CLI.
  *
- * Configure in any MCP client to expose the entire AgentMark library:
- *
- *   {
- *     "mcpServers": {
- *       "agentmark": {
- *         "command": "npx",
- *         "args": ["-y", "@thinkfleet/agentmark", "agentmark-mcp"]
- *       }
- *     }
- *   }
- *
- * (Or just `npx -y @thinkfleet/agentmark` once the bin name resolves on $PATH.)
+ *   agentmark-mcp                 → start the MCP server (default)
+ *   agentmark-mcp serve           → same; explicit
+ *   agentmark-mcp install         → auto-wire detected AI clients
+ *   agentmark-mcp install --client=cursor --dry-run
+ *   agentmark-mcp uninstall       → remove our entries
+ *   agentmark-mcp doctor          → diagnose what's wired up
+ *   agentmark-mcp --help          → usage
  */
 
+import * as path from 'node:path'
+import { spawnSync } from 'node:child_process'
+import { existsSync } from 'node:fs'
 import { startMcpServer } from './server'
+import {
+    installToClients,
+    uninstallFromClients,
+    allClients,
+    type McpServerEntry,
+} from './install'
+
+const HELP = `agentmark-mcp — Model Context Protocol server for AgentMark
+
+USAGE
+  agentmark-mcp [serve]                 Start the MCP server (default; speaks stdio)
+  agentmark-mcp install [options]       Wire detected AI clients to talk to this server
+  agentmark-mcp uninstall [options]     Remove our entry from those clients
+  agentmark-mcp doctor                  Diagnose what's installed + wired up
+  agentmark-mcp --help                  This help
+
+OPTIONS (install / uninstall)
+  --client=<id>     Target specific client(s). Repeat or comma-separate.
+                    Known: ${allClients().map((c) => c.id).join(', ')}
+  --name=<name>     Entry name to register under (default: agentmark)
+  --command=<path>  Absolute command path to register (default: auto-detected)
+  --dry-run         Show what would change without writing
+`
+
+async function main(argv: string[]): Promise<number> {
+    const sub = argv[0] ?? 'serve'
+
+    if (sub === '-h' || sub === '--help' || sub === 'help') {
+        process.stdout.write(HELP)
+        return 0
+    }
+
+    if (sub === 'serve') {
+        await startMcpServer({ name: 'agentmark', version: pkgVersion() })
+        // The MCP transport keeps the event loop alive via stdio.
+        return await new Promise<number>(() => { /* never resolves */ })
+    }
+
+    if (sub === 'install') {
+        return await runInstall(argv.slice(1))
+    }
+    if (sub === 'uninstall') {
+        return await runUninstall(argv.slice(1))
+    }
+    if (sub === 'doctor') {
+        return await runDoctor()
+    }
 
-async function main(): Promise<void> {
-    await startMcpServer({
-        name: 'agentmark',
-        // Version is read from package.json at build time; for now hardcoded.
-        version: '0.7.0',
+    process.stderr.write(`Unknown subcommand: ${sub}\n\n${HELP}`)
+    return 2
+}
+
+async function runInstall(args: string[]): Promise<number> {
+    const flags = parseFlags(args)
+    const entry = buildEntryFromFlags(flags)
+    const result = await installToClients({
+        clientIds: flags.client,
+        entry,
+        entryName: flags.name?.[0],
+        dryRun: flags.dryRun,
+    })
+    printInstallResult(result, flags.dryRun ? 'dry-run' : 'install')
+    return result.clients.every((r) => r.action !== 'error') ? 0 : 1
+}
+
+async function runUninstall(args: string[]): Promise<number> {
+    const flags = parseFlags(args)
+    const result = await uninstallFromClients({
+        clientIds: flags.client,
+        entryName: flags.name?.[0],
+        dryRun: flags.dryRun,
     })
-    // Stay alive — the MCP transport keeps the event loop busy via stdio.
+    printInstallResult(result, flags.dryRun ? 'dry-run' : 'uninstall')
+    return result.clients.every((r) => r.action !== 'error') ? 0 : 1
+}
+
+async function runDoctor(): Promise<number> {
+    process.stdout.write('agentmark-mcp doctor\n\n')
+    process.stdout.write(`Auto-detected command:  ${defaultCommand()}\n`)
+    process.stdout.write(`Node:                   ${process.execPath} (${process.version})\n\n`)
+    process.stdout.write('Clients:\n')
+    for (const c of allClients()) {
+        const cfg = c.configPath() ?? '(unsupported on this OS)'
+        const installed = await c.isInstalled()
+        process.stdout.write(`  ${c.id.padEnd(16)} ${installed ? '✓' : '·'}  ${cfg}\n`)
+    }
+    return 0
+}
+
+interface ParsedFlags {
+    client: string[] | undefined
+    name: string[] | undefined
+    command: string[] | undefined
+    dryRun: boolean
+}
+
+function parseFlags(args: string[]): ParsedFlags {
+    const client: string[] = []
+    const name: string[] = []
+    const command: string[] = []
+    let dryRun = false
+    for (const arg of args) {
+        if (arg === '--dry-run' || arg === '-n') { dryRun = true; continue }
+        const m = arg.match(/^--(client|name|command)(?:=(.*))?$/)
+        if (!m) continue
+        const value = m[2]
+        if (value === undefined) continue
+        if (m[1] === 'client') value.split(',').filter(Boolean).forEach((v) => client.push(v.trim()))
+        if (m[1] === 'name') name.push(value)
+        if (m[1] === 'command') command.push(value)
+    }
+    return {
+        client: client.length > 0 ? client : undefined,
+        name: name.length > 0 ? name : undefined,
+        command: command.length > 0 ? command : undefined,
+        dryRun,
+    }
+}
+
+function buildEntryFromFlags(flags: ParsedFlags): McpServerEntry {
+    const command = flags.command?.[0] ?? defaultCommand()
+    return { command, args: [] }
 }
 
-main().catch((err) => {
-    // eslint-disable-next-line no-console
-    console.error('Failed to start AgentMark MCP server:', err)
-    process.exit(1)
-})
+/**
+ * Best-effort detection of the agentmark-mcp launcher to register in
+ * client configs. In order of preference:
+ *   1. Same-directory launcher script (when running from the bundled
+ *      installer layout: <root>/bin/agentmark-mcp or <root>\agentmark-mcp.cmd)
+ *   2. `agentmark-mcp` on PATH
+ *   3. Fall back to `npx -y @thinkfleet/agentmark agentmark-mcp` form
+ */
+function defaultCommand(): string {
+    // 1. Bundled-installer layout: this script lives at
+    //    <root>/agentmark/dist/src/mcp/cli.js; the launcher is at
+    //    <root>/bin/agentmark-mcp (POSIX) or <root>\agentmark-mcp.cmd (Win).
+    try {
+        const here = path.dirname(__filename)
+        const installRoot = path.resolve(here, '..', '..', '..', '..')
+        const posixLauncher = path.join(installRoot, 'bin', 'agentmark-mcp')
+        if (existsSync(posixLauncher)) return posixLauncher
+        const winLauncher = path.join(installRoot, 'agentmark-mcp.cmd')
+        if (existsSync(winLauncher)) return winLauncher
+    } catch { /* swallow — fall through to PATH lookup */ }
+
+    // 2. PATH lookup.
+    const which = spawnSync(process.platform === 'win32' ? 'where' : 'which', ['agentmark-mcp'], { encoding: 'utf8' })
+    if (which.status === 0) {
+        const found = which.stdout.split(/\r?\n/).find((l) => l.trim().length > 0)
+        if (found) return found.trim()
+    }
+
+    // 3. Last resort.
+    return process.execPath
+}
+
+function pkgVersion(): string {
+    try {
+        // The compiled cli.js sits next to package.json in the dist tree
+        // when bundled by the installer; the source layout differs.
+        // Read it lazily and tolerate failure.
+        // eslint-disable-next-line @typescript-eslint/no-require-imports
+        return (require(path.resolve(__dirname, '..', '..', 'package.json')).version as string) ?? '0.0.0'
+    } catch {
+        return '0.0.0'
+    }
+}
+
+function printInstallResult(result: { clients: Array<{ id: string; name: string; path: string; action: string; message?: string }> }, label: string): void {
+    process.stdout.write(`agentmark-mcp ${label} — results:\n\n`)
+    for (const c of result.clients) {
+        const flag =
+            c.action === 'added' || c.action === 'updated' || c.action === 'removed' ? '✓'
+                : c.action === 'already_present' || c.action === 'not_present' ? '·'
+                    : c.action === 'skipped' ? '⏭'
+                        : '✗'
+        const padded = `${c.name} [${c.action}]`.padEnd(40)
+        process.stdout.write(`  ${flag}  ${padded}  ${c.path}\n`)
+        if (c.message) process.stdout.write(`     ${c.message}\n`)
+    }
+    process.stdout.write('\n')
+}
+
+main(process.argv.slice(2))
+    .then((code) => {
+        if (typeof code === 'number' && code !== 0) process.exit(code)
+    })
+    .catch((err) => {
+        // eslint-disable-next-line no-console
+        console.error(`agentmark-mcp: ${(err as Error).message ?? err}`)
+        process.exit(1)
+    })