codex-lens 0.1.27 → 0.1.29

package/dist/aggregator.js

@@ -1,6 +1,5 @@
 #!/usr/bin/env node
 import { createServer } from "http";
-import { createHash, createHash as cryptoCreateHash } from "crypto";
 import { WebSocketServer } from "ws";
 import express from "express";
 import { spawn } from "child_process";
@@ -8,6 +7,7 @@ import { createProxyServer } from "./proxy.js";
 import { FileWatcher, scanDirectory } from "./watcher.js";
 import { createLogger } from "./lib/logger.js";
 import { spawnCodex, writeToPty, resizePty, killPty, onPtyData, onPtyExit, getPtyState, getOutputBuffer } from "./pty-manager.js";
+import { createGitManager } from "./git-manager.js";
 import path from "path";
 import { fileURLToPath } from "url";
 import { readFileSync, existsSync, writeFileSync } from "fs";
@@ -50,6 +50,7 @@ class Aggregator {
     this.proxyServer = null;
     this.fileWatcher = null;
     this.ptyProcess = null;
+    this.gitManager = null;
   }
   async start(proxyPort) {
     await fetchLatestVersion();
@@ -135,6 +136,16 @@ class Aggregator {
     await this.proxyServer.start();
     this.fileWatcher = new FileWatcher(this.projectRoot, (event) => this.broadcast(event));
     await this.fileWatcher.start();
+    this.gitManager = createGitManager(this.projectRoot, (event) => this.broadcast(event));
+    if (this.gitManager.isGitRepo()) {
+      await this.gitManager.broadcastUpdate();
+      logger.info("Git repository detected, git status broadcasting enabled");
+      this.fileWatcher.setGitStatusCallback((filePath) => {
+        if (this.gitManager?.isGitRepo()) {
+          this.gitManager.scheduleStatusUpdate();
+        }
+      });
+    }
     await this.startCodex(proxyPort);
     return { httpPort: HTTP_PORT, proxyPort };
   }
@@ -210,7 +221,7 @@ class Aggregator {
       }
     }
   }
-  handleClientMessage(data, ws) {
+  async handleClientMessage(data, ws) {
     if (data.type === "user_message") {
       writeToPty(data.data + "\r");
     } else if (data.type === "open_file") {
@@ -228,6 +239,33 @@ class Aggregator {
           }
         }));
       }
+    } else if (data.type === "git_status_request") {
+      if (this.gitManager?.isGitRepo()) {
+        this.gitManager.broadcastUpdate();
+      }
+    } else if (data.type === "git_stage") {
+      if (this.gitManager?.isGitRepo()) {
+        if (data.filePath) {
+          await this.gitManager.stageFile(data.filePath);
+        } else {
+          await this.gitManager.stageAll();
+        }
+        this.gitManager.broadcastUpdate();
+      }
+    } else if (data.type === "git_unstage") {
+      if (this.gitManager?.isGitRepo()) {
+        if (data.filePath) {
+          await this.gitManager.unstageFile(data.filePath);
+        } else {
+          await this.gitManager.unstageAll();
+        }
+        this.gitManager.broadcastUpdate();
+      }
+    } else if (data.type === "git_commit") {
+      if (this.gitManager?.isGitRepo()) {
+        await this.gitManager.commit(data.message);
+        this.gitManager.broadcastUpdate();
+      }
     }
   }
   broadcast(event) {