register tool with mcp server

This commit is contained in:
Joe Danziger
2025-05-09 11:31:46 -04:00
parent 746fa90212
commit 21fcf92e0f
3 changed files with 96 additions and 0 deletions

View File

@@ -28,6 +28,7 @@ import { registerAddDependencyTool } from './add-dependency.js';
import { registerRemoveTaskTool } from './remove-task.js';
import { registerInitializeProjectTool } from './initialize-project.js';
import { registerModelsTool } from './models.js';
import { registerRulesTool } from './rules.js';
/**
* Register all Task Master tools with the MCP server
@@ -40,6 +41,7 @@ export function registerTaskMasterTools(server) {
// Group 1: Initialization & Setup
registerInitializeProjectTool(server);
registerModelsTool(server);
registerRulesTool(server);
registerParsePRDTool(server);
// Group 2: Task Listing & Viewing

View File

@@ -0,0 +1,39 @@
/**
* tools/rules.js
* Tool to add or remove brand rules from a project (MCP server)
*/
import { z } from 'zod';
import {
createErrorResponse,
handleApiResult,
withNormalizedProjectRoot
} from './utils.js';
import { rulesDirect } from '../core/direct-functions/rules.js';
/**
* Register the rules tool with the MCP server
* @param {Object} server - FastMCP server instance
*/
export function registerRulesTool(server) {
server.addTool({
name: 'rules',
description: 'Add or remove brand rules and MCP config from the project (mirrors CLI rules add/remove).',
parameters: z.object({
action: z.enum(['add', 'remove']).describe('Whether to add or remove brand rules.'),
rules: z.array(z.string()).min(1).describe('List of brand rules to add or remove (e.g., ["roo", "windsurf"]).'),
projectRoot: z.string().describe('The root directory of the project. Must be an absolute path.'),
yes: z.boolean().optional().default(true).describe('Run non-interactively (default: true).')
}),
execute: withNormalizedProjectRoot(async (args, { log, session }) => {
try {
log.info(`[rules tool] Executing action: ${args.action} for rules: ${args.rules.join(', ')} in ${args.projectRoot}`);
const result = await rulesDirect(args, log, { session });
return handleApiResult(result, log);
} catch (error) {
log.error(`[rules tool] Error: ${error.message}`);
return createErrorResponse(error.message, { details: error.stack });
}
})
});
}