oh-my-claude-sisyphus 3.5.7 → 3.6.0

package/dist/hooks/swarm/claiming.js

@@ -0,0 +1,460 @@
+/**
+ * Swarm Task Claiming
+ *
+ * Atomic task claiming with lease-based ownership.
+ * Uses SQLite transactions to ensure only one agent can claim a task.
+ * Implements 5-minute lease timeout and heartbeat monitoring.
+ */
+import { DEFAULT_SWARM_CONFIG } from './types.js';
+import { getDb, recordHeartbeat } from './state.js';
+/**
+ * Atomically claim the next available task
+ *
+ * Uses a transaction to ensure only one agent can claim a task.
+ * Claims are lease-based with a 5-minute timeout.
+ *
+ * @param agentId - Unique identifier for the claiming agent
+ * @returns ClaimResult indicating success/failure and task details
+ */
+export function claimTask(agentId) {
+    const db = getDb();
+    if (!db) {
+        return {
+            success: false,
+            taskId: null,
+            reason: 'Database not initialized'
+        };
+    }
+    try {
+        // Use a transaction for atomic claim
+        const claimTransaction = db.transaction(() => {
+            // Find the first pending task
+            const findStmt = db.prepare(`
+        SELECT id, description FROM tasks
+        WHERE status = 'pending'
+        ORDER BY id
+        LIMIT 1
+      `);
+            const task = findStmt.get();
+            if (!task) {
+                return {
+                    success: false,
+                    taskId: null,
+                    reason: 'No pending tasks available'
+                };
+            }
+            // Claim the task
+            const claimStmt = db.prepare(`
+        UPDATE tasks
+        SET status = 'claimed', claimed_by = ?, claimed_at = ?
+        WHERE id = ? AND status = 'pending'
+      `);
+            const result = claimStmt.run(agentId, Date.now(), task.id);
+            if (result.changes === 0) {
+                // Task was claimed by another agent between SELECT and UPDATE
+                return {
+                    success: false,
+                    taskId: null,
+                    reason: 'Task was claimed by another agent'
+                };
+            }
+            // Update heartbeat
+            const heartbeatStmt = db.prepare(`
+        INSERT OR REPLACE INTO heartbeats (agent_id, last_heartbeat, current_task_id)
+        VALUES (?, ?, ?)
+      `);
+            heartbeatStmt.run(agentId, Date.now(), task.id);
+            return {
+                success: true,
+                taskId: task.id,
+                description: task.description
+            };
+        });
+        return claimTransaction.immediate();
+    }
+    catch (error) {
+        return {
+            success: false,
+            taskId: null,
+            reason: `Claim failed: ${error instanceof Error ? error.message : String(error)}`
+        };
+    }
+}
+/**
+ * Release a claimed task back to the pending pool
+ *
+ * Used when an agent fails to complete a task or wants to give it up.
+ *
+ * @param agentId - Agent releasing the task
+ * @param taskId - Task to release
+ * @returns true if release was successful
+ */
+export function releaseTask(agentId, taskId) {
+    const db = getDb();
+    if (!db)
+        return false;
+    try {
+        const releaseTransaction = db.transaction(() => {
+            // Verify the agent owns this task
+            const verifyStmt = db.prepare(`
+        SELECT claimed_by FROM tasks WHERE id = ?
+      `);
+            const task = verifyStmt.get(taskId);
+            if (!task || task.claimed_by !== agentId) {
+                return false;
+            }
+            // Release the task
+            const releaseStmt = db.prepare(`
+        UPDATE tasks
+        SET status = 'pending', claimed_by = NULL, claimed_at = NULL
+        WHERE id = ? AND claimed_by = ?
+      `);
+            releaseStmt.run(taskId, agentId);
+            // Update heartbeat to show no current task
+            const heartbeatStmt = db.prepare(`
+        UPDATE heartbeats SET current_task_id = NULL WHERE agent_id = ?
+      `);
+            heartbeatStmt.run(agentId);
+            return true;
+        });
+        return releaseTransaction.immediate();
+    }
+    catch (error) {
+        console.error('Failed to release task:', error);
+        return false;
+    }
+}
+/**
+ * Mark a task as completed
+ *
+ * @param agentId - Agent that completed the task
+ * @param taskId - Completed task ID
+ * @param result - Optional result/output from the task
+ * @returns true if completion was recorded
+ */
+export function completeTask(agentId, taskId, result) {
+    const db = getDb();
+    if (!db)
+        return false;
+    try {
+        const completeTransaction = db.transaction(() => {
+            // Verify the agent owns this task
+            const verifyStmt = db.prepare(`
+        SELECT claimed_by FROM tasks WHERE id = ?
+      `);
+            const task = verifyStmt.get(taskId);
+            if (!task || task.claimed_by !== agentId) {
+                return false;
+            }
+            // Mark task as done
+            const completeStmt = db.prepare(`
+        UPDATE tasks
+        SET status = 'done', completed_at = ?, result = ?
+        WHERE id = ? AND claimed_by = ?
+      `);
+            completeStmt.run(Date.now(), result ?? null, taskId, agentId);
+            // Update heartbeat to show no current task
+            const heartbeatStmt = db.prepare(`
+        UPDATE heartbeats SET current_task_id = NULL WHERE agent_id = ?
+      `);
+            heartbeatStmt.run(agentId);
+            return true;
+        });
+        return completeTransaction.immediate();
+    }
+    catch (error) {
+        console.error('Failed to complete task:', error);
+        return false;
+    }
+}
+/**
+ * Mark a task as failed
+ *
+ * @param agentId - Agent that failed the task
+ * @param taskId - Failed task ID
+ * @param error - Error message/reason for failure
+ * @returns true if failure was recorded
+ */
+export function failTask(agentId, taskId, error) {
+    const db = getDb();
+    if (!db)
+        return false;
+    try {
+        const failTransaction = db.transaction(() => {
+            // Verify the agent owns this task
+            const verifyStmt = db.prepare(`
+        SELECT claimed_by FROM tasks WHERE id = ?
+      `);
+            const task = verifyStmt.get(taskId);
+            if (!task || task.claimed_by !== agentId) {
+                return false;
+            }
+            // Mark task as failed
+            const failStmt = db.prepare(`
+        UPDATE tasks
+        SET status = 'failed', completed_at = ?, error = ?
+        WHERE id = ? AND claimed_by = ?
+      `);
+            failStmt.run(Date.now(), error, taskId, agentId);
+            // Update heartbeat to show no current task
+            const heartbeatStmt = db.prepare(`
+        UPDATE heartbeats SET current_task_id = NULL WHERE agent_id = ?
+      `);
+            heartbeatStmt.run(agentId);
+            return true;
+        });
+        return failTransaction.immediate();
+    }
+    catch (error) {
+        console.error('Failed to fail task:', error);
+        return false;
+    }
+}
+/**
+ * Record agent heartbeat
+ *
+ * Agents should call this every 60 seconds to indicate they're still alive.
+ * Used by cleanupStaleClaims to detect dead agents.
+ *
+ * @param agentId - Agent sending heartbeat
+ * @returns true if heartbeat was recorded
+ */
+export function heartbeat(agentId) {
+    const db = getDb();
+    if (!db)
+        return false;
+    try {
+        // Get current task for this agent
+        const taskStmt = db.prepare(`
+      SELECT id FROM tasks WHERE claimed_by = ? AND status = 'claimed'
+    `);
+        const task = taskStmt.get(agentId);
+        return recordHeartbeat(agentId, task?.id ?? null);
+    }
+    catch (error) {
+        console.error('Failed to record heartbeat:', error);
+        return false;
+    }
+}
+/**
+ * Clean up stale claims from dead agents
+ *
+ * Releases tasks that have been claimed for longer than the lease timeout
+ * (default 5 minutes) by agents that haven't sent a heartbeat.
+ *
+ * @param leaseTimeout - Lease timeout in milliseconds (default: 5 minutes)
+ * @returns Number of tasks released
+ */
+export function cleanupStaleClaims(leaseTimeout = DEFAULT_SWARM_CONFIG.leaseTimeout) {
+    const db = getDb();
+    if (!db)
+        return 0;
+    try {
+        const cutoffTime = Date.now() - leaseTimeout;
+        const cleanupTransaction = db.transaction(() => {
+            // Find tasks claimed longer than the timeout
+            // Only release if the agent hasn't sent a heartbeat recently
+            const findStaleStmt = db.prepare(`
+        SELECT t.id, t.claimed_by
+        FROM tasks t
+        LEFT JOIN heartbeats h ON t.claimed_by = h.agent_id
+        WHERE t.status = 'claimed'
+          AND t.claimed_at < ?
+          AND (h.last_heartbeat IS NULL OR h.last_heartbeat < ?)
+      `);
+            const staleTasks = findStaleStmt.all(cutoffTime, cutoffTime);
+            if (staleTasks.length === 0) {
+                return 0;
+            }
+            // Release stale tasks
+            const releaseStmt = db.prepare(`
+        UPDATE tasks
+        SET status = 'pending', claimed_by = NULL, claimed_at = NULL
+        WHERE id = ?
+      `);
+            // Remove stale heartbeats
+            const removeHeartbeatStmt = db.prepare(`
+        DELETE FROM heartbeats WHERE agent_id = ?
+      `);
+            const staleAgents = new Set();
+            for (const task of staleTasks) {
+                releaseStmt.run(task.id);
+                if (task.claimed_by) {
+                    staleAgents.add(task.claimed_by);
+                }
+            }
+            for (const agentId of staleAgents) {
+                removeHeartbeatStmt.run(agentId);
+            }
+            return staleTasks.length;
+        });
+        return cleanupTransaction.immediate();
+    }
+    catch (error) {
+        console.error('Failed to cleanup stale claims:', error);
+        return 0;
+    }
+}
+/**
+ * Get tasks claimed by a specific agent
+ *
+ * @param agentId - Agent to query
+ * @returns Array of tasks claimed by the agent
+ */
+export function getTasksClaimedBy(agentId) {
+    const db = getDb();
+    if (!db)
+        return [];
+    try {
+        const stmt = db.prepare(`
+      SELECT * FROM tasks WHERE claimed_by = ? AND status = 'claimed'
+    `);
+        const rows = stmt.all(agentId);
+        return rows.map(row => ({
+            id: row.id,
+            description: row.description,
+            status: row.status,
+            claimedBy: row.claimed_by,
+            claimedAt: row.claimed_at,
+            completedAt: row.completed_at,
+            error: row.error ?? undefined,
+            result: row.result ?? undefined
+        }));
+    }
+    catch (error) {
+        console.error('Failed to get tasks claimed by agent:', error);
+        return [];
+    }
+}
+/**
+ * Check if there are any pending tasks
+ *
+ * @returns true if there are pending tasks
+ */
+export function hasPendingTasks() {
+    const db = getDb();
+    if (!db)
+        return false;
+    try {
+        const stmt = db.prepare(`
+      SELECT COUNT(*) as count FROM tasks WHERE status = 'pending'
+    `);
+        const result = stmt.get();
+        return result.count > 0;
+    }
+    catch (error) {
+        console.error('Failed to check pending tasks:', error);
+        return false;
+    }
+}
+/**
+ * Check if all tasks are complete (done or failed)
+ *
+ * @returns true if all tasks are complete
+ */
+export function allTasksComplete() {
+    const db = getDb();
+    if (!db)
+        return false;
+    try {
+        const stmt = db.prepare(`
+      SELECT COUNT(*) as count FROM tasks WHERE status IN ('pending', 'claimed')
+    `);
+        const result = stmt.get();
+        return result.count === 0;
+    }
+    catch (error) {
+        console.error('Failed to check if all tasks complete:', error);
+        return false;
+    }
+}
+/**
+ * Get the number of active agents (with recent heartbeats)
+ *
+ * @param heartbeatTimeout - How old a heartbeat can be to still be considered active
+ * @returns Number of active agents
+ */
+export function getActiveAgentCount(heartbeatTimeout = DEFAULT_SWARM_CONFIG.leaseTimeout) {
+    const db = getDb();
+    if (!db)
+        return 0;
+    try {
+        const cutoffTime = Date.now() - heartbeatTimeout;
+        const stmt = db.prepare(`
+      SELECT COUNT(*) as count FROM heartbeats WHERE last_heartbeat > ?
+    `);
+        const result = stmt.get(cutoffTime);
+        return result.count;
+    }
+    catch (error) {
+        console.error('Failed to get active agent count:', error);
+        return 0;
+    }
+}
+/**
+ * Attempt to reclaim a failed task
+ *
+ * @param agentId - Agent attempting to reclaim
+ * @param taskId - Task to reclaim
+ * @returns ClaimResult
+ */
+export function reclaimFailedTask(agentId, taskId) {
+    const db = getDb();
+    if (!db) {
+        return {
+            success: false,
+            taskId: null,
+            reason: 'Database not initialized'
+        };
+    }
+    try {
+        const reclaimTransaction = db.transaction(() => {
+            // Check if task is failed
+            const checkStmt = db.prepare(`
+        SELECT id, description, status FROM tasks WHERE id = ?
+      `);
+            const task = checkStmt.get(taskId);
+            if (!task) {
+                return {
+                    success: false,
+                    taskId: null,
+                    reason: 'Task not found'
+                };
+            }
+            if (task.status !== 'failed') {
+                return {
+                    success: false,
+                    taskId: null,
+                    reason: `Task is ${task.status}, not failed`
+                };
+            }
+            // Reclaim the task
+            const reclaimStmt = db.prepare(`
+        UPDATE tasks
+        SET status = 'claimed', claimed_by = ?, claimed_at = ?, error = NULL
+        WHERE id = ?
+      `);
+            reclaimStmt.run(agentId, Date.now(), taskId);
+            // Update heartbeat
+            const heartbeatStmt = db.prepare(`
+        INSERT OR REPLACE INTO heartbeats (agent_id, last_heartbeat, current_task_id)
+        VALUES (?, ?, ?)
+      `);
+            heartbeatStmt.run(agentId, Date.now(), taskId);
+            return {
+                success: true,
+                taskId: task.id,
+                description: task.description
+            };
+        });
+        return reclaimTransaction.immediate();
+    }
+    catch (error) {
+        return {
+            success: false,
+            taskId: null,
+            reason: `Reclaim failed: ${error instanceof Error ? error.message : String(error)}`
+        };
+    }
+}
+//# sourceMappingURL=claiming.js.map

package/dist/hooks/swarm/claiming.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"claiming.js","sourceRoot":"","sources":["../../../src/hooks/swarm/claiming.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAGH,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAClD,OAAO,EACL,KAAK,EAIL,eAAe,EAIhB,MAAM,YAAY,CAAC;AAEpB;;;;;;;;GAQG;AACH,MAAM,UAAU,SAAS,CAAC,OAAe;IACvC,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,IAAI,CAAC,EAAE,EAAE,CAAC;QACR,OAAO;YACL,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,IAAI;YACZ,MAAM,EAAE,0BAA0B;SACnC,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,qCAAqC;QACrC,MAAM,gBAAgB,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;YAC3C,8BAA8B;YAC9B,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CAAC;;;;;OAK3B,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,EAAqD,CAAC;YAE/E,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,IAAI;oBACZ,MAAM,EAAE,4BAA4B;iBACtB,CAAC;YACnB,CAAC;YAED,iBAAiB;YACjB,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,CAAC;;;;OAI5B,CAAC,CAAC;YACH,MAAM,MAAM,GAAG,SAAS,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;YAE3D,IAAI,MAAM,CAAC,OAAO,KAAK,CAAC,EAAE,CAAC;gBACzB,8DAA8D;gBAC9D,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,IAAI;oBACZ,MAAM,EAAE,mCAAmC;iBAC7B,CAAC;YACnB,CAAC;YAED,mBAAmB;YACnB,MAAM,aAAa,GAAG,EAAE,CAAC,OAAO,CAAC;;;OAGhC,CAAC,CAAC;YACH,aAAa,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;YAEhD,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,IAAI,CAAC,EAAE;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;aACf,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,OAAO,gBAAgB,CAAC,SAAS,EAAE,CAAC;IACtC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,IAAI;YACZ,MAAM,EAAE,iBAAiB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;SAClF,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,WAAW,CAAC,OAAe,EAAE,MAAc;IACzD,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,IAAI,CAAC,EAAE;QAAE,OAAO,KAAK,CAAC;IAEtB,IAAI,CAAC;QACH,MAAM,kBAAkB,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;YAC7C,kCAAkC;YAClC,MAAM,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC;;OAE7B,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC,MAAM,CAA8C,CAAC;YAEjF,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;gBACzC,OAAO,KAAK,CAAC;YACf,CAAC;YAED,mBAAmB;YACnB,MAAM,WAAW,GAAG,EAAE,CAAC,OAAO,CAAC;;;;OAI9B,CAAC,CAAC;YACH,WAAW,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAEjC,2CAA2C;YAC3C,MAAM,aAAa,GAAG,EAAE,CAAC,OAAO,CAAC;;OAEhC,CAAC,CAAC;YACH,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAE3B,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,OAAO,kBAAkB,CAAC,SAAS,EAAE,CAAC;IACxC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;QAChD,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,YAAY,CAAC,OAAe,EAAE,MAAc,EAAE,MAAe;IAC3E,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,IAAI,CAAC,EAAE;QAAE,OAAO,KAAK,CAAC;IAEtB,IAAI,CAAC;QACH,MAAM,mBAAmB,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;YAC9C,kCAAkC;YAClC,MAAM,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC;;OAE7B,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC,MAAM,CAA8C,CAAC;YAEjF,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;gBACzC,OAAO,KAAK,CAAC;YACf,CAAC;YAED,oBAAoB;YACpB,MAAM,YAAY,GAAG,EAAE,CAAC,OAAO,CAAC;;;;OAI/B,CAAC,CAAC;YACH,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,MAAM,IAAI,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YAE9D,2CAA2C;YAC3C,MAAM,aAAa,GAAG,EAAE,CAAC,OAAO,CAAC;;OAEhC,CAAC,CAAC;YACH,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAE3B,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,OAAO,mBAAmB,CAAC,SAAS,EAAE,CAAC;IACzC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;QACjD,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,QAAQ,CAAC,OAAe,EAAE,MAAc,EAAE,KAAa;IACrE,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,IAAI,CAAC,EAAE;QAAE,OAAO,KAAK,CAAC;IAEtB,IAAI,CAAC;QACH,MAAM,eAAe,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;YAC1C,kCAAkC;YAClC,MAAM,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC;;OAE7B,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC,MAAM,CAA8C,CAAC;YAEjF,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;gBACzC,OAAO,KAAK,CAAC;YACf,CAAC;YAED,sBAAsB;YACtB,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CAAC;;;;OAI3B,CAAC,CAAC;YACH,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YAEjD,2CAA2C;YAC3C,MAAM,aAAa,GAAG,EAAE,CAAC,OAAO,CAAC;;OAEhC,CAAC,CAAC;YACH,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAE3B,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,OAAO,eAAe,CAAC,SAAS,EAAE,CAAC;IACrC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,sBAAsB,EAAE,KAAK,CAAC,CAAC;QAC7C,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,SAAS,CAAC,OAAe;IACvC,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,IAAI,CAAC,EAAE;QAAE,OAAO,KAAK,CAAC;IAEtB,IAAI,CAAC;QACH,kCAAkC;QAClC,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CAAC;;KAE3B,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,CAAC,OAAO,CAA+B,CAAC;QAEjE,OAAO,eAAe,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,IAAI,IAAI,CAAC,CAAC;IACpD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;QACpD,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,kBAAkB,CAAC,eAAuB,oBAAoB,CAAC,YAAY;IACzF,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,IAAI,CAAC,EAAE;QAAE,OAAO,CAAC,CAAC;IAElB,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,YAAY,CAAC;QAE7C,MAAM,kBAAkB,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;YAC7C,6CAA6C;YAC7C,6DAA6D;YAC7D,MAAM,aAAa,GAAG,EAAE,CAAC,OAAO,CAAC;;;;;;;OAOhC,CAAC,CAAC;YACH,MAAM,UAAU,GAAG,aAAa,CAAC,GAAG,CAAC,UAAU,EAAE,UAAU,CAGzD,CAAC;YAEH,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC5B,OAAO,CAAC,CAAC;YACX,CAAC;YAED,sBAAsB;YACtB,MAAM,WAAW,GAAG,EAAE,CAAC,OAAO,CAAC;;;;OAI9B,CAAC,CAAC;YAEH,0BAA0B;YAC1B,MAAM,mBAAmB,GAAG,EAAE,CAAC,OAAO,CAAC;;OAEtC,CAAC,CAAC;YAEH,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;YACtC,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;gBAC9B,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACzB,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;oBACpB,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBACnC,CAAC;YACH,CAAC;YAED,KAAK,MAAM,OAAO,IAAI,WAAW,EAAE,CAAC;gBAClC,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACnC,CAAC;YAED,OAAO,UAAU,CAAC,MAAM,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,OAAO,kBAAkB,CAAC,SAAS,EAAE,CAAC;IACxC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;QACxD,OAAO,CAAC,CAAC;IACX,CAAC;AACH,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,iBAAiB,CAAC,OAAe;IAC/C,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,IAAI,CAAC,EAAE;QAAE,OAAO,EAAE,CAAC;IAEnB,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC;;KAEvB,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAS3B,CAAC;QAEH,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACtB,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,MAAM,EAAE,GAAG,CAAC,MAA6B;YACzC,SAAS,EAAE,GAAG,CAAC,UAAU;YACzB,SAAS,EAAE,GAAG,CAAC,UAAU;YACzB,WAAW,EAAE,GAAG,CAAC,YAAY;YAC7B,KAAK,EAAE,GAAG,CAAC,KAAK,IAAI,SAAS;YAC7B,MAAM,EAAE,GAAG,CAAC,MAAM,IAAI,SAAS;SAChC,CAAC,CAAC,CAAC;IACN,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,uCAAuC,EAAE,KAAK,CAAC,CAAC;QAC9D,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,eAAe;IAC7B,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,IAAI,CAAC,EAAE;QAAE,OAAO,KAAK,CAAC;IAEtB,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC;;KAEvB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAuB,CAAC;QAC/C,OAAO,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC;IAC1B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;QACvD,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,gBAAgB;IAC9B,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,IAAI,CAAC,EAAE;QAAE,OAAO,KAAK,CAAC;IAEtB,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC;;KAEvB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAuB,CAAC;QAC/C,OAAO,MAAM,CAAC,KAAK,KAAK,CAAC,CAAC;IAC5B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,wCAAwC,EAAE,KAAK,CAAC,CAAC;QAC/D,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,mBAAmB,CAAC,mBAA2B,oBAAoB,CAAC,YAAY;IAC9F,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,IAAI,CAAC,EAAE;QAAE,OAAO,CAAC,CAAC;IAElB,IAAI,CAAC;QACH,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,gBAAgB,CAAC;QACjD,MAAM,IAAI,GAAG,EAAE,CAAC,OAAO,CAAC;;KAEvB,CAAC,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAsB,CAAC;QACzD,OAAO,MAAM,CAAC,KAAK,CAAC;IACtB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;QAC1D,OAAO,CAAC,CAAC;IACX,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,iBAAiB,CAAC,OAAe,EAAE,MAAc;IAC/D,MAAM,EAAE,GAAG,KAAK,EAAE,CAAC;IACnB,IAAI,CAAC,EAAE,EAAE,CAAC;QACR,OAAO;YACL,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,IAAI;YACZ,MAAM,EAAE,0BAA0B;SACnC,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,kBAAkB,GAAG,EAAE,CAAC,WAAW,CAAC,GAAG,EAAE;YAC7C,0BAA0B;YAC1B,MAAM,SAAS,GAAG,EAAE,CAAC,OAAO,CAAC;;OAE5B,CAAC,CAAC;YACH,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,MAAM,CAAoE,CAAC;YAEtG,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,IAAI;oBACZ,MAAM,EAAE,gBAAgB;iBACV,CAAC;YACnB,CAAC;YAED,IAAI,IAAI,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC7B,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,IAAI;oBACZ,MAAM,EAAE,WAAW,IAAI,CAAC,MAAM,cAAc;iBAC9B,CAAC;YACnB,CAAC;YAED,mBAAmB;YACnB,MAAM,WAAW,GAAG,EAAE,CAAC,OAAO,CAAC;;;;OAI9B,CAAC,CAAC;YACH,WAAW,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,MAAM,CAAC,CAAC;YAE7C,mBAAmB;YACnB,MAAM,aAAa,GAAG,EAAE,CAAC,OAAO,CAAC;;;OAGhC,CAAC,CAAC;YACH,aAAa,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,MAAM,CAAC,CAAC;YAE/C,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,MAAM,EAAE,IAAI,CAAC,EAAE;gBACf,WAAW,EAAE,IAAI,CAAC,WAAW;aACf,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,OAAO,kBAAkB,CAAC,SAAS,EAAE,CAAC;IACxC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,IAAI;YACZ,MAAM,EAAE,mBAAmB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;SACpF,CAAC;IACJ,CAAC;AACH,CAAC"}

package/dist/hooks/swarm/index.d.ts

@@ -0,0 +1,221 @@
+/**
+ * Swarm Coordination System
+ *
+ * SQLite-based multi-agent task coordination.
+ * Enables N agents to claim and work on tasks from a shared pool
+ * with atomic claiming, lease-based ownership, and heartbeat monitoring.
+ *
+ * Usage:
+ * ```typescript
+ * import { startSwarm, claimTask, completeTask, stopSwarm } from './swarm';
+ *
+ * // Initialize swarm with tasks
+ * await startSwarm({
+ *   agentCount: 5,
+ *   tasks: ['fix error in auth.ts', 'add tests for api.ts', ...],
+ *   cwd: process.cwd()
+ * });
+ *
+ * // Each agent claims and works on tasks
+ * const claim = claimTask('agent-1');
+ * if (claim.success) {
+ *   // Do work...
+ *   completeTask('agent-1', claim.taskId!, 'Fixed the bug');
+ * }
+ *
+ * // Check status
+ * const status = getSwarmStatus();
+ *
+ * // Clean up when done
+ * stopSwarm();
+ * ```
+ */
+import type { SwarmConfig, SwarmState, SwarmTask, SwarmStats, ClaimResult, AgentHeartbeat } from './types.js';
+/**
+ * Start a new swarm session
+ *
+ * Initializes the SQLite database, creates the task pool,
+ * and starts the stale claim cleanup timer.
+ *
+ * @param config - Swarm configuration
+ * @returns true if swarm was started successfully
+ */
+export declare function startSwarm(config: SwarmConfig): Promise<boolean>;
+/**
+ * Stop the swarm and clean up resources
+ *
+ * Stops the cleanup timer and optionally deletes the database.
+ *
+ * @param deleteDatabase - Whether to delete the database file (default: false)
+ * @returns true if swarm was stopped successfully
+ */
+export declare function stopSwarm(deleteDatabase?: boolean): boolean;
+/**
+ * Get the current swarm status
+ *
+ * @returns SwarmState or null if swarm is not active
+ */
+export declare function getSwarmStatus(): SwarmState | null;
+/**
+ * Get swarm statistics
+ *
+ * @returns SwarmStats with task counts and timing info
+ */
+export declare function getSwarmStats(): SwarmStats | null;
+/**
+ * Claim the next available task
+ *
+ * @param agentId - Unique identifier for the claiming agent
+ * @returns ClaimResult with success status and task details
+ */
+export declare function claimTask(agentId: string): ClaimResult;
+/**
+ * Release a claimed task back to the pool
+ *
+ * @param agentId - Agent releasing the task
+ * @param taskId - Task to release
+ * @returns true if release was successful
+ */
+export declare function releaseTask(agentId: string, taskId: string): boolean;
+/**
+ * Mark a task as completed
+ *
+ * @param agentId - Agent that completed the task
+ * @param taskId - Completed task ID
+ * @param result - Optional result/output from the task
+ * @returns true if completion was recorded
+ */
+export declare function completeTask(agentId: string, taskId: string, result?: string): boolean;
+/**
+ * Mark a task as failed
+ *
+ * @param agentId - Agent that failed the task
+ * @param taskId - Failed task ID
+ * @param error - Error message/reason for failure
+ * @returns true if failure was recorded
+ */
+export declare function failTask(agentId: string, taskId: string, error: string): boolean;
+/**
+ * Send a heartbeat to indicate the agent is still alive
+ *
+ * Agents should call this every 60 seconds while working on tasks.
+ *
+ * @param agentId - Agent sending heartbeat
+ * @returns true if heartbeat was recorded
+ */
+export declare function heartbeat(agentId: string): boolean;
+/**
+ * Clean up stale claims from dead agents
+ *
+ * Called automatically by the cleanup timer, but can also be called manually.
+ *
+ * @param leaseTimeout - Lease timeout in milliseconds (default: 5 minutes)
+ * @returns Number of tasks released
+ */
+export declare function cleanupStaleClaims(leaseTimeout?: number): number;
+/**
+ * Check if there are pending tasks
+ *
+ * @returns true if there are tasks waiting to be claimed
+ */
+export declare function hasPendingWork(): boolean;
+/**
+ * Check if all tasks are complete
+ *
+ * @returns true if all tasks are done or failed
+ */
+export declare function isSwarmComplete(): boolean;
+/**
+ * Get the number of active agents
+ *
+ * @returns Number of agents with recent heartbeats
+ */
+export declare function getActiveAgents(): number;
+/**
+ * Get all tasks
+ *
+ * @returns Array of all tasks in the swarm
+ */
+export declare function getAllTasks(): SwarmTask[];
+/**
+ * Get tasks by status
+ *
+ * @param status - Task status to filter by
+ * @returns Array of tasks with the given status
+ */
+export declare function getTasksWithStatus(status: SwarmTask['status']): SwarmTask[];
+/**
+ * Get a specific task
+ *
+ * @param taskId - Task ID to retrieve
+ * @returns Task or null if not found
+ */
+export declare function getTaskById(taskId: string): SwarmTask | null;
+/**
+ * Get tasks claimed by a specific agent
+ *
+ * @param agentId - Agent to query
+ * @returns Array of tasks claimed by the agent
+ */
+export declare function getAgentTasks(agentId: string): SwarmTask[];
+/**
+ * Get all agent heartbeats
+ *
+ * @returns Array of agent heartbeat records
+ */
+export declare function getAllHeartbeats(): AgentHeartbeat[];
+/**
+ * Retry a failed task
+ *
+ * @param agentId - Agent attempting to retry
+ * @param taskId - Failed task to retry
+ * @returns ClaimResult
+ */
+export declare function retryTask(agentId: string, taskId: string): ClaimResult;
+/**
+ * Check if swarm database is initialized
+ *
+ * @returns true if database is ready
+ */
+export declare function isSwarmReady(): boolean;
+/**
+ * Initialize database without starting a swarm
+ *
+ * Useful for agents that join an existing swarm.
+ *
+ * @param cwd - Working directory
+ * @returns true if database was initialized
+ */
+export declare function connectToSwarm(cwd: string): Promise<boolean>;
+/**
+ * Disconnect from swarm without stopping it
+ *
+ * @returns true if disconnected successfully
+ */
+export declare function disconnectFromSwarm(): boolean;
+/**
+ * Check if swarm is currently active
+ *
+ * Used by cancel skill to detect active swarm.
+ * Checks if database exists and session is active.
+ *
+ * @param cwd - Working directory to check
+ * @returns true if swarm is active
+ */
+export declare function isSwarmActive(cwd: string): boolean;
+/**
+ * Cancel active swarm
+ *
+ * Used by cancel skill to stop swarm cleanly.
+ *
+ * @param cwd - Working directory
+ * @returns CancelResult with success status and message
+ */
+export declare function cancelSwarm(cwd: string): Promise<{
+    success: boolean;
+    message: string;
+    stats?: SwarmStats | null;
+}>;
+export type { SwarmTask, SwarmState, SwarmConfig, SwarmStats, ClaimResult, AgentHeartbeat } from './types.js';
+export { DEFAULT_SWARM_CONFIG } from './types.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/hooks/swarm/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/hooks/swarm/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+BG;AAKH,OAAO,KAAK,EACV,WAAW,EACX,UAAU,EACV,SAAS,EACT,UAAU,EACV,WAAW,EACX,cAAc,EACf,MAAM,YAAY,CAAC;AAyDpB;;;;;;;;GAQG;AACH,wBAAsB,UAAU,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,CAqEtE;AAED;;;;;;;GAOG;AACH,wBAAgB,SAAS,CAAC,cAAc,GAAE,OAAe,GAAG,OAAO,CAyBlE;AAED;;;;GAIG;AACH,wBAAgB,cAAc,IAAI,UAAU,GAAG,IAAI,CAElD;AAED;;;;GAIG;AACH,wBAAgB,aAAa,IAAI,UAAU,GAAG,IAAI,CAEjD;AAED;;;;;GAKG;AACH,wBAAgB,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,WAAW,CAEtD;AAED;;;;;;GAMG;AACH,wBAAgB,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAEpE;AAED;;;;;;;GAOG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,OAAO,CAStF;AAED;;;;;;;GAOG;AACH,wBAAgB,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAEhF;AAED;;;;;;;GAOG;AACH,wBAAgB,SAAS,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAElD;AAED;;;;;;;GAOG;AACH,wBAAgB,kBAAkB,CAAC,YAAY,CAAC,EAAE,MAAM,GAAG,MAAM,CAEhE;AAED;;;;GAIG;AACH,wBAAgB,cAAc,IAAI,OAAO,CAExC;AAED;;;;GAIG;AACH,wBAAgB,eAAe,IAAI,OAAO,CAEzC;AAED;;;;GAIG;AACH,wBAAgB,eAAe,IAAI,MAAM,CAExC;AAED;;;;GAIG;AACH,wBAAgB,WAAW,IAAI,SAAS,EAAE,CAEzC;AAED;;;;;GAKG;AACH,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,SAAS,CAAC,QAAQ,CAAC,GAAG,SAAS,EAAE,CAE3E;AAED;;;;;GAKG;AACH,wBAAgB,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI,CAE5D;AAED;;;;;GAKG;AACH,wBAAgB,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,SAAS,EAAE,CAE1D;AAED;;;;GAIG;AACH,wBAAgB,gBAAgB,IAAI,cAAc,EAAE,CAEnD;AAED;;;;;;GAMG;AACH,wBAAgB,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,WAAW,CAEtE;AAED;;;;GAIG;AACH,wBAAgB,YAAY,IAAI,OAAO,CAEtC;AAED;;;;;;;GAOG;AACH,wBAAsB,cAAc,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAUlE;AAED;;;;GAIG;AACH,wBAAgB,mBAAmB,IAAI,OAAO,CAI7C;AAED;;;;;;;;GAQG;AACH,wBAAgB,aAAa,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAmBlD;AAED;;;;;;;GAOG;AACH,wBAAsB,WAAW,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;IACtD,OAAO,EAAE,OAAO,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC;CAC3B,CAAC,CA8BD;AAGD,YAAY,EACV,SAAS,EACT,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,cAAc,EACf,MAAM,YAAY,CAAC;AAEpB,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC"}