import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js"
import { logger } from "../utils/logger.js"
import {
reloadPrompts,
getPromptStats,
getAllPromptRuntimes,
} from "./loaders.js"
import { syncRepo } from "./git.js"
import { setActiveRepo, STORAGE_DIR } from "../config/env.js"
/**
* Handle reload prompts request
* @param server - MCP Server instance
* @returns MCP Tool response with reload results
*/
export async function handleReload(
server: McpServer
): Promise<{ content: Array<{ type: "text"; text: string }> }> {
try {
logger.info("mcp_reload_prompts tool invoked")
const result = await reloadPrompts(server, STORAGE_DIR)
const message = `Reload completed. Loaded: ${result.loaded}, Errors: ${result.errors.length}`
return {
content: [
{
type: "text" as const,
text: message,
},
],
}
} catch (error) {
const reloadError =
error instanceof Error ? error : new Error(String(error))
logger.error({ error: reloadError }, "Reload prompts failed")
return {
content: [
{
type: "text" as const,
text: `Error: ${reloadError.message}`,
},
],
}
}
}
/**
* Handle prompt statistics request
* @returns MCP Tool response with prompt statistics
*/
// eslint-disable-next-line @typescript-eslint/require-await
export async function handlePromptStats(): Promise<{
content: Array<{ type: "text"; text: string }>
}> {
try {
logger.info("mcp_prompt_stats tool invoked")
const stats = getPromptStats()
return {
content: [
{
type: "text" as const,
text: JSON.stringify(stats, null, 2),
},
],
}
} catch (error) {
const statsError =
error instanceof Error ? error : new Error(String(error))
logger.error({ error: statsError }, "Failed to get prompt stats")
return {
content: [
{
type: "text" as const,
text: `Error: ${statsError.message}`,
},
],
}
}
}
/**
* Handle prompt list request
* @returns MCP Tool response with all prompt runtimes
*/
// eslint-disable-next-line @typescript-eslint/require-await
export async function handlePromptList(): Promise<{
content: Array<{ type: "text"; text: string }>
}> {
try {
logger.info("mcp_prompt_list tool invoked")
const runtimes = getAllPromptRuntimes()
return {
content: [
{
type: "text" as const,
text: JSON.stringify(runtimes, null, 2),
},
],
}
} catch (error) {
const listError =
error instanceof Error ? error : new Error(String(error))
logger.error({ error: listError }, "Failed to list prompts")
return {
content: [
{
type: "text" as const,
text: `Error: ${listError.message}`,
},
],
}
}
}
/**
* Handle repository switch request
* @param server - MCP Server instance
* @param args - Repository switch arguments
* @param args.repo_url - Repository URL
* @param args.branch - Branch name (optional)
* @returns MCP Tool response with switch results
*/
export async function handleRepoSwitch(
server: McpServer,
args: { repo_url: string; branch?: string | undefined }
): Promise<{ content: Array<{ type: "text"; text: string }> }> {
try {
logger.info(
{ repo_url: args.repo_url, branch: args.branch },
"mcp_repo_switch tool invoked"
)
// Validate repo_url is string
if (typeof args.repo_url !== "string" || args.repo_url.length === 0) {
throw new Error("repo_url must be a non-empty string")
}
// Set dynamic repo
setActiveRepo(args.repo_url, args.branch)
// Sync repo
await syncRepo()
// Reload prompts
const result = await reloadPrompts(server, STORAGE_DIR)
const message = `Repo switched and reloaded successfully. Loaded: ${result.loaded}`
return {
content: [
{
type: "text" as const,
text: message,
},
],
}
} catch (error) {
const switchError =
error instanceof Error ? error : new Error(String(error))
logger.error({ error: switchError }, "Failed to switch repository")
return {
content: [
{
type: "text" as const,
text: `Error: ${switchError.message}`,
},
],
}
}
}
