lens-engine 2.0.1 → 2.1.0

package/cli.js

@@ -368,21 +368,211 @@ async function runMain(cmd, opts = {}) {
 
 // packages/cli/src/commands/daemon.ts
 import { spawn } from "child_process";
-import { existsSync, mkdirSync, openSync, readFileSync, unlinkSync, writeFileSync } from "fs";
+import { existsSync as existsSync3, mkdirSync as mkdirSync2, openSync, readFileSync as readFileSync2, unlinkSync as unlinkSync2, writeFileSync as writeFileSync2 } from "fs";
+import { createRequire as createRequire2 } from "module";
+import { homedir as homedir2 } from "os";
+import { dirname as dirname3, join as join3 } from "path";
+import { fileURLToPath as fileURLToPath2 } from "url";
+
+// packages/cli/src/commands/daemon-install.ts
+import { existsSync as existsSync2 } from "fs";
 import { createRequire } from "module";
-import { homedir } from "os";
-import { dirname, join } from "path";
+import { dirname as dirname2, join as join2 } from "path";
 import { fileURLToPath } from "url";
-var DATA_DIR = join(homedir(), ".lens");
-var PID_FILE = join(DATA_DIR, "daemon.pid");
-var LOG_FILE = join(DATA_DIR, "daemon.log");
+
+// packages/cli/src/lib/launchd.ts
+import { execFileSync } from "child_process";
+import { existsSync, mkdirSync, readFileSync, unlinkSync, writeFileSync } from "fs";
+import { homedir, platform } from "os";
+import { dirname, join } from "path";
+var PLIST_LABEL = "com.lens.daemon";
+var PLIST_PATH = join(homedir(), "Library", "LaunchAgents", `${PLIST_LABEL}.plist`);
+var LOG_PATH = join(homedir(), ".lens", "daemon.log");
+function isMacOS() {
+  return platform() === "darwin";
+}
+function guiDomain() {
+  const uid = process.getuid?.();
+  if (typeof uid !== "number") {
+    throw new Error("process.getuid() unavailable \u2014 macOS required for launchd commands");
+  }
+  return `gui/${uid}`;
+}
+function renderPlist({ nodePath, daemonEntry, extraEnv = {} }) {
+  const env = {
+    PATH: `${dirname(nodePath)}:/usr/local/bin:/opt/homebrew/bin:/usr/bin:/bin:/usr/sbin:/sbin`,
+    HOME: homedir(),
+    LENS_DAEMON: "1",
+    ...extraEnv
+  };
+  const envEntries = Object.entries(env).map(([k, v]) => `      <key>${escapeXml(k)}</key>
+      <string>${escapeXml(v)}</string>`).join("\n");
+  return `<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+  <key>Label</key>
+  <string>${PLIST_LABEL}</string>
+  <key>ProgramArguments</key>
+  <array>
+    <string>${escapeXml(nodePath)}</string>
+    <string>${escapeXml(daemonEntry)}</string>
+  </array>
+  <key>RunAtLoad</key>
+  <true/>
+  <key>KeepAlive</key>
+  <true/>
+  <key>StandardOutPath</key>
+  <string>${escapeXml(LOG_PATH)}</string>
+  <key>StandardErrorPath</key>
+  <string>${escapeXml(LOG_PATH)}</string>
+  <key>EnvironmentVariables</key>
+  <dict>
+${envEntries}
+  </dict>
+</dict>
+</plist>
+`;
+}
+function escapeXml(s) {
+  return s.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;").replace(/"/g, "&quot;").replace(/'/g, "&apos;");
+}
+function writePlist(contents) {
+  mkdirSync(dirname(PLIST_PATH), { recursive: true });
+  writeFileSync(PLIST_PATH, contents);
+}
+function bootstrapPlist() {
+  const domain = guiDomain();
+  try {
+    execFileSync("launchctl", ["bootout", domain, PLIST_PATH], { stdio: "ignore" });
+  } catch {
+  }
+  try {
+    execFileSync("launchctl", ["bootstrap", domain, PLIST_PATH], { stdio: "inherit" });
+  } catch (e) {
+    throw new Error(
+      `launchctl bootstrap failed. Check Console.app (filter: com.lens.daemon) for details.
+${e.message}`
+    );
+  }
+}
+function unloadPlist() {
+  const domain = guiDomain();
+  try {
+    execFileSync("launchctl", ["bootout", domain, PLIST_PATH], { stdio: "ignore" });
+  } catch {
+  }
+  if (existsSync(PLIST_PATH)) unlinkSync(PLIST_PATH);
+}
+function plistStatus() {
+  const installed = existsSync(PLIST_PATH);
+  let loaded = false;
+  let pid = null;
+  let lastExit = null;
+  try {
+    const out = execFileSync("launchctl", ["list"], { encoding: "utf-8" });
+    for (const line of out.split("\n")) {
+      const parts = line.split(/\s+/);
+      if (parts[2] === PLIST_LABEL) {
+        loaded = true;
+        const pidParsed = parseInt(parts[0], 10);
+        const exitParsed = parseInt(parts[1], 10);
+        pid = Number.isNaN(pidParsed) ? null : pidParsed;
+        lastExit = Number.isNaN(exitParsed) ? null : exitParsed;
+        break;
+      }
+    }
+  } catch {
+  }
+  return { installed, loaded, pid, lastExit };
+}
+function readPlist() {
+  try {
+    return readFileSync(PLIST_PATH, "utf-8");
+  } catch {
+    return null;
+  }
+}
+
+// packages/cli/src/commands/daemon-install.ts
 function findDaemonEntry() {
   const self = fileURLToPath(import.meta.url);
-  const sibling = join(dirname(self), "daemon.js");
-  if (existsSync(sibling)) return sibling;
+  const sibling = join2(dirname2(self), "daemon.js");
+  if (existsSync2(sibling)) return sibling;
   const req = createRequire(import.meta.url);
   return req.resolve("@lens/daemon");
 }
+var install = defineCommand({
+  meta: {
+    description: "Install LENS daemon as a launchd service (auto-start at login, auto-restart on crash/sleep)."
+  },
+  async run() {
+    if (!isMacOS()) {
+      console.error("`lens daemon install` is macOS-only. On Linux use systemd; on Windows use Task Scheduler.");
+      process.exit(1);
+    }
+    const daemonEntry = findDaemonEntry();
+    const nodePath = process.execPath;
+    const plist = renderPlist({ nodePath, daemonEntry });
+    const existing = readPlist();
+    if (existing === plist) {
+      console.log(`Plist already up to date: ${PLIST_PATH}`);
+    } else {
+      writePlist(plist);
+      console.log(`Wrote plist: ${PLIST_PATH}`);
+    }
+    bootstrapPlist();
+    const status2 = plistStatus();
+    if (status2.loaded) {
+      console.log(`Loaded (PID ${status2.pid ?? "starting"}).`);
+      console.log("RunAtLoad=true, KeepAlive=true.");
+      console.log("Daemon will survive logout, reboot, sleep/wake, and crashes.");
+    } else {
+      console.error("launchctl bootstrap reported success but service not visible. Check Console.app for errors.");
+      process.exit(1);
+    }
+  }
+});
+var uninstall = defineCommand({
+  meta: { description: "Remove the LENS launchd service." },
+  async run() {
+    if (!isMacOS()) {
+      console.error("`lens daemon uninstall` is macOS-only.");
+      process.exit(1);
+    }
+    unloadPlist();
+    console.log("LENS daemon launchd service removed.");
+  }
+});
+var statusCmd = defineCommand({
+  meta: { description: "Show LENS daemon launchd status." },
+  async run() {
+    if (!isMacOS()) {
+      console.error("`lens daemon status` shows launchd state (macOS-only).");
+      process.exit(1);
+    }
+    const s = plistStatus();
+    console.log(`plist:    ${s.installed ? PLIST_PATH : "(not installed)"}`);
+    console.log(`loaded:   ${s.loaded}`);
+    console.log(`pid:      ${s.pid ?? "(none)"}`);
+    console.log(`lastExit: ${s.lastExit ?? "(none)"}`);
+    if (s.installed && !s.loaded) {
+      console.log("\nHint: run `lens daemon install` to load it.");
+    }
+  }
+});
+
+// packages/cli/src/commands/daemon.ts
+var DATA_DIR = join3(homedir2(), ".lens");
+var PID_FILE = join3(DATA_DIR, "daemon.pid");
+var LOG_FILE = join3(DATA_DIR, "daemon.log");
+function findDaemonEntry2() {
+  const self = fileURLToPath2(import.meta.url);
+  const sibling = join3(dirname3(self), "daemon.js");
+  if (existsSync3(sibling)) return sibling;
+  const req = createRequire2(import.meta.url);
+  return req.resolve("@lens/daemon");
+}
 function isRunning(pid) {
   try {
     process.kill(pid, 0);
@@ -393,7 +583,7 @@ function isRunning(pid) {
 }
 function readPid() {
   try {
-    const pid = parseInt(readFileSync(PID_FILE, "utf-8").trim(), 10);
+    const pid = parseInt(readFileSync2(PID_FILE, "utf-8").trim(), 10);
     return Number.isNaN(pid) ? null : isRunning(pid) ? pid : null;
   } catch {
     return null;
@@ -414,17 +604,17 @@ var start = defineCommand({
       console.log(`Daemon already running (PID ${existing})`);
       return;
     }
-    mkdirSync(DATA_DIR, { recursive: true });
-    const daemonEntry = findDaemonEntry();
+    mkdirSync2(DATA_DIR, { recursive: true });
+    const daemonEntry = findDaemonEntry2();
     if (args.foreground) {
       const child2 = spawn("node", [daemonEntry], {
         stdio: "inherit",
         env: { ...process.env, LENS_DATA_DIR: DATA_DIR }
       });
-      writeFileSync(PID_FILE, String(child2.pid));
+      writeFileSync2(PID_FILE, String(child2.pid));
       child2.on("exit", (code) => {
         try {
-          unlinkSync(PID_FILE);
+          unlinkSync2(PID_FILE);
         } catch {
         }
         process.exit(code ?? 1);
@@ -438,7 +628,7 @@ var start = defineCommand({
       env: { ...process.env, LENS_DATA_DIR: DATA_DIR }
     });
     child.unref();
-    writeFileSync(PID_FILE, String(child.pid));
+    writeFileSync2(PID_FILE, String(child.pid));
     const ok = await waitForHealth(1500);
     if (ok) {
       console.log(`Daemon started (PID ${child.pid})`);
@@ -473,7 +663,7 @@ var stop = defineCommand({
     }
     process.kill(pid, "SIGTERM");
     try {
-      unlinkSync(PID_FILE);
+      unlinkSync2(PID_FILE);
     } catch {
     }
     console.log(`Daemon stopped (PID ${pid})`);
@@ -481,7 +671,7 @@ var stop = defineCommand({
 });
 var daemon = defineCommand({
   meta: { description: "Manage the LENS daemon process." },
-  subCommands: { start, stop }
+  subCommands: { start, stop, install, uninstall, status: statusCmd }
 });
 
 // packages/cli/src/commands/dashboard.ts
@@ -727,12 +917,66 @@ var list = defineCommand({
   }
 });
 
+// packages/cli/src/commands/pattern.ts
+var pattern = defineCommand({
+  meta: {
+    description: "Structural AST search via ast-grep pattern."
+  },
+  args: {
+    pattern: {
+      type: "positional",
+      required: true,
+      description: 'ast-grep pattern, e.g. "function $N($$$) { $$$ }"'
+    },
+    lang: {
+      type: "string",
+      alias: "l",
+      default: "typescript",
+      description: "Source language: typescript | tsx | javascript | csharp"
+    },
+    repo: {
+      type: "string",
+      alias: "r",
+      description: "Repo root path (defaults to cwd)"
+    },
+    limit: {
+      type: "string",
+      default: "50",
+      description: "Max matches"
+    },
+    json: {
+      type: "boolean",
+      description: "Output full JSON instead of compact text"
+    }
+  },
+  async run({ args }) {
+    const repoPath = args.repo ?? process.cwd();
+    const format = args.json ? "json" : "text";
+    const res = await daemonFetch("/pattern", {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({
+        repoPath,
+        pattern: args.pattern,
+        language: args.lang,
+        limit: parseInt(args.limit, 10),
+        format
+      })
+    });
+    if (format === "json") {
+      console.log(JSON.stringify(await res.json(), null, 2));
+    } else {
+      process.stdout.write(await res.text());
+    }
+  }
+});
+
 // packages/cli/src/commands/register.ts
 import { resolve } from "path";
 
 // packages/cli/src/lib/inject-instructions.ts
-import { existsSync as existsSync2, mkdirSync as mkdirSync2, readFileSync as readFileSync2, writeFileSync as writeFileSync2 } from "fs";
-import { dirname as dirname2, join as join2 } from "path";
+import { existsSync as existsSync4, mkdirSync as mkdirSync3, readFileSync as readFileSync3, writeFileSync as writeFileSync3 } from "fs";
+import { dirname as dirname4, join as join4 } from "path";
 var MARKER = "<!-- lens-context -->";
 var CANDIDATE_FILES = [
   "CLAUDE.md",
@@ -750,6 +994,7 @@ This repo is indexed by LENS. **Prefer LENS MCP tools over built-in Grep/Glob**
 - \`lens_grep\` \u2014 Ranked search with import graph, co-change, hub scores. One call replaces multiple Grep+Read cycles.
 - \`lens_graph\` \u2014 Dependency map. Architecture, module relationships, change impact.
 - \`lens_graph_neighbors\` \u2014 Blast radius for a file. Importers, co-change partners, what breaks.
+- \`lens_pattern\` \u2014 Structural AST search via ast-grep. Pass a pattern with \`$NAME\` (single capture) or \`$$$\` (any sequence). Language: typescript | tsx | javascript | csharp. Use AFTER \`lens_grep\` when you know the exact AST shape. Example: \`class $C : IController { $$$ }\`.
 
 Load tools via \`ToolSearch\` before first use.
 `;
@@ -757,14 +1002,14 @@ function injectInstructions(repoRoot) {
   const results = [];
   const found = [];
   for (const file of CANDIDATE_FILES) {
-    const fullPath = join2(repoRoot, file);
-    if (existsSync2(fullPath)) {
+    const fullPath = join4(repoRoot, file);
+    if (existsSync4(fullPath)) {
       found.push(file);
-      const content = readFileSync2(fullPath, "utf-8");
+      const content = readFileSync3(fullPath, "utf-8");
       if (content.includes(MARKER)) {
         results.push({ file, action: "skipped" });
       } else {
-        writeFileSync2(fullPath, `${TEMPLATE}
+        writeFileSync3(fullPath, `${TEMPLATE}
 ${content}`);
         results.push({ file, action: "injected" });
       }
@@ -772,10 +1017,10 @@ ${content}`);
   }
   if (found.length === 0) {
     for (const file of ["CLAUDE.md", "AGENTS.md"]) {
-      const fullPath = join2(repoRoot, file);
-      const dir = dirname2(fullPath);
-      if (!existsSync2(dir)) mkdirSync2(dir, { recursive: true });
-      writeFileSync2(fullPath, `${TEMPLATE}
+      const fullPath = join4(repoRoot, file);
+      const dir = dirname4(fullPath);
+      if (!existsSync4(dir)) mkdirSync3(dir, { recursive: true });
+      writeFileSync3(fullPath, `${TEMPLATE}
 `);
       results.push({ file, action: "created" });
     }
@@ -784,24 +1029,24 @@ ${content}`);
 }
 
 // packages/cli/src/lib/inject-mcp.ts
-import { existsSync as existsSync3, readFileSync as readFileSync3, writeFileSync as writeFileSync3 } from "fs";
-import { join as join3 } from "path";
+import { existsSync as existsSync5, readFileSync as readFileSync4, writeFileSync as writeFileSync4 } from "fs";
+import { join as join5 } from "path";
 var LENS_MCP_ENTRY = {
   type: "http",
   url: "http://localhost:4111/mcp"
 };
 function injectMcp(repoRoot) {
-  const mcpPath = join3(repoRoot, ".mcp.json");
-  if (!existsSync3(mcpPath)) {
-    writeFileSync3(mcpPath, `${JSON.stringify({ mcpServers: { lens: LENS_MCP_ENTRY } }, null, 2)}
+  const mcpPath = join5(repoRoot, ".mcp.json");
+  if (!existsSync5(mcpPath)) {
+    writeFileSync4(mcpPath, `${JSON.stringify({ mcpServers: { lens: LENS_MCP_ENTRY } }, null, 2)}
 `);
     return "created";
   }
-  const raw = readFileSync3(mcpPath, "utf-8");
+  const raw = readFileSync4(mcpPath, "utf-8");
   const config = JSON.parse(raw);
   if (config.mcpServers?.lens) return "exists";
   config.mcpServers = { ...config.mcpServers, lens: LENS_MCP_ENTRY };
-  writeFileSync3(mcpPath, `${JSON.stringify(config, null, 2)}
+  writeFileSync4(mcpPath, `${JSON.stringify(config, null, 2)}
 `);
   return "updated";
 }
@@ -835,11 +1080,24 @@ var register = defineCommand({
   },
   async run({ args }) {
     const absPath = resolve(args.path);
-    const res = await daemonFetch("/repos", {
-      method: "POST",
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify({ path: absPath, name: args.name })
-    });
+    let res;
+    try {
+      res = await daemonFetch("/repos", {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ path: absPath, name: args.name })
+      });
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      if (msg.includes("Not a git repository")) {
+        console.error(`
+  \u2717 ${absPath} is not a git repository.
+    LENS requires a git repo to index.
+`);
+        process.exit(1);
+      }
+      throw err;
+    }
     const repo = await res.json();
     console.log();
     console.log(`  ${bold2("\u26A1 LENS")} ${dim(repo.name)}`);
@@ -941,7 +1199,8 @@ var main = defineCommand({
     remove,
     list,
     grep,
-    graph
+    graph,
+    pattern
   }
 });
 runMain(main);