open-codebase-index 0.22.4 → 0.22.5

package/dist/pi-extension.cjs

@@ -9011,6 +9011,29 @@ function extractPrimaryIdentifierQueryHint(query) {
   const best = codeTerms.find((term) => term.length >= 6);
   return best ?? null;
 }
+function pathSegmentsForAffinityMatch(filePath) {
+  const normalizedPath2 = normalizeRankingText(filePath).replace(/\\/g, "/");
+  const segments = normalizedPath2.split("/").filter((segment) => segment.length > 0);
+  if (segments.length === 0) {
+    return [];
+  }
+  const basename5 = segments[segments.length - 1] ?? "";
+  const basenameWithoutExt = basename5.replace(/\.[^/.]+$/u, "");
+  const normalizedSegments = segments.map((segment) => segment.toLowerCase());
+  return Array.from(/* @__PURE__ */ new Set([
+    ...normalizedSegments,
+    basenameWithoutExt.toLowerCase()
+  ]));
+}
+function hasModuleAffinity(filePath, exactIdentifierVariants) {
+  const haystack = pathSegmentsForAffinityMatch(filePath);
+  return exactIdentifierVariants.some((variant) => {
+    if (!variant || variant.length < 2) {
+      return false;
+    }
+    return haystack.includes(variant);
+  });
+}
 var FILE_PATH_HINT_EXTENSIONS = [
   "ts",
   "tsx",
@@ -9086,10 +9109,13 @@ function buildDeterministicIdentifierPass(query, candidates, limit, prioritizeSo
   ).map((candidate) => {
     const nameLower = (candidate.metadata.name ?? "").toLowerCase();
     const pathLower = candidate.metadata.filePath.toLowerCase();
-    let maxMatch = 0;
-    const nameMatchesPrimary = primaryVariants.some(
+    const exactIdentifierVariants = primaryVariants.filter((value) => value.length >= 2);
+    const exactMatch = exactIdentifierVariants.some(
       (variant) => nameLower === variant || nameLower.replace(/[^a-z0-9]/g, "") === variant.replace(/[^a-z0-9]/g, "")
     );
+    let maxMatch = 0;
+    const nameMatchesPrimary = exactMatch;
+    const pathAffinity = exactMatch ? hasModuleAffinity(candidate.metadata.filePath, exactIdentifierVariants) : false;
     const pathMatchesFileHint = filePathHint ? pathMatchesHint(candidate.metadata.filePath, filePathHint) : false;
     for (const hint of hints) {
       const variants = normalizeIdentifierVariants(hint);
@@ -9110,12 +9136,17 @@ function buildDeterministicIdentifierPass(query, candidates, limit, prioritizeSo
       candidate,
       maxMatch,
       pathMatchesFileHint,
-      nameMatchesPrimary
+      nameMatchesPrimary,
+      pathAffinity
     };
   }).filter((entry) => entry.maxMatch >= 0.7).sort((a, b) => {
     const aAnchored = a.pathMatchesFileHint && a.nameMatchesPrimary ? 1 : 0;
     const bAnchored = b.pathMatchesFileHint && b.nameMatchesPrimary ? 1 : 0;
     if (aAnchored !== bAnchored) return bAnchored - aAnchored;
+    if (a.nameMatchesPrimary !== b.nameMatchesPrimary) {
+      return b.nameMatchesPrimary ? 1 : -1;
+    }
+    if (a.pathAffinity !== b.pathAffinity) return b.pathAffinity ? 1 : -1;
     if (b.maxMatch !== a.maxMatch) return b.maxMatch - a.maxMatch;
     if (b.candidate.score !== a.candidate.score) return b.candidate.score - a.candidate.score;
     return a.candidate.id.localeCompare(b.candidate.id);