codex-lens 0.1.31 → 0.1.32

package/dist/aggregator.js

@@ -150,26 +150,14 @@ class Aggregator {
     return { httpPort: HTTP_PORT, proxyPort };
   }
   setupMainWebSocket() {
-    this.wss.on("connection", (ws) => {
+    this.wss.on("connection", async (ws) => {
       logger.info("API client connected");
       this.clients.add(ws);
       const fileTree = scanDirectory(this.projectRoot);
       logger.info(`Sending file_tree with ${fileTree.length} top-level items`);
       ws.send(JSON.stringify({ type: "file_tree", data: fileTree }));
       if (this.gitManager?.isGitRepo()) {
-        const branch = this.gitManager.getCurrentBranch();
-        const status = this.gitManager.currentStatus;
-        ws.send(JSON.stringify({
-          type: "git_status",
-          data: {
-            isRepo: true,
-            branch,
-            status,
-            stagedCount: status?.staged?.length || 0,
-            unstagedCount: (status?.unstaged?.length || 0) + (status?.untracked?.length || 0),
-            timestamp: (/* @__PURE__ */ new Date()).toISOString()
-          }
-        }));
+        await this.gitManager.broadcastUpdate();
       }
       ws.on("close", () => {
         logger.info("API client disconnected");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "codex-lens",
-  "version": "0.1.31",
+  "version": "0.1.32",
   "description": "A visualization tool for Codex that monitors API requests and file system changes",
   "license": "MIT",
   "type": "module",

package/src/aggregator.js

@@ -178,7 +178,7 @@ class Aggregator {
   }
 
   setupMainWebSocket() {
-    this.wss.on('connection', (ws) => {
+    this.wss.on('connection', async (ws) => {
       logger.info('API client connected');
       this.clients.add(ws);
 
@@ -188,19 +188,7 @@ class Aggregator {
 
       // Send initial git status if available
       if (this.gitManager?.isGitRepo()) {
-        const branch = this.gitManager.getCurrentBranch();
-        const status = this.gitManager.currentStatus;
-        ws.send(JSON.stringify({
-          type: 'git_status',
-          data: {
-            isRepo: true,
-            branch,
-            status,
-            stagedCount: status?.staged?.length || 0,
-            unstagedCount: (status?.unstaged?.length || 0) + (status?.untracked?.length || 0),
-            timestamp: new Date().toISOString()
-          }
-        }));
+        await this.gitManager.broadcastUpdate();
       }
 
       ws.on('close', () => {