token-goat 2.9.29 → 2.9.31

package/dist/{token-goat-chunk-435UQTKD.mjs → token-goat-chunk-IK2DRMFX.mjs}

@@ -1,18 +1,22 @@
 import { createRequire as __cjsRequire } from 'node:module';
 const require = __cjsRequire(import.meta.url);
 import {
-  getDb
-} from "./token-goat-chunk-7CPMDFGB.mjs";
+  filePathSpellingsClause,
+  fileRowByIndexKey,
+  getReadDb,
+  pathEqClause,
+  pathSuffixClause,
+  projectScopeClause
+} from "./token-goat-chunk-GVB2MUST.mjs";
 import {
-  foldPath,
-  isCaseInsensitiveFs
-} from "./token-goat-chunk-TCQANXYC.mjs";
+  foldPath
+} from "./token-goat-chunk-AM5V6GSX.mjs";
 import {
   normalizePath
-} from "./token-goat-chunk-NMTKNYGF.mjs";
+} from "./token-goat-chunk-FNFBKLRD.mjs";
 import {
   globalDbPath
-} from "./token-goat-chunk-MELOKFDJ.mjs";
+} from "./token-goat-chunk-CO3UYX6G.mjs";
 import {
   init_define_import_meta_env
 } from "./token-goat-chunk-A37V4PBF.mjs";
@@ -20,35 +24,95 @@ import {
 // src/index_reader.ts
 init_define_import_meta_env();
 
-// src/sql_path.ts
+// src/nested_worktrees.ts
 init_define_import_meta_env();
-function pathEqClause(column) {
-  return isCaseInsensitiveFs() ? `TG_LOWER(${column}) = ?` : `${column} = ?`;
+import fs from "node:fs";
+import path from "node:path";
+var cache = /* @__PURE__ */ new Map();
+function commonGitDir(root) {
+  const dotGit = path.join(root, ".git");
+  let st;
+  try {
+    st = fs.statSync(dotGit);
+  } catch {
+    return null;
+  }
+  if (st.isDirectory()) return dotGit;
+  if (!st.isFile()) return null;
+  try {
+    const m = /^gitdir:(.*)$/m.exec(fs.readFileSync(dotGit, "utf8"));
+    const target = m?.[1]?.trim();
+    if (target === void 0 || target === "") return null;
+    const gitDir = path.resolve(root, target);
+    const common = fs.readFileSync(path.join(gitDir, "commondir"), "utf8").trim();
+    return path.resolve(gitDir, common);
+  } catch {
+    return null;
+  }
 }
-function pathSuffixClause(column) {
-  const col = isCaseInsensitiveFs() ? `TG_LOWER(${column})` : column;
-  return {
-    // Both separators are accepted because not every writer into `symbols` stores a normalizePath'd (forward-slash) file_path, and the JS-side path-boundary test this narrowing feeds treats `/` and `\` alike.
-    clause: `(${col} = ? OR substr(${col}, -length(?)) = ? OR substr(${col}, -length(?)) = ?)`,
-    params: (baseName) => {
-      const folded = foldPath(baseName);
-      return [folded, `/${folded}`, `/${folded}`, `\\${folded}`, `\\${folded}`];
-    }
-  };
+function strictlyInside(child, parent) {
+  const base = parent.endsWith("/") ? parent : `${parent}/`;
+  return child.startsWith(base) && child.length > base.length;
 }
-function projectScopeClause(column) {
-  const caseInsensitive = isCaseInsensitiveFs();
-  const col = caseInsensitive ? `TG_LOWER(${column})` : column;
-  return {
-    // Parenthesised so an OR anywhere in the surrounding WHERE cannot bind tighter than the pair.
-    clause: `(${col} >= ? AND ${col} < ?)`,
-    params: (root) => {
-      const folded = caseInsensitive ? foldPath(normalizePath(root)) : normalizePath(root);
-      const lower = folded.endsWith("/") ? folded : `${folded}/`;
-      const upper = `${lower.slice(0, -1)}0`;
-      return [lower, upper];
+function nestedWorktreeRoots(root) {
+  const common = commonGitDir(root);
+  if (common === null) return [];
+  const listDir = path.join(common, "worktrees");
+  let listMtimeMs;
+  try {
+    listMtimeMs = fs.statSync(listDir).mtimeMs;
+  } catch {
+    return [];
+  }
+  const hit = cache.get(root);
+  if (hit?.listMtimeMs === listMtimeMs) return hit.roots;
+  const rootPath = normalizePath(root);
+  const rootKey = foldPath(rootPath);
+  let realPath;
+  try {
+    realPath = normalizePath(fs.realpathSync.native(root));
+  } catch {
+    realPath = null;
+  }
+  const roots = [];
+  let names;
+  try {
+    names = fs.readdirSync(listDir);
+  } catch {
+    names = [];
+  }
+  for (const name of names) {
+    const entryDir = path.join(listDir, name);
+    let target;
+    try {
+      target = fs.readFileSync(path.join(entryDir, "gitdir"), "utf8").trim();
+    } catch {
+      continue;
     }
-  };
+    if (target === "") continue;
+    const wtRoot = normalizePath(path.dirname(path.resolve(entryDir, target)));
+    if (strictlyInside(foldPath(wtRoot), rootKey)) roots.push(wtRoot);
+    else if (realPath !== null && strictlyInside(foldPath(wtRoot), foldPath(realPath))) roots.push(rootPath.replace(/\/+$/, "") + wtRoot.slice(realPath.replace(/\/+$/, "").length));
+  }
+  roots.sort();
+  cache.set(root, { listMtimeMs, roots });
+  return roots;
+}
+function ownProjectScope(column, root) {
+  const scope = projectScopeClause(column);
+  const excluded = nestedWorktreeExclusions(column, root);
+  if (excluded.clauses.length === 0) return { clause: scope.clause, params: scope.params(root) };
+  return { clause: `(${[scope.clause, ...excluded.clauses].join(" AND ")})`, params: [...scope.params(root), ...excluded.params] };
+}
+function nestedWorktreeExclusions(column, root) {
+  const scope = projectScopeClause(column);
+  const clauses = [];
+  const params = [];
+  for (const wt of nestedWorktreeRoots(root)) {
+    clauses.push(`NOT ${scope.clause}`);
+    params.push(...scope.params(wt));
+  }
+  return { clauses, params };
 }
 
 // src/index_reader.ts
@@ -78,9 +142,9 @@ function toRefEntry(row) {
 }
 function applyRootDirScope(rootDir, column, where, params) {
   if (rootDir === void 0) return;
-  const { clause, params: bounds } = projectScopeClause(column);
-  where.push(clause);
-  params.push(...bounds(indexKey(rootDir)));
+  const scope = ownProjectScope(column, indexKey(rootDir));
+  where.push(scope.clause);
+  params.push(...scope.params);
 }
 function buildSymbolWhere(opts) {
   const where = [];
@@ -90,15 +154,9 @@ function buildSymbolWhere(opts) {
     params.push(opts.name);
   }
   if (opts.filePath !== void 0) {
-    const key = indexKey(opts.filePath);
-    if (key.includes("/") || key.includes("\\")) {
-      const alt = key.includes("/") ? key.replace(/\//g, "\\") : key.replace(/\\/g, "/");
-      where.push(`(${pathEqClause("file_path")} OR ${pathEqClause("file_path")})`);
-      params.push(foldPath(key), foldPath(alt));
-    } else {
-      where.push(pathEqClause("file_path"));
-      params.push(foldPath(key));
-    }
+    const spellings = filePathSpellingsClause("file_path", opts.filePath);
+    where.push(spellings.clause);
+    params.push(...spellings.params);
   }
   if (opts.kind !== void 0) {
     where.push("kind = ?");
@@ -122,7 +180,7 @@ function querySymbols(opts = {}, dbPath = globalDbPath()) {
   const limit = opts.limit ?? DEFAULT_QUERY_LIMIT;
   const offset = opts.offset ?? 0;
   const sql = `SELECT file_path, name, kind, line_start, line_end, body, docstring, parent FROM symbols ${clause} ORDER BY file_path, line_start, rowid LIMIT ? OFFSET ?`;
-  const db = getDb(dbPath);
+  const db = getReadDb(dbPath);
   const rows = db.prepare(sql).all(...params, limit, offset);
   return rows.map(toSymbolEntry);
 }
@@ -131,14 +189,14 @@ function distinctSymbolKinds(rootDir, dbPath = globalDbPath()) {
   const params = [];
   applyRootDirScope(rootDir, "file_path", where, params);
   const clause = where.length > 0 ? `WHERE ${where.join(" AND ")}` : "";
-  const db = getDb(dbPath);
+  const db = getReadDb(dbPath);
   const rows = db.prepare(`SELECT DISTINCT kind FROM symbols ${clause} ORDER BY kind`).all(...params);
   return rows.map((r) => r.kind);
 }
 function countSymbols(opts = {}, dbPath = globalDbPath()) {
   const { clause, params } = buildSymbolWhere(opts);
   const sql = `SELECT COUNT(*) as cnt FROM symbols ${clause}`;
-  const db = getDb(dbPath);
+  const db = getReadDb(dbPath);
   const row = db.prepare(sql).get(...params);
   return row.cnt;
 }
@@ -156,20 +214,20 @@ function queryRefs(opts, dbPath = globalDbPath()) {
   const { clause, params } = buildRefsWhere(opts);
   const limit = opts.limit ?? DEFAULT_QUERY_LIMIT;
   const sql = `SELECT file_path, name, line, col, context FROM refs ${clause} ORDER BY file_path, line LIMIT ?`;
-  const db = getDb(dbPath);
+  const db = getReadDb(dbPath);
   const rows = db.prepare(sql).all(...params, limit);
   return rows.map(toRefEntry);
 }
 function countRefs(opts, dbPath = globalDbPath()) {
   const { clause, params } = buildRefsWhere(opts);
   const sql = `SELECT COUNT(*) as cnt FROM refs ${clause}`;
-  const db = getDb(dbPath);
+  const db = getReadDb(dbPath);
   const row = db.prepare(sql).get(...params);
   return row.cnt;
 }
 function queryRefsByContext(context, filePath, dbPath = globalDbPath()) {
   const sql = `SELECT file_path, name, line, col, context FROM refs WHERE context = ? AND ${pathEqClause("file_path")} ORDER BY line LIMIT 20`;
-  const db = getDb(dbPath);
+  const db = getReadDb(dbPath);
   const rows = db.prepare(sql).all(context, foldPath(indexKey(filePath)));
   return rows.map(toRefEntry);
 }
@@ -177,7 +235,7 @@ var REF_COUNT_BATCH = 900;
 function queryRefCounts(names, dbPath = globalDbPath(), rootDir) {
   const counts = /* @__PURE__ */ new Map();
   if (names.length === 0) return counts;
-  const db = getDb(dbPath);
+  const db = getReadDb(dbPath);
   const scopeWhere = [];
   const scopeParams = [];
   applyRootDirScope(rootDir, "file_path", scopeWhere, scopeParams);
@@ -196,9 +254,15 @@ function queryRefCounts(names, dbPath = globalDbPath(), rootDir) {
   return counts;
 }
 function getProjectFileEntries(rootDir, dbPath = globalDbPath()) {
-  const db = getDb(dbPath);
   const { clause, params } = projectScopeClause("path");
-  const rows = db.prepare(`SELECT path, sha, mtime, language, indexed_at, embed_sha, parser_sha FROM files WHERE ${clause}`).all(...params(indexKey(rootDir)));
+  return fileEntriesWhere(clause, params(indexKey(rootDir)), dbPath);
+}
+function getOwnProjectFileEntries(rootDir, dbPath = globalDbPath()) {
+  const { clause, params } = ownProjectScope("path", indexKey(rootDir));
+  return fileEntriesWhere(clause, params, dbPath);
+}
+function fileEntriesWhere(clause, params, dbPath) {
+  const rows = getReadDb(dbPath).prepare(`SELECT path, sha, mtime, language, indexed_at, embed_sha, parser_sha FROM files WHERE ${clause}`).all(...params);
   const out = /* @__PURE__ */ new Map();
   for (const row of rows) {
     out.set(foldPath(row.path), {
@@ -214,17 +278,8 @@ function getProjectFileEntries(rootDir, dbPath = globalDbPath()) {
   return out;
 }
 function getFileEntry(filePath, dbPath = globalDbPath()) {
-  const db = getDb(dbPath);
-  const key = indexKey(filePath);
-  let row = db.prepare(
-    `SELECT path, sha, mtime, language, indexed_at, embed_sha, parser_sha FROM files WHERE ${pathEqClause("path")}`
-  ).get(foldPath(key));
-  if (row === void 0 && (key.includes("/") || key.includes("\\"))) {
-    const altPath = key.includes("/") ? key.replace(/\//g, "\\") : key.replace(/\\/g, "/");
-    row = db.prepare(
-      `SELECT path, sha, mtime, language, indexed_at, embed_sha, parser_sha FROM files WHERE ${pathEqClause("path")}`
-    ).get(foldPath(altPath));
-  }
+  const db = getReadDb(dbPath);
+  const row = fileRowByIndexKey(db, filePath, "path, sha, mtime, language, indexed_at, embed_sha, parser_sha");
   if (row === void 0) return null;
   return {
     filePath: row.path,
@@ -240,12 +295,10 @@ function sanitizeFtsQuery(query, join = "AND") {
   const terms = query.split(/\s+/).filter((t) => t.length > 0).map((t) => '"' + t.replace(/"/g, '""') + '"');
   return terms.join(join === "OR" ? " OR " : " ");
 }
-function runFtsQuery(db, match, limit, scope, rootDir) {
+function runFtsQuery(db, match, limit, scope) {
   const sql = `SELECT s.file_path, s.name, s.kind, s.line_start, s.line_end, s.body, s.docstring, s.parent FROM symbols_fts JOIN symbols s ON s.id = symbols_fts.rowid WHERE symbols_fts MATCH ?${scope !== void 0 ? ` AND ${scope.clause}` : ""} ORDER BY bm25(symbols_fts) LIMIT ?`;
   const params = [match];
-  if (scope !== void 0 && rootDir !== void 0) {
-    params.push(...scope.params(rootDir));
-  }
+  if (scope !== void 0) params.push(...scope.params);
   params.push(limit);
   const rows = db.prepare(sql).all(...params);
   return rows.map(toSymbolEntry);
@@ -253,22 +306,23 @@ function runFtsQuery(db, match, limit, scope, rootDir) {
 function searchSymbolsFts(query, limit = 50, dbPath = globalDbPath(), rootDir) {
   const andMatch = sanitizeFtsQuery(query, "AND");
   if (andMatch === "") return [];
-  const db = getDb(dbPath);
-  const scope = rootDir !== void 0 ? projectScopeClause("s.file_path") : void 0;
+  const db = getReadDb(dbPath);
+  const scope = rootDir !== void 0 ? ownProjectScope("s.file_path", indexKey(rootDir)) : void 0;
   try {
-    const andResults = runFtsQuery(db, andMatch, limit, scope, rootDir);
+    const andResults = runFtsQuery(db, andMatch, limit, scope);
     if (andResults.length > 0) return andResults;
     const orMatch = sanitizeFtsQuery(query, "OR");
     if (orMatch === andMatch) return andResults;
-    return runFtsQuery(db, orMatch, limit, scope, rootDir);
+    return runFtsQuery(db, orMatch, limit, scope);
   } catch {
     return [];
   }
 }
 
 export {
-  pathEqClause,
-  projectScopeClause,
+  commonGitDir,
+  ownProjectScope,
+  nestedWorktreeExclusions,
   DEFAULT_QUERY_LIMIT,
   querySymbols,
   distinctSymbolKinds,
@@ -278,6 +332,7 @@ export {
   queryRefsByContext,
   queryRefCounts,
   getProjectFileEntries,
+  getOwnProjectFileEntries,
   getFileEntry,
   sanitizeFtsQuery,
   searchSymbolsFts