@toolbaux/guardian 0.1.21 → 0.1.22

package/dist/commands/mcp-serve.js

@@ -311,30 +311,63 @@ async function search(args) {
     // Frontend pages: match path or component
     const pages = (d.frontend_pages || []).filter((p) => p.path?.toLowerCase().includes(q) || p.component?.toLowerCase().includes(q) ||
         p.api_calls?.some((c) => c.toLowerCase().includes(q))).slice(0, 5).map((p) => `${p.path} → ${p.component}`);
-    // Functions: search literal_index (tactic:simp, sorry, string patterns) + function names
+    // Functions: ranked search across names, literals, calls — capped at 10 to prevent flooding
     const fnHits = [];
     const fi = await loadFuncIntel();
     if (fi) {
-        // Literal index: exact key match + partial key match
+        // Build a field map: file → field names (augments fn hits with model fields)
+        const fileToFields = new Map();
+        for (const m of Object.values(d.model_registry || {})) {
+            if (!m.file)
+                continue;
+            const existing = fileToFields.get(m.file) ?? [];
+            fileToFields.set(m.file, [...existing, ...(m.fields ?? [])]);
+        }
+        const scored = [];
+        const seen = new Set();
+        for (const fn of (fi.functions ?? [])) {
+            const nameNorm = (fn.name ?? "").toLowerCase();
+            const fileNorm = (fn.file ?? "").toLowerCase();
+            const callsNorm = (fn.calls ?? []).map((c) => c.toLowerCase());
+            const litsNorm = [...(fn.stringLiterals ?? []), ...(fn.regexPatterns ?? [])].map((l) => l.toLowerCase());
+            let score = 0;
+            if (nameNorm === q)
+                score = 1.0;
+            else if (nameNorm.includes(q))
+                score = 0.7;
+            else if (callsNorm.some((c) => c.includes(q)))
+                score = 0.5;
+            else if (litsNorm.some((l) => l.includes(q)))
+                score = 0.3;
+            else if (fileNorm.includes(q))
+                score = 0.2;
+            if (score > 0) {
+                scored.push({ fn, score });
+                seen.add(`${fn.file}:${fn.name}`);
+            }
+        }
+        // Also surface literal_index hits not already captured
         const litIndex = fi.literal_index ?? {};
         for (const [key, hits] of Object.entries(litIndex)) {
-            if (key.includes(q)) {
-                for (const h of hits.slice(0, 3)) {
-                    fnHits.push(`${h.function} [${h.file}:${h.line}] (lit:${key})`);
-                }
+            if (!key.includes(q))
+                continue;
+            for (const h of hits) {
+                const uid = `${h.file}:${h.function}`;
+                if (seen.has(uid))
+                    continue;
+                seen.add(uid);
+                const fn = fi.functions.find((f) => f.file === h.file && f.name === h.function);
+                scored.push({ fn: fn ?? { name: h.function, file: h.file, lines: [h.line, h.line], calls: [], stringLiterals: [], regexPatterns: [] }, score: 0.25 });
             }
-            if (fnHits.length >= 10)
-                break;
         }
-        // Function names: match by name
-        if (fnHits.length < 10) {
-            for (const fn of (fi.functions ?? [])) {
-                if (fn.name?.toLowerCase().includes(q)) {
-                    fnHits.push(`${fn.name} [${fn.file}:${fn.lines?.[0]}]`);
-                    if (fnHits.length >= 10)
-                        break;
-                }
-            }
+        // Sort by score desc, take top 10
+        scored.sort((a, b) => b.score - a.score);
+        const projectRoot = process.cwd();
+        for (const { fn } of scored.slice(0, 10)) {
+            const relFile = fn.file?.startsWith("/") ? require("path").relative(projectRoot, fn.file) : fn.file;
+            const fields = fileToFields.get(fn.file) ?? [];
+            const fieldSuffix = fields.length > 0 ? ` fields:${fields.slice(0, 5).join(",")}` : "";
+            fnHits.push(`${fn.name} [${relFile}:${fn.lines?.[0]}]${fieldSuffix}`);
         }
     }
     return compact({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@toolbaux/guardian",
-  "version": "0.1.21",
+  "version": "0.1.22",
   "type": "module",
   "description": "Architectural intelligence for codebases. Verify that AI-generated code matches your architectural intent.",
   "keywords": [