@nvhien20vn/git-mcp-server
Version:
85 lines (84 loc) • 4.56 kB
JavaScript
// Import utils from barrel (ErrorHandler from ../utils/internal/errorHandler.js)
import { ErrorHandler } from '../../../utils/index.js';
// Import utils from barrel (logger from ../utils/internal/logger.js)
import { logger } from '../../../utils/index.js';
// Import utils from barrel (requestContextService, RequestContext from ../utils/internal/requestContext.js)
import { BaseErrorCode } from '../../../types-global/errors.js'; // Keep direct import for types-global
import { requestContextService } from '../../../utils/index.js';
import { GitPullInputSchema, pullGitChanges } from './logic.js';
let _getWorkingDirectory;
let _getSessionId;
/**
* Initializes the state accessors needed by the tool registration.
* This should be called once during server setup.
* @param getWdFn - Function to get the working directory for a session.
* @param getSidFn - Function to get the session ID from context.
*/
export function initializeGitPullStateAccessors(getWdFn, getSidFn) {
_getWorkingDirectory = getWdFn;
_getSessionId = getSidFn;
logger.info('State accessors initialized for git_pull tool registration.');
}
const TOOL_NAME = 'git_pull';
const TOOL_DESCRIPTION = "Fetches from and integrates with another repository or a local branch (e.g., 'git pull origin main'). Supports rebase and fast-forward only options. Returns the pull result as a JSON object.";
/**
* Registers the git_pull tool with the MCP server.
*
* @param {McpServer} server - The MCP server instance.
* @throws {Error} If state accessors are not initialized.
*/
export async function registerGitPullTool(server) {
if (!_getWorkingDirectory || !_getSessionId) {
throw new Error('State accessors for git_pull must be initialized before registration.');
}
const operation = 'registerGitPullTool';
const context = requestContextService.createRequestContext({ operation });
await ErrorHandler.tryCatch(async () => {
server.tool(TOOL_NAME, TOOL_DESCRIPTION, GitPullInputSchema.shape, // Provide the Zod schema shape
async (validatedArgs, callContext) => {
const toolOperation = 'tool:git_pull';
const requestContext = requestContextService.createRequestContext({ operation: toolOperation, parentContext: callContext });
const sessionId = _getSessionId(requestContext);
const getWorkingDirectoryForSession = () => {
return _getWorkingDirectory(sessionId);
};
const logicContext = {
...requestContext,
sessionId: sessionId,
getWorkingDirectory: getWorkingDirectoryForSession,
};
logger.info(`Executing tool: ${TOOL_NAME}`, logicContext);
return await ErrorHandler.tryCatch(async () => {
// Call the core logic function
const pullResult = await pullGitChanges(validatedArgs, logicContext);
// Format the result as a JSON string within TextContent
const resultContent = {
type: 'text',
text: JSON.stringify(pullResult, null, 2), // Pretty-print JSON
contentType: 'application/json',
};
// Log based on the success flag in the result
if (pullResult.success) {
logger.info(`Tool ${TOOL_NAME} executed successfully, returning JSON`, logicContext);
}
else {
// Log non-fatal conditions like conflicts or already up-to-date differently
const logMessage = pullResult.conflict
? `Tool ${TOOL_NAME} completed with merge conflicts, returning JSON`
: `Tool ${TOOL_NAME} completed with status: ${pullResult.message}, returning JSON`;
logger.info(logMessage, logicContext);
}
// Even if success is false (e.g., conflicts), it's not necessarily a tool execution *error*
// unless the logic threw an McpError. The success flag in the JSON indicates the Git outcome.
return { content: [resultContent] };
}, {
operation: toolOperation,
context: logicContext,
input: validatedArgs,
errorCode: BaseErrorCode.INTERNAL_ERROR, // Default if unexpected error in logic
});
});
logger.info(`Tool registered: ${TOOL_NAME}`, context);
}, { operation, context, critical: true });
}
;