Claude Infinite Context

Instructions

Implementation Reference

• src/core/ProjectBrain.ts:291-327 (handler)

  Core handler function that performs the rollback by retrieving checkpoint history from Redis and restoring a previous project state.

  async rollback(steps: number = 1): Promise<string> {
    const sessionId = this.ensureInitialized();
  
    try {
      logger.info('Rolling back checkpoint', { sessionId, steps });
  
      const history = await this.redis.getCheckpointHistory(sessionId);
  
      if (history.length === 0) {
        return 'No checkpoint history available to rollback.';
      }
  
      if (steps > history.length) {
        return `Only ${history.length} checkpoints available. Cannot rollback ${steps} steps.`;
      }
  
      // Get the target checkpoint (index is steps because history[0] is current)
      const targetCheckpoint = history[steps];
  
      // Restore the state
      await this.redis.updateStateWithLock(sessionId, async () => {
        return targetCheckpoint.state;
      });
  
      logger.info('Rollback completed', {
        targetVersion: targetCheckpoint.version,
        targetTimestamp: targetCheckpoint.timestamp,
      });
  
      return `Rolled back to version ${targetCheckpoint.version} (${new Date(
        targetCheckpoint.timestamp
      ).toLocaleString()})`;
    } catch (error) {
      logger.error('Rollback failed', { error, sessionId });
      throw new Error(`Rollback failed: ${error}`);
    }
  }

• src/types/schema.ts:100-104 (schema)

  Zod schema for validating the rollback tool input (number of steps to rollback).

  export const RollbackInputSchema = z.object({
    steps: z.number().int().positive().default(1),
  });
  
  export type CheckpointInput = z.infer<typeof CheckpointInputSchema>;

• src/index.ts:86-99 (registration)

  Tool registration in the listTools handler, defining name, description, and input schema for MCP discovery.

    name: 'rollback',
    description:
      'Revert to a previous checkpoint version. Useful if a merge produced incorrect results.',
    inputSchema: {
      type: 'object',
      properties: {
        steps: {
          type: 'number',
          description: 'Number of versions to roll back (default: 1)',
          default: 1,
        },
      },
    },
  },

• src/index.ts:144-150 (handler)

  MCP tool call dispatcher case that parses input schema and delegates to ProjectBrain.rollback.

  case 'rollback': {
    const input = RollbackInputSchema.parse(args || {});
    const result = await this.brain.rollback(input.steps);
    return {
      content: [{ type: 'text', text: result }],
    };
  }