/**
* Code indexing tools registration
*/
import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import type { CodeIndexer } from "../code/indexer.js";
import * as schemas from "./schemas.js";
export interface CodeToolDependencies {
codeIndexer: CodeIndexer;
}
export function registerCodeTools(
server: McpServer,
deps: CodeToolDependencies,
): void {
const { codeIndexer } = deps;
// index_codebase
server.registerTool(
"index_codebase",
{
title: "Index Codebase",
description:
"Index a codebase for semantic code search. Automatically discovers files, chunks code intelligently using AST-aware parsing, and stores in vector database. Respects .gitignore and other ignore files.",
inputSchema: schemas.IndexCodebaseSchema,
},
async ({ path, forceReindex, extensions, ignorePatterns }) => {
const stats = await codeIndexer.indexCodebase(
path,
{ forceReindex, extensions, ignorePatterns },
(progress) => {
// Progress callback - could send progress updates via SSE in future
console.error(
`[${progress.phase}] ${progress.percentage}% - ${progress.message}`,
);
},
);
let statusMessage = `Indexed ${stats.filesIndexed}/${stats.filesScanned} files (${stats.chunksCreated} chunks) in ${(stats.durationMs / 1000).toFixed(1)}s`;
if (stats.status === "partial") {
statusMessage += `\n\nWarnings:\n${stats.errors?.join("\n")}`;
} else if (stats.status === "failed") {
statusMessage = `Indexing failed:\n${stats.errors?.join("\n")}`;
}
return {
content: [{ type: "text", text: statusMessage }],
isError: stats.status === "failed",
};
},
);
// search_code
server.registerTool(
"search_code",
{
title: "Search Code",
description:
"Search indexed codebase using natural language queries. Returns semantically relevant code chunks with file paths and line numbers.",
inputSchema: schemas.SearchCodeSchema,
},
async ({ path, query, limit, fileTypes, pathPattern }) => {
const results = await codeIndexer.searchCode(path, query, {
limit,
fileTypes,
pathPattern,
});
if (results.length === 0) {
return {
content: [
{ type: "text", text: `No results found for query: "${query}"` },
],
};
}
// Format results with file references
const formattedResults = results
.map(
(r, idx) =>
`\n--- Result ${idx + 1} (score: ${r.score.toFixed(3)}) ---\n` +
`File: ${r.filePath}:${r.startLine}-${r.endLine}\n` +
`Language: ${r.language}\n\n` +
`${r.content}\n`,
)
.join("\n");
return {
content: [
{
type: "text",
text: `Found ${results.length} result(s):\n${formattedResults}`,
},
],
};
},
);
// reindex_changes
server.registerTool(
"reindex_changes",
{
title: "Reindex Changes",
description:
"Incrementally re-index only changed files. Detects added, modified, and deleted files since last index. Requires previous indexing with index_codebase.",
inputSchema: schemas.ReindexChangesSchema,
},
async ({ path }) => {
const stats = await codeIndexer.reindexChanges(path, (progress) => {
console.error(
`[${progress.phase}] ${progress.percentage}% - ${progress.message}`,
);
});
let message = `Incremental re-index complete:\n`;
message += `- Files added: ${stats.filesAdded}\n`;
message += `- Files modified: ${stats.filesModified}\n`;
message += `- Files deleted: ${stats.filesDeleted}\n`;
message += `- Chunks added: ${stats.chunksAdded}\n`;
message += `- Duration: ${(stats.durationMs / 1000).toFixed(1)}s`;
if (
stats.filesAdded === 0 &&
stats.filesModified === 0 &&
stats.filesDeleted === 0
) {
message = `No changes detected. Codebase is up to date.`;
}
return {
content: [{ type: "text", text: message }],
};
},
);
// get_index_status
server.registerTool(
"get_index_status",
{
title: "Get Index Status",
description: "Get indexing status and statistics for a codebase.",
inputSchema: schemas.GetIndexStatusSchema,
},
async ({ path }) => {
const status = await codeIndexer.getIndexStatus(path);
if (status.status === "not_indexed") {
return {
content: [
{
type: "text",
text: `Codebase at "${path}" is not indexed. Use index_codebase to index it first.`,
},
],
};
}
if (status.status === "indexing") {
return {
content: [
{
type: "text",
text: `Codebase at "${path}" is currently being indexed. ${status.chunksCount || 0} chunks processed so far.`,
},
],
};
}
return {
content: [{ type: "text", text: JSON.stringify(status, null, 2) }],
};
},
);
// clear_index
server.registerTool(
"clear_index",
{
title: "Clear Index",
description:
"Delete all indexed data for a codebase. This is irreversible and will remove the entire collection.",
inputSchema: schemas.ClearIndexSchema,
},
async ({ path }) => {
await codeIndexer.clearIndex(path);
return {
content: [
{ type: "text", text: `Index cleared for codebase at "${path}".` },
],
};
},
);
}
