agent-relay 1.0.21 → 1.1.0

package/dist/cloud/services/coordinator.js

@@ -0,0 +1,389 @@
+/**
+ * Coordinator Agent Service
+ *
+ * Manages lifecycle of coordinator agents for project groups.
+ * Coordinators oversee and orchestrate work across repositories in a group.
+ */
+import { db } from '../db/index.js';
+import { createClient } from 'redis';
+import { getConfig } from '../config.js';
+/**
+ * In-memory coordinator state tracker
+ * In production, this would be persisted to database or Redis
+ */
+const coordinatorStates = new Map();
+/**
+ * Start a coordinator agent for a project group
+ */
+async function start(groupId) {
+    const group = await db.projectGroups.findById(groupId);
+    if (!group) {
+        throw new Error('Project group not found');
+    }
+    if (!group.coordinatorAgent?.enabled) {
+        throw new Error('Coordinator is not enabled for this group');
+    }
+    const repositories = await db.repositories.findByProjectGroupId(groupId);
+    if (repositories.length === 0) {
+        throw new Error('Cannot start coordinator for empty group');
+    }
+    const config = group.coordinatorAgent;
+    const agentName = config.name || `${group.name} Coordinator`;
+    const model = config.model || 'claude-sonnet-4-5';
+    // Check if already running
+    const existing = coordinatorStates.get(groupId);
+    if (existing && existing.status === 'running') {
+        console.log(`Coordinator for group ${groupId} is already running`);
+        return;
+    }
+    // Update state to starting
+    const state = {
+        groupId,
+        groupName: group.name,
+        agentName,
+        model,
+        status: 'starting',
+        repositories,
+    };
+    coordinatorStates.set(groupId, state);
+    try {
+        // Spawn the coordinator agent
+        // In a real implementation, this would:
+        // 1. Connect to agent-relay daemon or cloud workspace
+        // 2. Spawn agent with configured name and model
+        // 3. Provide system prompt with group context
+        // 4. Configure capabilities (read repos, create PRs, etc.)
+        await spawnCoordinatorAgent(group, config, repositories);
+        // Update state to running
+        state.status = 'running';
+        state.startedAt = new Date();
+        coordinatorStates.set(groupId, state);
+        console.log(`Coordinator agent started for group ${groupId}: ${agentName}`);
+    }
+    catch (error) {
+        state.status = 'error';
+        state.error = error instanceof Error ? error.message : 'Unknown error';
+        coordinatorStates.set(groupId, state);
+        throw error;
+    }
+}
+/**
+ * Stop a coordinator agent for a project group
+ */
+async function stop(groupId) {
+    const state = coordinatorStates.get(groupId);
+    if (!state) {
+        // Not running, nothing to do
+        return;
+    }
+    if (state.status === 'stopped') {
+        return;
+    }
+    // Update state to stopping
+    state.status = 'stopping';
+    coordinatorStates.set(groupId, state);
+    try {
+        // Stop the coordinator agent
+        // In a real implementation, this would:
+        // 1. Send stop signal to the agent
+        // 2. Wait for graceful shutdown
+        // 3. Clean up resources
+        await stopCoordinatorAgent(groupId, state);
+        // Update state to stopped
+        state.status = 'stopped';
+        state.stoppedAt = new Date();
+        coordinatorStates.set(groupId, state);
+        console.log(`Coordinator agent stopped for group ${groupId}`);
+    }
+    catch (error) {
+        state.status = 'error';
+        state.error = error instanceof Error ? error.message : 'Unknown error';
+        coordinatorStates.set(groupId, state);
+        throw error;
+    }
+}
+/**
+ * Restart a coordinator agent
+ */
+async function restart(groupId) {
+    await stop(groupId);
+    await start(groupId);
+}
+/**
+ * Get status of a coordinator agent
+ */
+async function getStatus(groupId) {
+    return coordinatorStates.get(groupId) || null;
+}
+/**
+ * List all active coordinators
+ */
+async function listActive() {
+    return Array.from(coordinatorStates.values()).filter((state) => state.status === 'running' || state.status === 'starting');
+}
+/**
+ * Redis pub/sub for cross-workspace messaging
+ */
+let pubClient = null;
+let subClient = null;
+const messageHandlers = new Map();
+/**
+ * Initialize Redis pub/sub clients
+ */
+async function initRedisClients() {
+    if (pubClient && subClient)
+        return;
+    const config = getConfig();
+    pubClient = createClient({ url: config.redisUrl });
+    subClient = createClient({ url: config.redisUrl });
+    await pubClient.connect();
+    await subClient.connect();
+    // Subscribe to coordinator channel
+    await subClient.subscribe('coordinator:messages', (message) => {
+        try {
+            const parsed = JSON.parse(message);
+            const handler = messageHandlers.get(parsed.toWorkspace || '*');
+            if (handler) {
+                handler(parsed);
+            }
+        }
+        catch (error) {
+            console.error('Failed to parse coordinator message:', error);
+        }
+    });
+    console.log('[coordinator] Redis pub/sub initialized');
+}
+/**
+ * Send a message from coordinator to a workspace
+ */
+export async function sendToWorkspace(coordinatorGroupId, targetWorkspaceId, agentName, message, thread) {
+    if (!pubClient) {
+        await initRedisClients();
+    }
+    const crossMessage = {
+        type: 'relay',
+        from: 'Coordinator',
+        fromWorkspace: `coordinator:${coordinatorGroupId}`,
+        to: agentName,
+        toWorkspace: targetWorkspaceId,
+        body: message,
+        thread,
+        timestamp: new Date().toISOString(),
+    };
+    await pubClient.publish('workspace:messages', JSON.stringify(crossMessage));
+    console.log(`[coordinator] Sent message to ${targetWorkspaceId}:${agentName}`);
+}
+/**
+ * Broadcast a message to all workspaces in a project group
+ */
+export async function broadcastToGroup(coordinatorGroupId, message, thread) {
+    if (!pubClient) {
+        await initRedisClients();
+    }
+    const group = await db.projectGroups.findById(coordinatorGroupId);
+    if (!group) {
+        throw new Error('Project group not found');
+    }
+    const repositories = await db.repositories.findByProjectGroupId(coordinatorGroupId);
+    // Get all workspaces containing these repositories
+    const workspaceIds = new Set();
+    for (const repo of repositories) {
+        if (repo.workspaceId) {
+            workspaceIds.add(repo.workspaceId);
+        }
+    }
+    const crossMessage = {
+        type: 'relay',
+        from: 'Coordinator',
+        fromWorkspace: `coordinator:${coordinatorGroupId}`,
+        to: '*',
+        body: message,
+        thread,
+        timestamp: new Date().toISOString(),
+    };
+    // Broadcast to each workspace
+    for (const workspaceId of workspaceIds) {
+        crossMessage.toWorkspace = workspaceId;
+        await pubClient.publish('workspace:messages', JSON.stringify(crossMessage));
+    }
+    console.log(`[coordinator] Broadcast to ${workspaceIds.size} workspace(s)`);
+}
+/**
+ * Spawn the actual coordinator agent
+ */
+async function spawnCoordinatorAgent(group, config, repositories) {
+    // Initialize Redis for cross-workspace messaging
+    await initRedisClients();
+    // Build system prompt for the coordinator
+    const systemPrompt = buildCoordinatorSystemPrompt(group, config, repositories);
+    // Get workspaces for the repositories
+    const workspaceIds = new Set();
+    for (const repo of repositories) {
+        if (repo.workspaceId) {
+            workspaceIds.add(repo.workspaceId);
+        }
+    }
+    const workspaces = await Promise.all(Array.from(workspaceIds).map(id => db.workspaces.findById(id)));
+    console.log(`[coordinator] Spawning coordinator agent: ${config.name || group.name}`);
+    console.log(`[coordinator] Model: ${config.model || 'claude-sonnet-4-5'}`);
+    console.log(`[coordinator] Repositories: ${repositories.map((r) => r.githubFullName).join(', ')}`);
+    console.log(`[coordinator] Connected workspaces: ${workspaces.filter(Boolean).length}`);
+    // Register message handler for this coordinator
+    messageHandlers.set(`coordinator:${group.id}`, async (message) => {
+        console.log(`[coordinator:${group.id}] Received: ${message.body.substring(0, 100)}...`);
+        // In a full implementation, this would:
+        // 1. Parse the message for coordinator commands
+        // 2. Route to appropriate workspace(s)
+        // 3. Track conversation state
+    });
+    // Store coordinator connection info in Redis for workspace discovery
+    if (pubClient) {
+        await pubClient.hSet(`coordinator:${group.id}`, {
+            groupId: group.id,
+            groupName: group.name,
+            agentName: config.name || `${group.name} Coordinator`,
+            model: config.model || 'claude-sonnet-4-5',
+            startedAt: new Date().toISOString(),
+            workspaces: JSON.stringify(Array.from(workspaceIds)),
+            systemPrompt,
+        });
+        await pubClient.expire(`coordinator:${group.id}`, 86400); // 24h TTL
+    }
+}
+/**
+ * Stop the actual coordinator agent
+ */
+async function stopCoordinatorAgent(groupId, state) {
+    console.log(`[coordinator] Stopping coordinator agent for group ${groupId}: ${state.agentName}`);
+    // Remove message handler
+    messageHandlers.delete(`coordinator:${groupId}`);
+    // Remove from Redis
+    if (pubClient) {
+        await pubClient.del(`coordinator:${groupId}`);
+    }
+}
+/**
+ * Route a message from a workspace to the coordinator
+ */
+export async function routeToCoordinator(workspaceId, agentName, message, thread) {
+    if (!pubClient) {
+        await initRedisClients();
+    }
+    // Find which coordinator is managing this workspace
+    // by scanning all coordinator keys
+    const coordinatorKeys = await pubClient.keys('coordinator:*');
+    for (const key of coordinatorKeys) {
+        if (key === 'coordinator:messages')
+            continue; // Skip the channel
+        const data = await pubClient.hGetAll(key);
+        if (data.workspaces) {
+            const workspaces = JSON.parse(data.workspaces);
+            if (workspaces.includes(workspaceId)) {
+                // Found the coordinator for this workspace
+                const crossMessage = {
+                    type: 'relay',
+                    from: agentName,
+                    fromWorkspace: workspaceId,
+                    to: 'Coordinator',
+                    toWorkspace: key,
+                    body: message,
+                    thread,
+                    timestamp: new Date().toISOString(),
+                };
+                await pubClient.publish('coordinator:messages', JSON.stringify(crossMessage));
+                console.log(`[coordinator] Routed message from ${workspaceId}:${agentName} to ${key}`);
+                return;
+            }
+        }
+    }
+    console.warn(`[coordinator] No coordinator found for workspace ${workspaceId}`);
+}
+/**
+ * Get all active coordinators
+ */
+export async function getActiveCoordinators() {
+    if (!pubClient) {
+        await initRedisClients();
+    }
+    const coordinatorKeys = await pubClient.keys('coordinator:*');
+    const coordinators = [];
+    for (const key of coordinatorKeys) {
+        if (key === 'coordinator:messages')
+            continue;
+        const data = await pubClient.hGetAll(key);
+        if (data.groupId) {
+            coordinators.push({
+                groupId: data.groupId,
+                groupName: data.groupName,
+                agentName: data.agentName,
+                model: data.model,
+                startedAt: data.startedAt,
+                workspaces: data.workspaces ? JSON.parse(data.workspaces) : [],
+            });
+        }
+    }
+    return coordinators;
+}
+/**
+ * Build system prompt for coordinator agent
+ */
+function buildCoordinatorSystemPrompt(group, config, repositories) {
+    const repoList = repositories.map((r) => r.githubFullName).join('\n- ');
+    let prompt = `You are the coordinator agent for the "${group.name}" project group.
+
+Your role is to oversee and orchestrate work across the following repositories:
+- ${repoList}
+
+`;
+    if (config.capabilities && config.capabilities.length > 0) {
+        prompt += `You have the following capabilities:
+${config.capabilities.map((c) => `- ${c}`).join('\n')}
+
+`;
+    }
+    if (config.systemPrompt) {
+        prompt += `${config.systemPrompt}\n\n`;
+    }
+    prompt += `When coordinating work:
+1. Monitor all repositories in your group
+2. Identify dependencies and coordination points
+3. Delegate tasks to project-specific agents when appropriate
+4. Ensure consistency across repositories
+5. Report status and blockers to the team
+
+Use the Agent Relay messaging system to communicate with other agents and team members.`;
+    return prompt;
+}
+/**
+ * Singleton instance
+ */
+let coordinatorServiceInstance = null;
+/**
+ * Get the coordinator service singleton
+ */
+export function getCoordinatorService() {
+    if (!coordinatorServiceInstance) {
+        coordinatorServiceInstance = {
+            start,
+            stop,
+            restart,
+            getStatus,
+            listActive,
+        };
+    }
+    return coordinatorServiceInstance;
+}
+/**
+ * Initialize coordinator service
+ * Restarts any coordinators that should be running
+ */
+export async function initializeCoordinatorService() {
+    console.log('Initializing coordinator service...');
+    // In a production system, this would:
+    // 1. Query database for all enabled coordinators
+    // 2. Check their expected state
+    // 3. Restart any that should be running
+    // For now, just log initialization
+    console.log('Coordinator service initialized');
+}
+//# sourceMappingURL=coordinator.js.map

package/dist/cloud/services/coordinator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"coordinator.js","sourceRoot":"","sources":["../../../src/cloud/services/coordinator.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,EAAE,EAAoD,MAAM,gBAAgB,CAAC;AACtF,OAAO,EAAE,YAAY,EAAmB,MAAM,OAAO,CAAC;AACtD,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAiBzC;;;GAGG;AACH,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAA4B,CAAC;AAU9D;;GAEG;AACH,KAAK,UAAU,KAAK,CAAC,OAAe;IAClC,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IACvD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;IAED,IAAI,CAAC,KAAK,CAAC,gBAAgB,EAAE,OAAO,EAAE,CAAC;QACrC,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;IAC/D,CAAC;IAED,MAAM,YAAY,GAAG,MAAM,EAAE,CAAC,YAAY,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;IACzE,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;IAC9D,CAAC;IAED,MAAM,MAAM,GAAG,KAAK,CAAC,gBAAgB,CAAC;IACtC,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,IAAI,GAAG,KAAK,CAAC,IAAI,cAAc,CAAC;IAC7D,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,mBAAmB,CAAC;IAElD,2BAA2B;IAC3B,MAAM,QAAQ,GAAG,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAChD,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QAC9C,OAAO,CAAC,GAAG,CAAC,yBAAyB,OAAO,qBAAqB,CAAC,CAAC;QACnE,OAAO;IACT,CAAC;IAED,2BAA2B;IAC3B,MAAM,KAAK,GAAqB;QAC9B,OAAO;QACP,SAAS,EAAE,KAAK,CAAC,IAAI;QACrB,SAAS;QACT,KAAK;QACL,MAAM,EAAE,UAAU;QAClB,YAAY;KACb,CAAC;IACF,iBAAiB,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IAEtC,IAAI,CAAC;QACH,8BAA8B;QAC9B,wCAAwC;QACxC,sDAAsD;QACtD,gDAAgD;QAChD,8CAA8C;QAC9C,2DAA2D;QAE3D,MAAM,qBAAqB,CAAC,KAAK,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;QAEzD,0BAA0B;QAC1B,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC;QACzB,KAAK,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,iBAAiB,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAEtC,OAAO,CAAC,GAAG,CAAC,uCAAuC,OAAO,KAAK,SAAS,EAAE,CAAC,CAAC;IAC9E,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,KAAK,CAAC,MAAM,GAAG,OAAO,CAAC;QACvB,KAAK,CAAC,KAAK,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;QACvE,iBAAiB,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACtC,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,IAAI,CAAC,OAAe;IACjC,MAAM,KAAK,GAAG,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC7C,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,6BAA6B;QAC7B,OAAO;IACT,CAAC;IAED,IAAI,KAAK,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QAC/B,OAAO;IACT,CAAC;IAED,2BAA2B;IAC3B,KAAK,CAAC,MAAM,GAAG,UAAU,CAAC;IAC1B,iBAAiB,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IAEtC,IAAI,CAAC;QACH,6BAA6B;QAC7B,wCAAwC;QACxC,mCAAmC;QACnC,gCAAgC;QAChC,wBAAwB;QAExB,MAAM,oBAAoB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAE3C,0BAA0B;QAC1B,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC;QACzB,KAAK,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;QAC7B,iBAAiB,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAEtC,OAAO,CAAC,GAAG,CAAC,uCAAuC,OAAO,EAAE,CAAC,CAAC;IAChE,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,KAAK,CAAC,MAAM,GAAG,OAAO,CAAC;QACvB,KAAK,CAAC,KAAK,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;QACvE,iBAAiB,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACtC,MAAM,KAAK,CAAC;IACd,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,OAAO,CAAC,OAAe;IACpC,MAAM,IAAI,CAAC,OAAO,CAAC,CAAC;IACpB,MAAM,KAAK,CAAC,OAAO,CAAC,CAAC;AACvB,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,SAAS,CAAC,OAAe;IACtC,OAAO,iBAAiB,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC;AAChD,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,UAAU;IACvB,OAAO,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAClD,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,KAAK,SAAS,IAAI,KAAK,CAAC,MAAM,KAAK,UAAU,CACrE,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,IAAI,SAAS,GAA2B,IAAI,CAAC;AAC7C,IAAI,SAAS,GAA2B,IAAI,CAAC;AAC7C,MAAM,eAAe,GAAG,IAAI,GAAG,EAAoD,CAAC;AAapF;;GAEG;AACH,KAAK,UAAU,gBAAgB;IAC7B,IAAI,SAAS,IAAI,SAAS;QAAE,OAAO;IAEnC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAE3B,SAAS,GAAG,YAAY,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IACnD,SAAS,GAAG,YAAY,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC;IAEnD,MAAM,SAAS,CAAC,OAAO,EAAE,CAAC;IAC1B,MAAM,SAAS,CAAC,OAAO,EAAE,CAAC;IAE1B,mCAAmC;IACnC,MAAM,SAAS,CAAC,SAAS,CAAC,sBAAsB,EAAE,CAAC,OAAO,EAAE,EAAE;QAC5D,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAA0B,CAAC;YAC5D,MAAM,OAAO,GAAG,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,WAAW,IAAI,GAAG,CAAC,CAAC;YAC/D,IAAI,OAAO,EAAE,CAAC;gBACZ,OAAO,CAAC,MAAM,CAAC,CAAC;YAClB,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,sCAAsC,EAAE,KAAK,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAC;AACzD,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,kBAA0B,EAC1B,iBAAyB,EACzB,SAAiB,EACjB,OAAe,EACf,MAAe;IAEf,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,gBAAgB,EAAE,CAAC;IAC3B,CAAC;IAED,MAAM,YAAY,GAA0B;QAC1C,IAAI,EAAE,OAAO;QACb,IAAI,EAAE,aAAa;QACnB,aAAa,EAAE,eAAe,kBAAkB,EAAE;QAClD,EAAE,EAAE,SAAS;QACb,WAAW,EAAE,iBAAiB;QAC9B,IAAI,EAAE,OAAO;QACb,MAAM;QACN,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC,CAAC;IAEF,MAAM,SAAU,CAAC,OAAO,CAAC,oBAAoB,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC;IAC7E,OAAO,CAAC,GAAG,CAAC,iCAAiC,iBAAiB,IAAI,SAAS,EAAE,CAAC,CAAC;AACjF,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,kBAA0B,EAC1B,OAAe,EACf,MAAe;IAEf,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,gBAAgB,EAAE,CAAC;IAC3B,CAAC;IAED,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,aAAa,CAAC,QAAQ,CAAC,kBAAkB,CAAC,CAAC;IAClE,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC7C,CAAC;IAED,MAAM,YAAY,GAAG,MAAM,EAAE,CAAC,YAAY,CAAC,oBAAoB,CAAC,kBAAkB,CAAC,CAAC;IAEpF,mDAAmD;IACnD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAU,CAAC;IACvC,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;QAChC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED,MAAM,YAAY,GAA0B;QAC1C,IAAI,EAAE,OAAO;QACb,IAAI,EAAE,aAAa;QACnB,aAAa,EAAE,eAAe,kBAAkB,EAAE;QAClD,EAAE,EAAE,GAAG;QACP,IAAI,EAAE,OAAO;QACb,MAAM;QACN,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;KACpC,CAAC;IAEF,8BAA8B;IAC9B,KAAK,MAAM,WAAW,IAAI,YAAY,EAAE,CAAC;QACvC,YAAY,CAAC,WAAW,GAAG,WAAW,CAAC;QACvC,MAAM,SAAU,CAAC,OAAO,CAAC,oBAAoB,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC;IAC/E,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,8BAA8B,YAAY,CAAC,IAAI,eAAe,CAAC,CAAC;AAC9E,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,qBAAqB,CAClC,KAAmB,EACnB,MAA8B,EAC9B,YAA0B;IAE1B,iDAAiD;IACjD,MAAM,gBAAgB,EAAE,CAAC;IAEzB,0CAA0C;IAC1C,MAAM,YAAY,GAAG,4BAA4B,CAAC,KAAK,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;IAE/E,sCAAsC;IACtC,MAAM,YAAY,GAAG,IAAI,GAAG,EAAU,CAAC;IACvC,KAAK,MAAM,IAAI,IAAI,YAAY,EAAE,CAAC;QAChC,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,GAAG,CAClC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAC/D,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,6CAA6C,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IACtF,OAAO,CAAC,GAAG,CAAC,wBAAwB,MAAM,CAAC,KAAK,IAAI,mBAAmB,EAAE,CAAC,CAAC;IAC3E,OAAO,CAAC,GAAG,CAAC,+BAA+B,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACnG,OAAO,CAAC,GAAG,CAAC,uCAAuC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;IAExF,gDAAgD;IAChD,eAAe,CAAC,GAAG,CAAC,eAAe,KAAK,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;QAC/D,OAAO,CAAC,GAAG,CAAC,gBAAgB,KAAK,CAAC,EAAE,eAAe,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;QAExF,wCAAwC;QACxC,gDAAgD;QAChD,uCAAuC;QACvC,8BAA8B;IAChC,CAAC,CAAC,CAAC;IAEH,qEAAqE;IACrE,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,SAAS,CAAC,IAAI,CAAC,eAAe,KAAK,CAAC,EAAE,EAAE,EAAE;YAC9C,OAAO,EAAE,KAAK,CAAC,EAAE;YACjB,SAAS,EAAE,KAAK,CAAC,IAAI;YACrB,SAAS,EAAE,MAAM,CAAC,IAAI,IAAI,GAAG,KAAK,CAAC,IAAI,cAAc;YACrD,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,mBAAmB;YAC1C,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACpD,YAAY;SACb,CAAC,CAAC;QACH,MAAM,SAAS,CAAC,MAAM,CAAC,eAAe,KAAK,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,UAAU;IACtE,CAAC;AACH,CAAC;AAED;;GAEG;AACH,KAAK,UAAU,oBAAoB,CAAC,OAAe,EAAE,KAAuB;IAC1E,OAAO,CAAC,GAAG,CAAC,sDAAsD,OAAO,KAAK,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC;IAEjG,yBAAyB;IACzB,eAAe,CAAC,MAAM,CAAC,eAAe,OAAO,EAAE,CAAC,CAAC;IAEjD,oBAAoB;IACpB,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,SAAS,CAAC,GAAG,CAAC,eAAe,OAAO,EAAE,CAAC,CAAC;IAChD,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,WAAmB,EACnB,SAAiB,EACjB,OAAe,EACf,MAAe;IAEf,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,gBAAgB,EAAE,CAAC;IAC3B,CAAC;IAED,oDAAoD;IACpD,mCAAmC;IACnC,MAAM,eAAe,GAAG,MAAM,SAAU,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAE/D,KAAK,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;QAClC,IAAI,GAAG,KAAK,sBAAsB;YAAE,SAAS,CAAC,mBAAmB;QAEjE,MAAM,IAAI,GAAG,MAAM,SAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC3C,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAa,CAAC;YAC3D,IAAI,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;gBACrC,2CAA2C;gBAC3C,MAAM,YAAY,GAA0B;oBAC1C,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,SAAS;oBACf,aAAa,EAAE,WAAW;oBAC1B,EAAE,EAAE,aAAa;oBACjB,WAAW,EAAE,GAAG;oBAChB,IAAI,EAAE,OAAO;oBACb,MAAM;oBACN,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACpC,CAAC;gBAEF,MAAM,SAAU,CAAC,OAAO,CAAC,sBAAsB,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC,CAAC;gBAC/E,OAAO,CAAC,GAAG,CAAC,qCAAqC,WAAW,IAAI,SAAS,OAAO,GAAG,EAAE,CAAC,CAAC;gBACvF,OAAO;YACT,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,CAAC,IAAI,CAAC,oDAAoD,WAAW,EAAE,CAAC,CAAC;AAClF,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB;IAQzC,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,MAAM,gBAAgB,EAAE,CAAC;IAC3B,CAAC;IAED,MAAM,eAAe,GAAG,MAAM,SAAU,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC/D,MAAM,YAAY,GAAG,EAAE,CAAC;IAExB,KAAK,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;QAClC,IAAI,GAAG,KAAK,sBAAsB;YAAE,SAAS;QAE7C,MAAM,IAAI,GAAG,MAAM,SAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAC3C,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,YAAY,CAAC,IAAI,CAAC;gBAChB,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,UAAU,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE;aAC/D,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC;AAED;;GAEG;AACH,SAAS,4BAA4B,CACnC,KAAmB,EACnB,MAA8B,EAC9B,YAA0B;IAE1B,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAExE,IAAI,MAAM,GAAG,0CAA0C,KAAK,CAAC,IAAI;;;IAG/D,QAAQ;;CAEX,CAAC;IAEA,IAAI,MAAM,CAAC,YAAY,IAAI,MAAM,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC1D,MAAM,IAAI;EACZ,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;CAEpD,CAAC;IACA,CAAC;IAED,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,MAAM,CAAC,YAAY,MAAM,CAAC;IACzC,CAAC;IAED,MAAM,IAAI;;;;;;;wFAO4E,CAAC;IAEvF,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,IAAI,0BAA0B,GAA8B,IAAI,CAAC;AAEjE;;GAEG;AACH,MAAM,UAAU,qBAAqB;IACnC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QAChC,0BAA0B,GAAG;YAC3B,KAAK;YACL,IAAI;YACJ,OAAO;YACP,SAAS;YACT,UAAU;SACX,CAAC;IACJ,CAAC;IACD,OAAO,0BAA0B,CAAC;AACpC,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,4BAA4B;IAChD,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAC;IAEnD,sCAAsC;IACtC,iDAAiD;IACjD,gCAAgC;IAChC,wCAAwC;IAExC,mCAAmC;IACnC,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;AACjD,CAAC"}

package/dist/cloud/services/planLimits.d.ts

@@ -0,0 +1,110 @@
+/**
+ * Plan Limits Service
+ *
+ * Defines resource limits for each plan tier and provides
+ * functions to check if users are within their limits.
+ */
+import { PlanType } from '../db/index.js';
+/**
+ * Resource limits for each plan tier
+ */
+export interface PlanLimits {
+    maxWorkspaces: number;
+    maxRepos: number;
+    maxConcurrentAgents: number;
+    maxComputeHoursPerMonth: number;
+    coordinatorsEnabled: boolean;
+}
+/**
+ * Plan limits configuration
+ *
+ * Free: Try it out on a side project
+ * Pro: Professional developers, coordinators enabled
+ * Team: Growing teams with advanced needs
+ * Enterprise: Unlimited everything
+ */
+export declare const PLAN_LIMITS: Record<PlanType, PlanLimits>;
+/**
+ * Get plan limits for a given plan type
+ */
+export declare function getPlanLimits(plan: PlanType): PlanLimits;
+/**
+ * Current usage for a user
+ */
+export interface UserUsage {
+    workspaceCount: number;
+    repoCount: number;
+    concurrentAgents: number;
+    computeHoursThisMonth: number;
+}
+/**
+ * Get current usage for a user
+ */
+export declare function getUserUsage(userId: string): Promise<UserUsage>;
+/**
+ * Check if user can create a new workspace
+ */
+export declare function canCreateWorkspace(userId: string): Promise<{
+    allowed: boolean;
+    reason?: string;
+    limit?: number;
+    current?: number;
+}>;
+/**
+ * Check if user can add a new repo
+ */
+export declare function canAddRepo(userId: string): Promise<{
+    allowed: boolean;
+    reason?: string;
+    limit?: number;
+    current?: number;
+}>;
+/**
+ * Check if user can spawn another agent (concurrent agent limit)
+ */
+export declare function canSpawnAgent(userId: string, currentRunningAgents?: number): Promise<{
+    allowed: boolean;
+    reason?: string;
+    limit?: number;
+    current?: number;
+}>;
+/**
+ * Check if user can use coordinator agents
+ */
+export declare function canUseCoordinator(userId: string): Promise<{
+    allowed: boolean;
+    reason?: string;
+    requiredPlan?: string;
+}>;
+/**
+ * Check if user has compute hours available
+ */
+export declare function hasComputeHoursAvailable(userId: string): Promise<{
+    available: boolean;
+    reason?: string;
+    limit?: number;
+    current?: number;
+}>;
+/**
+ * Get remaining quota for a user
+ */
+export declare function getRemainingQuota(userId: string): Promise<{
+    plan: PlanType;
+    limits: PlanLimits;
+    usage: UserUsage;
+    remaining: {
+        workspaces: number;
+        repos: number;
+        concurrentAgents: number;
+        computeHours: number;
+    };
+}>;
+/**
+ * Record compute usage
+ */
+export declare function recordComputeUsage(userId: string, workspaceId: string, hours: number): Promise<void>;
+/**
+ * Update active agent count for a user
+ */
+export declare function updateActiveAgentCount(userId: string, workspaceId: string, count: number): Promise<void>;
+//# sourceMappingURL=planLimits.d.ts.map

package/dist/cloud/services/planLimits.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"planLimits.d.ts","sourceRoot":"","sources":["../../../src/cloud/services/planLimits.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAM,QAAQ,EAAqB,MAAM,gBAAgB,CAAC;AAIjE;;GAEG;AACH,MAAM,WAAW,UAAU;IACzB,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,uBAAuB,EAAE,MAAM,CAAC;IAChC,mBAAmB,EAAE,OAAO,CAAC;CAC9B;AAED;;;;;;;GAOG;AACH,eAAO,MAAM,WAAW,EAAE,MAAM,CAAC,QAAQ,EAAE,UAAU,CA6BpD,CAAC;AAEF;;GAEG;AACH,wBAAgB,aAAa,CAAC,IAAI,EAAE,QAAQ,GAAG,UAAU,CAExD;AAED;;GAEG;AACH,MAAM,WAAW,SAAS;IACxB,cAAc,EAAE,MAAM,CAAC;IACvB,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,EAAE,MAAM,CAAC;IACzB,qBAAqB,EAAE,MAAM,CAAC;CAC/B;AAED;;GAEG;AACH,wBAAsB,YAAY,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAmDrE;AAED;;GAEG;AACH,wBAAsB,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC;IAChE,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC,CAoBD;AAED;;GAEG;AACH,wBAAsB,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC;IACxD,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC,CAoBD;AAED;;GAEG;AACH,wBAAsB,aAAa,CACjC,MAAM,EAAE,MAAM,EACd,oBAAoB,CAAC,EAAE,MAAM,GAC5B,OAAO,CAAC;IACT,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC,CA4BD;AAED;;GAEG;AACH,wBAAsB,iBAAiB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC;IAC/D,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC,CAkBD;AAED;;GAEG;AACH,wBAAsB,wBAAwB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC;IACtE,SAAS,EAAE,OAAO,CAAC;IACnB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,CAAC,CAyBD;AAED;;GAEG;AACH,wBAAsB,iBAAiB,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC;IAC/D,IAAI,EAAE,QAAQ,CAAC;IACf,MAAM,EAAE,UAAU,CAAC;IACnB,KAAK,EAAE,SAAS,CAAC;IACjB,SAAS,EAAE;QACT,UAAU,EAAE,MAAM,CAAC;QACnB,KAAK,EAAE,MAAM,CAAC;QACd,gBAAgB,EAAE,MAAM,CAAC;QACzB,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;CACH,CAAC,CAoBD;AAED;;GAEG;AACH,wBAAsB,kBAAkB,CACtC,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,MAAM,EACnB,KAAK,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC,CASf;AAED;;GAEG;AACH,wBAAsB,sBAAsB,CAC1C,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,MAAM,EACnB,KAAK,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC,CASf"}