claude-flow 2.0.0-alpha.61 → 2.0.0-alpha.62

package/src/cli/simple-commands/hive-mind/session-manager.js

@@ -15,10 +15,10 @@ export class HiveMindSessionManager {
     this.hiveMindDir = hiveMindDir || path.join(cwd(), '.hive-mind');
     this.sessionsDir = path.join(this.hiveMindDir, 'sessions');
     this.dbPath = path.join(this.hiveMindDir, 'hive.db');
-    
+
     // Ensure directories exist
     this.ensureDirectories();
-    
+
     // Initialize database connection
     this.db = new Database(this.dbPath);
     this.initializeSchema();
@@ -78,7 +78,7 @@ export class HiveMindSessionManager {
         FOREIGN KEY (session_id) REFERENCES sessions(id)
       );
     `);
-    
+
     // Run migrations to add new columns
     this.runMigrations();
   }
@@ -89,15 +89,15 @@ export class HiveMindSessionManager {
   runMigrations() {
     try {
       // Check if parent_pid column exists
-      const columns = this.db.prepare("PRAGMA table_info(sessions)").all();
-      const hasParentPid = columns.some(col => col.name === 'parent_pid');
-      const hasChildPids = columns.some(col => col.name === 'child_pids');
-      
+      const columns = this.db.prepare('PRAGMA table_info(sessions)').all();
+      const hasParentPid = columns.some((col) => col.name === 'parent_pid');
+      const hasChildPids = columns.some((col) => col.name === 'child_pids');
+
       if (!hasParentPid) {
         this.db.exec('ALTER TABLE sessions ADD COLUMN parent_pid INTEGER');
         console.log('Added parent_pid column to sessions table');
       }
-      
+
       if (!hasChildPids) {
         this.db.exec('ALTER TABLE sessions ADD COLUMN child_pids TEXT');
         console.log('Added child_pids column to sessions table');
@@ -112,22 +112,22 @@ export class HiveMindSessionManager {
    */
   createSession(swarmId, swarmName, objective, metadata = {}) {
     const sessionId = `session-${Date.now()}-${Math.random().toString(36).substring(2, 11)}`;
-    
+
     const stmt = this.db.prepare(`
       INSERT INTO sessions (id, swarm_id, swarm_name, objective, metadata, parent_pid)
       VALUES (?, ?, ?, ?, ?, ?)
     `);
-    
+
     stmt.run(sessionId, swarmId, swarmName, objective, JSON.stringify(metadata), process.pid);
-    
+
     // Log session creation
     this.logSessionEvent(sessionId, 'info', 'Session created', null, {
       swarmId,
       swarmName,
       objective,
-      parentPid: process.pid
+      parentPid: process.pid,
     });
-    
+
     return sessionId;
   }
 
@@ -136,38 +136,45 @@ export class HiveMindSessionManager {
    */
   async saveCheckpoint(sessionId, checkpointName, checkpointData) {
     const checkpointId = `checkpoint-${Date.now()}-${Math.random().toString(36).substring(2, 11)}`;
-    
+
     // Save to database
     const stmt = this.db.prepare(`
       INSERT INTO session_checkpoints (id, session_id, checkpoint_name, checkpoint_data)
       VALUES (?, ?, ?, ?)
     `);
-    
+
     stmt.run(checkpointId, sessionId, checkpointName, JSON.stringify(checkpointData));
-    
+
     // Update session checkpoint data and timestamp
     const updateStmt = this.db.prepare(`
       UPDATE sessions 
       SET checkpoint_data = ?, updated_at = CURRENT_TIMESTAMP
       WHERE id = ?
     `);
-    
+
     updateStmt.run(JSON.stringify(checkpointData), sessionId);
-    
+
     // Save checkpoint file for backup
     const checkpointFile = path.join(this.sessionsDir, `${sessionId}-${checkpointName}.json`);
-    await writeFile(checkpointFile, JSON.stringify({
-      sessionId,
-      checkpointId,
-      checkpointName,
-      timestamp: new Date().toISOString(),
-      data: checkpointData
-    }, null, 2));
-    
+    await writeFile(
+      checkpointFile,
+      JSON.stringify(
+        {
+          sessionId,
+          checkpointId,
+          checkpointName,
+          timestamp: new Date().toISOString(),
+          data: checkpointData,
+        },
+        null,
+        2,
+      ),
+    );
+
     this.logSessionEvent(sessionId, 'info', `Checkpoint saved: ${checkpointName}`, null, {
-      checkpointId
+      checkpointId,
     });
-    
+
     return checkpointId;
   }
 
@@ -187,17 +194,18 @@ export class HiveMindSessionManager {
       GROUP BY s.id
       ORDER BY s.updated_at DESC
     `);
-    
+
     const sessions = stmt.all();
-    
+
     // Parse JSON fields
-    return sessions.map(session => ({
+    return sessions.map((session) => ({
       ...session,
       metadata: session.metadata ? JSON.parse(session.metadata) : {},
       checkpoint_data: session.checkpoint_data ? JSON.parse(session.checkpoint_data) : null,
-      completion_percentage: session.task_count > 0 
-        ? Math.round((session.completed_tasks / session.task_count) * 100)
-        : 0
+      completion_percentage:
+        session.task_count > 0
+          ? Math.round((session.completed_tasks / session.task_count) * 100)
+          : 0,
     }));
   }
 
@@ -205,44 +213,68 @@ export class HiveMindSessionManager {
    * Get session by ID with full details
    */
   getSession(sessionId) {
-    const session = this.db.prepare(`
+    const session = this.db
+      .prepare(
+        `
       SELECT * FROM sessions WHERE id = ?
-    `).get(sessionId);
-    
+    `,
+      )
+      .get(sessionId);
+
     if (!session) {
       return null;
     }
-    
+
     // Get associated swarm data
-    const swarm = this.db.prepare(`
+    const swarm = this.db
+      .prepare(
+        `
       SELECT * FROM swarms WHERE id = ?
-    `).get(session.swarm_id);
-    
+    `,
+      )
+      .get(session.swarm_id);
+
     // Get agents
-    const agents = this.db.prepare(`
+    const agents = this.db
+      .prepare(
+        `
       SELECT * FROM agents WHERE swarm_id = ?
-    `).all(session.swarm_id);
-    
+    `,
+      )
+      .all(session.swarm_id);
+
     // Get tasks
-    const tasks = this.db.prepare(`
+    const tasks = this.db
+      .prepare(
+        `
       SELECT * FROM tasks WHERE swarm_id = ?
-    `).all(session.swarm_id);
-    
+    `,
+      )
+      .all(session.swarm_id);
+
     // Get checkpoints
-    const checkpoints = this.db.prepare(`
+    const checkpoints = this.db
+      .prepare(
+        `
       SELECT * FROM session_checkpoints 
       WHERE session_id = ? 
       ORDER BY created_at DESC
-    `).all(sessionId);
-    
+    `,
+      )
+      .all(sessionId);
+
     // Get recent logs
-    const recentLogs = this.db.prepare(`
+    const recentLogs = this.db
+      .prepare(
+        `
       SELECT * FROM session_logs 
       WHERE session_id = ? 
       ORDER BY timestamp DESC 
       LIMIT 50
-    `).all(sessionId);
-    
+    `,
+      )
+      .all(sessionId);
+
     return {
       ...session,
       metadata: session.metadata ? JSON.parse(session.metadata) : {},
@@ -250,22 +282,25 @@ export class HiveMindSessionManager {
       swarm,
       agents,
       tasks,
-      checkpoints: checkpoints.map(cp => ({
+      checkpoints: checkpoints.map((cp) => ({
         ...cp,
-        checkpoint_data: JSON.parse(cp.checkpoint_data)
+        checkpoint_data: JSON.parse(cp.checkpoint_data),
       })),
       recentLogs,
       statistics: {
         totalAgents: agents.length,
-        activeAgents: agents.filter(a => a.status === 'active' || a.status === 'busy').length,
+        activeAgents: agents.filter((a) => a.status === 'active' || a.status === 'busy').length,
         totalTasks: tasks.length,
-        completedTasks: tasks.filter(t => t.status === 'completed').length,
-        pendingTasks: tasks.filter(t => t.status === 'pending').length,
-        inProgressTasks: tasks.filter(t => t.status === 'in_progress').length,
-        completionPercentage: tasks.length > 0 
-          ? Math.round((tasks.filter(t => t.status === 'completed').length / tasks.length) * 100)
-          : 0
-      }
+        completedTasks: tasks.filter((t) => t.status === 'completed').length,
+        pendingTasks: tasks.filter((t) => t.status === 'pending').length,
+        inProgressTasks: tasks.filter((t) => t.status === 'in_progress').length,
+        completionPercentage:
+          tasks.length > 0
+            ? Math.round(
+                (tasks.filter((t) => t.status === 'completed').length / tasks.length) * 100,
+              )
+            : 0,
+      },
     };
   }
 
@@ -278,19 +313,21 @@ export class HiveMindSessionManager {
       SET status = 'paused', paused_at = CURRENT_TIMESTAMP, updated_at = CURRENT_TIMESTAMP
       WHERE id = ?
     `);
-    
+
     const result = stmt.run(sessionId);
-    
+
     if (result.changes > 0) {
       this.logSessionEvent(sessionId, 'info', 'Session paused');
-      
+
       // Update swarm status
       const session = this.db.prepare('SELECT swarm_id FROM sessions WHERE id = ?').get(sessionId);
       if (session) {
-        this.db.prepare('UPDATE swarms SET status = ? WHERE id = ?').run('paused', session.swarm_id);
+        this.db
+          .prepare('UPDATE swarms SET status = ? WHERE id = ?')
+          .run('paused', session.swarm_id);
       }
     }
-    
+
     return result.changes > 0;
   }
 
@@ -299,47 +336,53 @@ export class HiveMindSessionManager {
    */
   async resumeSession(sessionId) {
     const session = this.getSession(sessionId);
-    
+
     if (!session) {
       throw new Error(`Session ${sessionId} not found`);
     }
-    
+
     // Allow resuming any session regardless of status
     console.log(`Resuming session ${sessionId} from status: ${session.status}`);
-    
+
     // If session was stopped, log that we're restarting it
     if (session.status === 'stopped') {
-      this.logSessionEvent(sessionId, 'info', `Restarting stopped session with original configuration`);
+      this.logSessionEvent(
+        sessionId,
+        'info',
+        `Restarting stopped session with original configuration`,
+      );
     }
-    
+
     // Update session status
     const stmt = this.db.prepare(`
       UPDATE sessions 
       SET status = 'active', resumed_at = CURRENT_TIMESTAMP, updated_at = CURRENT_TIMESTAMP
       WHERE id = ?
     `);
-    
+
     stmt.run(sessionId);
-    
+
     // Update swarm status
     this.db.prepare('UPDATE swarms SET status = ? WHERE id = ?').run('active', session.swarm_id);
-    
+
     // Update agent statuses
-    this.db.prepare(`
+    this.db
+      .prepare(
+        `
       UPDATE agents 
       SET status = CASE 
         WHEN role = 'queen' THEN 'active'
         ELSE 'idle'
       END
       WHERE swarm_id = ?
-    `).run(session.swarm_id);
-    
+    `,
+      )
+      .run(session.swarm_id);
+
     this.logSessionEvent(sessionId, 'info', 'Session resumed', null, {
-      pausedDuration: session.paused_at 
-        ? new Date() - new Date(session.paused_at)
-        : null
+      pausedDuration: session.paused_at ? new Date() - new Date(session.paused_at) : null,
     });
-    
+
     return session;
   }
 
@@ -352,19 +395,21 @@ export class HiveMindSessionManager {
       SET status = 'completed', updated_at = CURRENT_TIMESTAMP, completion_percentage = 100
       WHERE id = ?
     `);
-    
+
     const result = stmt.run(sessionId);
-    
+
     if (result.changes > 0) {
       this.logSessionEvent(sessionId, 'info', 'Session completed');
-      
+
       // Update swarm status
       const session = this.db.prepare('SELECT swarm_id FROM sessions WHERE id = ?').get(sessionId);
       if (session) {
-        this.db.prepare('UPDATE swarms SET status = ? WHERE id = ?').run('completed', session.swarm_id);
+        this.db
+          .prepare('UPDATE swarms SET status = ? WHERE id = ?')
+          .run('completed', session.swarm_id);
       }
     }
-    
+
     return result.changes > 0;
   }
 
@@ -374,29 +419,33 @@ export class HiveMindSessionManager {
   async archiveSessions(daysOld = 30) {
     const cutoffDate = new Date();
     cutoffDate.setDate(cutoffDate.getDate() - daysOld);
-    
-    const sessionsToArchive = this.db.prepare(`
+
+    const sessionsToArchive = this.db
+      .prepare(
+        `
       SELECT * FROM sessions 
       WHERE status = 'completed' AND updated_at < ?
-    `).all(cutoffDate.toISOString());
-    
+    `,
+      )
+      .all(cutoffDate.toISOString());
+
     const archiveDir = path.join(this.sessionsDir, 'archive');
     if (!existsSync(archiveDir)) {
       mkdirSync(archiveDir, { recursive: true });
     }
-    
+
     for (const session of sessionsToArchive) {
       const sessionData = this.getSession(session.id);
       const archiveFile = path.join(archiveDir, `${session.id}-archive.json`);
-      
+
       await writeFile(archiveFile, JSON.stringify(sessionData, null, 2));
-      
+
       // Remove from database
       this.db.prepare('DELETE FROM session_logs WHERE session_id = ?').run(session.id);
       this.db.prepare('DELETE FROM session_checkpoints WHERE session_id = ?').run(session.id);
       this.db.prepare('DELETE FROM sessions WHERE id = ?').run(session.id);
     }
-    
+
     return sessionsToArchive.length;
   }
 
@@ -408,7 +457,7 @@ export class HiveMindSessionManager {
       INSERT INTO session_logs (session_id, log_level, message, agent_id, data)
       VALUES (?, ?, ?, ?, ?)
     `);
-    
+
     stmt.run(sessionId, logLevel, message, agentId, data ? JSON.stringify(data) : null);
   }
 
@@ -422,12 +471,12 @@ export class HiveMindSessionManager {
       ORDER BY timestamp DESC 
       LIMIT ? OFFSET ?
     `);
-    
+
     const logs = stmt.all(sessionId, limit, offset);
-    
-    return logs.map(log => ({
+
+    return logs.map((log) => ({
       ...log,
-      data: log.data ? JSON.parse(log.data) : null
+      data: log.data ? JSON.parse(log.data) : null,
     }));
   }
 
@@ -440,7 +489,7 @@ export class HiveMindSessionManager {
       SET completion_percentage = ?, updated_at = CURRENT_TIMESTAMP
       WHERE id = ?
     `);
-    
+
     stmt.run(completionPercentage, sessionId);
   }
 
@@ -449,32 +498,35 @@ export class HiveMindSessionManager {
    */
   generateSessionSummary(sessionId) {
     const session = this.getSession(sessionId);
-    
+
     if (!session) {
       return null;
     }
-    
-    const duration = session.paused_at && session.resumed_at
-      ? new Date(session.updated_at) - new Date(session.created_at) - (new Date(session.resumed_at) - new Date(session.paused_at))
-      : new Date(session.updated_at) - new Date(session.created_at);
-    
+
+    const duration =
+      session.paused_at && session.resumed_at
+        ? new Date(session.updated_at) -
+          new Date(session.created_at) -
+          (new Date(session.resumed_at) - new Date(session.paused_at))
+        : new Date(session.updated_at) - new Date(session.created_at);
+
     const tasksByType = session.agents.reduce((acc, agent) => {
-      const agentTasks = session.tasks.filter(t => t.agent_id === agent.id);
+      const agentTasks = session.tasks.filter((t) => t.agent_id === agent.id);
       if (!acc[agent.type]) {
         acc[agent.type] = {
           total: 0,
           completed: 0,
           inProgress: 0,
-          pending: 0
+          pending: 0,
         };
       }
       acc[agent.type].total += agentTasks.length;
-      acc[agent.type].completed += agentTasks.filter(t => t.status === 'completed').length;
-      acc[agent.type].inProgress += agentTasks.filter(t => t.status === 'in_progress').length;
-      acc[agent.type].pending += agentTasks.filter(t => t.status === 'pending').length;
+      acc[agent.type].completed += agentTasks.filter((t) => t.status === 'completed').length;
+      acc[agent.type].inProgress += agentTasks.filter((t) => t.status === 'in_progress').length;
+      acc[agent.type].pending += agentTasks.filter((t) => t.status === 'pending').length;
       return acc;
     }, {});
-    
+
     return {
       sessionId: session.id,
       swarmName: session.swarm_name,
@@ -489,8 +541,8 @@ export class HiveMindSessionManager {
         created: session.created_at,
         lastUpdated: session.updated_at,
         paused: session.paused_at,
-        resumed: session.resumed_at
-      }
+        resumed: session.resumed_at,
+      },
     };
   }
 
@@ -499,15 +551,15 @@ export class HiveMindSessionManager {
    */
   async exportSession(sessionId, exportPath = null) {
     const session = this.getSession(sessionId);
-    
+
     if (!session) {
       throw new Error(`Session ${sessionId} not found`);
     }
-    
+
     const exportFile = exportPath || path.join(this.sessionsDir, `${sessionId}-export.json`);
-    
+
     await writeFile(exportFile, JSON.stringify(session, null, 2));
-    
+
     return exportFile;
   }
 
@@ -516,20 +568,24 @@ export class HiveMindSessionManager {
    */
   async importSession(importPath) {
     const sessionData = JSON.parse(await readFile(importPath, 'utf8'));
-    
+
     // Create new session with imported data
     const newSessionId = this.createSession(
       sessionData.swarm_id,
       sessionData.swarm_name,
       sessionData.objective,
-      sessionData.metadata
+      sessionData.metadata,
     );
-    
+
     // Import checkpoints
     for (const checkpoint of sessionData.checkpoints || []) {
-      await this.saveCheckpoint(newSessionId, checkpoint.checkpoint_name, checkpoint.checkpoint_data);
+      await this.saveCheckpoint(
+        newSessionId,
+        checkpoint.checkpoint_name,
+        checkpoint.checkpoint_data,
+      );
     }
-    
+
     // Import logs
     for (const log of sessionData.recentLogs || []) {
       this.logSessionEvent(
@@ -537,10 +593,10 @@ export class HiveMindSessionManager {
         log.log_level,
         log.message,
         log.agent_id,
-        log.data ? JSON.parse(log.data) : null
+        log.data ? JSON.parse(log.data) : null,
       );
     }
-    
+
     return newSessionId;
   }
 
@@ -550,20 +606,20 @@ export class HiveMindSessionManager {
   addChildPid(sessionId, pid) {
     const session = this.db.prepare('SELECT child_pids FROM sessions WHERE id = ?').get(sessionId);
     if (!session) return false;
-    
+
     const childPids = session.child_pids ? JSON.parse(session.child_pids) : [];
     if (!childPids.includes(pid)) {
       childPids.push(pid);
     }
-    
+
     const stmt = this.db.prepare(`
       UPDATE sessions 
       SET child_pids = ?, updated_at = CURRENT_TIMESTAMP
       WHERE id = ?
     `);
-    
+
     stmt.run(JSON.stringify(childPids), sessionId);
-    
+
     this.logSessionEvent(sessionId, 'info', 'Child process added', null, { pid });
     return true;
   }
@@ -574,21 +630,21 @@ export class HiveMindSessionManager {
   removeChildPid(sessionId, pid) {
     const session = this.db.prepare('SELECT child_pids FROM sessions WHERE id = ?').get(sessionId);
     if (!session) return false;
-    
+
     const childPids = session.child_pids ? JSON.parse(session.child_pids) : [];
     const index = childPids.indexOf(pid);
     if (index > -1) {
       childPids.splice(index, 1);
     }
-    
+
     const stmt = this.db.prepare(`
       UPDATE sessions 
       SET child_pids = ?, updated_at = CURRENT_TIMESTAMP
       WHERE id = ?
     `);
-    
+
     stmt.run(JSON.stringify(childPids), sessionId);
-    
+
     this.logSessionEvent(sessionId, 'info', 'Child process removed', null, { pid });
     return true;
   }
@@ -599,7 +655,7 @@ export class HiveMindSessionManager {
   getChildPids(sessionId) {
     const session = this.db.prepare('SELECT child_pids FROM sessions WHERE id = ?').get(sessionId);
     if (!session || !session.child_pids) return [];
-    
+
     return JSON.parse(session.child_pids);
   }
 
@@ -611,10 +667,10 @@ export class HiveMindSessionManager {
     if (!session) {
       throw new Error(`Session ${sessionId} not found`);
     }
-    
+
     // Get child PIDs
     const childPids = this.getChildPids(sessionId);
-    
+
     // Terminate child processes
     for (const pid of childPids) {
       try {
@@ -622,27 +678,27 @@ export class HiveMindSessionManager {
         this.logSessionEvent(sessionId, 'info', 'Child process terminated', null, { pid });
       } catch (err) {
         // Process might already be dead
-        this.logSessionEvent(sessionId, 'warning', 'Failed to terminate child process', null, { 
-          pid, 
-          error: err.message 
+        this.logSessionEvent(sessionId, 'warning', 'Failed to terminate child process', null, {
+          pid,
+          error: err.message,
         });
       }
     }
-    
+
     // Update session status
     const stmt = this.db.prepare(`
       UPDATE sessions 
       SET status = 'stopped', updated_at = CURRENT_TIMESTAMP
       WHERE id = ?
     `);
-    
+
     stmt.run(sessionId);
-    
+
     // Update swarm status
     this.db.prepare('UPDATE swarms SET status = ? WHERE id = ?').run('stopped', session.swarm_id);
-    
+
     this.logSessionEvent(sessionId, 'info', 'Session stopped');
-    
+
     return true;
   }
 
@@ -651,12 +707,12 @@ export class HiveMindSessionManager {
    */
   getActiveSessionsWithProcessInfo() {
     const sessions = this.getActiveSessions();
-    
+
     // Add process info to each session
-    return sessions.map(session => {
+    return sessions.map((session) => {
       const childPids = session.child_pids ? JSON.parse(session.child_pids) : [];
       const aliveChildPids = [];
-      
+
       // Check which child processes are still alive
       for (const pid of childPids) {
         try {
@@ -666,12 +722,12 @@ export class HiveMindSessionManager {
           // Process is dead
         }
       }
-      
+
       return {
         ...session,
         parent_pid: session.parent_pid,
         child_pids: aliveChildPids,
-        total_processes: 1 + aliveChildPids.length
+        total_processes: 1 + aliveChildPids.length,
       };
     });
   }
@@ -680,13 +736,17 @@ export class HiveMindSessionManager {
    * Clean up orphaned processes
    */
   cleanupOrphanedProcesses() {
-    const sessions = this.db.prepare(`
+    const sessions = this.db
+      .prepare(
+        `
       SELECT * FROM sessions 
       WHERE status IN ('active', 'paused')
-    `).all();
-    
+    `,
+      )
+      .all();
+
     let cleanedCount = 0;
-    
+
     for (const session of sessions) {
       // Check if parent process is still alive
       try {
@@ -698,7 +758,7 @@ export class HiveMindSessionManager {
         this.logSessionEvent(session.id, 'info', 'Orphaned session cleaned up');
       }
     }
-    
+
     return cleanedCount;
   }
 
@@ -711,4 +771,4 @@ export class HiveMindSessionManager {
 }
 
 // Export for use in other modules
-export default HiveMindSessionManager;
+export default HiveMindSessionManager;