agent4discord 0.1.0

package/dist/sessions/sessionManager.js

@@ -0,0 +1,203 @@
+// Session lifecycle -- AGE-016
+import { EventEmitter } from 'node:events';
+import { query, } from '@anthropic-ai/claude-agent-sdk';
+import { resolvePlugins } from '../utils/plugins.js';
+class SessionManager extends EventEmitter {
+    sessions = new Map();
+    createSession(guildId, userId, channelId, cwd, model, canUseTool) {
+        const controller = new AbortController();
+        let resolveNext = null;
+        async function* messageStream() {
+            while (true) {
+                const msg = await new Promise((resolve) => {
+                    resolveNext = resolve;
+                });
+                yield msg;
+            }
+        }
+        const plugins = resolvePlugins();
+        const q = query({
+            prompt: messageStream(),
+            options: {
+                cwd,
+                model: model || 'opus',
+                permissionMode: 'default',
+                includePartialMessages: true,
+                abortController: controller,
+                canUseTool,
+                plugins,
+            },
+        });
+        const session = {
+            query: q,
+            channelId,
+            guildId,
+            userId,
+            sessionId: '',
+            cwd,
+            state: 'running',
+            totalCostUsd: 0,
+            createdAt: new Date().toISOString(),
+            resolveNext: null,
+            abortController: controller,
+        };
+        // Wire up resolveNext -- the generator captures the outer variable,
+        // but ActiveSession needs its own reference so sendMessage can call it.
+        // Because the generator closure mutates the local `resolveNext`, we
+        // need a proxy that always reads the latest value.
+        Object.defineProperty(session, 'resolveNext', {
+            get: () => resolveNext,
+            set: (v) => {
+                resolveNext = v;
+            },
+            enumerable: true,
+            configurable: true,
+        });
+        this.sessions.set(channelId, session);
+        void this._processEvents(session);
+        return session;
+    }
+    resumeSession(guildId, userId, channelId, sessionId, cwd, model, canUseTool) {
+        const controller = new AbortController();
+        let resolveNext = null;
+        async function* messageStream() {
+            while (true) {
+                const msg = await new Promise((resolve) => {
+                    resolveNext = resolve;
+                });
+                yield msg;
+            }
+        }
+        const plugins = resolvePlugins();
+        const q = query({
+            prompt: messageStream(),
+            options: {
+                cwd,
+                model: model || 'opus',
+                permissionMode: 'default',
+                includePartialMessages: true,
+                abortController: controller,
+                resume: sessionId,
+                canUseTool,
+                plugins,
+            },
+        });
+        const session = {
+            query: q,
+            channelId,
+            guildId,
+            userId,
+            sessionId,
+            cwd,
+            state: 'running',
+            totalCostUsd: 0,
+            createdAt: new Date().toISOString(),
+            resolveNext: null,
+            abortController: controller,
+        };
+        Object.defineProperty(session, 'resolveNext', {
+            get: () => resolveNext,
+            set: (v) => {
+                resolveNext = v;
+            },
+            enumerable: true,
+            configurable: true,
+        });
+        this.sessions.set(channelId, session);
+        void this._processEvents(session);
+        return session;
+    }
+    sendMessage(channelId, content) {
+        const session = this.sessions.get(channelId);
+        if (!session) {
+            throw new Error(`No session found for channel ${channelId}`);
+        }
+        if (session.state === 'stopped' || session.state === 'archived') {
+            throw new Error(`Session for channel ${channelId} is ${session.state}`);
+        }
+        session.state = 'running';
+        if (session.resolveNext) {
+            session.resolveNext({
+                type: 'user',
+                message: { role: 'user', content },
+                parent_tool_use_id: null,
+            });
+        }
+    }
+    getSession(channelId) {
+        return this.sessions.get(channelId) ?? null;
+    }
+    stopSession(channelId) {
+        const session = this.sessions.get(channelId);
+        if (!session)
+            return;
+        session.query.close();
+        session.state = 'stopped';
+    }
+    removeSession(channelId) {
+        const session = this.sessions.get(channelId);
+        if (!session)
+            return;
+        if (session.state !== 'stopped' && session.state !== 'archived') {
+            this.stopSession(channelId);
+        }
+        this.sessions.delete(channelId);
+    }
+    getAllSessions() {
+        return [...this.sessions.values()];
+    }
+    async _processEvents(session) {
+        try {
+            for await (const msg of session.query) {
+                // Log all non-stream messages for debugging
+                if (msg.type !== 'stream_event' && msg.type !== 'tool_progress') {
+                    console.log(`[sdk:${msg.type}]`, JSON.stringify(msg, null, 2).slice(0, 500));
+                }
+                switch (msg.type) {
+                    case 'system': {
+                        const sysMsg = msg;
+                        if (sysMsg.subtype === 'init' && sysMsg.session_id) {
+                            session.sessionId = sysMsg.session_id;
+                        }
+                        session.state = 'idle';
+                        break;
+                    }
+                    case 'assistant': {
+                        this.emit('assistant', session.channelId, msg);
+                        break;
+                    }
+                    case 'result': {
+                        const resultMsg = msg;
+                        console.log(`[sdk:result] subtype=${resultMsg.subtype}, cost=$${resultMsg.total_cost_usd}`);
+                        session.totalCostUsd = resultMsg.total_cost_usd ?? session.totalCostUsd;
+                        session.state = 'idle';
+                        this.emit('result', session.channelId, resultMsg);
+                        break;
+                    }
+                    case 'stream_event':
+                        this.emit('stream_event', session.channelId, msg);
+                        break;
+                    case 'tool_progress':
+                        this.emit('tool_progress', session.channelId, msg);
+                        break;
+                    default: {
+                        // Handle rate_limit_event
+                        const anyMsg = msg;
+                        if (anyMsg.type === 'rate_limit_event' && anyMsg.rate_limit_info) {
+                            console.log('[sdk:rate_limit]', JSON.stringify(anyMsg.rate_limit_info));
+                            this.emit('rate_limit', session.guildId, anyMsg.rate_limit_info);
+                        }
+                        break;
+                    }
+                }
+            }
+        }
+        catch (err) {
+            console.error(`[session] Event processing error for channel ${session.channelId}:`, err);
+            session.state = 'stopped';
+            this.emit('error', session.channelId, err);
+        }
+    }
+}
+export const sessionManager = new SessionManager();
+//# sourceMappingURL=sessionManager.js.map

package/dist/sessions/sessionManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionManager.js","sourceRoot":"","sources":["../../src/sessions/sessionManager.ts"],"names":[],"mappings":"AAAA,+BAA+B;AAC/B,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAC3C,OAAO,EACL,KAAK,GAON,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAkBrD,MAAM,cAAe,SAAQ,YAAY;IAC/B,QAAQ,GAAG,IAAI,GAAG,EAAyB,CAAC;IAEpD,aAAa,CACX,OAAe,EACf,MAAc,EACd,SAAiB,EACjB,GAAW,EACX,KAAc,EACd,UAAuB;QAEvB,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QAEzC,IAAI,WAAW,GAA2C,IAAI,CAAC;QAE/D,KAAK,SAAS,CAAC,CAAC,aAAa;YAC3B,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,GAAG,GAAG,MAAM,IAAI,OAAO,CAAiB,CAAC,OAAO,EAAE,EAAE;oBACxD,WAAW,GAAG,OAAO,CAAC;gBACxB,CAAC,CAAC,CAAC;gBACH,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GAAG,cAAc,EAAE,CAAC;QAEjC,MAAM,CAAC,GAAG,KAAK,CAAC;YACd,MAAM,EAAE,aAAa,EAAE;YACvB,OAAO,EAAE;gBACP,GAAG;gBACH,KAAK,EAAE,KAAK,IAAI,MAAM;gBACtB,cAAc,EAAE,SAAS;gBACzB,sBAAsB,EAAE,IAAI;gBAC5B,eAAe,EAAE,UAAU;gBAC3B,UAAU;gBACV,OAAO;aACR;SACF,CAAC,CAAC;QAEH,MAAM,OAAO,GAAkB;YAC7B,KAAK,EAAE,CAAC;YACR,SAAS;YACT,OAAO;YACP,MAAM;YACN,SAAS,EAAE,EAAE;YACb,GAAG;YACH,KAAK,EAAE,SAAS;YAChB,YAAY,EAAE,CAAC;YACf,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,WAAW,EAAE,IAAI;YACjB,eAAe,EAAE,UAAU;SAC5B,CAAC;QAEF,oEAAoE;QACpE,wEAAwE;QACxE,oEAAoE;QACpE,mDAAmD;QACnD,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,aAAa,EAAE;YAC5C,GAAG,EAAE,GAAG,EAAE,CAAC,WAAW;YACtB,GAAG,EAAE,CAAC,CAAyC,EAAE,EAAE;gBACjD,WAAW,GAAG,CAAC,CAAC;YAClB,CAAC;YACD,UAAU,EAAE,IAAI;YAChB,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QACtC,KAAK,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAClC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,aAAa,CACX,OAAe,EACf,MAAc,EACd,SAAiB,EACjB,SAAiB,EACjB,GAAW,EACX,KAAc,EACd,UAAuB;QAEvB,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QAEzC,IAAI,WAAW,GAA2C,IAAI,CAAC;QAE/D,KAAK,SAAS,CAAC,CAAC,aAAa;YAC3B,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,GAAG,GAAG,MAAM,IAAI,OAAO,CAAiB,CAAC,OAAO,EAAE,EAAE;oBACxD,WAAW,GAAG,OAAO,CAAC;gBACxB,CAAC,CAAC,CAAC;gBACH,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GAAG,cAAc,EAAE,CAAC;QAEjC,MAAM,CAAC,GAAG,KAAK,CAAC;YACd,MAAM,EAAE,aAAa,EAAE;YACvB,OAAO,EAAE;gBACP,GAAG;gBACH,KAAK,EAAE,KAAK,IAAI,MAAM;gBACtB,cAAc,EAAE,SAAS;gBACzB,sBAAsB,EAAE,IAAI;gBAC5B,eAAe,EAAE,UAAU;gBAC3B,MAAM,EAAE,SAAS;gBACjB,UAAU;gBACV,OAAO;aACR;SACF,CAAC,CAAC;QAEH,MAAM,OAAO,GAAkB;YAC7B,KAAK,EAAE,CAAC;YACR,SAAS;YACT,OAAO;YACP,MAAM;YACN,SAAS;YACT,GAAG;YACH,KAAK,EAAE,SAAS;YAChB,YAAY,EAAE,CAAC;YACf,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,WAAW,EAAE,IAAI;YACjB,eAAe,EAAE,UAAU;SAC5B,CAAC;QAEF,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,aAAa,EAAE;YAC5C,GAAG,EAAE,GAAG,EAAE,CAAC,WAAW;YACtB,GAAG,EAAE,CAAC,CAAyC,EAAE,EAAE;gBACjD,WAAW,GAAG,CAAC,CAAC;YAClB,CAAC;YACD,UAAU,EAAE,IAAI;YAChB,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QACtC,KAAK,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QAClC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,WAAW,CAAC,SAAiB,EAAE,OAAe;QAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,gCAAgC,SAAS,EAAE,CAAC,CAAC;QAC/D,CAAC;QACD,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,IAAI,OAAO,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;YAChE,MAAM,IAAI,KAAK,CAAC,uBAAuB,SAAS,OAAO,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;QAC1E,CAAC;QACD,OAAO,CAAC,KAAK,GAAG,SAAS,CAAC;QAC1B,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;YACxB,OAAO,CAAC,WAAW,CAAC;gBAClB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE;gBAClC,kBAAkB,EAAE,IAAI;aACP,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAED,UAAU,CAAC,SAAiB;QAC1B,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC;IAC9C,CAAC;IAED,WAAW,CAAC,SAAiB;QAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACtB,OAAO,CAAC,KAAK,GAAG,SAAS,CAAC;IAC5B,CAAC;IAED,aAAa,CAAC,SAAiB;QAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,IAAI,OAAO,CAAC,KAAK,KAAK,UAAU,EAAE,CAAC;YAChE,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;QAC9B,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC;IAED,cAAc;QACZ,OAAO,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IACrC,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,OAAsB;QACjD,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBACtC,4CAA4C;gBAC5C,IAAI,GAAG,CAAC,IAAI,KAAK,cAAc,IAAI,GAAG,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;oBAChE,OAAO,CAAC,GAAG,CAAC,QAAQ,GAAG,CAAC,IAAI,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;gBAC/E,CAAC;gBAED,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;oBACjB,KAAK,QAAQ,CAAC,CAAC,CAAC;wBACd,MAAM,MAAM,GAAG,GAAuB,CAAC;wBACvC,IAAI,MAAM,CAAC,OAAO,KAAK,MAAM,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;4BACnD,OAAO,CAAC,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC;wBACxC,CAAC;wBACD,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC;wBACvB,MAAM;oBACR,CAAC;oBACD,KAAK,WAAW,CAAC,CAAC,CAAC;wBACjB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC,SAAS,EAAE,GAA0B,CAAC,CAAC;wBACtE,MAAM;oBACR,CAAC;oBACD,KAAK,QAAQ,CAAC,CAAC,CAAC;wBACd,MAAM,SAAS,GAAG,GAAuB,CAAC;wBAC1C,OAAO,CAAC,GAAG,CAAC,wBAAwB,SAAS,CAAC,OAAO,WAAW,SAAS,CAAC,cAAc,EAAE,CAAC,CAAC;wBAC5F,OAAO,CAAC,YAAY,GAAG,SAAS,CAAC,cAAc,IAAI,OAAO,CAAC,YAAY,CAAC;wBACxE,OAAO,CAAC,KAAK,GAAG,MAAM,CAAC;wBACvB,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;wBAClD,MAAM;oBACR,CAAC;oBACD,KAAK,cAAc;wBACjB,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;wBAClD,MAAM;oBACR,KAAK,eAAe;wBAClB,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;wBACnD,MAAM;oBACR,OAAO,CAAC,CAAC,CAAC;wBACR,0BAA0B;wBAC1B,MAAM,MAAM,GAAG,GAAU,CAAC;wBAC1B,IAAI,MAAM,CAAC,IAAI,KAAK,kBAAkB,IAAI,MAAM,CAAC,eAAe,EAAE,CAAC;4BACjE,OAAO,CAAC,GAAG,CAAC,kBAAkB,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,CAAC;4BACxE,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC,OAAO,EAAE,MAAM,CAAC,eAAe,CAAC,CAAC;wBACnE,CAAC;wBACD,MAAM;oBACR,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CACX,gDAAgD,OAAO,CAAC,SAAS,GAAG,EACpE,GAAG,CACJ,CAAC;YACF,OAAO,CAAC,KAAK,GAAG,SAAS,CAAC;YAC1B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;CACF;AAED,MAAM,CAAC,MAAM,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC"}

package/dist/sessions/sessionStore.d.ts

@@ -0,0 +1,4 @@
+import { type SessionEntry } from '../guild.js';
+export declare function saveSessionToGuild(guildId: string, channelId: string, sessionId: string, cwd: string, userId: string): void;
+export declare function removeSessionFromGuild(guildId: string, channelId: string): void;
+export declare function getSessionsForGuild(guildId: string): Record<string, SessionEntry>;

package/dist/sessions/sessionStore.js

@@ -0,0 +1,29 @@
+// Persist session metadata to guild config -- AGE-016
+import { loadGuildConfig, saveGuildConfig, } from '../guild.js';
+export function saveSessionToGuild(guildId, channelId, sessionId, cwd, userId) {
+    const config = loadGuildConfig(guildId);
+    if (!config) {
+        throw new Error(`Guild config not found for guild ${guildId}`);
+    }
+    config.activeSessions[channelId] = {
+        sessionId,
+        cwd,
+        createdAt: new Date().toISOString(),
+        userId,
+    };
+    saveGuildConfig(config);
+}
+export function removeSessionFromGuild(guildId, channelId) {
+    const config = loadGuildConfig(guildId);
+    if (!config)
+        return;
+    delete config.activeSessions[channelId];
+    saveGuildConfig(config);
+}
+export function getSessionsForGuild(guildId) {
+    const config = loadGuildConfig(guildId);
+    if (!config)
+        return {};
+    return config.activeSessions;
+}
+//# sourceMappingURL=sessionStore.js.map

package/dist/sessions/sessionStore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sessionStore.js","sourceRoot":"","sources":["../../src/sessions/sessionStore.ts"],"names":[],"mappings":"AAAA,sDAAsD;AACtD,OAAO,EACL,eAAe,EACf,eAAe,GAEhB,MAAM,aAAa,CAAC;AAErB,MAAM,UAAU,kBAAkB,CAChC,OAAe,EACf,SAAiB,EACjB,SAAiB,EACjB,GAAW,EACX,MAAc;IAEd,MAAM,MAAM,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IACxC,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,oCAAoC,OAAO,EAAE,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,CAAC,cAAc,CAAC,SAAS,CAAC,GAAG;QACjC,SAAS;QACT,GAAG;QACH,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,MAAM;KACP,CAAC;IAEF,eAAe,CAAC,MAAM,CAAC,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,OAAe,EACf,SAAiB;IAEjB,MAAM,MAAM,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IACxC,IAAI,CAAC,MAAM;QAAE,OAAO;IAEpB,OAAO,MAAM,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;IACxC,eAAe,CAAC,MAAM,CAAC,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,OAAe;IAEf,MAAM,MAAM,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;IACxC,IAAI,CAAC,MAAM;QAAE,OAAO,EAAE,CAAC;IACvB,OAAO,MAAM,CAAC,cAAc,CAAC;AAC/B,CAAC"}

package/dist/sessions/streamHandler.d.ts

@@ -0,0 +1,18 @@
+import { type TextChannel } from 'discord.js';
+export declare class StreamHandler {
+    private channel;
+    private buffer;
+    private tokenCount;
+    private startTime;
+    private messageId;
+    private timer;
+    private debounceMs;
+    private type;
+    private finalized;
+    constructor(channel: TextChannel, type: 'text' | 'thinking', debounceMs?: number);
+    push(text: string): void;
+    private flush;
+    private sendOrEdit;
+    finalize(): Promise<void>;
+    getMessageId(): string | null;
+}

package/dist/sessions/streamHandler.js

@@ -0,0 +1,119 @@
+import { EmbedBuilder } from 'discord.js';
+import { COLORS } from '../formatters/embedBuilder.js';
+export class StreamHandler {
+    channel;
+    buffer = '';
+    tokenCount = 0;
+    startTime = Date.now();
+    messageId = null;
+    timer = null;
+    debounceMs;
+    type;
+    finalized = false;
+    constructor(channel, type, debounceMs) {
+        this.channel = channel;
+        this.type = type;
+        this.debounceMs = debounceMs ?? (type === 'text' ? 1000 : 2000);
+    }
+    push(text) {
+        this.buffer += text;
+        this.tokenCount++;
+        if (!this.timer) {
+            this.timer = setTimeout(() => void this.flush(), this.debounceMs);
+        }
+    }
+    async flush() {
+        this.timer = null;
+        const elapsed = ((Date.now() - this.startTime) / 1000).toFixed(1);
+        if (this.type === 'text') {
+            // Streaming text: yellow embed with accumulated text
+            const description = this.buffer.length > 4090 ? this.buffer.slice(-4090) + '...' : this.buffer;
+            const embed = new EmbedBuilder()
+                .setTitle('Responding...')
+                .setDescription(description)
+                .setColor(COLORS.STREAMING)
+                .setFooter({ text: `${elapsed}s \u00b7 ${this.tokenCount} tokens` });
+            await this.sendOrEdit(embed);
+        }
+        else {
+            // Thinking: purple embed with only token count and time
+            const embed = new EmbedBuilder()
+                .setTitle('Thinking...')
+                .setColor(COLORS.THINKING)
+                .setFooter({ text: `${elapsed}s \u00b7 ${this.tokenCount} tokens` });
+            await this.sendOrEdit(embed);
+        }
+    }
+    async sendOrEdit(embed) {
+        try {
+            if (this.messageId) {
+                const msg = await this.channel.messages.fetch(this.messageId).catch(() => null);
+                if (msg) {
+                    await msg.edit({ embeds: [embed] });
+                    return;
+                }
+            }
+            const msg = await this.channel.send({ embeds: [embed] });
+            this.messageId = msg.id;
+        }
+        catch (err) {
+            console.error('[stream] Failed to update embed:', err);
+        }
+    }
+    async finalize() {
+        if (this.finalized)
+            return;
+        this.finalized = true;
+        // Clear any pending timer
+        if (this.timer) {
+            clearTimeout(this.timer);
+            this.timer = null;
+        }
+        const elapsed = ((Date.now() - this.startTime) / 1000).toFixed(1);
+        if (this.type === 'text') {
+            // Replace streaming embed with plain message(s)
+            if (this.messageId) {
+                try {
+                    const msg = await this.channel.messages.fetch(this.messageId).catch(() => null);
+                    if (msg)
+                        await msg.delete().catch(() => { });
+                }
+                catch {
+                    /* best-effort */
+                }
+            }
+            // Send as plain text, chunked if needed
+            if (this.buffer.length > 0) {
+                const { chunkMessage } = await import('../formatters/chunker.js');
+                const chunks = chunkMessage(this.buffer);
+                for (const chunk of chunks) {
+                    await this.channel.send(chunk);
+                }
+            }
+        }
+        else {
+            // Thinking: finalize embed with "Thought for Xs"
+            const embed = new EmbedBuilder()
+                .setTitle(`Thought for ${elapsed}s`)
+                .setColor(COLORS.THINKING)
+                .setFooter({ text: `${this.tokenCount} tokens` });
+            if (this.messageId) {
+                try {
+                    const msg = await this.channel.messages.fetch(this.messageId).catch(() => null);
+                    if (msg) {
+                        await msg.edit({ embeds: [embed] });
+                        return;
+                    }
+                }
+                catch {
+                    /* best-effort */
+                }
+            }
+            await this.channel.send({ embeds: [embed] });
+        }
+    }
+    getMessageId() {
+        return this.messageId;
+    }
+}
+//# sourceMappingURL=streamHandler.js.map

package/dist/sessions/streamHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamHandler.js","sourceRoot":"","sources":["../../src/sessions/streamHandler.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAoB,MAAM,YAAY,CAAC;AAC5D,OAAO,EAAE,MAAM,EAAE,MAAM,+BAA+B,CAAC;AAEvD,MAAM,OAAO,aAAa;IAChB,OAAO,CAAc;IACrB,MAAM,GAAG,EAAE,CAAC;IACZ,UAAU,GAAG,CAAC,CAAC;IACf,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,SAAS,GAAkB,IAAI,CAAC;IAChC,KAAK,GAAyC,IAAI,CAAC;IACnD,UAAU,CAAS;IACnB,IAAI,CAAsB;IAC1B,SAAS,GAAG,KAAK,CAAC;IAE1B,YAAY,OAAoB,EAAE,IAAyB,EAAE,UAAmB;QAC9E,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,UAAU,GAAG,UAAU,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;IAClE,CAAC;IAED,IAAI,CAAC,IAAY;QACf,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC;QACpB,IAAI,CAAC,UAAU,EAAE,CAAC;QAClB,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,KAAK;QACjB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAElE,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,qDAAqD;YACrD,MAAM,WAAW,GACf,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;YAC7E,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE;iBAC7B,QAAQ,CAAC,eAAe,CAAC;iBACzB,cAAc,CAAC,WAAW,CAAC;iBAC3B,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC;iBAC1B,SAAS,CAAC,EAAE,IAAI,EAAE,GAAG,OAAO,YAAY,IAAI,CAAC,UAAU,SAAS,EAAE,CAAC,CAAC;YAEvE,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;aAAM,CAAC;YACN,wDAAwD;YACxD,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE;iBAC7B,QAAQ,CAAC,aAAa,CAAC;iBACvB,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC;iBACzB,SAAS,CAAC,EAAE,IAAI,EAAE,GAAG,OAAO,YAAY,IAAI,CAAC,UAAU,SAAS,EAAE,CAAC,CAAC;YAEvE,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,KAAmB;QAC1C,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;gBAChF,IAAI,GAAG,EAAE,CAAC;oBACR,MAAM,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;oBACpC,OAAO;gBACT,CAAC;YACH,CAAC;YACD,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACzD,IAAI,CAAC,SAAS,GAAG,GAAG,CAAC,EAAE,CAAC;QAC1B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,GAAG,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAEtB,0BAA0B;QAC1B,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QACpB,CAAC;QAED,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAElE,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACzB,gDAAgD;YAChD,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnB,IAAI,CAAC;oBACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;oBAChF,IAAI,GAAG;wBAAE,MAAM,GAAG,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;gBAC9C,CAAC;gBAAC,MAAM,CAAC;oBACP,iBAAiB;gBACnB,CAAC;YACH,CAAC;YACD,wCAAwC;YACxC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC3B,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,0BAA0B,CAAC,CAAC;gBAClE,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBACzC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;oBAC3B,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACjC,CAAC;YACH,CAAC;QACH,CAAC;aAAM,CAAC;YACN,iDAAiD;YACjD,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE;iBAC7B,QAAQ,CAAC,eAAe,OAAO,GAAG,CAAC;iBACnC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC;iBACzB,SAAS,CAAC,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,UAAU,SAAS,EAAE,CAAC,CAAC;YAEpD,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnB,IAAI,CAAC;oBACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;oBAChF,IAAI,GAAG,EAAE,CAAC;wBACR,MAAM,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;wBACpC,OAAO;oBACT,CAAC;gBACH,CAAC;gBAAC,MAAM,CAAC;oBACP,iBAAiB;gBACnB,CAAC;YACH,CAAC;YACD,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;CACF"}

package/dist/sessions/toolProgress.d.ts

@@ -0,0 +1,14 @@
+import { type TextChannel } from 'discord.js';
+export declare class ToolProgressHandler {
+    private channel;
+    private messageId;
+    private startTime;
+    private timer;
+    private toolName;
+    constructor(channel: TextChannel, toolName: string);
+    update(): void;
+    private flush;
+    private sendOrEdit;
+    finalize(): Promise<void>;
+    getMessageId(): string | null;
+}

package/dist/sessions/toolProgress.js

@@ -0,0 +1,65 @@
+import { EmbedBuilder } from 'discord.js';
+import { COLORS } from '../formatters/embedBuilder.js';
+export class ToolProgressHandler {
+    channel;
+    messageId = null;
+    startTime = Date.now();
+    timer = null;
+    toolName;
+    constructor(channel, toolName) {
+        this.channel = channel;
+        this.toolName = toolName;
+    }
+    update() {
+        // Debounce at ~3s
+        if (!this.timer) {
+            this.timer = setTimeout(() => void this.flush(), 3000);
+        }
+    }
+    async flush() {
+        this.timer = null;
+        const elapsed = ((Date.now() - this.startTime) / 1000).toFixed(1);
+        const embed = new EmbedBuilder()
+            .setTitle(`Executing: ${this.toolName}`)
+            .setColor(COLORS.TOOL_PROGRESS)
+            .setFooter({ text: `${elapsed}s` });
+        await this.sendOrEdit(embed);
+    }
+    async sendOrEdit(embed) {
+        try {
+            if (this.messageId) {
+                const msg = await this.channel.messages.fetch(this.messageId).catch(() => null);
+                if (msg) {
+                    await msg.edit({ embeds: [embed] });
+                    return;
+                }
+            }
+            const msg = await this.channel.send({ embeds: [embed] });
+            this.messageId = msg.id;
+        }
+        catch (err) {
+            console.error('[tool-progress] Failed to update embed:', err);
+        }
+    }
+    async finalize() {
+        if (this.timer) {
+            clearTimeout(this.timer);
+            this.timer = null;
+        }
+        // Delete the progress embed (tool result will replace it)
+        if (this.messageId) {
+            try {
+                const msg = await this.channel.messages.fetch(this.messageId).catch(() => null);
+                if (msg)
+                    await msg.delete().catch(() => { });
+            }
+            catch {
+                /* best-effort */
+            }
+        }
+    }
+    getMessageId() {
+        return this.messageId;
+    }
+}
+//# sourceMappingURL=toolProgress.js.map

package/dist/sessions/toolProgress.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"toolProgress.js","sourceRoot":"","sources":["../../src/sessions/toolProgress.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAoB,MAAM,YAAY,CAAC;AAC5D,OAAO,EAAE,MAAM,EAAE,MAAM,+BAA+B,CAAC;AAEvD,MAAM,OAAO,mBAAmB;IACtB,OAAO,CAAc;IACrB,SAAS,GAAkB,IAAI,CAAC;IAChC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,KAAK,GAAyC,IAAI,CAAC;IACnD,QAAQ,CAAS;IAEzB,YAAY,OAAoB,EAAE,QAAgB;QAChD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,MAAM;QACJ,kBAAkB;QAClB,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,KAAK;QACjB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QAClB,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAClE,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE;aAC7B,QAAQ,CAAC,cAAc,IAAI,CAAC,QAAQ,EAAE,CAAC;aACvC,QAAQ,CAAC,MAAM,CAAC,aAAa,CAAC;aAC9B,SAAS,CAAC,EAAE,IAAI,EAAE,GAAG,OAAO,GAAG,EAAE,CAAC,CAAC;QACtC,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,KAAmB;QAC1C,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnB,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;gBAChF,IAAI,GAAG,EAAE,CAAC;oBACR,MAAM,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;oBACpC,OAAO;gBACT,CAAC;YACH,CAAC;YACD,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACzD,IAAI,CAAC,SAAS,GAAG,GAAG,CAAC,EAAE,CAAC;QAC1B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,yCAAyC,EAAE,GAAG,CAAC,CAAC;QAChE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QACpB,CAAC;QACD,0DAA0D;QAC1D,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;gBAChF,IAAI,GAAG;oBAAE,MAAM,GAAG,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAC9C,CAAC;YAAC,MAAM,CAAC;gBACP,iBAAiB;YACnB,CAAC;QACH,CAAC;IACH,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;CACF"}

package/dist/sessions/usageTracker.d.ts

@@ -0,0 +1,12 @@
+import { EmbedBuilder, type Client } from 'discord.js';
+import type { SDKRateLimitInfo } from '@anthropic-ai/claude-agent-sdk';
+interface RateLimitState {
+    fiveHour?: SDKRateLimitInfo;
+    sevenDayOpus?: SDKRateLimitInfo;
+    sevenDaySonnet?: SDKRateLimitInfo;
+    sevenDay?: SDKRateLimitInfo;
+    lastUpdated: number;
+}
+export declare function buildUsageEmbed(state?: RateLimitState): EmbedBuilder;
+export declare function setupUsageTracker(client: Client): void;
+export {};