117 lines
3.1 KiB
TypeScript
117 lines
3.1 KiB
TypeScript
/**
|
|
* @fileoverview autopilot-finalize MCP tool
|
|
* Finalize and complete the workflow with working tree validation
|
|
*/
|
|
|
|
// TEMPORARY: Using zod/v3 for Draft-07 JSON Schema compatibility with FastMCP's zod-to-json-schema
|
|
// TODO: Revert to 'zod' when MCP spec issue is resolved (see PR #1323)
|
|
import { z } from 'zod/v3';
|
|
import {
|
|
handleApiResult,
|
|
withNormalizedProjectRoot
|
|
} from '../../shared/utils.js';
|
|
import type { MCPContext } from '../../shared/types.js';
|
|
import { WorkflowService } from '@tm/core';
|
|
import type { FastMCP } from 'fastmcp';
|
|
|
|
const FinalizeSchema = z.object({
|
|
projectRoot: z
|
|
.string()
|
|
.describe('Absolute path to the project root directory')
|
|
});
|
|
|
|
type FinalizeArgs = z.infer<typeof FinalizeSchema>;
|
|
|
|
/**
|
|
* Register the autopilot_finalize tool with the MCP server
|
|
*/
|
|
export function registerAutopilotFinalizeTool(server: FastMCP) {
|
|
server.addTool({
|
|
name: 'autopilot_finalize',
|
|
description:
|
|
'Finalize and complete the workflow. Validates that all changes are committed and working tree is clean before marking workflow as complete.',
|
|
parameters: FinalizeSchema,
|
|
execute: withNormalizedProjectRoot(
|
|
async (args: FinalizeArgs, context: MCPContext) => {
|
|
const { projectRoot } = args;
|
|
|
|
try {
|
|
context.log.info(`Finalizing workflow in ${projectRoot}`);
|
|
|
|
const workflowService = new WorkflowService(projectRoot);
|
|
|
|
// Check if workflow exists
|
|
if (!(await workflowService.hasWorkflow())) {
|
|
return handleApiResult({
|
|
result: {
|
|
success: false,
|
|
error: {
|
|
message:
|
|
'No active workflow found. Start a workflow with autopilot_start'
|
|
}
|
|
},
|
|
log: context.log,
|
|
projectRoot
|
|
});
|
|
}
|
|
|
|
// Resume workflow
|
|
await workflowService.resumeWorkflow();
|
|
const currentStatus = workflowService.getStatus();
|
|
|
|
// Verify we're in FINALIZE phase
|
|
if (currentStatus.phase !== 'FINALIZE') {
|
|
return handleApiResult({
|
|
result: {
|
|
success: false,
|
|
error: {
|
|
message: `Cannot finalize: workflow is in ${currentStatus.phase} phase. Complete all subtasks first.`
|
|
}
|
|
},
|
|
log: context.log,
|
|
projectRoot
|
|
});
|
|
}
|
|
|
|
// Finalize workflow (validates clean working tree)
|
|
const newStatus = await workflowService.finalizeWorkflow();
|
|
|
|
context.log.info('Workflow finalized successfully');
|
|
|
|
// Get next action
|
|
const nextAction = workflowService.getNextAction();
|
|
|
|
return handleApiResult({
|
|
result: {
|
|
success: true,
|
|
data: {
|
|
message: 'Workflow completed successfully',
|
|
...newStatus,
|
|
nextAction: nextAction.action,
|
|
nextSteps: nextAction.nextSteps
|
|
}
|
|
},
|
|
log: context.log,
|
|
projectRoot
|
|
});
|
|
} catch (error: any) {
|
|
context.log.error(`Error in autopilot-finalize: ${error.message}`);
|
|
if (error.stack) {
|
|
context.log.debug(error.stack);
|
|
}
|
|
return handleApiResult({
|
|
result: {
|
|
success: false,
|
|
error: {
|
|
message: `Failed to finalize workflow: ${error.message}`
|
|
}
|
|
},
|
|
log: context.log,
|
|
projectRoot
|
|
});
|
|
}
|
|
}
|
|
)
|
|
});
|
|
}
|