@liendev/lien 0.29.0 → 0.30.0

package/dist/index.js

@@ -8524,7 +8524,8 @@ var ListFunctionsSchema = external_exports.object({
   ),
   language: external_exports.string().optional().describe(
     "Filter by programming language.\n\nExamples: 'typescript', 'python', 'javascript', 'php'\n\nIf omitted, searches all languages."
-  )
+  ),
+  symbolType: external_exports.enum(["function", "method", "class", "interface"]).optional().describe("Filter by symbol type. If omitted, returns all types.")
 });
 
 // src/mcp/schemas/dependents.schema.ts
@@ -8647,6 +8648,10 @@ Batch calls are more efficient than multiple single-file calls.`
 Examples:
 - "Show all controllers" \u2192 list_functions({ pattern: ".*Controller.*" })
 - "Find service classes" \u2192 list_functions({ pattern: ".*Service$" })
+- "List all class methods" \u2192 list_functions({ symbolType: "method" })
+- "Find standalone functions" \u2192 list_functions({ symbolType: "function" })
+
+Filter by symbol type (function, method, class, interface) to narrow results.
 
 10x faster than semantic_search for structural/architectural queries. Use semantic_search instead when searching by what code DOES.`
   ),
@@ -8803,6 +8808,25 @@ function deduplicateResults(results) {
     return true;
   });
 }
+function cleanMetadataValue(key, value) {
+  if (value === void 0 || value === "") return null;
+  if (Array.isArray(value)) {
+    const filtered = value.filter((v) => v !== "");
+    return filtered.length > 0 ? filtered : null;
+  }
+  if (key === "symbols" && typeof value === "object" && value !== null) {
+    const symbols = value;
+    const filterArr = (arr) => Array.isArray(arr) ? arr.filter((s) => s !== "") : [];
+    const filtered = {
+      functions: filterArr(symbols.functions),
+      classes: filterArr(symbols.classes),
+      interfaces: filterArr(symbols.interfaces)
+    };
+    const hasAny = filtered.functions.length > 0 || filtered.classes.length > 0 || filtered.interfaces.length > 0;
+    return hasAny ? filtered : null;
+  }
+  return value;
+}
 function pickMetadata(metadata, allowlist) {
   const result = {
     file: metadata.file,
@@ -8812,8 +8836,9 @@ function pickMetadata(metadata, allowlist) {
   const out = result;
   for (const key of allowlist) {
     if (key === "file" || key === "startLine" || key === "endLine") continue;
-    if (metadata[key] !== void 0) {
-      out[key] = metadata[key];
+    const cleaned = cleanMetadataValue(key, metadata[key]);
+    if (cleaned !== null) {
+      out[key] = cleaned;
     }
   }
   return result;
@@ -8881,17 +8906,12 @@ async function handleSemanticSearch(args, ctx) {
       }
       log(`Returning ${results.length} results`);
       const shaped = shapeResults(results, "semantic_search");
-      const response = {
+      return {
         indexInfo: getIndexMetadata(),
-        results: shaped
+        results: shaped,
+        ...crossRepo && vectorDB instanceof QdrantDB && { groupedByRepo: groupResultsByRepo(shaped) },
+        ...notes.length > 0 && { note: notes.join(" ") }
       };
-      if (crossRepo && vectorDB instanceof QdrantDB) {
-        response.groupedByRepo = groupResultsByRepo(shaped);
-      }
-      if (notes.length > 0) {
-        response.note = notes.join(" ");
-      }
-      return response;
     }
   )(args);
 }
@@ -9086,7 +9106,9 @@ async function findRelatedChunks(filepaths, fileChunksMap, ctx) {
     const targetCanonical = getCanonicalPath(filepath, workspaceRoot);
     relatedChunksMap[index] = related.filter((r) => {
       const chunkCanonical = getCanonicalPath(r.metadata.file, workspaceRoot);
-      return chunkCanonical !== targetCanonical;
+      if (chunkCanonical === targetCanonical) return false;
+      if (r.metadata.language === "markdown") return false;
+      return true;
     });
   });
   return relatedChunksMap;
@@ -9122,23 +9144,15 @@ function findTestAssociations(filepaths, allChunks, ctx) {
     return Array.from(testFiles);
   });
 }
-function deduplicateChunks(fileChunks, relatedChunks, workspaceRoot) {
-  const seenChunks = /* @__PURE__ */ new Set();
-  return [...fileChunks, ...relatedChunks].filter((chunk) => {
-    const canonicalFile = getCanonicalPath(chunk.metadata.file, workspaceRoot);
-    const chunkId = `${canonicalFile}:${chunk.metadata.startLine}-${chunk.metadata.endLine}`;
-    if (seenChunks.has(chunkId)) return false;
-    seenChunks.add(chunkId);
-    return true;
-  });
+function deduplicateChunks(fileChunks, relatedChunks) {
+  return deduplicateResults([...fileChunks, ...relatedChunks]);
 }
-function buildFilesData(filepaths, fileChunksMap, relatedChunksMap, testAssociationsMap, workspaceRoot) {
+function buildFilesData(filepaths, fileChunksMap, relatedChunksMap, testAssociationsMap) {
   const filesData = {};
   filepaths.forEach((filepath, i) => {
     const dedupedChunks = deduplicateChunks(
       fileChunksMap[i],
-      relatedChunksMap[i] || [],
-      workspaceRoot
+      relatedChunksMap[i] || []
     );
     filesData[filepath] = {
       chunks: dedupedChunks,
@@ -9216,8 +9230,7 @@ async function handleGetFilesContext(args, ctx) {
         filepaths,
         fileChunksMap,
         relatedChunksMap,
-        testAssociationsMap,
-        workspaceRoot
+        testAssociationsMap
       );
       const totalChunks = Object.values(filesData).reduce(
         (sum, f) => sum + f.chunks.length,
@@ -9232,6 +9245,7 @@ async function handleGetFilesContext(args, ctx) {
 }
 
 // src/mcp/handlers/list-functions.ts
+import { SYMBOL_TYPE_MATCHES } from "@liendev/core";
 async function performContentScan(vectorDB, args, log) {
   log("Falling back to content scan...");
   let results = await vectorDB.scanWithFilter({
@@ -9246,6 +9260,13 @@ async function performContentScan(vectorDB, args, log) {
       return symbolName && regex.test(symbolName);
     });
   }
+  if (args.symbolType) {
+    const allowedTypes = SYMBOL_TYPE_MATCHES[args.symbolType];
+    results = results.filter((r) => {
+      const recordType2 = r.metadata?.symbolType;
+      return recordType2 && allowedTypes?.has(recordType2);
+    });
+  }
   return {
     results: results.slice(0, 50),
     method: "content"
@@ -9263,9 +9284,10 @@ async function handleListFunctions(args, ctx) {
         const results = await vectorDB.querySymbols({
           language: validatedArgs.language,
           pattern: validatedArgs.pattern,
+          symbolType: validatedArgs.symbolType,
           limit: 50
         });
-        if (results.length === 0 && (validatedArgs.language || validatedArgs.pattern)) {
+        if (results.length === 0 && (validatedArgs.language || validatedArgs.pattern || validatedArgs.symbolType)) {
           log("No symbol results, falling back to content scan...");
           queryResult = await performContentScan(vectorDB, validatedArgs, log);
         } else {
@@ -9796,7 +9818,11 @@ async function handleGetComplexity(args, ctx) {
         threshold,
         top ?? 10
       );
-      const response = {
+      const note = buildCrossRepoFallbackNote(fallback);
+      if (note) {
+        log("Warning: crossRepo=true requires Qdrant backend. Falling back to single-repo analysis.", "warning");
+      }
+      return {
         indexInfo: getIndexMetadata(),
         summary: {
           filesAnalyzed: report.summary.filesAnalyzed,
@@ -9805,17 +9831,12 @@ async function handleGetComplexity(args, ctx) {
           violationCount: violations.length,
           bySeverity
         },
-        violations: topViolations
+        violations: topViolations,
+        ...crossRepo && !fallback && allChunks.length > 0 && {
+          groupedByRepo: groupViolationsByRepo(topViolations, allChunks)
+        },
+        ...note && { note }
       };
-      if (crossRepo && !fallback && allChunks.length > 0) {
-        response.groupedByRepo = groupViolationsByRepo(topViolations, allChunks);
-      }
-      const note = buildCrossRepoFallbackNote(fallback);
-      if (note) {
-        log("Warning: crossRepo=true requires Qdrant backend. Falling back to single-repo analysis.", "warning");
-        response.note = note;
-      }
-      return response;
     }
   )(args);
 }