@sudocode-ai/local-server 0.1.0 → 0.1.1

package/dist/services/websocket.js

@@ -0,0 +1,389 @@
+import { WebSocketServer, WebSocket } from "ws";
+import { randomUUID } from "crypto";
+const LOG_CONNECTIONS = false;
+/**
+ * WebSocket server manager
+ */
+class WebSocketManager {
+    wss = null;
+    clients = new Map();
+    heartbeatInterval = null;
+    HEARTBEAT_INTERVAL = 30000; // 30 seconds
+    /**
+     * Initialize the WebSocket server
+     */
+    init(server, path = "/ws") {
+        if (this.wss) {
+            console.warn("[websocket] WebSocket server already initialized");
+            return;
+        }
+        this.wss = new WebSocketServer({ server, path });
+        console.log(`[websocket] WebSocket server initialized on path: ${path}`);
+        this.wss.on("connection", this.handleConnection.bind(this));
+        this.startHeartbeat();
+    }
+    /**
+     * Handle new WebSocket connection
+     */
+    handleConnection(ws, _req) {
+        const clientId = randomUUID();
+        const client = {
+            id: clientId,
+            ws,
+            subscriptions: new Set(),
+            isAlive: true,
+            connectedAt: new Date(),
+        };
+        this.clients.set(clientId, client);
+        if (LOG_CONNECTIONS) {
+            console.log(`[websocket] Client connected: ${clientId} (total: ${this.clients.size})`);
+        }
+        // Set up event handlers
+        ws.on("message", (data) => this.handleMessage(clientId, data));
+        ws.on("close", () => this.handleDisconnection(clientId));
+        ws.on("error", (error) => this.handleError(clientId, error));
+        ws.on("pong", () => this.handlePong(clientId));
+        // Send welcome message
+        this.sendToClient(clientId, {
+            type: "pong",
+            message: "Connected to sudocode server",
+        });
+    }
+    /**
+     * Handle client disconnection
+     */
+    handleDisconnection(clientId) {
+        const client = this.clients.get(clientId);
+        if (client) {
+            if (LOG_CONNECTIONS) {
+                console.log(`[websocket] Client disconnected: ${clientId} (subscriptions: ${client.subscriptions.size})`);
+            }
+            this.clients.delete(clientId);
+        }
+    }
+    /**
+     * Handle client errors
+     */
+    handleError(clientId, error) {
+        console.error(`[websocket] Client error (${clientId}):`, error.message);
+    }
+    /**
+     * Handle pong response from client
+     */
+    handlePong(clientId) {
+        const client = this.clients.get(clientId);
+        if (client) {
+            client.isAlive = true;
+        }
+    }
+    /**
+     * Handle incoming messages from clients
+     */
+    handleMessage(clientId, data) {
+        const client = this.clients.get(clientId);
+        if (!client) {
+            return;
+        }
+        try {
+            const message = JSON.parse(data.toString());
+            switch (message.type) {
+                case "subscribe":
+                    this.handleSubscribe(clientId, message);
+                    break;
+                case "unsubscribe":
+                    this.handleUnsubscribe(clientId, message);
+                    break;
+                case "ping":
+                    this.sendToClient(clientId, { type: "pong" });
+                    break;
+                default:
+                    this.sendToClient(clientId, {
+                        type: "error",
+                        message: `Unknown message type: ${message.type}`,
+                    });
+            }
+        }
+        catch (error) {
+            console.error(`[websocket] Failed to parse message from ${clientId}:`, error);
+            this.sendToClient(clientId, {
+                type: "error",
+                message: "Invalid message format",
+            });
+        }
+    }
+    /**
+     * Handle subscription request
+     */
+    handleSubscribe(clientId, message) {
+        const client = this.clients.get(clientId);
+        if (!client) {
+            return;
+        }
+        let subscription;
+        if (message.entity_type === "all") {
+            // Subscribe to all updates
+            subscription = "all";
+        }
+        else if (message.entity_type && message.entity_id) {
+            // Subscribe to a specific entity
+            subscription = `${message.entity_type}:${message.entity_id}`;
+        }
+        else if (message.entity_type) {
+            // Subscribe to all entities of a type
+            subscription = `${message.entity_type}:*`;
+        }
+        else {
+            this.sendToClient(clientId, {
+                type: "error",
+                message: "Invalid subscription request",
+            });
+            return;
+        }
+        client.subscriptions.add(subscription);
+        if (LOG_CONNECTIONS) {
+            console.log(`[websocket] Client ${clientId} subscribed to: ${subscription}`);
+        }
+        this.sendToClient(clientId, {
+            type: "subscribed",
+            subscription,
+            message: `Subscribed to ${subscription}`,
+        });
+    }
+    /**
+     * Handle unsubscription request
+     */
+    handleUnsubscribe(clientId, message) {
+        const client = this.clients.get(clientId);
+        if (!client) {
+            return;
+        }
+        let subscription;
+        if (message.entity_type === "all") {
+            subscription = "all";
+        }
+        else if (message.entity_type && message.entity_id) {
+            subscription = `${message.entity_type}:${message.entity_id}`;
+        }
+        else if (message.entity_type) {
+            subscription = `${message.entity_type}:*`;
+        }
+        else {
+            this.sendToClient(clientId, {
+                type: "error",
+                message: "Invalid unsubscription request",
+            });
+            return;
+        }
+        client.subscriptions.delete(subscription);
+        if (LOG_CONNECTIONS) {
+            console.log(`[websocket] Client ${clientId} unsubscribed from: ${subscription}`);
+        }
+        this.sendToClient(clientId, {
+            type: "unsubscribed",
+            subscription,
+            message: `Unsubscribed from ${subscription}`,
+        });
+    }
+    /**
+     * Send a message to a specific client
+     */
+    sendToClient(clientId, message) {
+        const client = this.clients.get(clientId);
+        if (!client || client.ws.readyState !== WebSocket.OPEN) {
+            return;
+        }
+        try {
+            client.ws.send(JSON.stringify(message));
+        }
+        catch (error) {
+            console.error(`[websocket] Failed to send message to ${clientId}:`, error);
+        }
+    }
+    /**
+     * Broadcast a message to all subscribed clients
+     */
+    broadcast(entityType, entityId, message) {
+        const subscription = `${entityType}:${entityId}`;
+        const typeSubscription = `${entityType}:*`;
+        let sentCount = 0;
+        this.clients.forEach((client) => {
+            if (client.ws.readyState !== WebSocket.OPEN) {
+                return;
+            }
+            // Check if client is subscribed to this specific entity, entity type, or all
+            if (client.subscriptions.has(subscription) ||
+                client.subscriptions.has(typeSubscription) ||
+                client.subscriptions.has("all")) {
+                try {
+                    client.ws.send(JSON.stringify(message));
+                    sentCount++;
+                }
+                catch (error) {
+                    console.error(`[websocket] Failed to broadcast to ${client.id}:`, error);
+                }
+            }
+        });
+        if (sentCount > 0) {
+            console.log(`[websocket] Broadcasted ${message.type} for ${subscription} to ${sentCount} clients`);
+        }
+    }
+    /**
+     * Broadcast a relationship or feedback update
+     * These don't have a specific entity type in the subscription, so we broadcast to 'all' subscribers
+     */
+    broadcastGeneric(message) {
+        let sentCount = 0;
+        this.clients.forEach((client) => {
+            if (client.ws.readyState !== WebSocket.OPEN) {
+                return;
+            }
+            // Only send to clients subscribed to 'all'
+            if (client.subscriptions.has("all")) {
+                try {
+                    client.ws.send(JSON.stringify(message));
+                    sentCount++;
+                }
+                catch (error) {
+                    console.error(`[websocket] Failed to broadcast to ${client.id}:`, error);
+                }
+            }
+        });
+        if (sentCount > 0) {
+            console.log(`[websocket] Broadcasted ${message.type} to ${sentCount} clients`);
+        }
+    }
+    /**
+     * Start the heartbeat mechanism to detect dead connections
+     */
+    startHeartbeat() {
+        if (this.heartbeatInterval) {
+            return;
+        }
+        this.heartbeatInterval = setInterval(() => {
+            this.clients.forEach((client, clientId) => {
+                if (!client.isAlive) {
+                    console.log(`[websocket] Terminating dead connection: ${clientId}`);
+                    client.ws.terminate();
+                    this.clients.delete(clientId);
+                    return;
+                }
+                client.isAlive = false;
+                try {
+                    client.ws.ping();
+                }
+                catch (error) {
+                    console.error(`[websocket] Failed to ping ${clientId}:`, error);
+                }
+            });
+        }, this.HEARTBEAT_INTERVAL);
+        console.log(`[websocket] Heartbeat started (interval: ${this.HEARTBEAT_INTERVAL}ms)`);
+    }
+    /**
+     * Stop the heartbeat mechanism
+     */
+    stopHeartbeat() {
+        if (this.heartbeatInterval) {
+            clearInterval(this.heartbeatInterval);
+            this.heartbeatInterval = null;
+            console.log("[websocket] Heartbeat stopped");
+        }
+    }
+    /**
+     * Get statistics about connected clients
+     */
+    getStats() {
+        return {
+            totalClients: this.clients.size,
+            clients: Array.from(this.clients.values()).map((client) => ({
+                id: client.id,
+                subscriptions: Array.from(client.subscriptions),
+                connectedAt: client.connectedAt.toISOString(),
+                isAlive: client.isAlive,
+            })),
+        };
+    }
+    /**
+     * Gracefully shutdown the WebSocket server
+     */
+    async shutdown() {
+        console.log("[websocket] Shutting down WebSocket server...");
+        this.stopHeartbeat();
+        // Close all client connections
+        this.clients.forEach((client) => {
+            try {
+                client.ws.close(1000, "Server shutting down");
+            }
+            catch (error) {
+                console.error(`[websocket] Error closing client ${client.id}:`, error);
+            }
+        });
+        this.clients.clear();
+        // Close the WebSocket server
+        if (this.wss) {
+            return new Promise((resolve) => {
+                this.wss.close(() => {
+                    console.log("[websocket] WebSocket server closed");
+                    this.wss = null;
+                    resolve();
+                });
+            });
+        }
+    }
+}
+// Export singleton instance
+export const websocketManager = new WebSocketManager();
+/**
+ * Initialize the WebSocket server
+ */
+export function initWebSocketServer(server, path) {
+    websocketManager.init(server, path);
+}
+/**
+ * Broadcast issue updates to subscribed clients
+ */
+export function broadcastIssueUpdate(issueId, action, data) {
+    websocketManager.broadcast("issue", issueId, {
+        type: `issue_${action}`,
+        data,
+    });
+}
+/**
+ * Broadcast spec updates to subscribed clients
+ */
+export function broadcastSpecUpdate(specId, action, data) {
+    websocketManager.broadcast("spec", specId, {
+        type: `spec_${action}`,
+        data,
+    });
+}
+/**
+ * Broadcast feedback updates to subscribed clients
+ */
+export function broadcastFeedbackUpdate(action, data) {
+    websocketManager.broadcastGeneric({
+        type: `feedback_${action}`,
+        data,
+    });
+}
+/**
+ * Broadcast relationship updates to subscribed clients
+ */
+export function broadcastRelationshipUpdate(action, data) {
+    websocketManager.broadcastGeneric({
+        type: `relationship_${action}`,
+        data,
+    });
+}
+/**
+ * Get WebSocket server statistics
+ */
+export function getWebSocketStats() {
+    return websocketManager.getStats();
+}
+/**
+ * Shutdown the WebSocket server
+ */
+export async function shutdownWebSocketServer() {
+    await websocketManager.shutdown();
+}
+//# sourceMappingURL=websocket.js.map

package/dist/services/websocket.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket.js","sourceRoot":"","sources":["../../src/services/websocket.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,SAAS,EAAW,MAAM,IAAI,CAAC;AAEzD,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAEpC,MAAM,eAAe,GAAG,KAAK,CAAC;AA+C9B;;GAEG;AACH,MAAM,gBAAgB;IACZ,GAAG,GAA2B,IAAI,CAAC;IACnC,OAAO,GAAwB,IAAI,GAAG,EAAE,CAAC;IACzC,iBAAiB,GAA0B,IAAI,CAAC;IACvC,kBAAkB,GAAG,KAAK,CAAC,CAAC,aAAa;IAE1D;;OAEG;IACH,IAAI,CAAC,MAAmB,EAAE,OAAe,KAAK;QAC5C,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,OAAO,CAAC,IAAI,CAAC,kDAAkD,CAAC,CAAC;YACjE,OAAO;QACT,CAAC;QAED,IAAI,CAAC,GAAG,GAAG,IAAI,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QACjD,OAAO,CAAC,GAAG,CAAC,qDAAqD,IAAI,EAAE,CAAC,CAAC;QAEzE,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5D,IAAI,CAAC,cAAc,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,EAAa,EAAE,IAA0B;QAChE,MAAM,QAAQ,GAAG,UAAU,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAW;YACrB,EAAE,EAAE,QAAQ;YACZ,EAAE;YACF,aAAa,EAAE,IAAI,GAAG,EAAE;YACxB,OAAO,EAAE,IAAI;YACb,WAAW,EAAE,IAAI,IAAI,EAAE;SACxB,CAAC;QAEF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACnC,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CACT,iCAAiC,QAAQ,YAAY,IAAI,CAAC,OAAO,CAAC,IAAI,GAAG,CAC1E,CAAC;QACJ,CAAC;QAED,wBAAwB;QACxB,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAa,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC,CAAC;QACxE,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC,CAAC;QACzD,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC,CAAC;QAC7D,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;QAE/C,uBAAuB;QACvB,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;YAC1B,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,8BAA8B;SACxC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,QAAgB;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,eAAe,EAAE,CAAC;gBACpB,OAAO,CAAC,GAAG,CACT,oCAAoC,QAAQ,oBAAoB,MAAM,CAAC,aAAa,CAAC,IAAI,GAAG,CAC7F,CAAC;YACJ,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,QAAgB,EAAE,KAAY;QAChD,OAAO,CAAC,KAAK,CAAC,6BAA6B,QAAQ,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IAC1E,CAAC;IAED;;OAEG;IACK,UAAU,CAAC,QAAgB;QACjC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC;QACxB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,QAAgB,EAAE,IAAa;QACnD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,OAAO,GAAkB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;YAE3D,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;gBACrB,KAAK,WAAW;oBACd,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;oBACxC,MAAM;gBAER,KAAK,aAAa;oBAChB,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;oBAC1C,MAAM;gBAER,KAAK,MAAM;oBACT,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;oBAC9C,MAAM;gBAER;oBACE,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;wBAC1B,IAAI,EAAE,OAAO;wBACb,OAAO,EAAE,yBAA0B,OAAe,CAAC,IAAI,EAAE;qBAC1D,CAAC,CAAC;YACP,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,4CAA4C,QAAQ,GAAG,EACvD,KAAK,CACN,CAAC;YACF,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;gBAC1B,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,wBAAwB;aAClC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,QAAgB,EAAE,OAAsB;QAC9D,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QAED,IAAI,YAAoB,CAAC;QAEzB,IAAI,OAAO,CAAC,WAAW,KAAK,KAAK,EAAE,CAAC;YAClC,2BAA2B;YAC3B,YAAY,GAAG,KAAK,CAAC;QACvB,CAAC;aAAM,IAAI,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YACpD,iCAAiC;YACjC,YAAY,GAAG,GAAG,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;QAC/D,CAAC;aAAM,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YAC/B,sCAAsC;YACtC,YAAY,GAAG,GAAG,OAAO,CAAC,WAAW,IAAI,CAAC;QAC5C,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;gBAC1B,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,8BAA8B;aACxC,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACvC,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CACT,sBAAsB,QAAQ,mBAAmB,YAAY,EAAE,CAChE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;YAC1B,IAAI,EAAE,YAAY;YAClB,YAAY;YACZ,OAAO,EAAE,iBAAiB,YAAY,EAAE;SACzC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,QAAgB,EAAE,OAAsB;QAChE,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QAED,IAAI,YAAoB,CAAC;QAEzB,IAAI,OAAO,CAAC,WAAW,KAAK,KAAK,EAAE,CAAC;YAClC,YAAY,GAAG,KAAK,CAAC;QACvB,CAAC;aAAM,IAAI,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;YACpD,YAAY,GAAG,GAAG,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;QAC/D,CAAC;aAAM,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YAC/B,YAAY,GAAG,GAAG,OAAO,CAAC,WAAW,IAAI,CAAC;QAC5C,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;gBAC1B,IAAI,EAAE,OAAO;gBACb,OAAO,EAAE,gCAAgC;aAC1C,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC1C,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,CAAC,GAAG,CACT,sBAAsB,QAAQ,uBAAuB,YAAY,EAAE,CACpE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,YAAY,CAAC,QAAQ,EAAE;YAC1B,IAAI,EAAE,cAAc;YACpB,YAAY;YACZ,OAAO,EAAE,qBAAqB,YAAY,EAAE;SAC7C,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,QAAgB,EAAE,OAAsB;QAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,IAAI,MAAM,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YACvD,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;QAC1C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CACX,yCAAyC,QAAQ,GAAG,EACpD,KAAK,CACN,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,SAAS,CACP,UAA4B,EAC5B,QAAgB,EAChB,OAAsB;QAEtB,MAAM,YAAY,GAAG,GAAG,UAAU,IAAI,QAAQ,EAAE,CAAC;QACjD,MAAM,gBAAgB,GAAG,GAAG,UAAU,IAAI,CAAC;QAC3C,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAC9B,IAAI,MAAM,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;gBAC5C,OAAO;YACT,CAAC;YAED,6EAA6E;YAC7E,IACE,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,YAAY,CAAC;gBACtC,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,gBAAgB,CAAC;gBAC1C,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAC/B,CAAC;gBACD,IAAI,CAAC;oBACH,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;oBACxC,SAAS,EAAE,CAAC;gBACd,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CACX,sCAAsC,MAAM,CAAC,EAAE,GAAG,EAClD,KAAK,CACN,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClB,OAAO,CAAC,GAAG,CACT,2BAA2B,OAAO,CAAC,IAAI,QAAQ,YAAY,OAAO,SAAS,UAAU,CACtF,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,gBAAgB,CAAC,OAAsB;QACrC,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAC9B,IAAI,MAAM,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;gBAC5C,OAAO;YACT,CAAC;YAED,2CAA2C;YAC3C,IAAI,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBACpC,IAAI,CAAC;oBACH,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;oBACxC,SAAS,EAAE,CAAC;gBACd,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CACX,sCAAsC,MAAM,CAAC,EAAE,GAAG,EAClD,KAAK,CACN,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;YAClB,OAAO,CAAC,GAAG,CACT,2BAA2B,OAAO,CAAC,IAAI,OAAO,SAAS,UAAU,CAClE,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,cAAc;QACpB,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,OAAO;QACT,CAAC;QAED,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE;YACxC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE;gBACxC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACpB,OAAO,CAAC,GAAG,CAAC,4CAA4C,QAAQ,EAAE,CAAC,CAAC;oBACpE,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC;oBACtB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;oBAC9B,OAAO;gBACT,CAAC;gBAED,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC;gBACvB,IAAI,CAAC;oBACH,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;gBACnB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,8BAA8B,QAAQ,GAAG,EAAE,KAAK,CAAC,CAAC;gBAClE,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAE5B,OAAO,CAAC,GAAG,CACT,4CAA4C,IAAI,CAAC,kBAAkB,KAAK,CACzE,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,aAAa;QACnB,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,aAAa,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACtC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;YAC9B,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IAED;;OAEG;IACH,QAAQ;QASN,OAAO;YACL,YAAY,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI;YAC/B,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBAC1D,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,aAAa,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC;gBAC/C,WAAW,EAAE,MAAM,CAAC,WAAW,CAAC,WAAW,EAAE;gBAC7C,OAAO,EAAE,MAAM,CAAC,OAAO;aACxB,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;QAE7D,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,+BAA+B;QAC/B,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAC9B,IAAI,CAAC;gBACH,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,sBAAsB,CAAC,CAAC;YAChD,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,MAAM,CAAC,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;YACzE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAErB,6BAA6B;QAC7B,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC7B,IAAI,CAAC,GAAI,CAAC,KAAK,CAAC,GAAG,EAAE;oBACnB,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAC;oBACnD,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC;oBAChB,OAAO,EAAE,CAAC;gBACZ,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;CACF;AAED,4BAA4B;AAC5B,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,gBAAgB,EAAE,CAAC;AAEvD;;GAEG;AACH,MAAM,UAAU,mBAAmB,CAAC,MAAmB,EAAE,IAAa;IACpE,gBAAgB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;AACtC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAClC,OAAe,EACf,MAAyC,EACzC,IAAU;IAEV,gBAAgB,CAAC,SAAS,CAAC,OAAO,EAAE,OAAO,EAAE;QAC3C,IAAI,EAAE,SAAS,MAAM,EAAS;QAC9B,IAAI;KACL,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,MAAc,EACd,MAAyC,EACzC,IAAU;IAEV,gBAAgB,CAAC,SAAS,CAAC,MAAM,EAAE,MAAM,EAAE;QACzC,IAAI,EAAE,QAAQ,MAAM,EAAS;QAC7B,IAAI;KACL,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB,CACrC,MAAyC,EACzC,IAAU;IAEV,gBAAgB,CAAC,gBAAgB,CAAC;QAChC,IAAI,EAAE,YAAY,MAAM,EAAS;QACjC,IAAI;KACL,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,2BAA2B,CACzC,MAA6B,EAC7B,IAAU;IAEV,gBAAgB,CAAC,gBAAgB,CAAC;QAChC,IAAI,EAAE,gBAAgB,MAAM,EAAS;QACrC,IAAI;KACL,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB;IAC/B,OAAO,gBAAgB,CAAC,QAAQ,EAAE,CAAC;AACrC,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,uBAAuB;IAC3C,MAAM,gBAAgB,CAAC,QAAQ,EAAE,CAAC;AACpC,CAAC"}

package/dist/utils/sudocode-dir.js

@@ -0,0 +1,9 @@
+/**
+ * Utility to get the SUDOCODE directory path
+ * Respects SUDOCODE_DIR environment variable for testing
+ */
+import * as path from "path";
+export function getSudocodeDir() {
+    return process.env.SUDOCODE_DIR || path.join(process.cwd(), ".sudocode");
+}
+//# sourceMappingURL=sudocode-dir.js.map

package/dist/utils/sudocode-dir.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sudocode-dir.js","sourceRoot":"","sources":["../../src/utils/sudocode-dir.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAE7B,MAAM,UAAU,cAAc;IAC5B,OAAO,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,WAAW,CAAC,CAAC;AAC3E,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sudocode-ai/local-server",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Local backend server for sudocode - provides REST API and WebSocket endpoints",
   "main": "dist/index.js",
   "type": "module",
@@ -9,8 +9,7 @@
   },
   "scripts": {
     "dev": "tsx watch src/index.ts",
-    "build": "node build.js",
-    "build:dev": "tsc && chmod +x dist/cli.js",
+    "build": "tsc && chmod +x dist/cli.js && node scripts/copy-frontend.js",
     "start": "node dist/index.js",
     "test": "NODE_ENV=test vitest",
     "test:e2e": "RUN_E2E_TESTS=true NODE_ENV=test vitest --run",
@@ -42,9 +41,8 @@
   },
   "dependencies": {
     "@ag-ui/core": "^0.0.39",
-    "@copilotkit/runtime": "^1.10.6",
-    "@sudocode-ai/cli": "^0.1.0",
-    "@sudocode-ai/types": "^0.1.0",
+    "@sudocode-ai/cli": "^0.1.1",
+    "@sudocode-ai/types": "^0.1.1",
     "@types/ws": "^8.18.1",
     "async-mutex": "^0.5.0",
     "better-sqlite3": "^11.8.1",