Taskmaster

/** * @fileoverview SetStatusCommand using Commander's native class pattern * Extends Commander.Command for better integration with the framework */ import { type TaskStatus, type TmCore, createTmCore, TaskIdSchema } from '@tm/core'; import type { StorageType } from '@tm/core'; import boxen from 'boxen'; import chalk from 'chalk'; import { Command } from 'commander'; import { displayError } from '../utils/error-handler.js'; import { getProjectRoot } from '../utils/project-root.js'; /** * Valid task status values for validation */ const VALID_TASK_STATUSES: TaskStatus[] = [ 'pending', 'in-progress', 'done', 'deferred', 'cancelled', 'blocked', 'review' ]; /** * Options interface for the set-status command */ export interface SetStatusCommandOptions { id?: string; status?: TaskStatus; format?: 'text' | 'json'; silent?: boolean; project?: string; } /** * Result type from set-status command */ export interface SetStatusResult { success: boolean; updatedTasks: Array<{ taskId: string; oldStatus: TaskStatus; newStatus: TaskStatus; }>; storageType: Exclude<StorageType, 'auto'>; } /** * SetStatusCommand extending Commander's Command class * This is a thin presentation layer over @tm/core */ export class SetStatusCommand extends Command { private tmCore?: TmCore; private lastResult?: SetStatusResult; constructor(name?: string) { super(name || 'set-status'); // Configure the command with positional arguments this.description('Update the status of one or more tasks') .alias('status') .argument( '[id]', 'Task ID(s) - comma-separated for multiple (e.g., 1 or 1,1.1,2)' ) .argument('[status]', `Status - ${VALID_TASK_STATUSES.join(', ')}`) .option('-i, --id <id>', 'Task ID(s) (fallback if not using positional)') .option( '-s, --status <status>', 'Status (fallback if not using positional)' ) .option('-f, --format <format>', 'Output format (text, json)', 'text') .option('--silent', 'Suppress output (useful for programmatic usage)') .option( '-p, --project <path>', 'Project root directory (auto-detected if not provided)' ) .action( async ( idArg?: string, statusArg?: string, options?: SetStatusCommandOptions ) => { // Prioritize positional arguments over options const mergedOptions: SetStatusCommandOptions = { ...options, id: idArg || options?.id, status: (statusArg as TaskStatus) || options?.status }; await this.executeCommand(mergedOptions); } ); } /** * Execute the set-status command */ private async executeCommand( options: SetStatusCommandOptions ): Promise<void> { let hasError = false; try { // Validate required options if (!options.id) { console.error(chalk.red('Error: Task ID is required')); console.log( chalk.yellow( 'Usage examples:\n' + ' tm set-status 1 done\n' + ' tm set-status 1.2 in-progress\n' + ' tm set-status --id=1 --status=done' ) ); process.exit(1); } if (!options.status) { console.error(chalk.red('Error: Status is required')); console.log( chalk.yellow( 'Usage examples:\n' + ' tm set-status 1 done\n' + ' tm set-status 1,1.1 in-progress\n' + ' tm set-status --id=1 --status=done' ) ); process.exit(1); } // Validate status if (!VALID_TASK_STATUSES.includes(options.status)) { console.error( chalk.red( `Error: Invalid status "${options.status}". Valid options: ${VALID_TASK_STATUSES.join(', ')}` ) ); process.exit(1); } // Initialize TaskMaster core this.tmCore = await createTmCore({ projectPath: getProjectRoot(options.project) }); // Parse and validate task IDs (handle comma-separated values) const rawIds = options.id.split(',').map((id) => id.trim()); const taskIds: string[] = []; for (const rawId of rawIds) { const parseResult = TaskIdSchema.safeParse(rawId); if (!parseResult.success) { console.error( chalk.red(`Invalid task ID: ${rawId}`), chalk.gray(`- ${parseResult.error.issues[0]?.message}`) ); process.exit(1); } taskIds.push(parseResult.data); } // Update each task const updatedTasks: Array<{ taskId: string; oldStatus: TaskStatus; newStatus: TaskStatus; }> = []; for (const taskId of taskIds) { try { const result = await this.tmCore.tasks.updateStatus( taskId, options.status ); updatedTasks.push({ taskId: result.taskId, oldStatus: result.oldStatus, newStatus: result.newStatus }); } catch (error: any) { hasError = true; if (options.format === 'json') { const errorMessage = error?.getSanitizedDetails ? error.getSanitizedDetails().message : error instanceof Error ? error.message : String(error); console.log( JSON.stringify({ success: false, error: errorMessage, taskId, timestamp: new Date().toISOString() }) ); } else if (!options.silent) { // Show which task failed with context console.error(chalk.red(`\nFailed to update task ${taskId}:`)); displayError(error, { skipExit: true }); } // Don't exit here - let finally block clean up first break; } } // Store result for potential reuse this.lastResult = { success: true, updatedTasks, storageType: this.tmCore.tasks.getStorageType() }; // Display results this.displayResults(this.lastResult, options); } catch (error: any) { hasError = true; if (options.format === 'json') { const errorMessage = error instanceof Error ? error.message : 'Unknown error occurred'; console.log(JSON.stringify({ success: false, error: errorMessage })); } else if (!options.silent) { displayError(error, { skipExit: true }); } } finally { // Clean up resources if (this.tmCore) { } } // Exit after cleanup completes if (hasError) { process.exit(1); } } /** * Display results based on format */ private displayResults( result: SetStatusResult, options: SetStatusCommandOptions ): void { const format = options.format || 'text'; switch (format) { case 'json': console.log(JSON.stringify(result, null, 2)); break; case 'text': default: if (!options.silent) { this.displayTextResults(result); } break; } } /** * Display results in text format */ private displayTextResults(result: SetStatusResult): void { if (result.updatedTasks.length === 1) { // Single task update const update = result.updatedTasks[0]; console.log( boxen( chalk.white.bold(`✅ Successfully updated task ${update.taskId}`) + '\n\n' + `${chalk.blue('From:')} ${this.getStatusDisplay(update.oldStatus)}\n` + `${chalk.blue('To:')} ${this.getStatusDisplay(update.newStatus)}`, { padding: 1, borderColor: 'green', borderStyle: 'round', margin: { top: 1 } } ) ); } else { // Multiple task updates console.log( boxen( chalk.white.bold( `✅ Successfully updated ${result.updatedTasks.length} tasks` ) + '\n\n' + result.updatedTasks .map( (update) => `${chalk.cyan(update.taskId)}: ${this.getStatusDisplay(update.oldStatus)} → ${this.getStatusDisplay(update.newStatus)}` ) .join('\n'), { padding: 1, borderColor: 'green', borderStyle: 'round', margin: { top: 1 } } ) ); } } /** * Get colored status display */ private getStatusDisplay(status: TaskStatus): string { const statusColors: Record<TaskStatus, (text: string) => string> = { pending: chalk.yellow, 'in-progress': chalk.blue, done: chalk.green, deferred: chalk.gray, cancelled: chalk.red, blocked: chalk.red, review: chalk.magenta, completed: chalk.green }; const colorFn = statusColors[status] || chalk.white; return colorFn(status); } /** * Get the last command result (useful for testing or chaining) */ getLastResult(): SetStatusResult | undefined { return this.lastResult; } /** * Register this command on an existing program */ static register(program: Command, name?: string): SetStatusCommand { const setStatusCommand = new SetStatusCommand(name); program.addCommand(setStatusCommand); return setStatusCommand; } } /** * Factory function to create and configure the set-status command */ export function createSetStatusCommand(): SetStatusCommand { return new SetStatusCommand(); }