kiri-mcp-server 0.2.0

package/dist/src/server/handlers.js

@@ -0,0 +1,954 @@
+import path from "node:path";
+import { generateEmbedding, structuralSimilarity } from "../shared/embedding.js";
+import { encode as encodeGPT } from "../shared/tokenizer.js";
+import { coerceProfileName, loadScoringProfile } from "./scoring.js";
+const DEFAULT_SEARCH_LIMIT = 50;
+const DEFAULT_SNIPPET_WINDOW = 150;
+const DEFAULT_BUNDLE_LIMIT = 12;
+const MAX_BUNDLE_LIMIT = 20;
+const MAX_KEYWORDS = 12;
+const MAX_MATCHES_PER_KEYWORD = 40;
+const MAX_DEPENDENCY_SEEDS = 8;
+const MAX_DEPENDENCY_SEEDS_QUERY_LIMIT = 100; // SQL injection防御用の上限
+const NEARBY_LIMIT = 6;
+const FALLBACK_SNIPPET_WINDOW = 120;
+const MAX_RERANK_LIMIT = 50;
+const STOP_WORDS = new Set([
+    "the",
+    "and",
+    "for",
+    "with",
+    "from",
+    "this",
+    "that",
+    "have",
+    "has",
+    "will",
+    "would",
+    "into",
+    "about",
+    "there",
+    "their",
+    "your",
+    "fix",
+    "test",
+    "tests",
+    "issue",
+    "error",
+    "bug",
+    "fail",
+    "failing",
+    "make",
+    "when",
+    "where",
+    "should",
+    "could",
+    "need",
+    "goal",
+]);
+function normalizeLimit(limit) {
+    if (!limit || Number.isNaN(limit)) {
+        return DEFAULT_SEARCH_LIMIT;
+    }
+    return Math.min(Math.max(1, Math.floor(limit)), 100);
+}
+function buildPreview(content, query) {
+    const lowerContent = content.toLowerCase();
+    const lowerQuery = query.toLowerCase();
+    const index = lowerContent.indexOf(lowerQuery);
+    if (index === -1) {
+        return { preview: content.slice(0, 240), line: 1 };
+    }
+    const prefix = content.slice(0, index);
+    const prefixLines = prefix.split(/\r?\n/);
+    const matchLine = prefix.length === 0 ? 1 : prefixLines.length;
+    const snippetStart = Math.max(0, index - 120);
+    const snippetEnd = Math.min(content.length, index + query.length + 120);
+    const preview = content.slice(snippetStart, snippetEnd);
+    return { preview, line: matchLine };
+}
+function normalizeBundleLimit(limit) {
+    if (!limit || Number.isNaN(limit)) {
+        return DEFAULT_BUNDLE_LIMIT;
+    }
+    return Math.min(Math.max(1, Math.floor(limit)), MAX_BUNDLE_LIMIT);
+}
+function extractKeywords(text) {
+    const words = text
+        .toLowerCase()
+        .split(/[^a-z0-9_]+/iu)
+        .map((word) => word.trim())
+        .filter((word) => word.length >= 3 && !STOP_WORDS.has(word));
+    const unique = [];
+    for (const word of words) {
+        if (!unique.includes(word)) {
+            unique.push(word);
+            if (unique.length >= MAX_KEYWORDS) {
+                break;
+            }
+        }
+    }
+    return unique;
+}
+function ensureCandidate(map, filePath) {
+    let candidate = map.get(filePath);
+    if (!candidate) {
+        candidate = {
+            path: filePath,
+            score: 0,
+            reasons: new Set(),
+            matchLine: null,
+            content: null,
+            totalLines: null,
+            lang: null,
+            ext: null,
+            embedding: null,
+            semanticSimilarity: null,
+        };
+        map.set(filePath, candidate);
+    }
+    return candidate;
+}
+function parseEmbedding(vectorJson, vectorDims) {
+    if (!vectorJson || !vectorDims || vectorDims <= 0) {
+        return null;
+    }
+    try {
+        const parsed = JSON.parse(vectorJson);
+        if (!Array.isArray(parsed)) {
+            return null;
+        }
+        const values = [];
+        for (let i = 0; i < parsed.length && i < vectorDims; i += 1) {
+            const raw = parsed[i];
+            const num = typeof raw === "number" ? raw : Number(raw);
+            if (!Number.isFinite(num)) {
+                return null;
+            }
+            values.push(num);
+        }
+        return values.length === vectorDims ? values : null;
+    }
+    catch {
+        return null;
+    }
+}
+function applyStructuralScores(candidates, queryEmbedding, structuralWeight) {
+    if (!queryEmbedding || structuralWeight <= 0) {
+        return;
+    }
+    for (const candidate of candidates) {
+        if (!candidate.embedding) {
+            continue;
+        }
+        const similarity = structuralSimilarity(queryEmbedding, candidate.embedding);
+        if (!Number.isFinite(similarity) || similarity <= 0) {
+            continue;
+        }
+        candidate.semanticSimilarity = similarity;
+        candidate.score += structuralWeight * similarity;
+        candidate.reasons.add(`structural:${similarity.toFixed(2)}`);
+    }
+}
+async function fetchEmbeddingMap(db, repoId, paths) {
+    const map = new Map();
+    if (paths.length === 0) {
+        return map;
+    }
+    const placeholders = paths.map(() => "?").join(", ");
+    const rows = await db.all(`
+      SELECT path, vector_json, dims AS vector_dims
+      FROM file_embedding
+      WHERE repo_id = ? AND path IN (${placeholders})
+    `, [repoId, ...paths]);
+    for (const row of rows) {
+        const embedding = parseEmbedding(row.vector_json, row.vector_dims);
+        if (embedding) {
+            map.set(row.path, embedding);
+        }
+    }
+    return map;
+}
+async function loadFileContent(db, repoId, filePath) {
+    const rows = await db.all(`
+      SELECT f.path, f.lang, f.ext, f.is_binary, b.content, fe.vector_json, fe.dims AS vector_dims
+      FROM file f
+      JOIN blob b ON b.hash = f.blob_hash
+      LEFT JOIN file_embedding fe
+        ON fe.repo_id = f.repo_id
+       AND fe.path = f.path
+      WHERE f.repo_id = ? AND f.path = ?
+      LIMIT 1
+    `, [repoId, filePath]);
+    const row = rows[0];
+    if (!row || row.is_binary || row.content === null) {
+        return null;
+    }
+    const totalLines = row.content.length === 0 ? 0 : row.content.split(/\r?\n/).length;
+    return {
+        content: row.content,
+        lang: row.lang,
+        ext: row.ext,
+        totalLines,
+        embedding: parseEmbedding(row.vector_json ?? null, row.vector_dims ?? null),
+    };
+}
+function selectSnippet(snippets, matchLine) {
+    const firstSnippet = snippets[0];
+    if (!firstSnippet) {
+        return null;
+    }
+    if (matchLine === null) {
+        return firstSnippet;
+    }
+    const containing = snippets.find((snippet) => matchLine >= snippet.start_line && matchLine <= snippet.end_line);
+    if (containing) {
+        return containing;
+    }
+    if (matchLine < firstSnippet.start_line) {
+        return firstSnippet;
+    }
+    const lastSnippet = snippets[snippets.length - 1];
+    return lastSnippet ?? firstSnippet;
+}
+function buildSnippetPreview(content, startLine, endLine) {
+    const lines = content.split(/\r?\n/);
+    const startIndex = Math.max(0, Math.min(startLine - 1, lines.length));
+    const endIndex = Math.max(startIndex, Math.min(endLine, lines.length));
+    const snippet = lines.slice(startIndex, endIndex).join("\n");
+    if (snippet.length <= 480) {
+        return snippet;
+    }
+    return `${snippet.slice(0, 479)}…`;
+}
+/**
+ * トークン数を推定（コンテンツベース）
+ * 実際のGPTトークナイザーを使用して正確にカウント
+ *
+ * @param content - ファイル全体のコンテンツ
+ * @param startLine - 開始行（1-indexed）
+ * @param endLine - 終了行（1-indexed）
+ * @returns 推定トークン数
+ */
+function estimateTokensFromContent(content, startLine, endLine) {
+    const lines = content.split(/\r?\n/);
+    const startIndex = Math.max(0, startLine - 1);
+    const endIndex = Math.min(endLine, lines.length);
+    const selectedLines = lines.slice(startIndex, endIndex);
+    const text = selectedLines.join("\n");
+    try {
+        // 実際のGPTトークナイザーを使用
+        return encodeGPT(text).length;
+    }
+    catch (error) {
+        // フォールバック: 平均的な英語テキストで4文字 ≈ 1トークン
+        console.warn("Token encoding failed, using character-based fallback", error);
+        return Math.max(1, Math.ceil(text.length / 4));
+    }
+}
+/**
+ * 複数単語クエリを単語分割してOR検索条件を構築
+ * @param query - 検索クエリ文字列
+ * @returns 単語配列（2文字以下を除外）
+ */
+function splitQueryWords(query) {
+    // 空白、スラッシュ、ハイフン、アンダースコアで分割
+    const words = query.split(/[\s/\-_]+/).filter((w) => w.length > 2);
+    return words.length > 0 ? words : [query]; // 全て除外された場合は元のクエリを使用
+}
+/**
+ * ファイルタイプに基づいてスコアをブーストする
+ * プロファイルに応じて実装ファイルまたはドキュメントを優遇
+ * @param path - ファイルパス
+ * @param baseScore - 元のスコア
+ * @param profile - ブーストプロファイル ("default" | "docs" | "none")
+ * @returns ブースト適用後のスコア
+ */
+function applyFileTypeBoost(path, baseScore, profile = "default") {
+    // ブースト無効
+    if (profile === "none") {
+        return baseScore;
+    }
+    // ドキュメントモード: ドキュメントを優遇、実装ファイルを軽度減点
+    if (profile === "docs") {
+        if (path.endsWith(".md") || path.endsWith(".yaml") || path.endsWith(".yml")) {
+            return baseScore * 1.5; // ドキュメント優遇
+        }
+        if (path.startsWith("src/") && (path.endsWith(".ts") || path.endsWith(".js"))) {
+            return baseScore * 0.7; // 実装ファイル軽度減点
+        }
+        return baseScore;
+    }
+    // デフォルトモード: 実装ファイルを優遇、ドキュメントを減点
+    if (path.startsWith("src/") && (path.endsWith(".ts") || path.endsWith(".js"))) {
+        return baseScore * 1.5; // 実装ファイル優遇
+    }
+    if (path.startsWith("tests/") && path.endsWith(".ts")) {
+        return baseScore * 1.2; // テストファイル軽度優遇
+    }
+    if (path.endsWith(".md") || path.endsWith(".yaml") || path.endsWith(".yml")) {
+        return baseScore * 0.5; // ドキュメント減点
+    }
+    return baseScore;
+}
+export async function filesSearch(context, params) {
+    const { db, repoId } = context;
+    const { query } = params;
+    if (!query || query.trim().length === 0) {
+        throw new Error("files.search requires a non-empty query. Provide a search keyword to continue.");
+    }
+    const limit = normalizeLimit(params.limit);
+    const hasFTS = context.features?.fts ?? false;
+    let sql;
+    let values;
+    if (hasFTS) {
+        // FTS拡張利用可能: fts_main_blob.match_bm25 を使用
+        const conditions = ["f.repo_id = ?"];
+        values = [repoId];
+        // 言語・拡張子フィルタ
+        if (params.lang) {
+            conditions.push("COALESCE(f.lang, '') = ?");
+            values.push(params.lang);
+        }
+        if (params.ext) {
+            conditions.push("COALESCE(f.ext, '') = ?");
+            values.push(params.ext);
+        }
+        if (params.path_prefix) {
+            conditions.push("f.path LIKE ?");
+            values.push(`${params.path_prefix}%`);
+        }
+        // FTS検索（BM25スコアリング）
+        sql = `
+      SELECT f.path, f.lang, f.ext, b.content, fts.score
+      FROM file f
+      JOIN blob b ON b.hash = f.blob_hash
+      JOIN (
+        SELECT hash, fts_main_blob.match_bm25(hash, ?) AS score
+        FROM blob
+        WHERE score IS NOT NULL
+      ) fts ON fts.hash = b.hash
+      WHERE ${conditions.join(" AND ")}
+      ORDER BY fts.score DESC
+      LIMIT ?
+    `;
+        values.unshift(query); // FTSクエリを先頭に追加
+        values.push(limit);
+    }
+    else {
+        // FTS拡張利用不可: ILIKE検索（Phase 1の単語分割ロジック）
+        const conditions = ["f.repo_id = ?", "b.content IS NOT NULL"];
+        values = [repoId];
+        const words = splitQueryWords(query);
+        if (words.length === 1) {
+            conditions.push("b.content ILIKE '%' || ? || '%'");
+            values.push(query);
+        }
+        else {
+            const wordConditions = words.map(() => "b.content ILIKE '%' || ? || '%'");
+            conditions.push(`(${wordConditions.join(" OR ")})`);
+            values.push(...words);
+        }
+        if (params.lang) {
+            conditions.push("COALESCE(f.lang, '') = ?");
+            values.push(params.lang);
+        }
+        if (params.ext) {
+            conditions.push("COALESCE(f.ext, '') = ?");
+            values.push(params.ext);
+        }
+        if (params.path_prefix) {
+            conditions.push("f.path LIKE ?");
+            values.push(`${params.path_prefix}%`);
+        }
+        sql = `
+      SELECT f.path, f.lang, f.ext, b.content
+      FROM file f
+      JOIN blob b ON b.hash = f.blob_hash
+      WHERE ${conditions.join(" AND ")}
+      ORDER BY f.path
+      LIMIT ?
+    `;
+        values.push(limit);
+    }
+    const rows = await db.all(sql, values);
+    const boostProfile = params.boost_profile ?? "default";
+    return rows
+        .map((row) => {
+        const { preview, line } = buildPreview(row.content ?? "", query);
+        const baseScore = row.score ?? 1.0; // FTS時はBM25スコア、ILIKE時は1.0
+        const boostedScore = applyFileTypeBoost(row.path, baseScore, boostProfile);
+        return {
+            path: row.path,
+            preview,
+            matchLine: line,
+            lang: row.lang,
+            ext: row.ext,
+            score: boostedScore,
+        };
+    })
+        .sort((a, b) => b.score - a.score); // スコアの高い順に再ソート
+}
+export async function snippetsGet(context, params) {
+    const { db, repoId } = context;
+    if (!params.path) {
+        throw new Error("snippets.get requires a file path. Specify a tracked text file path to continue.");
+    }
+    const rows = await db.all(`
+      SELECT f.path, f.lang, f.ext, f.is_binary, b.content
+      FROM file f
+      JOIN blob b ON b.hash = f.blob_hash
+      WHERE f.repo_id = ? AND f.path = ?
+      LIMIT 1
+    `, [repoId, params.path]);
+    if (rows.length === 0) {
+        throw new Error("Requested snippet file was not indexed. Re-run the indexer or choose another path.");
+    }
+    const row = rows[0];
+    if (!row) {
+        throw new Error("Requested snippet file was not indexed. Re-run the indexer or choose another path.");
+    }
+    if (row.is_binary) {
+        throw new Error("Binary snippets are not supported. Choose a text file to preview its content.");
+    }
+    if (row.content === null) {
+        throw new Error("Snippet content is unavailable. Re-run the indexer to refresh DuckDB state.");
+    }
+    const lines = row.content.split(/\r?\n/);
+    const totalLines = lines.length;
+    const snippetRows = await db.all(`
+      SELECT s.snippet_id, s.start_line, s.end_line, s.symbol_id, sym.name AS symbol_name, sym.kind AS symbol_kind
+      FROM snippet s
+      LEFT JOIN symbol sym
+        ON sym.repo_id = s.repo_id
+       AND sym.path = s.path
+       AND sym.symbol_id = s.symbol_id
+      WHERE s.repo_id = ? AND s.path = ?
+      ORDER BY s.start_line
+    `, [repoId, params.path]);
+    const requestedStart = params.start_line ?? 1;
+    const requestedEnd = params.end_line ?? Math.min(totalLines, requestedStart + DEFAULT_SNIPPET_WINDOW - 1);
+    const useSymbolSnippets = snippetRows.length > 0 && params.end_line === undefined;
+    let snippetSelection = null;
+    if (useSymbolSnippets) {
+        snippetSelection =
+            snippetRows.find((snippet) => requestedStart >= snippet.start_line && requestedStart <= snippet.end_line) ?? null;
+        if (!snippetSelection) {
+            const firstSnippet = snippetRows[0];
+            if (firstSnippet && requestedStart < firstSnippet.start_line) {
+                snippetSelection = firstSnippet;
+            }
+            else {
+                snippetSelection = snippetRows[snippetRows.length - 1] ?? null;
+            }
+        }
+    }
+    let startLine;
+    let endLine;
+    let symbolName = null;
+    let symbolKind = null;
+    if (snippetSelection) {
+        startLine = snippetSelection.start_line;
+        endLine = snippetSelection.end_line;
+        symbolName = snippetSelection.symbol_name;
+        symbolKind = snippetSelection.symbol_kind;
+    }
+    else {
+        startLine = Math.max(1, Math.min(totalLines, requestedStart));
+        endLine = Math.max(startLine, Math.min(totalLines, requestedEnd));
+    }
+    const snippetContent = lines.slice(startLine - 1, endLine).join("\n");
+    return {
+        path: row.path,
+        startLine,
+        endLine,
+        content: snippetContent,
+        totalLines,
+        symbolName,
+        symbolKind,
+    };
+}
+export async function contextBundle(context, params) {
+    const { db, repoId } = context;
+    const goal = params.goal?.trim() ?? "";
+    if (goal.length === 0) {
+        throw new Error("context.bundle requires a non-empty goal. Describe your objective to receive context.");
+    }
+    const limit = normalizeBundleLimit(params.limit);
+    const artifacts = params.artifacts ?? {};
+    // スコアリング重みをロード（将来的には設定ファイルや引数から）
+    const profileName = coerceProfileName(params.profile ?? null);
+    const weights = loadScoringProfile(profileName);
+    const keywordSources = [goal];
+    if (artifacts.failing_tests && artifacts.failing_tests.length > 0) {
+        keywordSources.push(artifacts.failing_tests.join(" "));
+    }
+    if (artifacts.last_diff) {
+        keywordSources.push(artifacts.last_diff);
+    }
+    if (artifacts.editing_path) {
+        keywordSources.push(artifacts.editing_path);
+    }
+    const semanticSeed = keywordSources.join(" ");
+    const queryEmbedding = generateEmbedding(semanticSeed)?.values ?? null;
+    let keywords = extractKeywords(semanticSeed);
+    if (keywords.length === 0 && artifacts.editing_path) {
+        const pathSegments = artifacts.editing_path
+            .split(/[/_.-]/)
+            .map((segment) => segment.toLowerCase())
+            .filter((segment) => segment.length >= 3 && !STOP_WORDS.has(segment));
+        keywords = pathSegments.slice(0, MAX_KEYWORDS);
+    }
+    const candidates = new Map();
+    const stringMatchSeeds = new Set();
+    const fileCache = new Map();
+    for (const keyword of keywords) {
+        const rows = await db.all(`
+        SELECT f.path, f.lang, f.ext, f.is_binary, b.content, fe.vector_json, fe.dims AS vector_dims
+        FROM file f
+        JOIN blob b ON b.hash = f.blob_hash
+        LEFT JOIN file_embedding fe
+          ON fe.repo_id = f.repo_id
+         AND fe.path = f.path
+        WHERE f.repo_id = ?
+          AND f.is_binary = FALSE
+          AND b.content ILIKE '%' || ? || '%'
+        ORDER BY f.path
+        LIMIT ?
+      `, [repoId, keyword, MAX_MATCHES_PER_KEYWORD]);
+        for (const row of rows) {
+            if (row.content === null) {
+                continue;
+            }
+            const candidate = ensureCandidate(candidates, row.path);
+            candidate.score += weights.textMatch;
+            candidate.reasons.add(`text:${keyword}`);
+            // ファイルタイプブーストを適用（boost_profileに応じて）
+            const boostProfile = params.boost_profile ?? "default";
+            if (boostProfile === "docs") {
+                // ドキュメントモード
+                if (row.path.endsWith(".md") || row.path.endsWith(".yaml") || row.path.endsWith(".yml")) {
+                    candidate.score += 0.5; // ドキュメントに追加ボーナス
+                    candidate.reasons.add("boost:doc-file");
+                }
+                if (row.path.startsWith("src/") && row.ext === ".ts") {
+                    candidate.score -= 0.2; // 実装ファイルに軽度ペナルティ
+                    candidate.reasons.add("penalty:impl-file");
+                }
+            }
+            else if (boostProfile === "default") {
+                // デフォルトモード
+                if (row.path.startsWith("src/") && row.ext === ".ts") {
+                    candidate.score += 0.5; // 実装ファイルに追加ボーナス
+                    candidate.reasons.add("boost:impl-file");
+                }
+                if (row.path.endsWith(".md") || row.path.endsWith(".yaml") || row.path.endsWith(".yml")) {
+                    candidate.score -= 0.3; // ドキュメントにペナルティ
+                    candidate.reasons.add("penalty:doc-file");
+                }
+            }
+            // boostProfile === "none" の場合はブースト適用なし
+            const { line } = buildPreview(row.content, keyword);
+            candidate.matchLine =
+                candidate.matchLine === null ? line : Math.min(candidate.matchLine, line);
+            candidate.content ??= row.content;
+            candidate.lang ??= row.lang;
+            candidate.ext ??= row.ext;
+            candidate.totalLines ??= row.content.length === 0 ? 0 : row.content.split(/\r?\n/).length;
+            candidate.embedding ??= parseEmbedding(row.vector_json ?? null, row.vector_dims ?? null);
+            stringMatchSeeds.add(row.path);
+            if (!fileCache.has(row.path)) {
+                fileCache.set(row.path, {
+                    content: row.content,
+                    lang: row.lang,
+                    ext: row.ext,
+                    totalLines: candidate.totalLines ?? 0,
+                    embedding: candidate.embedding,
+                });
+            }
+        }
+    }
+    if (artifacts.editing_path) {
+        const editingCandidate = ensureCandidate(candidates, artifacts.editing_path);
+        editingCandidate.score += weights.editingPath;
+        editingCandidate.reasons.add("artifact:editing_path");
+        editingCandidate.matchLine ??= 1;
+    }
+    // SQL injection防御: ファイルパスの検証パターン
+    const SAFE_PATH_PATTERN = /^[a-zA-Z0-9_.\-/]+$/;
+    const dependencySeeds = new Set();
+    for (const pathSeed of stringMatchSeeds) {
+        if (!SAFE_PATH_PATTERN.test(pathSeed)) {
+            console.warn(`Skipping potentially unsafe path in dependency seeds: ${pathSeed}`);
+            continue;
+        }
+        dependencySeeds.add(pathSeed);
+        if (dependencySeeds.size >= MAX_DEPENDENCY_SEEDS) {
+            break;
+        }
+    }
+    if (artifacts.editing_path) {
+        if (!SAFE_PATH_PATTERN.test(artifacts.editing_path)) {
+            throw new Error(`Invalid editing_path format. Path must contain only alphanumeric characters, underscores, dots, hyphens, and forward slashes.`);
+        }
+        dependencySeeds.add(artifacts.editing_path);
+    }
+    if (dependencySeeds.size > 0) {
+        // SQL injection防御: プレースホルダー生成前にサイズを検証
+        if (dependencySeeds.size > MAX_DEPENDENCY_SEEDS_QUERY_LIMIT) {
+            throw new Error(`Too many dependency seeds: ${dependencySeeds.size} (max ${MAX_DEPENDENCY_SEEDS_QUERY_LIMIT}). Narrow your search criteria.`);
+        }
+        const placeholders = Array.from(dependencySeeds, () => "?").join(", ");
+        // 防御的チェック: プレースホルダーが正しい形式であることを確認
+        // 期待される形式: "?, ?, ..." (クエスチョンマーク、カンマ、スペースのみ)
+        if (!/^(\?)(,\s*\?)*$/.test(placeholders)) {
+            throw new Error("Invalid placeholder generation detected. Operation aborted for safety.");
+        }
+        const depRows = await db.all(`
+        SELECT src_path, dst_kind, dst, rel
+        FROM dependency
+        WHERE repo_id = ? AND src_path IN (${placeholders})
+      `, [repoId, ...dependencySeeds]);
+        for (const dep of depRows) {
+            if (dep.dst_kind !== "path") {
+                continue;
+            }
+            const candidate = ensureCandidate(candidates, dep.dst);
+            candidate.score += weights.dependency;
+            candidate.reasons.add(`dep:${dep.src_path}`);
+        }
+    }
+    if (artifacts.editing_path) {
+        const directory = path.posix.dirname(artifacts.editing_path);
+        if (directory && directory !== ".") {
+            const nearRows = await db.all(`
+          SELECT path
+          FROM file
+          WHERE repo_id = ?
+            AND is_binary = FALSE
+            AND path LIKE ?
+          ORDER BY path
+          LIMIT ?
+        `, [repoId, `${directory}/%`, NEARBY_LIMIT + 1]);
+            for (const near of nearRows) {
+                if (near.path === artifacts.editing_path) {
+                    continue;
+                }
+                const candidate = ensureCandidate(candidates, near.path);
+                candidate.score += weights.proximity;
+                candidate.reasons.add(`near:${directory}`);
+            }
+        }
+    }
+    const materializedCandidates = [];
+    for (const candidate of candidates.values()) {
+        if (!candidate.content) {
+            const cached = fileCache.get(candidate.path);
+            if (cached) {
+                candidate.content = cached.content;
+                candidate.lang = cached.lang;
+                candidate.ext = cached.ext;
+                candidate.totalLines = cached.totalLines;
+                candidate.embedding = cached.embedding;
+            }
+            else {
+                const loaded = await loadFileContent(db, repoId, candidate.path);
+                if (!loaded) {
+                    continue;
+                }
+                candidate.content = loaded.content;
+                candidate.lang = loaded.lang;
+                candidate.ext = loaded.ext;
+                candidate.totalLines = loaded.totalLines;
+                candidate.embedding = loaded.embedding;
+                fileCache.set(candidate.path, loaded);
+            }
+        }
+        materializedCandidates.push(candidate);
+    }
+    if (materializedCandidates.length === 0) {
+        return { context: [], tokens_estimate: 0 };
+    }
+    applyStructuralScores(materializedCandidates, queryEmbedding, weights.structural);
+    const sortedCandidates = materializedCandidates
+        .sort((a, b) => {
+        if (b.score === a.score) {
+            return a.path.localeCompare(b.path);
+        }
+        return b.score - a.score;
+    })
+        .slice(0, limit);
+    const maxScore = Math.max(...sortedCandidates.map((candidate) => candidate.score));
+    const results = [];
+    for (const candidate of sortedCandidates) {
+        if (!candidate.content) {
+            continue;
+        }
+        const snippets = await db.all(`
+        SELECT s.snippet_id, s.start_line, s.end_line, s.symbol_id, sym.name AS symbol_name, sym.kind AS symbol_kind
+        FROM snippet s
+        LEFT JOIN symbol sym
+          ON sym.repo_id = s.repo_id
+         AND sym.path = s.path
+         AND sym.symbol_id = s.symbol_id
+        WHERE s.repo_id = ? AND s.path = ?
+        ORDER BY s.start_line
+      `, [repoId, candidate.path]);
+        const selected = selectSnippet(snippets, candidate.matchLine);
+        let startLine;
+        let endLine;
+        if (selected) {
+            startLine = selected.start_line;
+            endLine = selected.end_line;
+        }
+        else {
+            const totalLines = candidate.totalLines ?? 0;
+            const matchLine = candidate.matchLine ?? 1;
+            const windowHalf = Math.floor(FALLBACK_SNIPPET_WINDOW / 2);
+            startLine = Math.max(1, matchLine - windowHalf);
+            endLine = Math.min(totalLines === 0 ? matchLine + windowHalf : totalLines, startLine + FALLBACK_SNIPPET_WINDOW - 1);
+        }
+        if (endLine < startLine) {
+            endLine = startLine;
+        }
+        const preview = buildSnippetPreview(candidate.content, startLine, endLine);
+        const reasons = new Set(candidate.reasons);
+        if (selected && selected.symbol_name) {
+            reasons.add(`symbol:${selected.symbol_name}`);
+        }
+        const normalizedScore = maxScore > 0 ? candidate.score / maxScore : 0;
+        results.push({
+            path: candidate.path,
+            range: [startLine, endLine],
+            preview,
+            why: Array.from(reasons).sort(),
+            score: Number.isFinite(normalizedScore) ? normalizedScore : 0,
+        });
+    }
+    // コンテンツベースのトークン推定を使用（より正確）
+    const tokensEstimate = results.reduce((acc, item) => {
+        const candidate = sortedCandidates.find((c) => c.path === item.path);
+        if (candidate && candidate.content) {
+            return acc + estimateTokensFromContent(candidate.content, item.range[0], item.range[1]);
+        }
+        // フォールバック: 行ベース推定（コンテンツが利用不可の場合）
+        const lineCount = Math.max(1, item.range[1] - item.range[0] + 1);
+        return acc + lineCount * 4;
+    }, 0);
+    return { context: results, tokens_estimate: tokensEstimate };
+}
+export async function semanticRerank(context, params) {
+    const text = params.text?.trim() ?? "";
+    if (text.length === 0) {
+        throw new Error("semantic.rerank requires non-empty text. Describe the intent to compute semantic similarity.");
+    }
+    if (!Array.isArray(params.candidates) || params.candidates.length === 0) {
+        return { candidates: [] };
+    }
+    const uniqueCandidates = [];
+    const seenPaths = new Set();
+    for (const candidate of params.candidates) {
+        if (!candidate || typeof candidate.path !== "string" || candidate.path.length === 0) {
+            continue;
+        }
+        if (seenPaths.has(candidate.path)) {
+            continue;
+        }
+        seenPaths.add(candidate.path);
+        uniqueCandidates.push(candidate);
+        if (uniqueCandidates.length >= MAX_RERANK_LIMIT) {
+            break;
+        }
+    }
+    if (uniqueCandidates.length === 0) {
+        return { candidates: [] };
+    }
+    const limitRaw = params.k ?? uniqueCandidates.length;
+    const limit = Math.max(1, Math.min(MAX_RERANK_LIMIT, Math.floor(limitRaw)));
+    const profileName = coerceProfileName(params.profile ?? null);
+    const weights = loadScoringProfile(profileName);
+    const structuralWeight = weights.structural;
+    const queryEmbedding = generateEmbedding(text)?.values ?? null;
+    let embeddingMap = new Map();
+    if (queryEmbedding && structuralWeight > 0) {
+        const paths = uniqueCandidates.map((candidate) => candidate.path);
+        embeddingMap = await fetchEmbeddingMap(context.db, context.repoId, paths);
+    }
+    const scored = uniqueCandidates.map((candidate) => {
+        const base = typeof candidate.score === "number" && Number.isFinite(candidate.score) ? candidate.score : 0;
+        let semantic = 0;
+        if (queryEmbedding && structuralWeight > 0) {
+            const embedding = embeddingMap.get(candidate.path);
+            if (embedding) {
+                const similarity = structuralSimilarity(queryEmbedding, embedding);
+                if (Number.isFinite(similarity) && similarity > 0) {
+                    semantic = similarity;
+                }
+            }
+        }
+        const combined = base + structuralWeight * semantic;
+        return {
+            path: candidate.path,
+            base,
+            semantic,
+            combined,
+        };
+    });
+    const sorted = scored.sort((a, b) => {
+        if (b.combined === a.combined) {
+            if (b.semantic === a.semantic) {
+                return a.path.localeCompare(b.path);
+            }
+            return b.semantic - a.semantic;
+        }
+        return b.combined - a.combined;
+    });
+    return { candidates: sorted.slice(0, limit) };
+}
+export async function depsClosure(context, params) {
+    const { db, repoId } = context;
+    if (!params.path) {
+        throw new Error("deps.closure requires a file path. Provide a tracked source file path to continue.");
+    }
+    const direction = params.direction ?? "outbound";
+    const maxDepth = params.max_depth ?? 3;
+    const includePackages = params.include_packages ?? true;
+    const dependencyRows = await db.all(`
+      SELECT src_path, dst_kind, dst, rel
+      FROM dependency
+      WHERE repo_id = ?
+    `, [repoId]);
+    // outbound: このファイルが使用する依存関係
+    const outbound = new Map();
+    // inbound: このファイルを使用しているファイル
+    const inbound = new Map();
+    for (const row of dependencyRows) {
+        // outbound マップ構築
+        if (!outbound.has(row.src_path)) {
+            outbound.set(row.src_path, []);
+        }
+        outbound.get(row.src_path)?.push(row);
+        // inbound マップ構築（dst が path の場合のみ）
+        if (row.dst_kind === "path") {
+            if (!inbound.has(row.dst)) {
+                inbound.set(row.dst, []);
+            }
+            inbound.get(row.dst)?.push(row);
+        }
+    }
+    const queue = [{ path: params.path, depth: 0 }];
+    const visitedPaths = new Set([params.path]);
+    const nodeDepth = new Map();
+    const edgeSet = new Map();
+    const recordNode = (node) => {
+        const key = `${node.kind}:${node.target}`;
+        const existing = nodeDepth.get(key);
+        if (!existing || node.depth < existing.depth) {
+            nodeDepth.set(key, { ...node });
+        }
+    };
+    const recordEdge = (edge) => {
+        const key = `${edge.from}->${edge.to}:${edge.kind}:${edge.rel}`;
+        const existing = edgeSet.get(key);
+        if (!existing || edge.depth < existing.depth) {
+            edgeSet.set(key, { ...edge });
+        }
+    };
+    recordNode({ kind: "path", target: params.path, depth: 0 });
+    while (queue.length > 0) {
+        const current = queue.shift();
+        if (current.depth >= maxDepth) {
+            continue;
+        }
+        // direction に応じて使用するマップを選択
+        const edgeMap = direction === "inbound" ? inbound : outbound;
+        const edges = edgeMap.get(current.path) ?? [];
+        for (const edge of edges) {
+            const nextDepth = current.depth + 1;
+            if (direction === "inbound") {
+                // inbound: edge.src_path がこのファイルを使用している
+                recordEdge({
+                    from: edge.src_path,
+                    to: current.path,
+                    kind: "path",
+                    rel: edge.rel,
+                    depth: nextDepth,
+                });
+                recordNode({ kind: "path", target: edge.src_path, depth: nextDepth });
+                if (!visitedPaths.has(edge.src_path)) {
+                    visitedPaths.add(edge.src_path);
+                    queue.push({ path: edge.src_path, depth: nextDepth });
+                }
+            }
+            else {
+                // outbound: このファイルが edge.dst を使用している
+                if (edge.dst_kind === "path") {
+                    recordEdge({
+                        from: current.path,
+                        to: edge.dst,
+                        kind: "path",
+                        rel: edge.rel,
+                        depth: nextDepth,
+                    });
+                    recordNode({ kind: "path", target: edge.dst, depth: nextDepth });
+                    if (!visitedPaths.has(edge.dst)) {
+                        visitedPaths.add(edge.dst);
+                        queue.push({ path: edge.dst, depth: nextDepth });
+                    }
+                }
+                else if (edge.dst_kind === "package" && includePackages) {
+                    recordEdge({
+                        from: current.path,
+                        to: edge.dst,
+                        kind: "package",
+                        rel: edge.rel,
+                        depth: nextDepth,
+                    });
+                    recordNode({ kind: "package", target: edge.dst, depth: nextDepth });
+                }
+            }
+        }
+    }
+    const nodes = Array.from(nodeDepth.values()).sort((a, b) => {
+        if (a.depth === b.depth) {
+            return a.target.localeCompare(b.target);
+        }
+        return a.depth - b.depth;
+    });
+    const edges = Array.from(edgeSet.values()).sort((a, b) => {
+        if (a.depth === b.depth) {
+            const fromCmp = a.from.localeCompare(b.from);
+            if (fromCmp !== 0) {
+                return fromCmp;
+            }
+            return a.to.localeCompare(b.to);
+        }
+        return a.depth - b.depth;
+    });
+    return {
+        root: params.path,
+        direction,
+        nodes,
+        edges,
+    };
+}
+export async function resolveRepoId(db, repoRoot) {
+    try {
+        const rows = await db.all("SELECT id FROM repo WHERE root = ?", [repoRoot]);
+        if (rows.length === 0) {
+            throw new Error("Target repository is missing from DuckDB. Run the indexer before starting the server.");
+        }
+        const row = rows[0];
+        if (!row) {
+            throw new Error("Failed to retrieve repository record. Database returned empty result.");
+        }
+        return row.id;
+    }
+    catch (error) {
+        if (error instanceof Error && error.message.includes("Table with name repo")) {
+            throw new Error("Target repository is missing from DuckDB. Run the indexer before starting the server.");
+        }
+        throw error;
+    }
+}
+//# sourceMappingURL=handlers.js.map