@wrongstack/tools 0.7.7 → 0.8.0

package/dist/codebase-index/index.js

@@ -1,11 +1,12 @@
 import * as fs2 from 'node:fs/promises';
-import * as path3 from 'node:path';
+import * as path4 from 'node:path';
 import { compileGlob } from '@wrongstack/core';
+import { createRequire } from 'node:module';
 import * as fs from 'node:fs';
-import { mkdirSync, writeFileSync, unlinkSync } from 'node:fs';
-import { DatabaseSync } from 'node:sqlite';
+import { mkdirSync, writeFileSync } from 'node:fs';
 import * as ts from 'typescript';
 import { execSync, spawnSync } from 'node:child_process';
+import * as os from 'node:os';
 
 var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require : typeof Proxy !== "undefined" ? new Proxy(x, {
   get: (a, b) => (typeof require !== "undefined" ? require : a)[b]
@@ -82,12 +83,39 @@ function internalKindToLspKind(k) {
 // src/codebase-index/writer.ts
 var INDEX_DIR = ".codebase-index";
 var DB_FILE = "index.db";
+var warningSilenced = false;
+function silenceSqliteExperimentalWarning() {
+  if (warningSilenced) return;
+  warningSilenced = true;
+  const original = process.emitWarning.bind(process);
+  process.emitWarning = ((warning, ...rest) => {
+    const msg = typeof warning === "string" ? warning : warning?.message ?? "";
+    const name = typeof warning === "string" ? String(rest[0] ?? "") : warning?.name ?? "";
+    if (/sqlite/i.test(msg) && /experimental/i.test(`${name} ${msg}`)) return;
+    original(warning, ...rest);
+  });
+}
+var DatabaseSyncCtor;
+function loadDatabaseSync() {
+  if (DatabaseSyncCtor) return DatabaseSyncCtor;
+  silenceSqliteExperimentalWarning();
+  try {
+    const req = createRequire(import.meta.url);
+    DatabaseSyncCtor = req("node:sqlite").DatabaseSync;
+  } catch (err) {
+    throw new Error(
+      `The codebase index needs Node's built-in SQLite (node:sqlite), available since Node 22.5. This runtime doesn't provide it: ${err instanceof Error ? err.message : String(err)}`
+    );
+  }
+  return DatabaseSyncCtor;
+}
 var IndexStore = class {
   constructor(projectRoot) {
     this.projectRoot = projectRoot;
-    const dir = path3.join(projectRoot, INDEX_DIR);
+    const dir = path4.join(projectRoot, INDEX_DIR);
     fs.mkdirSync(dir, { recursive: true });
-    this.db = new DatabaseSync(path3.join(dir, DB_FILE));
+    const Database = loadDatabaseSync();
+    this.db = new Database(path4.join(dir, DB_FILE));
     this.initSchema();
   }
   projectRoot;
@@ -282,7 +310,7 @@ var IndexStore = class {
       totalFiles,
       byLang,
       byKind,
-      indexPath: path3.join(this.projectRoot, INDEX_DIR),
+      indexPath: path4.join(this.projectRoot, INDEX_DIR),
       lastIndexed,
       sizeBytes,
       version: SCHEMA_VERSION
@@ -375,7 +403,7 @@ var IndexStore = class {
     }));
   }
   sizeBytes() {
-    const dbPath = path3.join(this.projectRoot, INDEX_DIR, DB_FILE);
+    const dbPath = path4.join(this.projectRoot, INDEX_DIR, DB_FILE);
     try {
       return fs.statSync(dbPath).size;
     } catch {
@@ -590,6 +618,7 @@ import (
 	"go/ast"
 	"go/parser"
 	"go/token"
+	"io"
 	"os"
 	"strings"
 )
@@ -604,17 +633,13 @@ type Sym struct {
 }
 
 func main() {
-	if len(os.Args) < 2 {
-		fmt.Print("[]")
-		return
-	}
-	src, err := os.ReadFile(os.Args[1])
+	src, err := io.ReadAll(os.Stdin)
 	if err != nil {
 		fmt.Print("[]")
 		return
 	}
 	fset := token.NewFileSet()
-	node, err := parser.ParseFile(fset, os.Args[1], src, 0)
+	node, err := parser.ParseFile(fset, "src.go", src, 0)
 	if err != nil {
 		fmt.Print("[]")
 		return
@@ -758,7 +783,7 @@ func formatMethods(fields []*ast.Field) string {
 	return formatFields(fields)
 }
 
-func formatTypeParams(tp *ast.TypeParams) string {
+func formatTypeParams(tp *ast.FieldList) string {
 	if tp == nil || len(tp.List) == 0 {
 		return ""
 	}
@@ -801,30 +826,33 @@ func formatType(t ast.Expr) string {
 		return "chan " + formatType(v.Value)
 	case *ast.BasicLit:
 		return v.Value
+	case *ast.IndexExpr:
+		// Generic instantiation with one type arg, e.g. Logger[int].
+		return formatType(v.X) + "[" + formatType(v.Index) + "]"
+	case *ast.IndexListExpr:
+		// Generic instantiation with multiple type args, e.g. Map[K, V].
+		args := make([]string, len(v.Indices))
+		for i, idx := range v.Indices {
+			args[i] = formatType(idx)
+		}
+		return formatType(v.X) + "[" + strings.Join(args, ", ") + "]"
 	default:
 		return "?"
 	}
 }
 `;
-function syncGoParse(filePath, _content, lang) {
-  const tmpDir = path3.join(process.env.TEMP ?? "/tmp", "ws-go-parse");
+function syncGoParse(filePath, content, lang) {
+  const tmpDir = path4.join(os.tmpdir(), "ws-go-parse");
   try {
     mkdirSync(tmpDir, { recursive: true });
-    const scriptPath = path3.join(tmpDir, "parse.go");
+    const scriptPath = path4.join(tmpDir, "parse.go");
     writeFileSync(scriptPath, GO_PARSE_SCRIPT, "utf8");
-    let stdout;
-    try {
-      stdout = execSync(`go run "${scriptPath}" "${filePath}"`, {
-        timeout: 15e3,
-        encoding: "utf8",
-        windowsHide: true
-      });
-    } finally {
-      try {
-        unlinkSync(scriptPath);
-      } catch {
-      }
-    }
+    const stdout = execSync(`go run "${scriptPath}"`, {
+      input: content,
+      timeout: 15e3,
+      encoding: "utf8",
+      windowsHide: true
+    });
     if (!stdout.trim()) {
       return { file: filePath, lang, symbols: [], mtimeMs: Date.now() };
     }
@@ -1061,7 +1089,11 @@ print(json.dumps([s.to_dict() for s in syms]))
 `;
 function syncPyParse(filePath, lang) {
   try {
-    const stdout = execSync(`python -c "${PY_PARSE_SCRIPT.replace(/"/g, '\\"')}" "${filePath}"`, {
+    const tmpDir = path4.join(os.tmpdir(), "ws-py-parse");
+    mkdirSync(tmpDir, { recursive: true });
+    const scriptPath = path4.join(tmpDir, "parse.py");
+    writeFileSync(scriptPath, PY_PARSE_SCRIPT, "utf8");
+    const stdout = execSync(`python "${scriptPath}" "${filePath}"`, {
       timeout: 15e3,
       encoding: "utf8",
       windowsHide: true
@@ -1100,9 +1132,12 @@ function parseSymbols4(opts) {
 function checkNativeParser() {
   try {
     execSync("rustc --version", { stdio: "pipe" });
-    const toolsDir = path3.join(process.cwd(), "tools");
+    const toolsDir = path4.join(process.cwd(), "tools");
     try {
-      execSync("cargo metadata --no-deps --format-version 1 --manifest-path " + path3.join(toolsDir, "Cargo.toml"), { stdio: "pipe" });
+      execSync(
+        "cargo metadata --no-deps --format-version 1 --manifest-path " + path4.join(toolsDir, "Cargo.toml"),
+        { stdio: "pipe" }
+      );
       return true;
     } catch {
       return false;
@@ -1113,17 +1148,21 @@ function checkNativeParser() {
 }
 function tryNativeParse(file, content) {
   try {
-    const toolsDir = path3.join(process.cwd(), "tools");
-    const crateDir = path3.join(toolsDir, "syn-parser");
-    const tmpFile = path3.join(crateDir, "src", "input.rs");
-    const { writeFileSync: writeFileSync2 } = __require("node:fs");
-    writeFileSync2(tmpFile, content, "utf8");
-    const result = spawnSync("cargo", ["run", "--manifest-path", path3.join(toolsDir, "Cargo.toml")], {
-      cwd: process.cwd(),
-      encoding: "utf8",
-      timeout: 15e3,
-      stdio: ["pipe", "pipe", "pipe"]
-    });
+    const toolsDir = path4.join(process.cwd(), "tools");
+    const crateDir = path4.join(toolsDir, "syn-parser");
+    const tmpFile = path4.join(crateDir, "src", "input.rs");
+    const { writeFileSync: writeFileSync3 } = __require("node:fs");
+    writeFileSync3(tmpFile, content, "utf8");
+    const result = spawnSync(
+      "cargo",
+      ["run", "--manifest-path", path4.join(toolsDir, "Cargo.toml")],
+      {
+        cwd: process.cwd(),
+        encoding: "utf8",
+        timeout: 15e3,
+        stdio: ["pipe", "pipe", "pipe"]
+      }
+    );
     if (result.status === 0 && result.stdout) {
       const symbols = JSON.parse(result.stdout);
       return {
@@ -1157,7 +1196,8 @@ function regexParse(opts) {
     lineOffsets.push(lineOffsets[i] + lines[i].length + 1);
   }
   function lineFromOffset(offset) {
-    let lo = 0, hi = lineOffsets.length - 1;
+    let lo = 0;
+    let hi = lineOffsets.length - 1;
     while (lo < hi) {
       const mid = lo + hi + 1 >>> 1;
       if (lineOffsets[mid] <= offset) lo = mid;
@@ -1171,8 +1211,7 @@ function regexParse(opts) {
   }
   for (const pattern of RS_PATTERNS) {
     pattern.regex.lastIndex = 0;
-    let match;
-    while ((match = pattern.regex.exec(content)) !== null) {
+    for (let match = pattern.regex.exec(content); match !== null; match = pattern.regex.exec(content)) {
       const name = match[1];
       const offset = match.index;
       const line = lineFromOffset(offset);
@@ -1214,7 +1253,7 @@ function parseSymbols5(opts) {
 function regexParse2(opts) {
   const { file, content, lang } = opts;
   const symbols = [];
-  const basename2 = path3.basename(file).toLowerCase();
+  const basename2 = path4.basename(file).toLowerCase();
   const isPackageJson = basename2 === "package.json";
   const isTsconfig = basename2 === "tsconfig.json" || basename2 === "tsconfig.build.json";
   const isJsonSchema = content.includes("$schema") || content.includes("$id") || content.includes("$ref");
@@ -1225,7 +1264,8 @@ function regexParse2(opts) {
     lineOffsets.push(lineOffsets[i] + lines[i].length + 1);
   }
   function lineFromOffset(offset) {
-    let lo = 0, hi = lineOffsets.length - 1;
+    let lo = 0;
+    let hi = lineOffsets.length - 1;
     while (lo < hi) {
       const mid = lo + hi + 1 >>> 1;
       if (lineOffsets[mid] <= offset) lo = mid;
@@ -1237,19 +1277,20 @@ function regexParse2(opts) {
   if (rootMatch) {
     const offset = rootMatch.index;
     const line = lineFromOffset(offset);
-    symbols.push(makeSymbol({
-      name: path3.basename(file),
-      kind: "object",
-      line,
-      col: 0,
-      signature: `"${path3.basename(file)}" = { ... }`,
-      file,
-      lang
-    }));
+    symbols.push(
+      makeSymbol({
+        name: path4.basename(file),
+        kind: "object",
+        line,
+        col: 0,
+        signature: `"${path4.basename(file)}" = { ... }`,
+        file,
+        lang
+      })
+    );
   }
   const topLevelKeyRegex = /^\s*"([^"]+)"\s*:/gm;
-  let match;
-  while ((match = topLevelKeyRegex.exec(content)) !== null) {
+  for (let match = topLevelKeyRegex.exec(content); match !== null; match = topLevelKeyRegex.exec(content)) {
     const key = match[1];
     const offset = match.index;
     const line = lineFromOffset(offset);
@@ -1276,15 +1317,17 @@ function regexParse2(opts) {
         signature = `"$ref": "..."`;
       }
     }
-    symbols.push(makeSymbol({
-      name: key,
-      kind,
-      line,
-      col,
-      signature,
-      file,
-      lang
-    }));
+    symbols.push(
+      makeSymbol({
+        name: key,
+        kind,
+        line,
+        col,
+        signature,
+        file,
+        lang
+      })
+    );
     if (isPackageJson && key === "scripts") {
       extractPackageScripts(content, symbols, file, lang, lineOffsets, lineFromOffset);
     }
@@ -1293,20 +1336,21 @@ function regexParse2(opts) {
     }
   }
   const defsRegex = /"\$defs"\s*:|"\$defs"\s*:/g;
-  let defsMatch;
-  while ((defsMatch = defsRegex.exec(content)) !== null) {
+  const defsMatch = defsRegex.exec(content);
+  if (defsMatch !== null) {
     const offset = defsMatch.index;
     const line = lineFromOffset(offset);
-    symbols.push(makeSymbol({
-      name: "$defs",
-      kind: "property",
-      line,
-      col: offset - (lineOffsets[line - 1] ?? 0),
-      signature: '"$defs": { ... }',
-      file,
-      lang
-    }));
-    break;
+    symbols.push(
+      makeSymbol({
+        name: "$defs",
+        kind: "property",
+        line,
+        col: offset - (lineOffsets[line - 1] ?? 0),
+        signature: '"$defs": { ... }',
+        file,
+        lang
+      })
+    );
   }
   const defsPatterns = [
     /"\$defs"\s*:/g,
@@ -1316,69 +1360,71 @@ function regexParse2(opts) {
   ];
   for (const pat of defsPatterns) {
     pat.lastIndex = 0;
-    while ((match = pat.exec(content)) !== null) {
+    for (let match = pat.exec(content); match !== null; match = pat.exec(content)) {
       const offset = match.index;
       const line = lineFromOffset(offset);
       const key = match[0].match(/"([^"]+)"/)?.[1] ?? match[0];
-      symbols.push(makeSymbol({
-        name: key,
-        kind: "property",
-        line,
-        col: offset - (lineOffsets[line - 1] ?? 0),
-        signature: `"${key}": { ... }`,
-        file,
-        lang
-      }));
+      symbols.push(
+        makeSymbol({
+          name: key,
+          kind: "property",
+          line,
+          col: offset - (lineOffsets[line - 1] ?? 0),
+          signature: `"${key}": { ... }`,
+          file,
+          lang
+        })
+      );
     }
   }
   return { file, lang, symbols, mtimeMs: Date.now() };
 }
 function extractPackageScripts(content, symbols, file, lang, lineOffsets, lineFromOffset) {
   const scriptsBlockRegex = /"scripts"\s*:\s*\{([^}]+)\}/g;
-  let match;
-  while ((match = scriptsBlockRegex.exec(content)) !== null) {
+  for (let match = scriptsBlockRegex.exec(content); match !== null; match = scriptsBlockRegex.exec(content)) {
     const blockContent = match[0];
     const blockOffset = match.index;
     const scriptKeyRegex = /"(\w[\w-]*)"\s*:/g;
-    let scriptMatch;
-    while ((scriptMatch = scriptKeyRegex.exec(blockContent)) !== null) {
+    for (let scriptMatch = scriptKeyRegex.exec(blockContent); scriptMatch !== null; scriptMatch = scriptKeyRegex.exec(blockContent)) {
       const key = scriptMatch[1];
       const keyOffset = blockOffset + scriptMatch.index;
       const line = lineFromOffset(keyOffset);
-      symbols.push(makeSymbol({
-        name: key,
-        kind: "function",
-        line,
-        col: keyOffset - (lineOffsets[line - 1] ?? 0),
-        signature: `"${key}": "..."`,
-        file,
-        lang
-      }));
+      symbols.push(
+        makeSymbol({
+          name: key,
+          kind: "function",
+          line,
+          col: keyOffset - (lineOffsets[line - 1] ?? 0),
+          signature: `"${key}": "..."`,
+          file,
+          lang
+        })
+      );
     }
   }
 }
 function extractCompilerOptions(content, symbols, file, lang, lineOffsets, parentLine, lineFromOffset) {
   const optsBlockRegex = /"compilerOptions"\s*:\s*\{([^}]+)\}/g;
-  let match;
-  while ((match = optsBlockRegex.exec(content)) !== null) {
+  for (let match = optsBlockRegex.exec(content); match !== null; match = optsBlockRegex.exec(content)) {
     const blockContent = match[0];
     const blockOffset = match.index;
     const optKeyRegex = /"(\w[\w]*)"\s*:/g;
-    let optMatch;
-    while ((optMatch = optKeyRegex.exec(blockContent)) !== null) {
+    for (let optMatch = optKeyRegex.exec(blockContent); optMatch !== null; optMatch = optKeyRegex.exec(blockContent)) {
       const key = optMatch[1];
       const keyOffset = blockOffset + optMatch.index;
       const line = lineFromOffset(keyOffset);
       if (line <= parentLine) continue;
-      symbols.push(makeSymbol({
-        name: key,
-        kind: "property",
-        line,
-        col: keyOffset - (lineOffsets[line - 1] ?? 0),
-        signature: `"${key}": ...`,
-        file,
-        lang
-      }));
+      symbols.push(
+        makeSymbol({
+          name: key,
+          kind: "property",
+          line,
+          col: keyOffset - (lineOffsets[line - 1] ?? 0),
+          signature: `"${key}": ...`,
+          file,
+          lang
+        })
+      );
     }
   }
 }
@@ -1416,7 +1462,8 @@ function regexParse3(opts) {
     lineOffsets.push(lineOffsets[i] + lines[i].length + 1);
   }
   function lineFromOffset(offset) {
-    let lo = 0, hi = lineOffsets.length - 1;
+    let lo = 0;
+    let hi = lineOffsets.length - 1;
     while (lo < hi) {
       const mid = lo + hi + 1 >>> 1;
       if (lineOffsets[mid] <= offset) lo = mid;
@@ -1425,40 +1472,43 @@ function regexParse3(opts) {
     return lo + 1;
   }
   const anchorRegex = /&(\w[\w-]*)/g;
-  let match;
-  while ((match = anchorRegex.exec(content)) !== null) {
+  for (let match = anchorRegex.exec(content); match !== null; match = anchorRegex.exec(content)) {
     const name = match[1];
     const offset = match.index;
     const line = lineFromOffset(offset);
     const col = offset - (lineOffsets[line - 1] ?? 0);
-    symbols.push(makeSymbol2({
-      name,
-      kind: "const",
-      line,
-      col,
-      signature: `&${name}`,
-      file,
-      lang
-    }));
+    symbols.push(
+      makeSymbol2({
+        name,
+        kind: "const",
+        line,
+        col,
+        signature: `&${name}`,
+        file,
+        lang
+      })
+    );
   }
   const aliasRegex = /\*(\w[\w-]*)/g;
-  while ((match = aliasRegex.exec(content)) !== null) {
+  for (let match = aliasRegex.exec(content); match !== null; match = aliasRegex.exec(content)) {
     const name = match[1];
     const offset = match.index;
     const line = lineFromOffset(offset);
     const col = offset - (lineOffsets[line - 1] ?? 0);
-    symbols.push(makeSymbol2({
-      name,
-      kind: "const",
-      line,
-      col,
-      signature: `*${name}`,
-      file,
-      lang
-    }));
+    symbols.push(
+      makeSymbol2({
+        name,
+        kind: "const",
+        line,
+        col,
+        signature: `*${name}`,
+        file,
+        lang
+      })
+    );
   }
   const kvRegex = /^(\s*)([^:#\s][^:#\s]*)\s*:/gm;
-  while ((match = kvRegex.exec(content)) !== null) {
+  for (let match = kvRegex.exec(content); match !== null; match = kvRegex.exec(content)) {
     const indent = match[1].length;
     const key = match[2];
     const offset = match.index;
@@ -1474,38 +1524,42 @@ function regexParse3(opts) {
     symbols.push(makeSymbol2({ name: key, kind, line, col, signature, file, lang }));
   }
   const listItemRegex = /^-(\s+)([^:#\s][^:#\s]*)\s*:/gm;
-  while ((match = listItemRegex.exec(content)) !== null) {
+  for (let match = listItemRegex.exec(content); match !== null; match = listItemRegex.exec(content)) {
     const key = match[2];
     const offset = match.index;
     const line = lineFromOffset(offset);
     const col = offset - (lineOffsets[line - 1] ?? 0);
     const value = extractValue(content, offset + match[0].length);
     const kind = isScalar(value) ? "literal" : "property";
-    symbols.push(makeSymbol2({
-      name: key,
-      kind,
-      line,
-      col,
-      signature: `- ${key}: ${truncate(value, 60)}`,
-      file,
-      lang
-    }));
+    symbols.push(
+      makeSymbol2({
+        name: key,
+        kind,
+        line,
+        col,
+        signature: `- ${key}: ${truncate(value, 60)}`,
+        file,
+        lang
+      })
+    );
   }
   const blockScalarRegex = /^(\s*)([^:#\s][^:#\s]*)\s*:\s*[|>](\s|$)/gm;
-  while ((match = blockScalarRegex.exec(content)) !== null) {
+  for (let match = blockScalarRegex.exec(content); match !== null; match = blockScalarRegex.exec(content)) {
     const key = match[2];
     const offset = match.index;
     const line = lineFromOffset(offset);
     const col = offset - (lineOffsets[line - 1] ?? 0);
-    symbols.push(makeSymbol2({
-      name: key,
-      kind: "property",
-      line,
-      col,
-      signature: `${key}: | ...`,
-      file,
-      lang
-    }));
+    symbols.push(
+      makeSymbol2({
+        name: key,
+        kind: "property",
+        line,
+        col,
+        signature: `${key}: | ...`,
+        file,
+        lang
+      })
+    );
   }
   return { file, lang, symbols, mtimeMs: Date.now() };
 }
@@ -1577,12 +1631,12 @@ async function findSourceFiles(projectRoot, ignore) {
     }
     for (const e of entries) {
       if (ignoreSet.has(e.name)) continue;
-      const full = path3.join(dir, e.name);
+      const full = path4.join(dir, e.name);
       if (e.isDirectory()) {
         await walk(full);
       } else if (e.isFile()) {
-        const rel = path3.relative(projectRoot, full).replace(/\\/g, "/");
-        const ext = path3.extname(e.name);
+        const rel = path4.relative(projectRoot, full).replace(/\\/g, "/");
+        const ext = path4.extname(e.name);
         for (const { ext: extName, pat } of globs) {
           if (ext === extName && (pat.test(rel) || pat.test(e.name))) {
             results.push(full);
@@ -1626,7 +1680,7 @@ async function runIndexer(ctx, opts) {
   let symbolsIndexed = 0;
   let files;
   if (opts.files && opts.files.length > 0) {
-    files = opts.files.map((f) => path3.resolve(projectRoot, f));
+    files = opts.files.map((f) => path4.resolve(projectRoot, f));
   } else {
     files = await findSourceFiles(projectRoot, ignore);
   }