agent-relay 2.0.22 → 2.0.24

package/packages/cloud/dist/services/ci-agent-spawner.js

@@ -1,373 +0,0 @@
-/**
- * CI Agent Spawner Service
- *
- * Spawns agents to fix CI failures automatically.
- * Called by the webhook handler when CI checks fail on PRs.
- *
- * Flow:
- * 1. App posts acknowledgment comment on the PR
- * 2. Finds a linked daemon for the repository
- * 3. Queues spawn command for the daemon
- * 4. Agent works and posts response comment
- */
-import { db } from '../db/index.js';
-import { nangoService } from './nango.js';
-/**
- * Get the GitHub App name for comments
- */
-function getAppName() {
-    return process.env.GITHUB_APP_NAME || 'Agent Relay';
-}
-/**
- * Post a CI failure acknowledgment comment on GitHub
- */
-async function postCIAcknowledgmentComment(repository, prNumber, checkName, failureTitle) {
-    if (!repository.nangoConnectionId) {
-        console.warn(`[ci-spawner] Repository ${repository.githubFullName} has no Nango connection`);
-        return null;
-    }
-    const [owner, repo] = repository.githubFullName.split('/');
-    const appName = getAppName();
-    const body = `🔴 **CI Failure Detected**
-
-The \`${checkName}\` check has failed${failureTitle ? `: ${failureTitle}` : ''}.
-
-I'm spawning an agent to investigate and fix this issue. The **@ci-fix** agent will analyze the failure and attempt to resolve it.
-
-You'll be notified when the fix is ready or if manual intervention is needed.
-
-_— ${appName}_`;
-    try {
-        const result = await nangoService.addGithubIssueComment(repository.nangoConnectionId, owner, repo, prNumber, body);
-        console.log(`[ci-spawner] Posted CI acknowledgment comment: ${result.html_url}`);
-        return { id: result.id, url: result.html_url };
-    }
-    catch (error) {
-        console.error(`[ci-spawner] Failed to post CI acknowledgment comment:`, error);
-        return null;
-    }
-}
-/**
- * Post a completion comment on GitHub
- */
-async function postCompletionComment(repository, prNumber, success, summary, commitSha) {
-    if (!repository.nangoConnectionId) {
-        return;
-    }
-    const [owner, repo] = repository.githubFullName.split('/');
-    const appName = getAppName();
-    let body;
-    if (success) {
-        body = `✅ **CI Fix Applied**
-
-${summary}
-
-${commitSha ? `**Commit:** ${commitSha.substring(0, 7)}` : ''}
-
-Please review the changes and re-run the CI checks.
-
-_— ${appName}_`;
-    }
-    else {
-        body = `⚠️ **CI Fix Unsuccessful**
-
-${summary}
-
-Manual intervention may be required. Please check the failure details and fix the issue manually.
-
-_— ${appName}_`;
-    }
-    try {
-        await nangoService.addGithubIssueComment(repository.nangoConnectionId, owner, repo, prNumber, body);
-        console.log(`[ci-spawner] Posted completion comment for PR #${prNumber}`);
-    }
-    catch (error) {
-        console.error(`[ci-spawner] Failed to post completion comment:`, error);
-    }
-}
-/**
- * Find a linked daemon that can handle this repository
- */
-async function findAvailableDaemon(repository) {
-    if (!repository.userId) {
-        console.warn(`[ci-spawner] Repository ${repository.githubFullName} has no userId`);
-        return null;
-    }
-    const daemons = await db.linkedDaemons.findByUserId(repository.userId);
-    const onlineDaemon = daemons.find(d => d.status === 'online');
-    if (!onlineDaemon) {
-        console.warn(`[ci-spawner] No online daemon found for user ${repository.userId}`);
-        return null;
-    }
-    return { id: onlineDaemon.id, userId: repository.userId };
-}
-/**
- * Queue a spawn command for a linked daemon
- */
-async function queueSpawnCommand(daemonId, agentName, prompt, metadata) {
-    const command = {
-        type: 'spawn_agent',
-        agentName,
-        cli: 'claude',
-        task: prompt,
-        metadata,
-        timestamp: new Date().toISOString(),
-    };
-    await db.linkedDaemons.queueMessage(daemonId, {
-        from: { daemonId: 'cloud', daemonName: 'Agent Relay Cloud', agent: 'system' },
-        to: '__spawner__',
-        content: JSON.stringify(command),
-        metadata: { type: 'spawn_command' },
-        timestamp: new Date().toISOString(),
-    });
-    console.log(`[ci-spawner] Queued spawn command for daemon ${daemonId}`);
-}
-/**
- * Spawn an agent to fix CI failures
- *
- * This function:
- * 1. Finds the workspace for the repository
- * 2. Posts acknowledgment comment
- * 3. Creates a fix attempt record
- * 4. Queues spawn command for a linked daemon
- *
- * @param failureEvent - The CI failure event from the database
- */
-export async function spawnCIFixAgent(failureEvent) {
-    console.log(`[ci-spawner] Spawning agent for failure: ${failureEvent.id}`);
-    console.log(`[ci-spawner] Repository: ${failureEvent.repository}`);
-    console.log(`[ci-spawner] Check: ${failureEvent.checkName}`);
-    console.log(`[ci-spawner] PR: #${failureEvent.prNumber}`);
-    // Only handle failures on PRs
-    if (failureEvent.prNumber === null) {
-        console.log(`[ci-spawner] Failure not on a PR, skipping`);
-        return;
-    }
-    const prNumber = failureEvent.prNumber;
-    // Find the repository
-    const repository = await db.repositories.findByFullName(failureEvent.repository);
-    if (!repository) {
-        console.error(`[ci-spawner] Repository not found: ${failureEvent.repository}`);
-        return;
-    }
-    // Generate agent name and ID
-    const agentName = `ci-fix-${failureEvent.checkName.replace(/[^a-zA-Z0-9-]/g, '-')}-${prNumber}`;
-    const agentId = `ci-${failureEvent.id}`;
-    // Create fix attempt record
-    const fixAttempt = await db.ciFixAttempts.create({
-        failureEventId: failureEvent.id,
-        agentId,
-        agentName,
-        status: 'pending',
-    });
-    console.log(`[ci-spawner] Created fix attempt: ${fixAttempt.id}`);
-    try {
-        // Step 1: Post acknowledgment comment
-        await postCIAcknowledgmentComment(repository, prNumber, failureEvent.checkName, failureEvent.failureTitle);
-        // Step 2: Find a linked daemon
-        const daemon = await findAvailableDaemon(repository);
-        if (!daemon) {
-            console.warn(`[ci-spawner] No available daemon for ${failureEvent.repository}`);
-            // Post a comment explaining the situation
-            if (repository.nangoConnectionId) {
-                const [owner, repo] = repository.githubFullName.split('/');
-                try {
-                    await nangoService.addGithubIssueComment(repository.nangoConnectionId, owner, repo, prNumber, `⚠️ I couldn't spawn an agent to fix this CI failure because no Agent Relay daemon is available.
-
-Please ensure you have a linked daemon running by executing \`agent-relay cloud link\` on your development machine.
-
-You can also fix this issue manually by reviewing the failure output above.
-
-_— ${getAppName()}_`);
-                }
-                catch (error) {
-                    console.error(`[ci-spawner] Failed to post error comment:`, error);
-                }
-            }
-            await db.ciFixAttempts.complete(fixAttempt.id, 'failed', undefined, 'No available daemon to spawn agent');
-            return;
-        }
-        // Step 3: Build the agent prompt
-        const prompt = buildAgentPrompt(failureEvent, repository);
-        // Step 4: Update status to in_progress
-        await db.ciFixAttempts.updateStatus(fixAttempt.id, 'in_progress');
-        // Step 5: Queue spawn command for the daemon
-        await queueSpawnCommand(daemon.id, agentName, prompt, {
-            failureEventId: failureEvent.id,
-            fixAttemptId: fixAttempt.id,
-            repository: failureEvent.repository,
-            prNumber,
-            checkName: failureEvent.checkName,
-        });
-        console.log(`[ci-spawner] Successfully queued CI fix agent for ${failureEvent.repository}#${prNumber}`);
-    }
-    catch (error) {
-        console.error(`[ci-spawner] Failed to spawn agent:`, error);
-        await db.ciFixAttempts.complete(fixAttempt.id, 'failed', undefined, error instanceof Error ? error.message : 'Unknown error');
-        throw error;
-    }
-}
-/**
- * Build the prompt for the CI fix agent
- */
-function buildAgentPrompt(failureEvent, _repository) {
-    const annotations = failureEvent.annotations;
-    const annotationsList = annotations && annotations.length > 0
-        ? annotations
-            .slice(0, 20) // Limit to first 20 annotations
-            .map(a => `- ${a.path}:${a.startLine} - ${a.message}`)
-            .join('\n')
-        : null;
-    const responseInstructions = `
-## Response Instructions
-
-When you complete your work:
-1. Commit and push your changes
-2. Post a comment on the PR summarizing what you fixed
-
-Use the GitHub CLI (\`gh\`) to post your response:
-\`\`\`bash
-gh pr comment ${failureEvent.prNumber} --repo ${failureEvent.repository} --body "## CI Fix Applied
-
-Summary of changes...
-
-**Files modified:**
-- file1.ts
-- file2.ts
-
-Please re-run the CI checks to verify the fix."
-\`\`\`
-`;
-    return `
-# CI Failure Fix Task
-
-A CI check has failed on PR #${failureEvent.prNumber} in ${failureEvent.repository}.
-
-## Failure Details
-
-**Check Name:** ${failureEvent.checkName}
-**Branch:** ${failureEvent.branch || 'unknown'}
-**Commit:** ${failureEvent.commitSha || 'unknown'}
-
-${failureEvent.failureTitle ? `**Title:** ${failureEvent.failureTitle}` : ''}
-
-${failureEvent.failureSummary ? `**Summary:**\n${failureEvent.failureSummary}` : ''}
-
-${failureEvent.failureDetails ? `**Details:**\n${failureEvent.failureDetails}` : ''}
-
-${annotationsList ? `## Annotations\n\n${annotationsList}` : ''}
-
-## Your Task
-
-1. Checkout the branch: \`${failureEvent.branch || 'unknown'}\`
-2. Analyze the failure based on the annotations and error messages
-3. Fix the issues in the affected files
-4. Run the relevant checks locally to verify the fix
-5. Commit and push your changes with a clear commit message
-6. Report back with a summary of what was fixed
-
-${responseInstructions}
-
-## Important
-
-- Only fix the specific issues causing the CI failure
-- Do not refactor or improve unrelated code
-- If you cannot fix the issue, explain why and what manual intervention is needed
-- Keep your commit message descriptive and reference the CI check name
-`.trim();
-}
-/**
- * Notify an existing agent about a CI failure
- *
- * Used when an agent is already working on a PR and a new failure occurs.
- *
- * @param agentId - The ID of the existing agent
- * @param failureEvent - The new CI failure event
- */
-export async function notifyAgentOfCIFailure(agentId, failureEvent) {
-    console.log(`[ci-spawner] Notifying agent ${agentId} of new failure`);
-    // Find the repository
-    const repository = await db.repositories.findByFullName(failureEvent.repository);
-    if (!repository || !repository.userId) {
-        console.warn(`[ci-spawner] Repository not found or has no userId: ${failureEvent.repository}`);
-        return;
-    }
-    // Find the daemon that should have this agent
-    const daemons = await db.linkedDaemons.findByUserId(repository.userId);
-    const onlineDaemon = daemons.find(d => d.status === 'online');
-    if (!onlineDaemon) {
-        console.warn(`[ci-spawner] No online daemon to notify agent ${agentId}`);
-        return;
-    }
-    // Build notification message
-    const annotations = failureEvent.annotations;
-    const annotationsList = annotations && annotations.length > 0
-        ? annotations
-            .slice(0, 10)
-            .map(a => `  - ${a.path}:${a.startLine}: ${a.message}`)
-            .join('\n')
-        : null;
-    const message = `
-CI FAILURE: ${failureEvent.checkName}
-
-${failureEvent.failureTitle || 'Check failed'}
-
-${failureEvent.failureSummary || ''}
-
-${annotationsList ? `Issues:\n${annotationsList}` : ''}
-
-Please investigate and fix these issues, then push your changes.
-`.trim();
-    // Queue message for the agent via daemon
-    await db.linkedDaemons.queueMessage(onlineDaemon.id, {
-        from: { daemonId: 'cloud', daemonName: 'Agent Relay Cloud', agent: 'system' },
-        to: agentId,
-        content: message,
-        metadata: { type: 'ci_failure_notification', failureEventId: failureEvent.id },
-        timestamp: new Date().toISOString(),
-    });
-    console.log(`[ci-spawner] Queued CI failure notification for agent ${agentId}`);
-}
-/**
- * Mark a fix attempt as complete
- *
- * Called when an agent reports completion (success or failure)
- */
-export async function completeFixAttempt(fixAttemptId, success, commitSha, errorMessage) {
-    console.log(`[ci-spawner] Completing fix attempt ${fixAttemptId}: ${success ? 'success' : 'failed'}`);
-    // Update the fix attempt record
-    await db.ciFixAttempts.complete(fixAttemptId, success ? 'success' : 'failed', commitSha, errorMessage);
-    // Get the fix attempt to find the failure event
-    const fixAttempt = await db.ciFixAttempts.findById(fixAttemptId);
-    if (!fixAttempt) {
-        console.warn(`[ci-spawner] Fix attempt not found: ${fixAttemptId}`);
-        return;
-    }
-    // Get the failure event to find the repository and PR
-    const failureEvent = await db.ciFailureEvents.findById(fixAttempt.failureEventId);
-    if (!failureEvent) {
-        console.warn(`[ci-spawner] Failure event not found: ${fixAttempt.failureEventId}`);
-        return;
-    }
-    // Find the repository to post completion comment
-    const repository = await db.repositories.findByFullName(failureEvent.repository);
-    if (repository && failureEvent.prNumber !== null) {
-        const summary = success
-            ? `The @ci-fix agent has fixed the \`${failureEvent.checkName}\` check failure.`
-            : errorMessage || 'The @ci-fix agent was unable to fix the issue.';
-        await postCompletionComment(repository, failureEvent.prNumber, success, summary, commitSha);
-    }
-}
-/**
- * Get failure history for a repository
- */
-export async function getFailureHistory(repository, limit = 50) {
-    return db.ciFailureEvents.findByRepository(repository, limit);
-}
-/**
- * Get failure history for a specific PR
- */
-export async function getPRFailureHistory(repository, prNumber) {
-    return db.ciFailureEvents.findByPR(repository, prNumber);
-}
-//# sourceMappingURL=ci-agent-spawner.js.map

package/packages/cloud/dist/services/cloud-message-bus.d.ts

@@ -1,28 +0,0 @@
-/**
- * Cloud Message Bus - Event-based message delivery for cloud users.
- *
- * This module provides a simple pub/sub mechanism for delivering messages
- * to users connected via the cloud dashboard. The daemons API publishes
- * messages here, and the presence WebSocket handler subscribes to deliver them.
- */
-import { EventEmitter } from 'events';
-export interface CloudMessage {
-    from: {
-        daemonId: string;
-        daemonName: string;
-        agent: string;
-    };
-    to: string;
-    body: string;
-    timestamp: string;
-    metadata?: Record<string, unknown>;
-}
-declare class CloudMessageBus extends EventEmitter {
-    /**
-     * Send a message to a cloud user
-     */
-    sendToUser(username: string, message: CloudMessage): void;
-}
-export declare const cloudMessageBus: CloudMessageBus;
-export {};
-//# sourceMappingURL=cloud-message-bus.d.ts.map

package/packages/cloud/dist/services/cloud-message-bus.js

@@ -1,19 +0,0 @@
-/**
- * Cloud Message Bus - Event-based message delivery for cloud users.
- *
- * This module provides a simple pub/sub mechanism for delivering messages
- * to users connected via the cloud dashboard. The daemons API publishes
- * messages here, and the presence WebSocket handler subscribes to deliver them.
- */
-import { EventEmitter } from 'events';
-class CloudMessageBus extends EventEmitter {
-    /**
-     * Send a message to a cloud user
-     */
-    sendToUser(username, message) {
-        this.emit('user-message', { username, message });
-    }
-}
-// Singleton instance
-export const cloudMessageBus = new CloudMessageBus();
-//# sourceMappingURL=cloud-message-bus.js.map

package/packages/cloud/dist/services/compute-enforcement.d.ts

@@ -1,57 +0,0 @@
-/**
- * Compute Enforcement Service
- *
- * Enforces compute hour limits for free tier users.
- * Runs periodically to check usage and stop workspaces that have exceeded limits.
- */
-import { PlanType } from '../db/index.js';
-export interface ComputeEnforcementConfig {
-    enabled: boolean;
-    checkIntervalMs: number;
-    warningThresholdPercent: number;
-}
-export interface EnforcementResult {
-    userId: string;
-    plan: PlanType;
-    computeHoursUsed: number;
-    computeHoursLimit: number;
-    action: 'none' | 'warning' | 'stopped';
-    workspacesStopped: string[];
-}
-export declare class ComputeEnforcementService {
-    private config;
-    private checkTimer;
-    private isRunning;
-    constructor(config?: Partial<ComputeEnforcementConfig>);
-    /**
-     * Start the enforcement service
-     */
-    start(): void;
-    /**
-     * Stop the enforcement service
-     */
-    stop(): void;
-    /**
-     * Run enforcement check for all free tier users
-     */
-    runEnforcement(): Promise<EnforcementResult[]>;
-    /**
-     * Enforce limits for a specific user
-     */
-    enforceUserLimits(userId: string): Promise<EnforcementResult>;
-    /**
-     * Manually trigger enforcement for a specific user
-     */
-    enforceUser(userId: string): Promise<EnforcementResult>;
-    /**
-     * Get service status
-     */
-    getStatus(): {
-        enabled: boolean;
-        isRunning: boolean;
-        checkIntervalMs: number;
-    };
-}
-export declare function getComputeEnforcementService(): ComputeEnforcementService;
-export declare function createComputeEnforcementService(config?: Partial<ComputeEnforcementConfig>): ComputeEnforcementService;
-//# sourceMappingURL=compute-enforcement.d.ts.map

package/packages/cloud/dist/services/compute-enforcement.js

@@ -1,175 +0,0 @@
-/**
- * Compute Enforcement Service
- *
- * Enforces compute hour limits for free tier users.
- * Runs periodically to check usage and stop workspaces that have exceeded limits.
- */
-import { db } from '../db/index.js';
-import { getProvisioner } from '../provisioner/index.js';
-import { getUserUsage, PLAN_LIMITS } from './planLimits.js';
-const DEFAULT_CONFIG = {
-    enabled: true,
-    checkIntervalMs: 15 * 60 * 1000, // 15 minutes
-    warningThresholdPercent: 80,
-};
-export class ComputeEnforcementService {
-    config;
-    checkTimer = null;
-    isRunning = false;
-    constructor(config = {}) {
-        this.config = { ...DEFAULT_CONFIG, ...config };
-    }
-    /**
-     * Start the enforcement service
-     */
-    start() {
-        if (!this.config.enabled) {
-            console.log('[compute-enforcement] Service disabled');
-            return;
-        }
-        if (this.isRunning) {
-            console.warn('[compute-enforcement] Service already running');
-            return;
-        }
-        this.isRunning = true;
-        console.log(`[compute-enforcement] Started (checking every ${this.config.checkIntervalMs / 1000}s)`);
-        // Run immediately on start
-        this.runEnforcement().catch((err) => {
-            console.error('[compute-enforcement] Initial run failed:', err);
-        });
-        // Then run periodically
-        this.checkTimer = setInterval(() => {
-            this.runEnforcement().catch((err) => {
-                console.error('[compute-enforcement] Periodic run failed:', err);
-            });
-        }, this.config.checkIntervalMs);
-    }
-    /**
-     * Stop the enforcement service
-     */
-    stop() {
-        if (this.checkTimer) {
-            clearInterval(this.checkTimer);
-            this.checkTimer = null;
-        }
-        this.isRunning = false;
-        console.log('[compute-enforcement] Stopped');
-    }
-    /**
-     * Run enforcement check for all free tier users
-     */
-    async runEnforcement() {
-        const results = [];
-        try {
-            // Get all users on free tier
-            const freeUsers = await db.users.findByPlan('free');
-            console.log(`[compute-enforcement] Checking ${freeUsers.length} free tier users`);
-            for (const user of freeUsers) {
-                try {
-                    const result = await this.enforceUserLimits(user.id);
-                    results.push(result);
-                    if (result.action !== 'none') {
-                        console.log(`[compute-enforcement] User ${user.id.substring(0, 8)}: ${result.action} ` +
-                            `(${result.computeHoursUsed.toFixed(2)}/${result.computeHoursLimit}h)`);
-                    }
-                }
-                catch (err) {
-                    console.error(`[compute-enforcement] Error for user ${user.id}:`, err);
-                }
-            }
-            const stopped = results.filter((r) => r.action === 'stopped').length;
-            const warned = results.filter((r) => r.action === 'warning').length;
-            if (stopped > 0 || warned > 0) {
-                console.log(`[compute-enforcement] Summary: ${stopped} stopped, ${warned} warned, ` +
-                    `${results.length - stopped - warned} ok`);
-            }
-        }
-        catch (err) {
-            console.error('[compute-enforcement] Failed to run enforcement:', err);
-        }
-        return results;
-    }
-    /**
-     * Enforce limits for a specific user
-     */
-    async enforceUserLimits(userId) {
-        const user = await db.users.findById(userId);
-        const plan = user?.plan || 'free';
-        const limits = PLAN_LIMITS[plan];
-        const usage = await getUserUsage(userId);
-        const result = {
-            userId,
-            plan,
-            computeHoursUsed: usage.computeHoursThisMonth,
-            computeHoursLimit: limits.maxComputeHoursPerMonth,
-            action: 'none',
-            workspacesStopped: [],
-        };
-        // Skip if user has unlimited compute (paid plans may have high limits)
-        if (limits.maxComputeHoursPerMonth === Infinity) {
-            return result;
-        }
-        // Check if user has exceeded limit
-        if (usage.computeHoursThisMonth >= limits.maxComputeHoursPerMonth) {
-            // Stop all running workspaces
-            const workspaces = await db.workspaces.findByUserId(userId);
-            const runningWorkspaces = workspaces.filter((w) => w.status === 'running');
-            if (runningWorkspaces.length > 0) {
-                const provisioner = getProvisioner();
-                for (const workspace of runningWorkspaces) {
-                    try {
-                        await provisioner.stop(workspace.id);
-                        result.workspacesStopped.push(workspace.id);
-                        console.log(`[compute-enforcement] Stopped workspace ${workspace.id.substring(0, 8)} ` +
-                            `for user ${userId.substring(0, 8)} (limit exceeded)`);
-                    }
-                    catch (err) {
-                        console.error(`[compute-enforcement] Failed to stop workspace ${workspace.id}:`, err);
-                    }
-                }
-                result.action = 'stopped';
-                // TODO: Send notification email to user
-                // await sendLimitReachedEmail(userId, usage.computeHoursThisMonth, limits.maxComputeHoursPerMonth);
-            }
-        }
-        else {
-            // Check if approaching limit (warning)
-            const usagePercent = (usage.computeHoursThisMonth / limits.maxComputeHoursPerMonth) * 100;
-            if (usagePercent >= this.config.warningThresholdPercent) {
-                result.action = 'warning';
-                // TODO: Send warning email to user (once per day)
-                // await sendLimitWarningEmail(userId, usage.computeHoursThisMonth, limits.maxComputeHoursPerMonth);
-            }
-        }
-        return result;
-    }
-    /**
-     * Manually trigger enforcement for a specific user
-     */
-    async enforceUser(userId) {
-        return this.enforceUserLimits(userId);
-    }
-    /**
-     * Get service status
-     */
-    getStatus() {
-        return {
-            enabled: this.config.enabled,
-            isRunning: this.isRunning,
-            checkIntervalMs: this.config.checkIntervalMs,
-        };
-    }
-}
-// Singleton instance
-let _computeEnforcement = null;
-export function getComputeEnforcementService() {
-    if (!_computeEnforcement) {
-        _computeEnforcement = new ComputeEnforcementService();
-    }
-    return _computeEnforcement;
-}
-export function createComputeEnforcementService(config = {}) {
-    _computeEnforcement = new ComputeEnforcementService(config);
-    return _computeEnforcement;
-}
-//# sourceMappingURL=compute-enforcement.js.map