ugly-app 0.1.0

package/dist/server/Socket.js

@@ -0,0 +1,278 @@
+import { WebSocket, WebSocketServer } from 'ws';
+import './audio/index.js';
+import { selectSTTProvider } from './audio/stt/registry.js';
+import { createSTTStreamManager } from './audio/STTStream.js';
+import { selectTTSProvider } from './audio/tts/registry.js';
+import { createTTSStreamHandler } from './audio/TTSStream.js';
+import { verifyToken } from './Auth.js';
+import { subscribeCollection, subscribeDoc } from './NatsStore.js';
+import { registerPushConnection } from './PushNotification.js';
+function send(ws, msg) {
+    if (ws.readyState === WebSocket.OPEN)
+        ws.send(JSON.stringify(msg));
+}
+export function initWebSocketServer(server, router, collections, storage, 
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+typedDb, 
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+userHelper, wsPath = '/rpc', onSocketMessage) {
+    const wss = new WebSocketServer({ server, path: wsPath });
+    const sessions = new Map();
+    wss.on('connection', (ws) => {
+        let session = null;
+        const tts = createTTSStreamHandler({
+            getProvider: () => selectTTSProvider(),
+        });
+        const stt = createSTTStreamManager(() => selectSTTProvider());
+        const sendToClient = (data) => ws.send(JSON.stringify(data));
+        ws.on('message', async (data) => {
+            let msg;
+            try {
+                msg = JSON.parse(String(data));
+            }
+            catch {
+                ws.close();
+                return;
+            }
+            // Auth handshake must be first message
+            if (!session) {
+                if (msg.type !== 'auth') {
+                    ws.close();
+                    return;
+                }
+                const { token } = msg.payload;
+                try {
+                    const userId = await verifyToken(token);
+                    const user = await userHelper.get(typedDb, userId);
+                    if (!user) {
+                        send(ws, { id: msg.id, type: 'error', error: 'User not found' });
+                        ws.close();
+                        return;
+                    }
+                    const unregisterPush = await registerPushConnection(userId, ws);
+                    session = { userId, ws, subscriptions: new Map(), unregisterPush };
+                    sessions.set(ws, session);
+                    send(ws, { id: msg.id, type: 'auth', ok: true, result: { user } });
+                }
+                catch {
+                    send(ws, { id: msg.id, type: 'error', error: 'Invalid token' });
+                    ws.close();
+                }
+                return;
+            }
+            const s = session;
+            // Let custom handler intercept before framework processing
+            if (onSocketMessage) {
+                const handled = onSocketMessage(ws, s.userId, msg);
+                if (handled)
+                    return;
+            }
+            try {
+                if (msg.type === 'tts:start') {
+                    void tts.handle(msg, sendToClient);
+                    return;
+                }
+                if (msg.type === 'tts:cancel') {
+                    tts.cancel(msg.streamId);
+                    return;
+                }
+                if (msg.type === 'stt:start') {
+                    void stt.handleStart(msg, sendToClient);
+                    return;
+                }
+                if (msg.type === 'stt:audio') {
+                    stt.handleAudio(msg);
+                    return;
+                }
+                if (msg.type === 'stt:stop') {
+                    void stt.handleStop(msg);
+                    return;
+                }
+                if (msg.type === 'call' || msg.type === 'request') {
+                    const { name, input } = msg.payload;
+                    const result = await router.dispatch(msg.type === 'call' ? 'function' : 'request', name, input, s.userId);
+                    send(ws, { id: msg.id, type: 'response', result });
+                    return;
+                }
+                if (msg.type === 'getDoc') {
+                    const { collection, id } = msg.payload;
+                    const meta = collections[collection];
+                    if (!meta?.public) {
+                        send(ws, {
+                            id: msg.id,
+                            type: 'error',
+                            error: 'Collection not public',
+                        });
+                        return;
+                    }
+                    const doc = await typedDb.rawGetDoc(collection, id);
+                    send(ws, { id: msg.id, type: 'response', result: doc });
+                    return;
+                }
+                if (msg.type === 'trackDoc') {
+                    const { collection, id, subId } = msg.payload;
+                    const meta = collections[collection];
+                    if (!meta?.public || !meta?.trackable) {
+                        send(ws, {
+                            id: msg.id,
+                            type: 'error',
+                            error: 'Collection not trackable/public',
+                        });
+                        return;
+                    }
+                    // Send current value immediately
+                    const doc = await typedDb.rawGetDoc(collection, id);
+                    send(ws, { type: 'track', subId, data: doc });
+                    // Clean up any existing subscription for this subId before reusing it.
+                    // Without this, the old NatsStore subscription leaks and fires duplicate updates.
+                    s.subscriptions.get(subId)?.();
+                    // Subscribe to changes (synchronous — no await)
+                    const unsub = subscribeDoc(collection, id, (change) => {
+                        if (change.type === 'delete')
+                            send(ws, { type: 'track', subId, data: null });
+                        else
+                            send(ws, { type: 'track', subId, data: change.item });
+                    });
+                    s.subscriptions.set(subId, unsub);
+                    send(ws, { id: msg.id, type: 'response', result: null });
+                    return;
+                }
+                if (msg.type === 'trackDocs') {
+                    const { collection, filter, subId, sort, limit, skip } = msg.payload;
+                    const meta = collections[collection];
+                    if (!meta?.public || !meta?.trackable) {
+                        send(ws, {
+                            id: msg.id,
+                            type: 'error',
+                            error: 'Collection not trackable/public',
+                        });
+                        return;
+                    }
+                    // Validate filter — equality only, no MongoDB operator keys
+                    for (const [k, v] of Object.entries(filter)) {
+                        if (k.startsWith('$')) {
+                            send(ws, {
+                                id: msg.id,
+                                type: 'error',
+                                error: 'Operator keys not allowed in filter',
+                            });
+                            return;
+                        }
+                        if (typeof v === 'object' && v !== null) {
+                            send(ws, {
+                                id: msg.id,
+                                type: 'error',
+                                error: 'Only equality filters allowed',
+                            });
+                            return;
+                        }
+                    }
+                    // Validate sort — keys must not be operators; values must be 1 or -1
+                    if (sort !== undefined) {
+                        for (const [k, v] of Object.entries(sort)) {
+                            if (k.startsWith('$')) {
+                                send(ws, {
+                                    id: msg.id,
+                                    type: 'error',
+                                    error: 'Operator keys not allowed in sort',
+                                });
+                                return;
+                            }
+                            if (v !== 1 && v !== -1) {
+                                send(ws, {
+                                    id: msg.id,
+                                    type: 'error',
+                                    error: 'Sort values must be 1 (asc) or -1 (desc)',
+                                });
+                                return;
+                            }
+                        }
+                    }
+                    // Validate limit and skip — must be non-negative integers if provided
+                    if (limit !== undefined) {
+                        if (typeof limit !== 'number' ||
+                            !Number.isInteger(limit) ||
+                            limit < 0) {
+                            send(ws, {
+                                id: msg.id,
+                                type: 'error',
+                                error: 'limit must be a non-negative integer',
+                            });
+                            return;
+                        }
+                    }
+                    if (skip !== undefined) {
+                        if (typeof skip !== 'number' ||
+                            !Number.isInteger(skip) ||
+                            skip < 0) {
+                            send(ws, {
+                                id: msg.id,
+                                type: 'error',
+                                error: 'skip must be a non-negative integer',
+                            });
+                            return;
+                        }
+                    }
+                    // MongoDB treats limit=0 as "no limit" — short-circuit to [] instead
+                    // so callers get the intuitive behaviour of "limit 0 → empty result".
+                    if (limit === 0) {
+                        send(ws, { type: 'track', subId, data: [] });
+                        send(ws, { id: msg.id, type: 'response', result: null });
+                        return;
+                    }
+                    const queryOptions = {
+                        sort: sort,
+                        limit: limit,
+                        skip: skip,
+                    };
+                    const docs = await typedDb.rawGetDocs(collection, filter, queryOptions);
+                    send(ws, { type: 'track', subId, data: docs });
+                    // Clean up any existing subscription for this subId before reusing it.
+                    s.subscriptions.get(subId)?.();
+                    const unsub = subscribeCollection(collection, () => {
+                        void (async () => {
+                            const updated = await typedDb.rawGetDocs(collection, filter, queryOptions);
+                            send(ws, { type: 'track', subId, data: updated });
+                        })().catch((err) => {
+                            console.error('[Socket] trackDocs re-query failed', {
+                                collection,
+                                err,
+                            });
+                        });
+                    });
+                    s.subscriptions.set(subId, unsub);
+                    send(ws, { id: msg.id, type: 'response', result: null });
+                    return;
+                }
+                if (msg.type === 'unsubscribe') {
+                    const { subId } = msg.payload;
+                    s.subscriptions.get(subId)?.();
+                    s.subscriptions.delete(subId);
+                    send(ws, { id: msg.id, type: 'response', result: null });
+                    return;
+                }
+                if (msg.type === 'uploadUrl') {
+                    const { key } = msg.payload;
+                    const result = await storage.presignedPut('temp', key);
+                    send(ws, { id: msg.id, type: 'response', result });
+                    return;
+                }
+            }
+            catch (e) {
+                const err = e;
+                send(ws, { id: msg.id, type: 'error', error: err.message });
+            }
+        });
+        ws.on('close', () => {
+            stt.cleanup().catch(() => { });
+            if (session) {
+                for (const unsub of session.subscriptions.values())
+                    unsub();
+                session.unregisterPush();
+                sessions.delete(ws);
+            }
+        });
+    });
+    return wss;
+}
+//# sourceMappingURL=Socket.js.map

package/dist/server/Socket.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Socket.js","sourceRoot":"","sources":["../../src/server/Socket.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,IAAI,CAAC;AAShD,OAAO,kBAAkB,CAAC;AAC1B,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AACxC,OAAO,EAAE,mBAAmB,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AACnE,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AA6C/D,SAAS,IAAI,CAAC,EAAa,EAAE,GAAW;IACtC,IAAI,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI;QAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC;AACrE,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,MAAc,EACd,MAAiB,EACjB,WAA+B,EAC/B,OAAsB;AACtB,8DAA8D;AAC9D,OAAqB;AACrB,8DAA8D;AAC9D,UAA2B,EAC3B,MAAM,GAAG,MAAM,EACf,eAA0E;IAE1E,MAAM,GAAG,GAAG,IAAI,eAAe,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;IAC1D,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAsB,CAAC;IAE/C,GAAG,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,EAAE,EAAE;QAC1B,IAAI,OAAO,GAAmB,IAAI,CAAC;QACnC,MAAM,GAAG,GAAG,sBAAsB,CAAC;YACjC,WAAW,EAAE,GAAG,EAAE,CAAC,iBAAiB,EAAE;SACvC,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,sBAAsB,CAAC,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC,CAAC;QAC9D,MAAM,YAAY,GAAG,CAAC,IAAa,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;QAEtE,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE;YAC9B,IAAI,GAAoB,CAAC;YACzB,IAAI,CAAC;gBACH,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAoB,CAAC;YACpD,CAAC;YAAC,MAAM,CAAC;gBACP,EAAE,CAAC,KAAK,EAAE,CAAC;gBACX,OAAO;YACT,CAAC;YAED,uCAAuC;YACvC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBACxB,EAAE,CAAC,KAAK,EAAE,CAAC;oBACX,OAAO;gBACT,CAAC;gBACD,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAC,OAA4B,CAAC;gBACnD,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,KAAK,CAAC,CAAC;oBACxC,MAAM,IAAI,GAAG,MAAM,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;oBACnD,IAAI,CAAC,IAAI,EAAE,CAAC;wBACV,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,gBAAgB,EAAE,CAAC,CAAC;wBACjE,EAAE,CAAC,KAAK,EAAE,CAAC;wBACX,OAAO;oBACT,CAAC;oBACD,MAAM,cAAc,GAAG,MAAM,sBAAsB,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;oBAChE,OAAO,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,aAAa,EAAE,IAAI,GAAG,EAAE,EAAE,cAAc,EAAE,CAAC;oBACnE,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;oBAC1B,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC;gBACrE,CAAC;gBAAC,MAAM,CAAC;oBACP,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;oBAChE,EAAE,CAAC,KAAK,EAAE,CAAC;gBACb,CAAC;gBACD,OAAO;YACT,CAAC;YAED,MAAM,CAAC,GAAG,OAAO,CAAC;YAElB,2DAA2D;YAC3D,IAAI,eAAe,EAAE,CAAC;gBACpB,MAAM,OAAO,GAAG,eAAe,CAAC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;gBACnD,IAAI,OAAO;oBAAE,OAAO;YACtB,CAAC;YAED,IAAI,CAAC;gBACH,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC7B,KAAK,GAAG,CAAC,MAAM,CAAC,GAA6B,EAAE,YAAY,CAAC,CAAC;oBAC7D,OAAO;gBACT,CAAC;gBACD,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;oBAC9B,GAAG,CAAC,MAAM,CAAE,GAA+B,CAAC,QAAQ,CAAC,CAAC;oBACtD,OAAO;gBACT,CAAC;gBACD,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC7B,KAAK,GAAG,CAAC,WAAW,CAAC,GAA6B,EAAE,YAAY,CAAC,CAAC;oBAClE,OAAO;gBACT,CAAC;gBACD,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC7B,GAAG,CAAC,WAAW,CAAC,GAA6B,CAAC,CAAC;oBAC/C,OAAO;gBACT,CAAC;gBACD,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC5B,KAAK,GAAG,CAAC,UAAU,CAAC,GAA4B,CAAC,CAAC;oBAClD,OAAO;gBACT,CAAC;gBAED,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,IAAI,GAAG,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;oBAClD,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,GAAG,CAAC,OAG3B,CAAC;oBACF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAClC,GAAG,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,EAC5C,IAAI,EACJ,KAAK,EACL,CAAC,CAAC,MAAM,CACT,CAAC;oBACF,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC;oBACnD,OAAO;gBACT,CAAC;gBAED,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBAC1B,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,GAAG,GAAG,CAAC,OAG9B,CAAC;oBACF,MAAM,IAAI,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC;oBACrC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC;wBAClB,IAAI,CAAC,EAAE,EAAE;4BACP,EAAE,EAAE,GAAG,CAAC,EAAE;4BACV,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,uBAAuB;yBAC/B,CAAC,CAAC;wBACH,OAAO;oBACT,CAAC;oBACD,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;oBACpD,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;oBACxD,OAAO;gBACT,CAAC;gBAED,IAAI,GAAG,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC5B,MAAM,EAAE,UAAU,EAAE,EAAE,EAAE,KAAK,EAAE,GAAG,GAAG,CAAC,OAA0B,CAAC;oBACjE,MAAM,IAAI,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC;oBACrC,IAAI,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;wBACtC,IAAI,CAAC,EAAE,EAAE;4BACP,EAAE,EAAE,GAAG,CAAC,EAAE;4BACV,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,iCAAiC;yBACzC,CAAC,CAAC;wBACH,OAAO;oBACT,CAAC;oBACD,iCAAiC;oBACjC,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;oBACpD,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;oBAC9C,uEAAuE;oBACvE,kFAAkF;oBAClF,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC;oBAC/B,gDAAgD;oBAChD,MAAM,KAAK,GAAG,YAAY,CAAC,UAAU,EAAE,EAAE,EAAE,CAAC,MAAM,EAAE,EAAE;wBACpD,IAAI,MAAM,CAAC,IAAI,KAAK,QAAQ;4BAC1B,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;;4BAC5C,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC7D,CAAC,CAAC,CAAC;oBACH,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;oBAClC,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;oBACzD,OAAO;gBACT,CAAC;gBAED,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC7B,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,GACpD,GAAG,CAAC,OAA2B,CAAC;oBAClC,MAAM,IAAI,GAAG,WAAW,CAAC,UAAU,CAAC,CAAC;oBACrC,IAAI,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;wBACtC,IAAI,CAAC,EAAE,EAAE;4BACP,EAAE,EAAE,GAAG,CAAC,EAAE;4BACV,IAAI,EAAE,OAAO;4BACb,KAAK,EAAE,iCAAiC;yBACzC,CAAC,CAAC;wBACH,OAAO;oBACT,CAAC;oBACD,4DAA4D;oBAC5D,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC5C,IAAI,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;4BACtB,IAAI,CAAC,EAAE,EAAE;gCACP,EAAE,EAAE,GAAG,CAAC,EAAE;gCACV,IAAI,EAAE,OAAO;gCACb,KAAK,EAAE,qCAAqC;6BAC7C,CAAC,CAAC;4BACH,OAAO;wBACT,CAAC;wBACD,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;4BACxC,IAAI,CAAC,EAAE,EAAE;gCACP,EAAE,EAAE,GAAG,CAAC,EAAE;gCACV,IAAI,EAAE,OAAO;gCACb,KAAK,EAAE,+BAA+B;6BACvC,CAAC,CAAC;4BACH,OAAO;wBACT,CAAC;oBACH,CAAC;oBACD,qEAAqE;oBACrE,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;wBACvB,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;4BAC1C,IAAI,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gCACtB,IAAI,CAAC,EAAE,EAAE;oCACP,EAAE,EAAE,GAAG,CAAC,EAAE;oCACV,IAAI,EAAE,OAAO;oCACb,KAAK,EAAE,mCAAmC;iCAC3C,CAAC,CAAC;gCACH,OAAO;4BACT,CAAC;4BACD,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;gCACxB,IAAI,CAAC,EAAE,EAAE;oCACP,EAAE,EAAE,GAAG,CAAC,EAAE;oCACV,IAAI,EAAE,OAAO;oCACb,KAAK,EAAE,0CAA0C;iCAClD,CAAC,CAAC;gCACH,OAAO;4BACT,CAAC;wBACH,CAAC;oBACH,CAAC;oBACD,sEAAsE;oBACtE,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;wBACxB,IACE,OAAO,KAAK,KAAK,QAAQ;4BACzB,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC;4BACxB,KAAK,GAAG,CAAC,EACT,CAAC;4BACD,IAAI,CAAC,EAAE,EAAE;gCACP,EAAE,EAAE,GAAG,CAAC,EAAE;gCACV,IAAI,EAAE,OAAO;gCACb,KAAK,EAAE,sCAAsC;6BAC9C,CAAC,CAAC;4BACH,OAAO;wBACT,CAAC;oBACH,CAAC;oBACD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;wBACvB,IACE,OAAO,IAAI,KAAK,QAAQ;4BACxB,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC;4BACvB,IAAI,GAAG,CAAC,EACR,CAAC;4BACD,IAAI,CAAC,EAAE,EAAE;gCACP,EAAE,EAAE,GAAG,CAAC,EAAE;gCACV,IAAI,EAAE,OAAO;gCACb,KAAK,EAAE,qCAAqC;6BAC7C,CAAC,CAAC;4BACH,OAAO;wBACT,CAAC;oBACH,CAAC;oBACD,qEAAqE;oBACrE,sEAAsE;oBACtE,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;wBAChB,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;wBAC7C,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;wBACzD,OAAO;oBACT,CAAC;oBACD,MAAM,YAAY,GAAG;wBACnB,IAAI,EAAE,IAA0C;wBAChD,KAAK,EAAE,KAA2B;wBAClC,IAAI,EAAE,IAA0B;qBACjC,CAAC;oBACF,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,UAAU,CACnC,UAAU,EACV,MAAiC,EACjC,YAAY,CACb,CAAC;oBACF,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;oBAC/C,uEAAuE;oBACvE,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC;oBAC/B,MAAM,KAAK,GAAG,mBAAmB,CAAC,UAAU,EAAE,GAAG,EAAE;wBACjD,KAAK,CAAC,KAAK,IAAI,EAAE;4BACf,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CACtC,UAAU,EACV,MAAiC,EACjC,YAAY,CACb,CAAC;4BACF,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;wBACpD,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;4BAC1B,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE;gCAClD,UAAU;gCACV,GAAG;6BACJ,CAAC,CAAC;wBACL,CAAC,CAAC,CAAC;oBACL,CAAC,CAAC,CAAC;oBACH,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;oBAClC,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;oBACzD,OAAO;gBACT,CAAC;gBAED,IAAI,GAAG,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;oBAC/B,MAAM,EAAE,KAAK,EAAE,GAAG,GAAG,CAAC,OAA4B,CAAC;oBACnD,CAAC,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC;oBAC/B,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBAC9B,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;oBACzD,OAAO;gBACT,CAAC;gBAED,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC7B,MAAM,EAAE,GAAG,EAAE,GAAG,GAAG,CAAC,OAA0B,CAAC;oBAC/C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;oBACvD,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC;oBACnD,OAAO;gBACT,CAAC;YACH,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,MAAM,GAAG,GAAG,CAAU,CAAC;gBACvB,IAAI,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YAClB,GAAG,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAC9B,IAAI,OAAO,EAAE,CAAC;gBACZ,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,aAAa,CAAC,MAAM,EAAE;oBAAE,KAAK,EAAE,CAAC;gBAC5D,OAAO,CAAC,cAAc,EAAE,CAAC;gBACzB,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACtB,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO,GAAG,CAAC;AACb,CAAC"}

package/dist/server/SpendTracker.d.ts

@@ -0,0 +1,18 @@
+export declare class SpendTracker {
+  private maxPerHour;
+  private maxQueueDepth;
+  private queue;
+  private retryTimer;
+  constructor(opts?: { maxPerHour?: number; maxQueueDepth?: number });
+  private getWindowSpend;
+  private recordSpend;
+  private startRetryLoop;
+  private drainQueue;
+  checkAndRecord(
+    userId: string,
+    costUsd: number,
+    model?: string,
+  ): Promise<void>;
+}
+export declare function getSpendTracker(): SpendTracker;
+//# sourceMappingURL=SpendTracker.d.ts.map

package/dist/server/SpendTracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"SpendTracker.d.ts","sourceRoot":"","sources":["../../src/server/SpendTracker.ts"],"names":[],"mappings":"AA4BA,qBAAa,YAAY;IACvB,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,KAAK,CAAoB;IACjC,OAAO,CAAC,UAAU,CAA+C;gBAErD,IAAI,GAAE;QAAE,UAAU,CAAC,EAAE,MAAM,CAAC;QAAC,aAAa,CAAC,EAAE,MAAM,CAAA;KAAO;YAOxD,cAAc;YAYd,WAAW;IAezB,OAAO,CAAC,cAAc;YAOR,UAAU;IA0ClB,cAAc,CAClB,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,MAAM,EACf,KAAK,SAAY,GAChB,OAAO,CAAC,IAAI,CAAC;CAuBjB;AAID,wBAAgB,eAAe,IAAI,YAAY,CAG9C"}

package/dist/server/SpendTracker.js

@@ -0,0 +1,110 @@
+import { nanoid } from 'nanoid';
+import { dbDefaults } from '../shared/DB.js';
+import { SpendLimitError } from '../shared/Errors.js';
+import { getDocs, setDoc } from './DB.js';
+const RETRY_INTERVAL_MS = 5_000;
+const MAX_WAIT_MS = 5 * 60 * 1000; // 5 minutes
+const WINDOW_MS = 60 * 60 * 1000; // 60 minutes
+export class SpendTracker {
+  maxPerHour;
+  maxQueueDepth;
+  queue = [];
+  retryTimer = null;
+  constructor(opts = {}) {
+    this.maxPerHour =
+      opts.maxPerHour ??
+      parseFloat(process.env.AI_MAX_SPEND_PER_HOUR ?? '1.00');
+    this.maxQueueDepth = opts.maxQueueDepth ?? 100;
+  }
+  async getWindowSpend() {
+    const since = new Date(Date.now() - WINDOW_MS);
+    try {
+      const records = await getDocs('aiSpend', {
+        created: { $gte: since },
+      });
+      return records.reduce((sum, r) => sum + r.costUsd, 0);
+    } catch {
+      return 0;
+    }
+  }
+  async recordSpend(userId, model, costUsd) {
+    const doc = {
+      ...dbDefaults(),
+      id: nanoid(),
+      model,
+      costUsd,
+      userId,
+    };
+    await setDoc('aiSpend', doc);
+  }
+  startRetryLoop() {
+    if (this.retryTimer) return;
+    this.retryTimer = setInterval(() => {
+      void this.drainQueue();
+    }, RETRY_INTERVAL_MS);
+  }
+  async drainQueue() {
+    if (this.queue.length === 0) {
+      if (this.retryTimer) {
+        clearInterval(this.retryTimer);
+        this.retryTimer = null;
+      }
+      return;
+    }
+    const now = Date.now();
+    // Expire timed-out entries
+    this.queue = this.queue.filter((entry) => {
+      if (now - entry.enqueuedAt > MAX_WAIT_MS) {
+        entry.reject(new SpendLimitError());
+        return false;
+      }
+      return true;
+    });
+    const windowSpend = await this.getWindowSpend();
+    const available = this.maxPerHour - windowSpend;
+    let remaining = available;
+    const toProcess = [];
+    const toKeep = [];
+    for (const entry of this.queue) {
+      if (entry.costUsd <= remaining) {
+        toProcess.push(entry);
+        remaining -= entry.costUsd;
+      } else {
+        toKeep.push(entry);
+      }
+    }
+    this.queue = toKeep;
+    for (const entry of toProcess) {
+      await this.recordSpend(entry.userId, entry.model, entry.costUsd);
+      entry.resolve();
+    }
+  }
+  async checkAndRecord(userId, costUsd, model = 'unknown') {
+    const windowSpend = await this.getWindowSpend();
+    if (windowSpend + costUsd <= this.maxPerHour) {
+      await this.recordSpend(userId, model, costUsd);
+      return;
+    }
+    if (this.queue.length >= this.maxQueueDepth) {
+      throw new SpendLimitError();
+    }
+    await new Promise((resolve, reject) => {
+      this.queue.push({
+        costUsd,
+        userId,
+        model,
+        enqueuedAt: Date.now(),
+        resolve,
+        reject,
+      });
+      this.startRetryLoop();
+    });
+  }
+}
+// Singleton for use in TextGen/ImageGen
+let _tracker = null;
+export function getSpendTracker() {
+  if (!_tracker) _tracker = new SpendTracker();
+  return _tracker;
+}
+//# sourceMappingURL=SpendTracker.js.map

package/dist/server/SpendTracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SpendTracker.js","sourceRoot":"","sources":["../../src/server/SpendTracker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAqB1C,MAAM,iBAAiB,GAAG,KAAK,CAAC;AAChC,MAAM,WAAW,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,YAAY;AAC/C,MAAM,SAAS,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,aAAa;AAE/C,MAAM,OAAO,YAAY;IACf,UAAU,CAAS;IACnB,aAAa,CAAS;IACtB,KAAK,GAAiB,EAAE,CAAC;IACzB,UAAU,GAA0C,IAAI,CAAC;IAEjE,YAAY,OAAwD,EAAE;QACpE,IAAI,CAAC,UAAU;YACb,IAAI,CAAC,UAAU;gBACf,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,MAAM,CAAC,CAAC;QAC1D,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,IAAI,GAAG,CAAC;IACjD,CAAC;IAEO,KAAK,CAAC,cAAc;QAC1B,MAAM,KAAK,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC;QAC/C,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,OAAO,CAAc,SAAS,EAAE;gBACpD,OAAO,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE;aACzB,CAAC,CAAC;YACH,OAAO,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QACxD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW,CACvB,MAAc,EACd,KAAa,EACb,OAAe;QAEf,MAAM,GAAG,GAAgB;YACvB,GAAG,UAAU,EAAE;YACf,EAAE,EAAE,MAAM,EAAE;YACZ,KAAK;YACL,OAAO;YACP,MAAM;SACP,CAAC;QACF,MAAM,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IAC/B,CAAC;IAEO,cAAc;QACpB,IAAI,IAAI,CAAC,UAAU;YAAE,OAAO;QAC5B,IAAI,CAAC,UAAU,GAAG,WAAW,CAAC,GAAG,EAAE;YACjC,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC;QACzB,CAAC,EAAE,iBAAiB,CAAC,CAAC;IACxB,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACpB,aAAa,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBAC/B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACzB,CAAC;YACD,OAAO;QACT,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,2BAA2B;QAC3B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE;YACvC,IAAI,GAAG,GAAG,KAAK,CAAC,UAAU,GAAG,WAAW,EAAE,CAAC;gBACzC,KAAK,CAAC,MAAM,CAAC,IAAI,eAAe,EAAE,CAAC,CAAC;gBACpC,OAAO,KAAK,CAAC;YACf,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAChD,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,GAAG,WAAW,CAAC;QAEhD,IAAI,SAAS,GAAG,SAAS,CAAC;QAC1B,MAAM,SAAS,GAAiB,EAAE,CAAC;QACnC,MAAM,MAAM,GAAiB,EAAE,CAAC;QAEhC,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC/B,IAAI,KAAK,CAAC,OAAO,IAAI,SAAS,EAAE,CAAC;gBAC/B,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;gBACtB,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC;YAC7B,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC;QAEpB,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;YAC9B,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YACjE,KAAK,CAAC,OAAO,EAAE,CAAC;QAClB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,MAAc,EACd,OAAe,EACf,KAAK,GAAG,SAAS;QAEjB,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAChD,IAAI,WAAW,GAAG,OAAO,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YAC7C,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;YAC/C,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YAC5C,MAAM,IAAI,eAAe,EAAE,CAAC;QAC9B,CAAC;QAED,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC1C,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;gBACd,OAAO;gBACP,MAAM;gBACN,KAAK;gBACL,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;gBACtB,OAAO;gBACP,MAAM;aACP,CAAC,CAAC;YACH,IAAI,CAAC,cAAc,EAAE,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC;CACF;AAED,wCAAwC;AACxC,IAAI,QAAQ,GAAwB,IAAI,CAAC;AACzC,MAAM,UAAU,eAAe;IAC7B,IAAI,CAAC,QAAQ;QAAE,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAC;IAC7C,OAAO,QAAQ,CAAC;AAClB,CAAC"}

package/dist/server/Storage.d.ts

@@ -0,0 +1,16 @@
+export interface StorageClient {
+    put(bucket: 'public' | 'temp', key: string, body: Buffer, contentType: string): Promise<string>;
+    moveToPublic(tempKey: string, destKey: string): Promise<string>;
+    url(bucket: 'public' | 'temp', key: string): string;
+    presignedPut(bucket: 'temp', key: string): Promise<{
+        uploadUrl: string;
+        resultUrl: string;
+    }>;
+}
+/**
+ * Creates a storage client backed by Cloudflare R2 (production) or MinIO (dev).
+ *
+ * Reads configuration from env vars (STORAGE_* preferred, CLOUDFLARE_R2_* accepted as aliases).
+ */
+export declare function createStorageClient(): StorageClient;
+//# sourceMappingURL=Storage.d.ts.map

package/dist/server/Storage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"Storage.d.ts","sourceRoot":"","sources":["../../src/server/Storage.ts"],"names":[],"mappings":"AAuEA,MAAM,WAAW,aAAa;IAC5B,GAAG,CACD,MAAM,EAAE,QAAQ,GAAG,MAAM,EACzB,GAAG,EAAE,MAAM,EACX,IAAI,EAAE,MAAM,EACZ,WAAW,EAAE,MAAM,GAClB,OAAO,CAAC,MAAM,CAAC,CAAC;IACnB,YAAY,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAChE,GAAG,CAAC,MAAM,EAAE,QAAQ,GAAG,MAAM,EAAE,GAAG,EAAE,MAAM,GAAG,MAAM,CAAC;IACpD,YAAY,CACV,MAAM,EAAE,MAAM,EACd,GAAG,EAAE,MAAM,GACV,OAAO,CAAC;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACtD;AAED;;;;GAIG;AACH,wBAAgB,mBAAmB,IAAI,aAAa,CA8CnD"}

package/dist/server/Storage.js

@@ -0,0 +1,95 @@
+import { CopyObjectCommand, DeleteObjectCommand, PutObjectCommand, S3Client, } from '@aws-sdk/client-s3';
+import { getSignedUrl } from '@aws-sdk/s3-request-presigner';
+function isDev() {
+    return process.env.NODE_ENV !== 'production';
+}
+function getS3Client() {
+    if (isDev()) {
+        return new S3Client({
+            region: 'us-east-1',
+            endpoint: process.env['MINIO_ENDPOINT'] ?? 'http://localhost:9000',
+            credentials: { accessKeyId: 'minioadmin', secretAccessKey: 'minioadmin' },
+            forcePathStyle: true,
+        });
+    }
+    const accountId = process.env['STORAGE_ACCOUNT_ID'] ?? process.env['CLOUDFLARE_ACCOUNT_ID'];
+    const accessKeyId = process.env['STORAGE_ACCESS_KEY_ID'] ??
+        process.env['CLOUDFLARE_R2_ACCESS_KEY_ID'];
+    const secretAccessKey = process.env['STORAGE_SECRET_ACCESS_KEY'] ??
+        process.env['CLOUDFLARE_R2_SECRET_ACCESS_KEY'];
+    if (!accountId || !accessKeyId || !secretAccessKey) {
+        throw new Error('[Storage] STORAGE_ACCOUNT_ID, STORAGE_ACCESS_KEY_ID, STORAGE_SECRET_ACCESS_KEY required in production');
+    }
+    return new S3Client({
+        region: 'auto',
+        endpoint: `https://${accountId}.r2.cloudflarestorage.com`,
+        credentials: { accessKeyId, secretAccessKey },
+    });
+}
+function getBucketName(bucket) {
+    if (isDev())
+        return bucket;
+    const name = bucket === 'public'
+        ? process.env['STORAGE_PUBLIC_BUCKET'] ??
+            process.env['CLOUDFLARE_R2_PUBLIC_BUCKET']
+        : process.env['STORAGE_TEMP_BUCKET'] ??
+            process.env['CLOUDFLARE_R2_TEMP_BUCKET'];
+    if (!name)
+        throw new Error(`[Storage] STORAGE_${bucket.toUpperCase()}_BUCKET required in production`);
+    return name;
+}
+function getBaseUrl(bucket) {
+    if (isDev())
+        return `http://localhost:9000/${bucket}/`;
+    const url = bucket === 'public'
+        ? process.env['STORAGE_PUBLIC_URL'] ??
+            process.env['CLOUDFLARE_R2_PUBLIC_URL']
+        : process.env['STORAGE_TEMP_URL'] ??
+            process.env['CLOUDFLARE_R2_TEMP_URL'];
+    if (!url)
+        throw new Error(`[Storage] STORAGE_${bucket.toUpperCase()}_URL required in production`);
+    return url.endsWith('/') ? url : url + '/';
+}
+/**
+ * Creates a storage client backed by Cloudflare R2 (production) or MinIO (dev).
+ *
+ * Reads configuration from env vars (STORAGE_* preferred, CLOUDFLARE_R2_* accepted as aliases).
+ */
+export function createStorageClient() {
+    const s3 = getS3Client();
+    return {
+        async put(bucket, key, body, contentType) {
+            await s3.send(new PutObjectCommand({
+                Bucket: getBucketName(bucket),
+                Key: key,
+                Body: body,
+                ContentType: contentType,
+            }));
+            return getBaseUrl(bucket) + key;
+        },
+        async moveToPublic(tempKey, destKey) {
+            const srcBucket = getBucketName('temp');
+            const destBucket = getBucketName('public');
+            await s3.send(new CopyObjectCommand({
+                CopySource: `${srcBucket}/${tempKey}`,
+                Bucket: destBucket,
+                Key: destKey,
+            }));
+            await s3.send(new DeleteObjectCommand({ Bucket: srcBucket, Key: tempKey }));
+            return getBaseUrl('public') + destKey;
+        },
+        url(bucket, key) {
+            return getBaseUrl(bucket) + key;
+        },
+        async presignedPut(bucket, key) {
+            const command = new PutObjectCommand({
+                Bucket: getBucketName(bucket),
+                Key: key,
+            });
+            const uploadUrl = await getSignedUrl(s3, command, { expiresIn: 3600 });
+            const resultUrl = getBaseUrl(bucket) + key;
+            return { uploadUrl, resultUrl };
+        },
+    };
+}
+//# sourceMappingURL=Storage.js.map

package/dist/server/Storage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Storage.js","sourceRoot":"","sources":["../../src/server/Storage.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,iBAAiB,EACjB,mBAAmB,EACnB,gBAAgB,EAChB,QAAQ,GACT,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAE7D,SAAS,KAAK;IACZ,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,YAAY,CAAC;AAC/C,CAAC;AAED,SAAS,WAAW;IAClB,IAAI,KAAK,EAAE,EAAE,CAAC;QACZ,OAAO,IAAI,QAAQ,CAAC;YAClB,MAAM,EAAE,WAAW;YACnB,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,uBAAuB;YAClE,WAAW,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE;YACzE,cAAc,EAAE,IAAI;SACrB,CAAC,CAAC;IACL,CAAC;IACD,MAAM,SAAS,GACb,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IAC5E,MAAM,WAAW,GACf,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC;QACpC,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;IAC7C,MAAM,eAAe,GACnB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC;QACxC,OAAO,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAC;IACjD,IAAI,CAAC,SAAS,IAAI,CAAC,WAAW,IAAI,CAAC,eAAe,EAAE,CAAC;QACnD,MAAM,IAAI,KAAK,CACb,uGAAuG,CACxG,CAAC;IACJ,CAAC;IACD,OAAO,IAAI,QAAQ,CAAC;QAClB,MAAM,EAAE,MAAM;QACd,QAAQ,EAAE,WAAW,SAAS,2BAA2B;QACzD,WAAW,EAAE,EAAE,WAAW,EAAE,eAAe,EAAE;KAC9C,CAAC,CAAC;AACL,CAAC;AAED,SAAS,aAAa,CAAC,MAAyB;IAC9C,IAAI,KAAK,EAAE;QAAE,OAAO,MAAM,CAAC;IAC3B,MAAM,IAAI,GACR,MAAM,KAAK,QAAQ;QACjB,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC;YACpC,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC;QAC5C,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;YAClC,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;IAC/C,IAAI,CAAC,IAAI;QACP,MAAM,IAAI,KAAK,CACb,qBAAqB,MAAM,CAAC,WAAW,EAAE,gCAAgC,CAC1E,CAAC;IACJ,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,UAAU,CAAC,MAAyB;IAC3C,IAAI,KAAK,EAAE;QAAE,OAAO,yBAAyB,MAAM,GAAG,CAAC;IACvD,MAAM,GAAG,GACP,MAAM,KAAK,QAAQ;QACjB,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;YACjC,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC;QACzC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;YAC/B,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;IAC5C,IAAI,CAAC,GAAG;QACN,MAAM,IAAI,KAAK,CACb,qBAAqB,MAAM,CAAC,WAAW,EAAE,6BAA6B,CACvE,CAAC;IACJ,OAAO,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,CAAC;AAC7C,CAAC;AAiBD;;;;GAIG;AACH,MAAM,UAAU,mBAAmB;IACjC,MAAM,EAAE,GAAG,WAAW,EAAE,CAAC;IAEzB,OAAO;QACL,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,WAAW;YACtC,MAAM,EAAE,CAAC,IAAI,CACX,IAAI,gBAAgB,CAAC;gBACnB,MAAM,EAAE,aAAa,CAAC,MAAM,CAAC;gBAC7B,GAAG,EAAE,GAAG;gBACR,IAAI,EAAE,IAAI;gBACV,WAAW,EAAE,WAAW;aACzB,CAAC,CACH,CAAC;YACF,OAAO,UAAU,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC;QAClC,CAAC;QAED,KAAK,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO;YACjC,MAAM,SAAS,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;YACxC,MAAM,UAAU,GAAG,aAAa,CAAC,QAAQ,CAAC,CAAC;YAC3C,MAAM,EAAE,CAAC,IAAI,CACX,IAAI,iBAAiB,CAAC;gBACpB,UAAU,EAAE,GAAG,SAAS,IAAI,OAAO,EAAE;gBACrC,MAAM,EAAE,UAAU;gBAClB,GAAG,EAAE,OAAO;aACb,CAAC,CACH,CAAC;YACF,MAAM,EAAE,CAAC,IAAI,CACX,IAAI,mBAAmB,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAC7D,CAAC;YACF,OAAO,UAAU,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC;QACxC,CAAC;QAED,GAAG,CAAC,MAAM,EAAE,GAAG;YACb,OAAO,UAAU,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC;QAClC,CAAC;QAED,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,GAAG;YAC5B,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC;gBACnC,MAAM,EAAE,aAAa,CAAC,MAAM,CAAC;gBAC7B,GAAG,EAAE,GAAG;aACT,CAAC,CAAC;YACH,MAAM,SAAS,GAAG,MAAM,YAAY,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACvE,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC;YAC3C,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;QAClC,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/server/TextGen.d.ts

@@ -0,0 +1,5 @@
+import './ai/index.js';
+export { createTextGenClient as createTextGen } from './ai/TextGenClient.js';
+export type { TextGenClient } from './ai/TextGenClient.js';
+export type { Message, TextGenOptions } from './ai/types.js';
+//# sourceMappingURL=TextGen.d.ts.map

package/dist/server/TextGen.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TextGen.d.ts","sourceRoot":"","sources":["../../src/server/TextGen.ts"],"names":[],"mappings":"AAGA,OAAO,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,IAAI,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAC7E,YAAY,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AAC3D,YAAY,EAAE,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC"}

package/dist/server/TextGen.js

@@ -0,0 +1,6 @@
+// src/server/TextGen.ts
+// Thin re-export — AI functionality lives in src/server/ai/
+// Import side-effect registers all built-in providers
+import './ai/index.js';
+export { createTextGenClient as createTextGen } from './ai/TextGenClient.js';
+//# sourceMappingURL=TextGen.js.map

package/dist/server/TextGen.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TextGen.js","sourceRoot":"","sources":["../../src/server/TextGen.ts"],"names":[],"mappings":"AAAA,wBAAwB;AACxB,4DAA4D;AAC5D,sDAAsD;AACtD,OAAO,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,IAAI,aAAa,EAAE,MAAM,uBAAuB,CAAC"}

package/dist/server/WorkerQueue.d.ts

@@ -0,0 +1,27 @@
+import type { WorkerQueue } from './App.js';
+export interface Job<T = unknown> {
+    id: string;
+    type: string;
+    payload: T;
+    createdAt: Date;
+    attempts: number;
+    scheduledFor?: Date;
+    /** Call periodically in long-running jobs to extend the NATS ack deadline. */
+    working: () => void;
+}
+export interface JobHandler<T = unknown> {
+    (job: Job<T>): Promise<void>;
+}
+export interface WorkerQueueConfig {
+    streamName?: string;
+    clockServerOnly?: boolean;
+    concurrency?: number;
+    maxRetries?: number;
+}
+export declare function createWorkerQueue(config?: WorkerQueueConfig): WorkerQueue & {
+    enqueue<T>(type: string, payload: T, options?: {
+        delay?: number;
+    }): Promise<void>;
+    registerHandler<T>(type: string, handler: JobHandler<T>): void;
+};
+//# sourceMappingURL=WorkerQueue.d.ts.map

package/dist/server/WorkerQueue.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"WorkerQueue.d.ts","sourceRoot":"","sources":["../../src/server/WorkerQueue.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,UAAU,CAAC;AAG5C,MAAM,WAAW,GAAG,CAAC,CAAC,GAAG,OAAO;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,CAAC,CAAC;IACX,SAAS,EAAE,IAAI,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,IAAI,CAAC;IACpB,8EAA8E;IAC9E,OAAO,EAAE,MAAM,IAAI,CAAC;CACrB;AAED,MAAM,WAAW,UAAU,CAAC,CAAC,GAAG,OAAO;IACrC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAC9B;AAED,MAAM,WAAW,iBAAiB;IAEhC,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB,eAAe,CAAC,EAAE,OAAO,CAAC;IAE1B,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,wBAAgB,iBAAiB,CAC/B,MAAM,GAAE,iBAAsB,GAC7B,WAAW,GAAG;IACf,OAAO,CAAC,CAAC,EACP,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,CAAC,EACV,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GAC3B,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,eAAe,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC;CAChE,CA8KA"}