@gowelle/stint-agent 1.0.8 → 1.2.0

package/dist/chunk-IJUJ6NEL.js

@@ -0,0 +1,369 @@
+import {
+  apiService
+} from "./chunk-W4JGOGR7.js";
+import {
+  gitService,
+  projectService
+} from "./chunk-FBQA4K5J.js";
+import {
+  authService,
+  config,
+  logger
+} from "./chunk-RHMTZK2J.js";
+
+// src/daemon/queue.ts
+var CommitQueueProcessor = class {
+  queue = [];
+  isProcessing = false;
+  /**
+   * Add commit to processing queue
+   */
+  addToQueue(commit, project) {
+    this.queue.push({ commit, project });
+    logger.info("queue", `Added commit ${commit.id} to queue (position: ${this.queue.length})`);
+    if (!this.isProcessing) {
+      this.processQueue();
+    }
+  }
+  /**
+   * Process commits sequentially
+   */
+  async processQueue() {
+    if (this.isProcessing) {
+      return;
+    }
+    this.isProcessing = true;
+    while (this.queue.length > 0) {
+      const item = this.queue.shift();
+      if (!item) break;
+      try {
+        await this.executeCommit(item.commit, item.project);
+      } catch (error) {
+        logger.error("queue", `Failed to execute commit ${item.commit.id}`, error);
+      }
+    }
+    this.isProcessing = false;
+  }
+  /**
+   * Execute a single commit
+   */
+  async executeCommit(commit, project, onProgress) {
+    logger.info("queue", `Processing commit: ${commit.id} - ${commit.message}`);
+    try {
+      onProgress?.("Finding project directory...");
+      const projectPath = this.findProjectPath(project.id);
+      if (!projectPath) {
+        throw new Error(`Project ${project.id} is not linked to any local directory`);
+      }
+      logger.info("queue", `Executing in directory: ${projectPath}`);
+      onProgress?.("Validating repository...");
+      const isRepo = await gitService.isRepo(projectPath);
+      if (!isRepo) {
+        throw new Error(`Directory ${projectPath} is not a git repository`);
+      }
+      onProgress?.("Checking repository status...");
+      const status = await gitService.getStatus(projectPath);
+      const hasStagedChanges = status.staged.length > 0;
+      if (!hasStagedChanges) {
+        throw new Error('No staged changes to commit. Please stage files using "git add" before committing.');
+      }
+      logger.info("queue", `Committing ${status.staged.length} staged files.`);
+      onProgress?.("Creating commit...");
+      logger.info("queue", `Creating commit with message: "${commit.message}"`);
+      const sha = await gitService.commit(projectPath, commit.message);
+      logger.success("queue", `Commit created successfully: ${sha}`);
+      onProgress?.("Reporting to server...");
+      await this.reportSuccess(commit.id, sha);
+      return sha;
+    } catch (error) {
+      const errorMessage = error.message;
+      logger.error("queue", `Commit execution failed: ${errorMessage}`);
+      await this.reportFailure(commit.id, errorMessage);
+      throw error;
+    }
+  }
+  /**
+   * Report successful execution to API
+   */
+  async reportSuccess(commitId, sha) {
+    try {
+      await apiService.markCommitExecuted(commitId, sha);
+      logger.success("queue", `Reported commit execution to API: ${commitId} -> ${sha}`);
+    } catch (error) {
+      logger.error("queue", "Failed to report commit success to API", error);
+    }
+  }
+  /**
+   * Report failed execution to API
+   */
+  async reportFailure(commitId, error) {
+    try {
+      await apiService.markCommitFailed(commitId, error);
+      logger.info("queue", `Reported commit failure to API: ${commitId}`);
+    } catch (apiError) {
+      logger.error("queue", "Failed to report commit failure to API", apiError);
+    }
+  }
+  /**
+   * Find local path for a project ID
+   */
+  findProjectPath(projectId) {
+    const allProjects = projectService.getAllLinkedProjects();
+    for (const [path, linkedProject] of Object.entries(allProjects)) {
+      if (linkedProject.projectId === projectId) {
+        return path;
+      }
+    }
+    return null;
+  }
+  /**
+   * Check if queue is currently processing
+   */
+  isCurrentlyProcessing() {
+    return this.isProcessing;
+  }
+  /**
+   * Get queue length
+   */
+  getQueueLength() {
+    return this.queue.length;
+  }
+};
+var commitQueue = new CommitQueueProcessor();
+
+// src/services/websocket.ts
+import WebSocket from "ws";
+var WebSocketServiceImpl = class {
+  ws = null;
+  userId = null;
+  reconnectAttempts = 0;
+  maxReconnectAttempts = 10;
+  reconnectTimer = null;
+  isManualDisconnect = false;
+  // Event handlers
+  commitApprovedHandlers = [];
+  commitPendingHandlers = [];
+  suggestionCreatedHandlers = [];
+  projectUpdatedHandlers = [];
+  disconnectHandlers = [];
+  agentDisconnectedHandlers = [];
+  syncRequestedHandlers = [];
+  /**
+   * Connect to the WebSocket server
+   * @throws Error if connection fails or no auth token available
+   */
+  async connect() {
+    try {
+      const token = await authService.getToken();
+      if (!token) {
+        throw new Error("No authentication token available");
+      }
+      const wsUrl = config.getWsUrl();
+      const url = `${wsUrl}?token=${encodeURIComponent(token)}`;
+      logger.info("websocket", `Connecting to ${wsUrl}...`);
+      this.ws = new WebSocket(url);
+      return new Promise((resolve, reject) => {
+        if (!this.ws) {
+          reject(new Error("WebSocket not initialized"));
+          return;
+        }
+        this.ws.on("open", () => {
+          logger.success("websocket", "WebSocket connected");
+          this.reconnectAttempts = 0;
+          this.isManualDisconnect = false;
+          resolve();
+        });
+        this.ws.on("message", (data) => {
+          this.handleMessage(data);
+        });
+        this.ws.on("close", () => {
+          logger.warn("websocket", "WebSocket disconnected");
+          this.handleDisconnect();
+        });
+        this.ws.on("error", (error) => {
+          logger.error("websocket", "WebSocket error", error);
+          reject(error);
+        });
+      });
+    } catch (error) {
+      logger.error("websocket", "Failed to connect", error);
+      throw error;
+    }
+  }
+  /**
+   * Disconnect from the WebSocket server
+   * Prevents automatic reconnection
+   */
+  disconnect() {
+    this.isManualDisconnect = true;
+    if (this.reconnectTimer) {
+      clearTimeout(this.reconnectTimer);
+      this.reconnectTimer = null;
+    }
+    if (this.ws) {
+      if (this.userId) {
+        this.sendMessage({
+          event: "pusher:unsubscribe",
+          data: {
+            channel: `private-user.${this.userId}`
+          }
+        });
+      }
+      this.ws.close();
+      this.ws = null;
+      logger.info("websocket", "WebSocket disconnected");
+    }
+  }
+  /**
+   * Check if WebSocket is currently connected
+   * @returns True if connected and ready
+   */
+  isConnected() {
+    return this.ws !== null && this.ws.readyState === WebSocket.OPEN;
+  }
+  /**
+   * Subscribe to user-specific channel for real-time updates
+   * @param userId - User ID to subscribe to
+   */
+  subscribeToUserChannel(userId) {
+    this.userId = userId;
+    if (!this.isConnected()) {
+      logger.warn("websocket", "Cannot subscribe: not connected");
+      return;
+    }
+    const channel = `private-user.${userId}`;
+    logger.info("websocket", `Subscribing to channel: ${channel}`);
+    this.sendMessage({
+      event: "pusher:subscribe",
+      data: {
+        channel
+      }
+    });
+  }
+  /**
+   * Register handler for commit approved events
+   * @param handler - Callback function
+   */
+  onCommitApproved(handler) {
+    this.commitApprovedHandlers.push(handler);
+  }
+  onCommitPending(handler) {
+    this.commitPendingHandlers.push(handler);
+  }
+  onSuggestionCreated(handler) {
+    this.suggestionCreatedHandlers.push(handler);
+  }
+  onProjectUpdated(handler) {
+    this.projectUpdatedHandlers.push(handler);
+  }
+  onDisconnect(handler) {
+    this.disconnectHandlers.push(handler);
+  }
+  onAgentDisconnected(handler) {
+    this.agentDisconnectedHandlers.push(handler);
+  }
+  onSyncRequested(handler) {
+    this.syncRequestedHandlers.push(handler);
+  }
+  sendMessage(message) {
+    if (!this.isConnected()) {
+      logger.warn("websocket", "Cannot send message: not connected");
+      return;
+    }
+    this.ws.send(JSON.stringify(message));
+  }
+  handleMessage(data) {
+    try {
+      const message = JSON.parse(data.toString());
+      logger.debug("websocket", `Received message: ${message.event}`);
+      if (message.event === "pusher:connection_established") {
+        logger.success("websocket", "Connection established");
+        return;
+      }
+      if (message.event === "pusher_internal:subscription_succeeded") {
+        logger.success("websocket", `Subscribed to channel: ${message.channel}`);
+        return;
+      }
+      if (message.event === "commit.approved") {
+        const { commit, project } = message.data;
+        logger.info("websocket", `Commit approved: ${commit.id}`);
+        this.commitApprovedHandlers.forEach((handler) => handler(commit, project));
+        return;
+      }
+      if (message.event === "commit.pending") {
+        const { pendingCommit } = message.data;
+        logger.info("websocket", `Commit pending: ${pendingCommit.id}`);
+        this.commitPendingHandlers.forEach((handler) => handler(pendingCommit));
+        return;
+      }
+      if (message.event === "suggestion.created") {
+        const { suggestion } = message.data;
+        logger.info("websocket", `Suggestion created: ${suggestion.id}`);
+        this.suggestionCreatedHandlers.forEach((handler) => handler(suggestion));
+        return;
+      }
+      if (message.event === "project.updated") {
+        const { project } = message.data;
+        logger.info("websocket", `Project updated: ${project.id}`);
+        this.projectUpdatedHandlers.forEach((handler) => handler(project));
+        return;
+      }
+      if (message.event === "sync.requested") {
+        const { projectId } = message.data;
+        logger.info("websocket", `Sync requested for project: ${projectId}`);
+        this.syncRequestedHandlers.forEach((handler) => handler(projectId));
+        return;
+      }
+      if (message.event === "agent.disconnected") {
+        const reason = message.data?.reason || "Server requested disconnect";
+        logger.warn("websocket", `Agent disconnected by server: ${reason}`);
+        this.agentDisconnectedHandlers.forEach((handler) => handler(reason));
+        return;
+      }
+      logger.debug("websocket", `Unhandled event: ${message.event}`);
+    } catch (error) {
+      logger.error("websocket", "Failed to parse message", error);
+    }
+  }
+  handleDisconnect() {
+    this.ws = null;
+    this.disconnectHandlers.forEach((handler) => handler());
+    if (this.isManualDisconnect) {
+      return;
+    }
+    if (this.reconnectAttempts < this.maxReconnectAttempts) {
+      const delay = this.getReconnectDelay();
+      this.reconnectAttempts++;
+      logger.info("websocket", `Reconnecting in ${delay}ms (attempt ${this.reconnectAttempts}/${this.maxReconnectAttempts})`);
+      this.reconnectTimer = setTimeout(async () => {
+        try {
+          await this.connect();
+          if (this.userId) {
+            this.subscribeToUserChannel(this.userId);
+          }
+        } catch (error) {
+          logger.error("websocket", "Reconnection failed", error);
+        }
+      }, delay);
+    } else {
+      logger.error("websocket", "Max reconnection attempts reached");
+    }
+  }
+  /**
+   * Get reconnect delay with exponential backoff and jitter
+   * Jitter prevents thundering herd problem when many clients reconnect simultaneously
+   */
+  getReconnectDelay() {
+    const delays = [1e3, 2e3, 4e3, 8e3, 16e3, 3e4];
+    const index = Math.min(this.reconnectAttempts, delays.length - 1);
+    const baseDelay = delays[index];
+    const jitter = baseDelay * (Math.random() * 0.3);
+    return Math.floor(baseDelay + jitter);
+  }
+};
+var websocketService = new WebSocketServiceImpl();
+
+export {
+  commitQueue,
+  websocketService
+};