claudedesk 1.0.0

package/dist/core/usage-manager.js

@@ -0,0 +1,363 @@
+import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'fs';
+import { join } from 'path';
+const USAGE_DIR = join(process.cwd(), 'config', 'usage');
+const GLOBAL_USAGE_FILE = join(USAGE_DIR, 'global-usage.json');
+const WEEKLY_USAGE_FILE = join(USAGE_DIR, 'weekly-usage.json');
+const SESSIONS_USAGE_DIR = join(USAGE_DIR, 'sessions');
+// Model pricing per 1M tokens (input / output)
+const MODEL_PRICING = {
+    'claude-opus-4-5-20251101': { input: 15, output: 75 },
+    'claude-sonnet-4-20250514': { input: 3, output: 15 },
+    'claude-3-5-sonnet-20241022': { input: 3, output: 15 },
+    'claude-3-5-haiku-20241022': { input: 1, output: 5 },
+    // Default pricing if model not found
+    'default': { input: 3, output: 15 },
+};
+/**
+ * Get the start of the week (Sunday 00:00:00) for a given date
+ */
+function getWeekStart(date) {
+    const d = new Date(date);
+    const day = d.getDay(); // 0 = Sunday
+    d.setDate(d.getDate() - day);
+    d.setHours(0, 0, 0, 0);
+    return d;
+}
+/**
+ * Get the end of the week (Saturday 23:59:59) for a given date
+ */
+function getWeekEnd(date) {
+    const weekStart = getWeekStart(date);
+    const weekEnd = new Date(weekStart);
+    weekEnd.setDate(weekEnd.getDate() + 6);
+    weekEnd.setHours(23, 59, 59, 999);
+    return weekEnd;
+}
+class UsageManager {
+    constructor() {
+        this.ensureDirectories();
+    }
+    ensureDirectories() {
+        if (!existsSync(USAGE_DIR)) {
+            mkdirSync(USAGE_DIR, { recursive: true });
+        }
+        if (!existsSync(SESSIONS_USAGE_DIR)) {
+            mkdirSync(SESSIONS_USAGE_DIR, { recursive: true });
+        }
+    }
+    /**
+     * Calculate cost based on model and token usage
+     */
+    calculateCost(model, inputTokens, outputTokens) {
+        // Find pricing for the model
+        const pricing = MODEL_PRICING[model] || MODEL_PRICING['default'];
+        // Cost = (tokens / 1M) * price per 1M tokens
+        const inputCost = (inputTokens / 1_000_000) * pricing.input;
+        const outputCost = (outputTokens / 1_000_000) * pricing.output;
+        return inputCost + outputCost;
+    }
+    /**
+     * Get model short name for display
+     */
+    getModelShortName(model) {
+        if (model.includes('opus'))
+            return 'Opus';
+        if (model.includes('sonnet'))
+            return 'Sonnet';
+        if (model.includes('haiku'))
+            return 'Haiku';
+        return model.split('-').slice(0, 2).join(' ');
+    }
+    /**
+     * Record usage for a message in a session
+     */
+    recordMessageUsage(sessionId, data, toolUseCount = 0, filesChanged = 0) {
+        if (!data.usage)
+            return;
+        const model = data.model || 'unknown';
+        const cost = data.costUsd ?? this.calculateCost(model, data.usage.inputTokens, data.usage.outputTokens);
+        const messageUsage = {
+            messageId: data.messageId,
+            timestamp: new Date().toISOString(),
+            model,
+            usage: data.usage,
+            costUsd: cost,
+            durationMs: data.durationMs || 0,
+        };
+        // Update session usage
+        this.updateSessionUsage(sessionId, messageUsage, toolUseCount, filesChanged);
+        // Update global usage (daily)
+        this.updateGlobalUsage(messageUsage);
+        // Update weekly usage
+        this.updateWeeklyUsage(messageUsage);
+    }
+    /**
+     * Update session-specific usage data
+     */
+    updateSessionUsage(sessionId, messageUsage, toolUseCount, filesChanged) {
+        const sessionFile = join(SESSIONS_USAGE_DIR, `${sessionId}.json`);
+        let sessionData = {
+            sessionId,
+            messages: [],
+            toolUseCount: 0,
+            filesChanged: 0,
+        };
+        if (existsSync(sessionFile)) {
+            try {
+                sessionData = JSON.parse(readFileSync(sessionFile, 'utf-8'));
+            }
+            catch {
+                // Use default
+            }
+        }
+        sessionData.messages.push(messageUsage);
+        sessionData.toolUseCount += toolUseCount;
+        sessionData.filesChanged += filesChanged;
+        writeFileSync(sessionFile, JSON.stringify(sessionData, null, 2));
+    }
+    /**
+     * Update global usage data
+     */
+    updateGlobalUsage(messageUsage) {
+        const today = new Date().toISOString().split('T')[0];
+        let globalData = {
+            periodStart: today,
+            messages: [],
+            byModel: {},
+        };
+        if (existsSync(GLOBAL_USAGE_FILE)) {
+            try {
+                globalData = JSON.parse(readFileSync(GLOBAL_USAGE_FILE, 'utf-8'));
+                // Reset if it's a new day
+                if (globalData.periodStart !== today) {
+                    globalData = {
+                        periodStart: today,
+                        messages: [],
+                        byModel: {},
+                    };
+                }
+            }
+            catch {
+                // Use default
+            }
+        }
+        globalData.messages.push(messageUsage);
+        // Update by-model stats
+        const model = messageUsage.model;
+        if (!globalData.byModel[model]) {
+            globalData.byModel[model] = { inputTokens: 0, outputTokens: 0, costUsd: 0 };
+        }
+        globalData.byModel[model].inputTokens += messageUsage.usage.inputTokens;
+        globalData.byModel[model].outputTokens += messageUsage.usage.outputTokens;
+        globalData.byModel[model].costUsd += messageUsage.costUsd;
+        writeFileSync(GLOBAL_USAGE_FILE, JSON.stringify(globalData, null, 2));
+    }
+    /**
+     * Update weekly usage data
+     */
+    updateWeeklyUsage(messageUsage) {
+        const now = new Date();
+        const weekStart = getWeekStart(now);
+        const weekEnd = getWeekEnd(now);
+        const weekStartStr = weekStart.toISOString();
+        const weekEndStr = weekEnd.toISOString();
+        const today = now.toISOString().split('T')[0];
+        let weeklyData = {
+            weekStart: weekStartStr,
+            weekEnd: weekEndStr,
+            messages: [],
+            byModel: {},
+            dailyBreakdown: {},
+        };
+        if (existsSync(WEEKLY_USAGE_FILE)) {
+            try {
+                weeklyData = JSON.parse(readFileSync(WEEKLY_USAGE_FILE, 'utf-8'));
+                // Reset if it's a new week
+                const storedWeekStart = new Date(weeklyData.weekStart);
+                if (storedWeekStart.getTime() !== weekStart.getTime()) {
+                    weeklyData = {
+                        weekStart: weekStartStr,
+                        weekEnd: weekEndStr,
+                        messages: [],
+                        byModel: {},
+                        dailyBreakdown: {},
+                    };
+                }
+            }
+            catch {
+                // Use default
+            }
+        }
+        weeklyData.messages.push(messageUsage);
+        // Update by-model stats
+        const model = messageUsage.model;
+        if (!weeklyData.byModel[model]) {
+            weeklyData.byModel[model] = { inputTokens: 0, outputTokens: 0, costUsd: 0, apiCalls: 0 };
+        }
+        weeklyData.byModel[model].inputTokens += messageUsage.usage.inputTokens;
+        weeklyData.byModel[model].outputTokens += messageUsage.usage.outputTokens;
+        weeklyData.byModel[model].costUsd += messageUsage.costUsd;
+        weeklyData.byModel[model].apiCalls += 1;
+        // Update daily breakdown
+        if (!weeklyData.dailyBreakdown[today]) {
+            weeklyData.dailyBreakdown[today] = { inputTokens: 0, outputTokens: 0, costUsd: 0, apiCalls: 0 };
+        }
+        weeklyData.dailyBreakdown[today].inputTokens += messageUsage.usage.inputTokens;
+        weeklyData.dailyBreakdown[today].outputTokens += messageUsage.usage.outputTokens;
+        weeklyData.dailyBreakdown[today].costUsd += messageUsage.costUsd;
+        weeklyData.dailyBreakdown[today].apiCalls += 1;
+        writeFileSync(WEEKLY_USAGE_FILE, JSON.stringify(weeklyData, null, 2));
+    }
+    /**
+     * Get usage stats for a specific session
+     */
+    getSessionUsage(sessionId) {
+        const sessionFile = join(SESSIONS_USAGE_DIR, `${sessionId}.json`);
+        const defaultStats = {
+            totalInputTokens: 0,
+            totalOutputTokens: 0,
+            totalCostUsd: 0,
+            messageCount: 0,
+            toolUseCount: 0,
+            filesChanged: 0,
+        };
+        if (!existsSync(sessionFile)) {
+            return defaultStats;
+        }
+        try {
+            const sessionData = JSON.parse(readFileSync(sessionFile, 'utf-8'));
+            let totalInput = 0;
+            let totalOutput = 0;
+            let totalCost = 0;
+            let lastModel;
+            for (const msg of sessionData.messages) {
+                totalInput += msg.usage.inputTokens;
+                totalOutput += msg.usage.outputTokens;
+                totalCost += msg.costUsd;
+                lastModel = msg.model;
+            }
+            return {
+                totalInputTokens: totalInput,
+                totalOutputTokens: totalOutput,
+                totalCostUsd: totalCost,
+                messageCount: sessionData.messages.length,
+                toolUseCount: sessionData.toolUseCount,
+                filesChanged: sessionData.filesChanged,
+                model: lastModel,
+            };
+        }
+        catch {
+            return defaultStats;
+        }
+    }
+    /**
+     * Get global usage stats for today
+     */
+    getGlobalUsage() {
+        const today = new Date().toISOString().split('T')[0];
+        const defaultStats = {
+            periodStart: today,
+            totalInputTokens: 0,
+            totalOutputTokens: 0,
+            totalCostUsd: 0,
+            totalApiCalls: 0,
+            byModel: {},
+        };
+        if (!existsSync(GLOBAL_USAGE_FILE)) {
+            return defaultStats;
+        }
+        try {
+            const globalData = JSON.parse(readFileSync(GLOBAL_USAGE_FILE, 'utf-8'));
+            // Return default if data is from a different day
+            if (globalData.periodStart !== today) {
+                return defaultStats;
+            }
+            let totalInput = 0;
+            let totalOutput = 0;
+            let totalCost = 0;
+            for (const msg of globalData.messages) {
+                totalInput += msg.usage.inputTokens;
+                totalOutput += msg.usage.outputTokens;
+                totalCost += msg.costUsd;
+            }
+            return {
+                periodStart: globalData.periodStart,
+                totalInputTokens: totalInput,
+                totalOutputTokens: totalOutput,
+                totalCostUsd: totalCost,
+                totalApiCalls: globalData.messages.length,
+                byModel: globalData.byModel,
+            };
+        }
+        catch {
+            return defaultStats;
+        }
+    }
+    /**
+     * Get weekly usage stats (resets on Sunday)
+     */
+    getWeeklyUsage() {
+        const now = new Date();
+        const weekStart = getWeekStart(now);
+        const weekEnd = getWeekEnd(now);
+        const defaultStats = {
+            weekStart: weekStart.toISOString(),
+            weekEnd: weekEnd.toISOString(),
+            totalInputTokens: 0,
+            totalOutputTokens: 0,
+            totalCostUsd: 0,
+            totalApiCalls: 0,
+            byModel: {},
+            dailyBreakdown: {},
+        };
+        if (!existsSync(WEEKLY_USAGE_FILE)) {
+            return defaultStats;
+        }
+        try {
+            const weeklyData = JSON.parse(readFileSync(WEEKLY_USAGE_FILE, 'utf-8'));
+            // Return default if data is from a different week
+            const storedWeekStart = new Date(weeklyData.weekStart);
+            if (storedWeekStart.getTime() !== weekStart.getTime()) {
+                return defaultStats;
+            }
+            let totalInput = 0;
+            let totalOutput = 0;
+            let totalCost = 0;
+            for (const msg of weeklyData.messages) {
+                totalInput += msg.usage.inputTokens;
+                totalOutput += msg.usage.outputTokens;
+                totalCost += msg.costUsd;
+            }
+            return {
+                weekStart: weeklyData.weekStart,
+                weekEnd: weeklyData.weekEnd,
+                totalInputTokens: totalInput,
+                totalOutputTokens: totalOutput,
+                totalCostUsd: totalCost,
+                totalApiCalls: weeklyData.messages.length,
+                byModel: weeklyData.byModel,
+                dailyBreakdown: weeklyData.dailyBreakdown,
+            };
+        }
+        catch {
+            return defaultStats;
+        }
+    }
+    /**
+     * Delete session usage data when session is closed
+     */
+    deleteSessionUsage(sessionId) {
+        const sessionFile = join(SESSIONS_USAGE_DIR, `${sessionId}.json`);
+        if (existsSync(sessionFile)) {
+            try {
+                const fs = require('fs');
+                fs.unlinkSync(sessionFile);
+            }
+            catch {
+                // Ignore deletion errors
+            }
+        }
+    }
+}
+export const usageManager = new UsageManager();
+//# sourceMappingURL=usage-manager.js.map

package/dist/core/usage-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-manager.js","sourceRoot":"","sources":["../../src/core/usage-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAG5B,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;AACzD,MAAM,iBAAiB,GAAG,IAAI,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;AAC/D,MAAM,iBAAiB,GAAG,IAAI,CAAC,SAAS,EAAE,mBAAmB,CAAC,CAAC;AAC/D,MAAM,kBAAkB,GAAG,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;AAEvD,+CAA+C;AAC/C,MAAM,aAAa,GAAsD;IACvE,0BAA0B,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;IACrD,0BAA0B,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;IACpD,4BAA4B,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;IACtD,2BAA2B,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE;IACpD,qCAAqC;IACrC,SAAS,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;CACpC,CAAC;AAuBF;;GAEG;AACH,SAAS,YAAY,CAAC,IAAU;IAC9B,MAAM,CAAC,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC;IACzB,MAAM,GAAG,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,aAAa;IACrC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EAAE,GAAG,GAAG,CAAC,CAAC;IAC7B,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IACvB,OAAO,CAAC,CAAC;AACX,CAAC;AAED;;GAEG;AACH,SAAS,UAAU,CAAC,IAAU;IAC5B,MAAM,SAAS,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IACrC,MAAM,OAAO,GAAG,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC;IACpC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC,CAAC;IACvC,OAAO,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,CAAC;IAClC,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,YAAY;IAChB;QACE,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC;IAEO,iBAAiB;QACvB,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC3B,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5C,CAAC;QACD,IAAI,CAAC,UAAU,CAAC,kBAAkB,CAAC,EAAE,CAAC;YACpC,SAAS,CAAC,kBAAkB,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,KAAa,EAAE,WAAmB,EAAE,YAAoB;QACpE,6BAA6B;QAC7B,MAAM,OAAO,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,aAAa,CAAC,SAAS,CAAC,CAAC;QAEjE,6CAA6C;QAC7C,MAAM,SAAS,GAAG,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC;QAC5D,MAAM,UAAU,GAAG,CAAC,YAAY,GAAG,SAAS,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;QAE/D,OAAO,SAAS,GAAG,UAAU,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,KAAa;QAC7B,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,MAAM,CAAC;QAC1C,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,QAAQ,CAAC;QAC9C,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,OAAO,OAAO,CAAC;QAC5C,OAAO,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAChD,CAAC;IAED;;OAEG;IACH,kBAAkB,CAChB,SAAiB,EACjB,IASC,EACD,eAAuB,CAAC,EACxB,eAAuB,CAAC;QAExB,IAAI,CAAC,IAAI,CAAC,KAAK;YAAE,OAAO;QAExB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,SAAS,CAAC;QACtC,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAExG,MAAM,YAAY,GAAiB;YACjC,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,KAAK;YACL,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,OAAO,EAAE,IAAI;YACb,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,CAAC;SACjC,CAAC;QAEF,uBAAuB;QACvB,IAAI,CAAC,kBAAkB,CAAC,SAAS,EAAE,YAAY,EAAE,YAAY,EAAE,YAAY,CAAC,CAAC;QAE7E,8BAA8B;QAC9B,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAErC,sBAAsB;QACtB,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACK,kBAAkB,CACxB,SAAiB,EACjB,YAA0B,EAC1B,YAAoB,EACpB,YAAoB;QAEpB,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,EAAE,GAAG,SAAS,OAAO,CAAC,CAAC;QAClE,IAAI,WAAW,GAAqB;YAClC,SAAS;YACT,QAAQ,EAAE,EAAE;YACZ,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,CAAC;SAChB,CAAC;QAEF,IAAI,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC;gBACH,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC;YAC/D,CAAC;YAAC,MAAM,CAAC;gBACP,cAAc;YAChB,CAAC;QACH,CAAC;QAED,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACxC,WAAW,CAAC,YAAY,IAAI,YAAY,CAAC;QACzC,WAAW,CAAC,YAAY,IAAI,YAAY,CAAC;QAEzC,aAAa,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACnE,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,YAA0B;QAClD,MAAM,KAAK,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrD,IAAI,UAAU,GAAoB;YAChC,WAAW,EAAE,KAAK;YAClB,QAAQ,EAAE,EAAE;YACZ,OAAO,EAAE,EAAE;SACZ,CAAC;QAEF,IAAI,UAAU,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAClC,IAAI,CAAC;gBACH,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC,CAAC;gBAElE,0BAA0B;gBAC1B,IAAI,UAAU,CAAC,WAAW,KAAK,KAAK,EAAE,CAAC;oBACrC,UAAU,GAAG;wBACX,WAAW,EAAE,KAAK;wBAClB,QAAQ,EAAE,EAAE;wBACZ,OAAO,EAAE,EAAE;qBACZ,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,cAAc;YAChB,CAAC;QACH,CAAC;QAED,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAEvC,wBAAwB;QACxB,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC;QACjC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/B,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;QAC9E,CAAC;QACD,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,WAAW,IAAI,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC;QACxE,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,YAAY,IAAI,YAAY,CAAC,KAAK,CAAC,YAAY,CAAC;QAC1E,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,IAAI,YAAY,CAAC,OAAO,CAAC;QAE1D,aAAa,CAAC,iBAAiB,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACxE,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,YAA0B;QAClD,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;QAChC,MAAM,YAAY,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC;QAC7C,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACzC,MAAM,KAAK,GAAG,GAAG,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAE9C,IAAI,UAAU,GAAoB;YAChC,SAAS,EAAE,YAAY;YACvB,OAAO,EAAE,UAAU;YACnB,QAAQ,EAAE,EAAE;YACZ,OAAO,EAAE,EAAE;YACX,cAAc,EAAE,EAAE;SACnB,CAAC;QAEF,IAAI,UAAU,CAAC,iBAAiB,CAAC,EAAE,CAAC;YAClC,IAAI,CAAC;gBACH,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC,CAAC;gBAElE,2BAA2B;gBAC3B,MAAM,eAAe,GAAG,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;gBACvD,IAAI,eAAe,CAAC,OAAO,EAAE,KAAK,SAAS,CAAC,OAAO,EAAE,EAAE,CAAC;oBACtD,UAAU,GAAG;wBACX,SAAS,EAAE,YAAY;wBACvB,OAAO,EAAE,UAAU;wBACnB,QAAQ,EAAE,EAAE;wBACZ,OAAO,EAAE,EAAE;wBACX,cAAc,EAAE,EAAE;qBACnB,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,MAAM,CAAC;gBACP,cAAc;YAChB,CAAC;QACH,CAAC;QAED,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAEvC,wBAAwB;QACxB,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC;QACjC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/B,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QAC3F,CAAC;QACD,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,WAAW,IAAI,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC;QACxE,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,YAAY,IAAI,YAAY,CAAC,KAAK,CAAC,YAAY,CAAC;QAC1E,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,IAAI,YAAY,CAAC,OAAO,CAAC;QAC1D,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC;QAExC,yBAAyB;QACzB,IAAI,CAAC,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;YACtC,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,GAAG,EAAE,WAAW,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;QAClG,CAAC;QACD,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,WAAW,IAAI,YAAY,CAAC,KAAK,CAAC,WAAW,CAAC;QAC/E,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,YAAY,IAAI,YAAY,CAAC,KAAK,CAAC,YAAY,CAAC;QACjF,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,OAAO,IAAI,YAAY,CAAC,OAAO,CAAC;QACjE,UAAU,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC;QAE/C,aAAa,CAAC,iBAAiB,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACxE,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,SAAiB;QAC/B,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,EAAE,GAAG,SAAS,OAAO,CAAC,CAAC;QAClE,MAAM,YAAY,GAAsB;YACtC,gBAAgB,EAAE,CAAC;YACnB,iBAAiB,EAAE,CAAC;YACpB,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,CAAC;SAChB,CAAC;QAEF,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;YAC7B,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,WAAW,GAAqB,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC;YAErF,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,IAAI,SAA6B,CAAC;YAElC,KAAK,MAAM,GAAG,IAAI,WAAW,CAAC,QAAQ,EAAE,CAAC;gBACvC,UAAU,IAAI,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC;gBACpC,WAAW,IAAI,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC;gBACtC,SAAS,IAAI,GAAG,CAAC,OAAO,CAAC;gBACzB,SAAS,GAAG,GAAG,CAAC,KAAK,CAAC;YACxB,CAAC;YAED,OAAO;gBACL,gBAAgB,EAAE,UAAU;gBAC5B,iBAAiB,EAAE,WAAW;gBAC9B,YAAY,EAAE,SAAS;gBACvB,YAAY,EAAE,WAAW,CAAC,QAAQ,CAAC,MAAM;gBACzC,YAAY,EAAE,WAAW,CAAC,YAAY;gBACtC,YAAY,EAAE,WAAW,CAAC,YAAY;gBACtC,KAAK,EAAE,SAAS;aACjB,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,YAAY,CAAC;QACtB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,MAAM,KAAK,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACrD,MAAM,YAAY,GAAqB;YACrC,WAAW,EAAE,KAAK;YAClB,gBAAgB,EAAE,CAAC;YACnB,iBAAiB,EAAE,CAAC;YACpB,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,OAAO,EAAE,EAAE;SACZ,CAAC;QAEF,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACnC,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,UAAU,GAAoB,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC,CAAC;YAEzF,iDAAiD;YACjD,IAAI,UAAU,CAAC,WAAW,KAAK,KAAK,EAAE,CAAC;gBACrC,OAAO,YAAY,CAAC;YACtB,CAAC;YAED,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,IAAI,SAAS,GAAG,CAAC,CAAC;YAElB,KAAK,MAAM,GAAG,IAAI,UAAU,CAAC,QAAQ,EAAE,CAAC;gBACtC,UAAU,IAAI,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC;gBACpC,WAAW,IAAI,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC;gBACtC,SAAS,IAAI,GAAG,CAAC,OAAO,CAAC;YAC3B,CAAC;YAED,OAAO;gBACL,WAAW,EAAE,UAAU,CAAC,WAAW;gBACnC,gBAAgB,EAAE,UAAU;gBAC5B,iBAAiB,EAAE,WAAW;gBAC9B,YAAY,EAAE,SAAS;gBACvB,aAAa,EAAE,UAAU,CAAC,QAAQ,CAAC,MAAM;gBACzC,OAAO,EAAE,UAAU,CAAC,OAAO;aAC5B,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,YAAY,CAAC;QACtB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;QACpC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;QAEhC,MAAM,YAAY,GAAqB;YACrC,SAAS,EAAE,SAAS,CAAC,WAAW,EAAE;YAClC,OAAO,EAAE,OAAO,CAAC,WAAW,EAAE;YAC9B,gBAAgB,EAAE,CAAC;YACnB,iBAAiB,EAAE,CAAC;YACpB,YAAY,EAAE,CAAC;YACf,aAAa,EAAE,CAAC;YAChB,OAAO,EAAE,EAAE;YACX,cAAc,EAAE,EAAE;SACnB,CAAC;QAEF,IAAI,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,CAAC;YACnC,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,UAAU,GAAoB,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC,CAAC;YAEzF,kDAAkD;YAClD,MAAM,eAAe,GAAG,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YACvD,IAAI,eAAe,CAAC,OAAO,EAAE,KAAK,SAAS,CAAC,OAAO,EAAE,EAAE,CAAC;gBACtD,OAAO,YAAY,CAAC;YACtB,CAAC;YAED,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,IAAI,SAAS,GAAG,CAAC,CAAC;YAElB,KAAK,MAAM,GAAG,IAAI,UAAU,CAAC,QAAQ,EAAE,CAAC;gBACtC,UAAU,IAAI,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC;gBACpC,WAAW,IAAI,GAAG,CAAC,KAAK,CAAC,YAAY,CAAC;gBACtC,SAAS,IAAI,GAAG,CAAC,OAAO,CAAC;YAC3B,CAAC;YAED,OAAO;gBACL,SAAS,EAAE,UAAU,CAAC,SAAS;gBAC/B,OAAO,EAAE,UAAU,CAAC,OAAO;gBAC3B,gBAAgB,EAAE,UAAU;gBAC5B,iBAAiB,EAAE,WAAW;gBAC9B,YAAY,EAAE,SAAS;gBACvB,aAAa,EAAE,UAAU,CAAC,QAAQ,CAAC,MAAM;gBACzC,OAAO,EAAE,UAAU,CAAC,OAAO;gBAC3B,cAAc,EAAE,UAAU,CAAC,cAAc;aAC1C,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,YAAY,CAAC;QACtB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,SAAiB;QAClC,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,EAAE,GAAG,SAAS,OAAO,CAAC,CAAC;QAClE,IAAI,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;YAC5B,IAAI,CAAC;gBACH,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;gBACzB,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;YAC7B,CAAC;YAAC,MAAM,CAAC;gBACP,yBAAyB;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC"}

package/dist/core/ws-manager.d.ts

@@ -0,0 +1,39 @@
+import { WebSocket } from 'ws';
+import { Server } from 'http';
+export interface WSMessage {
+    type: string;
+    sessionId?: string;
+    content?: string;
+    mode?: 'plan' | 'direct';
+    [key: string]: unknown;
+}
+export interface WSClient {
+    id: string;
+    ws: WebSocket;
+    token: string;
+    subscribedSessions: Set<string>;
+    isAlive: boolean;
+}
+type MessageHandler = (client: WSClient, message: WSMessage) => void | Promise<void>;
+declare class WebSocketManager {
+    private wss;
+    private clients;
+    private handlers;
+    private heartbeatInterval;
+    initialize(server: Server, authToken: string): void;
+    private startHeartbeat;
+    private handleMessage;
+    on(type: string, handler: MessageHandler): void;
+    send(client: WSClient, message: WSMessage): void;
+    broadcastToSession(sessionId: string, message: WSMessage): void;
+    broadcastAll(message: WSMessage): void;
+    subscribeToSession(client: WSClient, sessionId: string): void;
+    unsubscribeFromSession(client: WSClient, sessionId: string): void;
+    getSessionSubscribers(sessionId: string): WSClient[];
+    getClientCount(): number;
+    shutdown(): void;
+    private generateId;
+}
+export declare const wsManager: WebSocketManager;
+export {};
+//# sourceMappingURL=ws-manager.d.ts.map

package/dist/core/ws-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-manager.d.ts","sourceRoot":"","sources":["../../src/core/ws-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAmB,SAAS,EAAE,MAAM,IAAI,CAAC;AAChD,OAAO,EAAE,MAAM,EAAmB,MAAM,MAAM,CAAC;AAG/C,MAAM,WAAW,SAAS;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,GAAG,QAAQ,CAAC;IACzB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,EAAE,EAAE,SAAS,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,kBAAkB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAChC,OAAO,EAAE,OAAO,CAAC;CAClB;AAED,KAAK,cAAc,GAAG,CAAC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAErF,cAAM,gBAAgB;IACpB,OAAO,CAAC,GAAG,CAAgC;IAC3C,OAAO,CAAC,OAAO,CAAoC;IACnD,OAAO,CAAC,QAAQ,CAA0C;IAC1D,OAAO,CAAC,iBAAiB,CAA+B;IAExD,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,IAAI;IAuGnD,OAAO,CAAC,cAAc;YAeR,aAAa;IAkB3B,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,GAAG,IAAI;IAK/C,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,GAAG,IAAI;IAOhD,kBAAkB,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,GAAG,IAAI;IAS/D,YAAY,CAAC,OAAO,EAAE,SAAS,GAAG,IAAI;IAOtC,kBAAkB,CAAC,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,GAAG,IAAI;IAM7D,sBAAsB,CAAC,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,GAAG,IAAI;IAMjE,qBAAqB,CAAC,SAAS,EAAE,MAAM,GAAG,QAAQ,EAAE;IAOpD,cAAc,IAAI,MAAM;IAKxB,QAAQ,IAAI,IAAI;IAiBhB,OAAO,CAAC,UAAU;CAGnB;AAGD,eAAO,MAAM,SAAS,kBAAyB,CAAC"}

package/dist/core/ws-manager.js

@@ -0,0 +1,190 @@
+import { WebSocketServer, WebSocket } from 'ws';
+import { parse as parseUrl } from 'url';
+class WebSocketManager {
+    wss = null;
+    clients = new Map();
+    handlers = new Map();
+    heartbeatInterval = null;
+    initialize(server, authToken) {
+        this.wss = new WebSocketServer({ noServer: true });
+        // Handle HTTP upgrade requests
+        server.on('upgrade', (request, socket, head) => {
+            const { pathname, query } = parseUrl(request.url || '', true);
+            if (pathname === '/ws') {
+                // SEC-05: Prefer Sec-WebSocket-Protocol header over query string to avoid token in logs
+                // Protocol header format: "token, <actual-token>" - the first value identifies the protocol,
+                // the second is the actual authentication token
+                let token;
+                let useProtocolHeader = false;
+                const protocolHeader = request.headers['sec-websocket-protocol'];
+                if (protocolHeader) {
+                    // Parse "token, <actual-token>" format
+                    const protocols = typeof protocolHeader === 'string'
+                        ? protocolHeader.split(',').map(p => p.trim())
+                        : protocolHeader;
+                    if (protocols.length >= 2 && protocols[0] === 'token') {
+                        token = protocols[1];
+                        useProtocolHeader = true;
+                    }
+                }
+                // Fall back to query param for backward compatibility (deprecated)
+                if (!token) {
+                    token = query.token;
+                }
+                if (token !== authToken) {
+                    socket.write('HTTP/1.1 401 Unauthorized\r\n\r\n');
+                    socket.destroy();
+                    return;
+                }
+                this.wss.handleUpgrade(request, socket, head, (ws) => {
+                    // If client used protocol header auth, echo back the protocol name
+                    // This is required by WebSocket spec when using Sec-WebSocket-Protocol
+                    if (useProtocolHeader) {
+                        // Note: ws library handles this automatically if we pass protocol to handleUpgrade
+                        // but we emit connection event manually, so client will receive the echo
+                    }
+                    this.wss.emit('connection', ws, request, token);
+                });
+            }
+            else {
+                socket.destroy();
+            }
+        });
+        // Handle new connections
+        this.wss.on('connection', (ws, _request, token) => {
+            const clientId = this.generateId();
+            const client = {
+                id: clientId,
+                ws,
+                token,
+                subscribedSessions: new Set(),
+                isAlive: true,
+            };
+            this.clients.set(clientId, client);
+            console.log(`[WS] Client connected: ${clientId}`);
+            // Send welcome message
+            this.send(client, { type: 'connected', clientId });
+            // Handle incoming messages
+            ws.on('message', async (data) => {
+                try {
+                    const message = JSON.parse(data.toString());
+                    await this.handleMessage(client, message);
+                }
+                catch (error) {
+                    console.error('[WS] Failed to parse message:', error);
+                    this.send(client, { type: 'error', error: 'Invalid message format' });
+                }
+            });
+            // Handle pong for heartbeat
+            ws.on('pong', () => {
+                client.isAlive = true;
+            });
+            // Handle close
+            ws.on('close', () => {
+                console.log(`[WS] Client disconnected: ${clientId}`);
+                this.clients.delete(clientId);
+            });
+            // Handle errors
+            ws.on('error', (error) => {
+                console.error(`[WS] Client error ${clientId}:`, error);
+                this.clients.delete(clientId);
+            });
+        });
+        // Start heartbeat to detect dead connections
+        this.startHeartbeat();
+        console.log('[WS] WebSocket server initialized');
+    }
+    startHeartbeat() {
+        this.heartbeatInterval = setInterval(() => {
+            this.clients.forEach((client, id) => {
+                if (!client.isAlive) {
+                    console.log(`[WS] Terminating inactive client: ${id}`);
+                    client.ws.terminate();
+                    this.clients.delete(id);
+                    return;
+                }
+                client.isAlive = false;
+                client.ws.ping();
+            });
+        }, 30000); // Check every 30 seconds
+    }
+    async handleMessage(client, message) {
+        const handler = this.handlers.get(message.type);
+        if (handler) {
+            try {
+                await handler(client, message);
+            }
+            catch (error) {
+                const errorMsg = error instanceof Error ? error.message : String(error);
+                console.error(`[WS] Handler error for ${message.type}:`, errorMsg);
+                this.send(client, { type: 'error', error: errorMsg });
+            }
+        }
+        else {
+            console.warn(`[WS] Unknown message type: ${message.type}`);
+            this.send(client, { type: 'error', error: `Unknown message type: ${message.type}` });
+        }
+    }
+    // Register a message handler
+    on(type, handler) {
+        this.handlers.set(type, handler);
+    }
+    // Send message to a specific client
+    send(client, message) {
+        if (client.ws.readyState === WebSocket.OPEN) {
+            client.ws.send(JSON.stringify(message));
+        }
+    }
+    // Broadcast to all clients subscribed to a session
+    broadcastToSession(sessionId, message) {
+        this.clients.forEach((client) => {
+            if (client.subscribedSessions.has(sessionId)) {
+                this.send(client, { ...message, sessionId });
+            }
+        });
+    }
+    // Broadcast to all connected clients
+    broadcastAll(message) {
+        this.clients.forEach((client) => {
+            this.send(client, message);
+        });
+    }
+    // Subscribe client to a session
+    subscribeToSession(client, sessionId) {
+        client.subscribedSessions.add(sessionId);
+        console.log(`[WS] Client ${client.id} subscribed to session ${sessionId}`);
+    }
+    // Unsubscribe client from a session
+    unsubscribeFromSession(client, sessionId) {
+        client.subscribedSessions.delete(sessionId);
+        console.log(`[WS] Client ${client.id} unsubscribed from session ${sessionId}`);
+    }
+    // Get all clients subscribed to a session
+    getSessionSubscribers(sessionId) {
+        return Array.from(this.clients.values()).filter((client) => client.subscribedSessions.has(sessionId));
+    }
+    // Get client count
+    getClientCount() {
+        return this.clients.size;
+    }
+    // Cleanup
+    shutdown() {
+        if (this.heartbeatInterval) {
+            clearInterval(this.heartbeatInterval);
+        }
+        this.clients.forEach((client) => {
+            client.ws.close();
+        });
+        this.clients.clear();
+        if (this.wss) {
+            this.wss.close();
+        }
+        console.log('[WS] WebSocket server shut down');
+    }
+    generateId() {
+        return Math.random().toString(36).substring(2, 15);
+    }
+}
+// Singleton instance
+export const wsManager = new WebSocketManager();
+//# sourceMappingURL=ws-manager.js.map

package/dist/core/ws-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-manager.js","sourceRoot":"","sources":["../../src/core/ws-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAEhD,OAAO,EAAE,KAAK,IAAI,QAAQ,EAAE,MAAM,KAAK,CAAC;AAoBxC,MAAM,gBAAgB;IACZ,GAAG,GAA2B,IAAI,CAAC;IACnC,OAAO,GAA0B,IAAI,GAAG,EAAE,CAAC;IAC3C,QAAQ,GAAgC,IAAI,GAAG,EAAE,CAAC;IAClD,iBAAiB,GAA0B,IAAI,CAAC;IAExD,UAAU,CAAC,MAAc,EAAE,SAAiB;QAC1C,IAAI,CAAC,GAAG,GAAG,IAAI,eAAe,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QAEnD,+BAA+B;QAC/B,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE;YAC7C,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,IAAI,EAAE,EAAE,IAAI,CAAC,CAAC;YAE9D,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;gBACvB,wFAAwF;gBACxF,6FAA6F;gBAC7F,gDAAgD;gBAChD,IAAI,KAAyB,CAAC;gBAC9B,IAAI,iBAAiB,GAAG,KAAK,CAAC;gBAE9B,MAAM,cAAc,GAAG,OAAO,CAAC,OAAO,CAAC,wBAAwB,CAAC,CAAC;gBACjE,IAAI,cAAc,EAAE,CAAC;oBACnB,uCAAuC;oBACvC,MAAM,SAAS,GAAG,OAAO,cAAc,KAAK,QAAQ;wBAClD,CAAC,CAAC,cAAc,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;wBAC9C,CAAC,CAAC,cAAc,CAAC;oBACnB,IAAI,SAAS,CAAC,MAAM,IAAI,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,KAAK,OAAO,EAAE,CAAC;wBACtD,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;wBACrB,iBAAiB,GAAG,IAAI,CAAC;oBAC3B,CAAC;gBACH,CAAC;gBAED,mEAAmE;gBACnE,IAAI,CAAC,KAAK,EAAE,CAAC;oBACX,KAAK,GAAG,KAAK,CAAC,KAAe,CAAC;gBAChC,CAAC;gBAED,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;oBACxB,MAAM,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;oBAClD,MAAM,CAAC,OAAO,EAAE,CAAC;oBACjB,OAAO;gBACT,CAAC;gBAED,IAAI,CAAC,GAAI,CAAC,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE;oBACpD,mEAAmE;oBACnE,uEAAuE;oBACvE,IAAI,iBAAiB,EAAE,CAAC;wBACtB,mFAAmF;wBACnF,yEAAyE;oBAC3E,CAAC;oBACD,IAAI,CAAC,GAAI,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;gBACnD,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,yBAAyB;QACzB,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAa,EAAE,QAAyB,EAAE,KAAa,EAAE,EAAE;YACpF,MAAM,QAAQ,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;YACnC,MAAM,MAAM,GAAa;gBACvB,EAAE,EAAE,QAAQ;gBACZ,EAAE;gBACF,KAAK;gBACL,kBAAkB,EAAE,IAAI,GAAG,EAAE;gBAC7B,OAAO,EAAE,IAAI;aACd,CAAC;YAEF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YACnC,OAAO,CAAC,GAAG,CAAC,0BAA0B,QAAQ,EAAE,CAAC,CAAC;YAElD,uBAAuB;YACvB,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,CAAC;YAEnD,2BAA2B;YAC3B,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;gBAC9B,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAc,CAAC;oBACzD,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;gBAC5C,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;oBACtD,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,wBAAwB,EAAE,CAAC,CAAC;gBACxE,CAAC;YACH,CAAC,CAAC,CAAC;YAEH,4BAA4B;YAC5B,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;gBACjB,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC;YACxB,CAAC,CAAC,CAAC;YAEH,eAAe;YACf,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;gBAClB,OAAO,CAAC,GAAG,CAAC,6BAA6B,QAAQ,EAAE,CAAC,CAAC;gBACrD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAChC,CAAC,CAAC,CAAC;YAEH,gBAAgB;YAChB,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;gBACvB,OAAO,CAAC,KAAK,CAAC,qBAAqB,QAAQ,GAAG,EAAE,KAAK,CAAC,CAAC;gBACvD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAChC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,6CAA6C;QAC7C,IAAI,CAAC,cAAc,EAAE,CAAC;QAEtB,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;IACnD,CAAC;IAEO,cAAc;QACpB,IAAI,CAAC,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE;YACxC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,EAAE,EAAE;gBAClC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACpB,OAAO,CAAC,GAAG,CAAC,qCAAqC,EAAE,EAAE,CAAC,CAAC;oBACvD,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC;oBACtB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBACxB,OAAO;gBACT,CAAC;gBACD,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC;gBACvB,MAAM,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;YACnB,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,yBAAyB;IACtC,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,MAAgB,EAAE,OAAkB;QAC9D,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAEhD,IAAI,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC;gBACH,MAAM,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACjC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,QAAQ,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACxE,OAAO,CAAC,KAAK,CAAC,0BAA0B,OAAO,CAAC,IAAI,GAAG,EAAE,QAAQ,CAAC,CAAC;gBACnE,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,IAAI,CAAC,8BAA8B,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;YAC3D,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,yBAAyB,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACvF,CAAC;IACH,CAAC;IAED,6BAA6B;IAC7B,EAAE,CAAC,IAAY,EAAE,OAAuB;QACtC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC;IAED,oCAAoC;IACpC,IAAI,CAAC,MAAgB,EAAE,OAAkB;QACvC,IAAI,MAAM,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YAC5C,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAED,mDAAmD;IACnD,kBAAkB,CAAC,SAAiB,EAAE,OAAkB;QACtD,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAC9B,IAAI,MAAM,CAAC,kBAAkB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;gBAC7C,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,GAAG,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED,qCAAqC;IACrC,YAAY,CAAC,OAAkB;QAC7B,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAC9B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC;IAED,gCAAgC;IAChC,kBAAkB,CAAC,MAAgB,EAAE,SAAiB;QACpD,MAAM,CAAC,kBAAkB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACzC,OAAO,CAAC,GAAG,CAAC,eAAe,MAAM,CAAC,EAAE,0BAA0B,SAAS,EAAE,CAAC,CAAC;IAC7E,CAAC;IAED,oCAAoC;IACpC,sBAAsB,CAAC,MAAgB,EAAE,SAAiB;QACxD,MAAM,CAAC,kBAAkB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC5C,OAAO,CAAC,GAAG,CAAC,eAAe,MAAM,CAAC,EAAE,8BAA8B,SAAS,EAAE,CAAC,CAAC;IACjF,CAAC;IAED,0CAA0C;IAC1C,qBAAqB,CAAC,SAAiB;QACrC,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAC7C,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,GAAG,CAAC,SAAS,CAAC,CACrD,CAAC;IACJ,CAAC;IAED,mBAAmB;IACnB,cAAc;QACZ,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAC3B,CAAC;IAED,UAAU;IACV,QAAQ;QACN,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,aAAa,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACxC,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAC9B,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;QACpB,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAErB,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;QACnB,CAAC;QAED,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;IACjD,CAAC;IAEO,UAAU;QAChB,OAAO,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IACrD,CAAC;CACF;AAED,qBAAqB;AACrB,MAAM,CAAC,MAAM,SAAS,GAAG,IAAI,gBAAgB,EAAE,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,7 @@
+export interface StartServerOptions {
+    port?: number;
+    host?: string;
+    skipWizard?: boolean;
+}
+export declare function startServer(options?: StartServerOptions): Promise<void>;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAgBA,MAAM,WAAW,kBAAkB;IACjC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,UAAU,CAAC,EAAE,OAAO,CAAC;CACtB;AAeD,wBAAsB,WAAW,CAAC,OAAO,GAAE,kBAAuB,GAAG,OAAO,CAAC,IAAI,CAAC,CA6MjF"}