feat(mcp): register ast-grep as built-in MCP

Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
2026-05-18 21:19:24 +09:00
parent 499aff011a
commit ef09880e26
9 changed files with 273 additions and 22 deletions
@@ -0,0 +1,119 @@
+import { afterEach, describe, expect, it } from "bun:test"
+import { mkdirSync, mkdtempSync, rmSync, writeFileSync } from "node:fs"
+import { tmpdir } from "node:os"
+import { join } from "node:path"
+import { pathToFileURL } from "node:url"
+import { createAstGrepMcpConfig } from "./ast-grep"
+
+const temporaryDirectories: string[] = []
+
+function createTemporaryDirectory(prefix: string): string {
+  const directory = mkdtempSync(join(tmpdir(), prefix))
+  temporaryDirectories.push(directory)
+  return directory
+}
+
+afterEach(() => {
+  for (const directory of temporaryDirectories.splice(0)) {
+    rmSync(directory, { recursive: true, force: true })
+  }
+})
+
+describe("createAstGrepMcpConfig", () => {
+  it("resolves bundled dist cli from module root when cwd is unrelated", () => {
+    // given
+    const packageRoot = createTemporaryDirectory("omo-ast-grep-package-root-")
+    const unrelatedCwd = createTemporaryDirectory("omo-ast-grep-unrelated-cwd-")
+    const moduleFilePath = join(packageRoot, "dist", "index.js")
+    const cliPath = join(packageRoot, "packages", "ast-grep-mcp", "dist", "cli.js")
+    mkdirSync(join(packageRoot, "dist"), { recursive: true })
+    mkdirSync(join(packageRoot, "packages", "ast-grep-mcp", "dist"), { recursive: true })
+    writeFileSync(cliPath, "#!/usr/bin/env node\n", "utf-8")
+
+    // when
+    const config = createAstGrepMcpConfig({
+      cwd: unrelatedCwd,
+      moduleUrl: pathToFileURL(moduleFilePath).href,
+    })
+
+    // then
+    expect(config.command).toEqual(["node", cliPath, "mcp"])
+    expect(config.environment?.OMO_AST_GREP_WORKSPACE).toBe(unrelatedCwd)
+  })
+
+  it("falls back to bun source cli for source checkouts before build", () => {
+    // given
+    const packageRoot = createTemporaryDirectory("omo-ast-grep-source-root-")
+    const moduleFilePath = join(packageRoot, "src", "mcp", "ast-grep.ts")
+    const sourceCliPath = join(packageRoot, "packages", "ast-grep-mcp", "src", "cli.ts")
+    mkdirSync(join(packageRoot, "src", "mcp"), { recursive: true })
+    mkdirSync(join(packageRoot, "packages", "ast-grep-mcp", "src"), { recursive: true })
+    writeFileSync(sourceCliPath, "console.log('mcp')\n", "utf-8")
+
+    // when
+    const config = createAstGrepMcpConfig({
+      cwd: createTemporaryDirectory("omo-ast-grep-source-cwd-"),
+      moduleUrl: pathToFileURL(moduleFilePath).href,
+    })
+
+    // then
+    expect(config.command).toEqual(["bun", sourceCliPath, "mcp"])
+  })
+
+  it("still returns a built-in MCP config when the cli has not been built yet", () => {
+    // given
+    const packageRoot = createTemporaryDirectory("omo-ast-grep-missing-root-")
+    const moduleFilePath = join(packageRoot, "dist", "index.js")
+    mkdirSync(join(packageRoot, "dist"), { recursive: true })
+
+    // when
+    const config = createAstGrepMcpConfig({
+      cwd: createTemporaryDirectory("omo-ast-grep-missing-cwd-"),
+      moduleUrl: pathToFileURL(moduleFilePath).href,
+    })
+
+    // then
+    expect(config.enabled).toBe(true)
+    expect(config.command[0]).toBe("node")
+    expect(config.command[1]).toContain(join("packages", "ast-grep-mcp", "dist", "cli.js"))
+    expect(config.command[2]).toBe("mcp")
+  })
+
+  it("does not resolve the MCP command from the opened workspace", () => {
+    // given
+    const packageRoot = createTemporaryDirectory("omo-ast-grep-safe-package-root-")
+    const workspaceRoot = createTemporaryDirectory("omo-ast-grep-malicious-workspace-")
+    const moduleFilePath = join(packageRoot, "dist", "index.js")
+    const workspaceCliPath = join(workspaceRoot, "packages", "ast-grep-mcp", "dist", "cli.js")
+    mkdirSync(join(packageRoot, "dist"), { recursive: true })
+    mkdirSync(join(workspaceRoot, "packages", "ast-grep-mcp", "dist"), { recursive: true })
+    writeFileSync(workspaceCliPath, "console.log('malicious')\n", "utf-8")
+
+    // when
+    const config = createAstGrepMcpConfig({
+      cwd: workspaceRoot,
+      moduleUrl: pathToFileURL(moduleFilePath).href,
+    })
+
+    // then
+    expect(config.command[1]).not.toBe(workspaceCliPath)
+    expect(config.command[1]).toContain(packageRoot)
+  })
+
+  it("maps disabled ast-grep tool names to MCP subtools", () => {
+    // given
+    const packageRoot = createTemporaryDirectory("omo-ast-grep-disabled-root-")
+    const moduleFilePath = join(packageRoot, "dist", "index.js")
+    mkdirSync(join(packageRoot, "dist"), { recursive: true })
+
+    // when
+    const config = createAstGrepMcpConfig({
+      cwd: createTemporaryDirectory("omo-ast-grep-disabled-cwd-"),
+      disabledTools: ["ast_grep_replace", "glob"],
+      moduleUrl: pathToFileURL(moduleFilePath).href,
+    })
+
+    // then
+    expect(config.environment?.OMO_AST_GREP_DISABLED_TOOLS).toBe("replace")
+  })
+})