claude-task-master/apps/mcp/src/tools/autopilot/next.tool.ts

/**
 * @fileoverview autopilot-next MCP tool
 * Get the next action to perform in the TDD workflow
 */

import type { FastMCP } from 'fastmcp';
import { z } from 'zod';
import type { ToolContext } from '../../shared/types.js';
import { handleApiResult, withToolContext } from '../../shared/utils.js';

const NextActionSchema = z.object({
	projectRoot: z
		.string()
		.describe('Absolute path to the project root directory')
});

type NextActionArgs = z.infer<typeof NextActionSchema>;

/**
 * Register the autopilot_next tool with the MCP server
 */
export function registerAutopilotNextTool(server: FastMCP) {
	server.addTool({
		name: 'autopilot_next',
		description:
			'Get the next action to perform in the TDD workflow. Returns detailed context about what needs to be done next, including the current phase, subtask, and expected actions.',
		parameters: NextActionSchema,
		annotations: {
			title: 'Get Next Autopilot Action',
			readOnlyHint: true
		},
		execute: withToolContext(
			'autopilot-next',
			async (args: NextActionArgs, { log, tmCore }: ToolContext) => {
				const { projectRoot } = args;

				try {
					log.info(`Getting next action for workflow in ${projectRoot}`);

					// Check if workflow exists
					if (!(await tmCore.workflow.hasWorkflow())) {
						return handleApiResult({
							result: {
								success: false,
								error: {
									message:
										'No active workflow found. Start a workflow with autopilot_start'
								}
							},
							log,
							projectRoot
						});
					}

					// Resume to load state
					await tmCore.workflow.resume();

					// Get next action
					const nextAction = tmCore.workflow.getNextAction();
					const status = tmCore.workflow.getStatus();

					log.info(`Next action determined: ${nextAction.action}`);

					return handleApiResult({
						result: {
							success: true,
							data: {
								action: nextAction.action,
								actionDescription: nextAction.description,
								...status,
								nextSteps: nextAction.nextSteps
							}
						},
						log,
						projectRoot
					});
				} catch (error: any) {
					log.error(`Error in autopilot-next: ${error.message}`);
					if (error.stack) {
						log.debug(error.stack);
					}
					return handleApiResult({
						result: {
							success: false,
							error: {
								message: `Failed to get next action: ${error.message}`
							}
						},
						log,
						projectRoot
					});
				}
			}
		)
	});
}