@nocobase/app-server 0.11.1-alpha.5 → 2.0.0-beta.2

package/dist/realtime/service.d.ts

@@ -0,0 +1,9 @@
+import type { Logger } from '@nocobase/logging';
+import type { RealtimeService } from './types.js';
+export interface RealtimeServiceOptions {
+    logger?: Pick<Logger, 'error'>;
+    maxSubscriptionsPerConnection?: number;
+}
+export type { RealtimeConnection, RealtimeConnectionContext, RealtimePublishResult, RealtimeService, RealtimeSubscription, } from './types.js';
+export declare function createRealtimeService(options?: RealtimeServiceOptions): RealtimeService;
+//# sourceMappingURL=service.d.ts.map

package/dist/realtime/service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"service.d.ts","sourceRoot":"","sources":["../../src/realtime/service.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAGhD,OAAO,KAAK,EAGV,eAAe,EAIhB,MAAM,YAAY,CAAC;AAapB,MAAM,WAAW,sBAAsB;IACrC,MAAM,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC/B,6BAA6B,CAAC,EAAE,MAAM,CAAC;CACxC;AAED,YAAY,EACV,kBAAkB,EAClB,yBAAyB,EACzB,qBAAqB,EACrB,eAAe,EACf,oBAAoB,GACrB,MAAM,YAAY,CAAC;AAEpB,wBAAgB,qBAAqB,CACnC,OAAO,GAAE,sBAA2B,GACnC,eAAe,CAwbjB"}

package/dist/realtime/service.js

@@ -0,0 +1,347 @@
+import { randomUUID } from 'node:crypto';
+import { encodeRealtimeServerMessage, parseRealtimeClientMessage, RealtimeProtocolError, validateRealtimeTopic, } from '@nocobase/realtime/protocol';
+const WEB_SOCKET_OPEN = 1;
+const CLOSE_GOING_AWAY = 1001;
+export function createRealtimeService(options = {}) {
+    const maxSubscriptionsPerConnection = options.maxSubscriptionsPerConnection ?? 64;
+    const connections = new Map();
+    const subscriptions = new Map();
+    const subscriptionsByTopic = new Map();
+    const listenersByTopic = new Map();
+    const service = {
+        connect(ws, context = {}) {
+            const connection = {
+                id: randomUUID(),
+                ws,
+                connectedAt: new Date(),
+                request: context.request,
+                principal: context.principal,
+                subscriptions: new Set(),
+            };
+            connections.set(connection.id, connection);
+            return connection;
+        },
+        disconnect(connection) {
+            const stored = connections.get(connection.id);
+            if (!stored) {
+                return;
+            }
+            removeConnectionSubscriptions(stored);
+            connections.delete(stored.id);
+        },
+        disconnectUser(userId) {
+            let disconnected = 0;
+            for (const connection of Array.from(connections.values())) {
+                if (connection.principal?.userId !== userId)
+                    continue;
+                if (connection.ws.readyState === WEB_SOCKET_OPEN) {
+                    connection.ws.close(CLOSE_GOING_AWAY, 'account session revoked');
+                }
+                service.disconnect(connection);
+                disconnected += 1;
+            }
+            return disconnected;
+        },
+        subscribe(connection, topic) {
+            validateRealtimeTopic(topic);
+            const existing = findSubscription(connection, { topic });
+            if (existing) {
+                return existing;
+            }
+            if (connection.subscriptions.size >= maxSubscriptionsPerConnection) {
+                throw new RealtimeProtocolError('TOO_MANY_SUBSCRIPTIONS', 'Realtime connection has too many subscriptions.');
+            }
+            const subscription = {
+                id: randomUUID(),
+                connectionId: connection.id,
+                topic,
+                userId: resolveSubscriptionUserId(connection, topic),
+                createdAt: new Date(),
+            };
+            subscriptions.set(subscription.id, subscription);
+            connection.subscriptions.add(subscription.id);
+            addTopicSubscription(subscription);
+            return subscription;
+        },
+        unsubscribe(connection, selector) {
+            const removed = [];
+            for (const subscriptionId of Array.from(connection.subscriptions)) {
+                const subscription = subscriptions.get(subscriptionId);
+                if (!subscription || !matchesSubscription(subscription, selector)) {
+                    continue;
+                }
+                removeSubscription(connection, subscription);
+                removed.push(subscription);
+            }
+            return removed;
+        },
+        handleClientMessage(connection, data) {
+            let message;
+            try {
+                message = parseRealtimeClientMessage(data);
+            }
+            catch (error) {
+                sendProtocolError(connection, error);
+                return;
+            }
+            try {
+                switch (message.type) {
+                    case 'subscribe': {
+                        const subscription = service.subscribe(connection, message.topic);
+                        send(connection, {
+                            type: 'subscribed',
+                            id: message.id,
+                            topic: subscription.topic,
+                            subscriptionId: subscription.id,
+                        });
+                        return;
+                    }
+                    case 'unsubscribe': {
+                        const removed = service.unsubscribe(connection, {
+                            subscriptionId: message.subscriptionId,
+                            topic: message.topic,
+                        });
+                        const first = removed[0];
+                        send(connection, {
+                            type: 'unsubscribed',
+                            id: message.id,
+                            subscriptionId: message.subscriptionId ?? first?.id,
+                            topic: message.topic ?? first?.topic,
+                        });
+                        return;
+                    }
+                    case 'publish':
+                        send(connection, {
+                            type: 'error',
+                            id: message.id,
+                            code: 'PUBLISH_FORBIDDEN',
+                            message: 'Client publish is not allowed.',
+                        });
+                        return;
+                    case 'ping':
+                        send(connection, {
+                            type: 'pong',
+                            id: message.id,
+                        });
+                        return;
+                }
+            }
+            catch (error) {
+                sendProtocolError(connection, error);
+            }
+        },
+        registerTopic(topic, topicDefinition) {
+            validateRealtimeTopic(topic);
+            if (topicOptions.has(topic)) {
+                throw new RealtimeProtocolError('TOPIC_ALREADY_REGISTERED', `Realtime topic "${topic}" is already registered.`);
+            }
+            topicOptions.set(topic, topicDefinition);
+            return () => {
+                if (topicOptions.get(topic) !== topicDefinition) {
+                    return;
+                }
+                removeTopicSubscriptions(topic);
+                topicOptions.delete(topic);
+            };
+        },
+        defineTopic(topic, topicDefinition) {
+            const close = service.registerTopic(topic, topicDefinition);
+            const definedTopic = topicDefinition.audience === 'user'
+                ? {
+                    name: topic,
+                    audience: 'user',
+                    publishFor(userId, payload) {
+                        return service.publish(topic, payload, { userId });
+                    },
+                    close,
+                }
+                : {
+                    name: topic,
+                    audience: 'public',
+                    publish(payload) {
+                        return service.publish(topic, payload);
+                    },
+                    close,
+                };
+            return definedTopic;
+        },
+        publish(topic, payload, options = {}) {
+            validateRealtimeTopic(topic);
+            const audience = topicOptions.get(topic)?.audience ?? 'public';
+            if (audience === 'user' && !options.userId) {
+                throw new RealtimeProtocolError('USER_AUDIENCE_REQUIRED', `Realtime topic "${topic}" requires a user audience.`);
+            }
+            if (audience === 'public' && options.userId) {
+                throw new RealtimeProtocolError('INVALID_AUDIENCE', `Realtime topic "${topic}" does not accept a user audience.`);
+            }
+            const subscriptionIds = subscriptionsByTopic.get(topic);
+            if (!subscriptionIds?.size) {
+                return {
+                    topic,
+                    subscriberCount: 0,
+                };
+            }
+            const message = {
+                type: 'event',
+                topic,
+                payload,
+                publishedAt: new Date().toISOString(),
+            };
+            const sentConnections = new Set();
+            for (const subscriptionId of Array.from(subscriptionIds)) {
+                const subscription = subscriptions.get(subscriptionId);
+                const connection = subscription
+                    ? connections.get(subscription.connectionId)
+                    : undefined;
+                if (!subscription ||
+                    !connection ||
+                    (audience === 'user' && subscription.userId !== options.userId) ||
+                    sentConnections.has(connection.id)) {
+                    continue;
+                }
+                if (connection.ws.readyState !== WEB_SOCKET_OPEN) {
+                    service.disconnect(connection);
+                    continue;
+                }
+                sentConnections.add(connection.id);
+                send(connection, message);
+            }
+            return {
+                topic,
+                subscriberCount: sentConnections.size,
+            };
+        },
+        subscriptionCount(topic) {
+            return subscriptionsByTopic.get(topic)?.size ?? 0;
+        },
+        onTopicSubscriptionChange(topic, listener) {
+            validateRealtimeTopic(topic);
+            const listeners = listenersByTopic.get(topic) ?? new Set();
+            listeners.add(listener);
+            listenersByTopic.set(topic, listeners);
+            return () => {
+                listeners.delete(listener);
+                if (listeners.size === 0) {
+                    listenersByTopic.delete(topic);
+                }
+            };
+        },
+        close() {
+            for (const connection of Array.from(connections.values())) {
+                if (connection.ws.readyState === WEB_SOCKET_OPEN) {
+                    connection.ws.close(CLOSE_GOING_AWAY, 'realtime service closed');
+                }
+                service.disconnect(connection);
+            }
+            subscriptions.clear();
+            subscriptionsByTopic.clear();
+            listenersByTopic.clear();
+            topicOptions.clear();
+        },
+    };
+    const topicOptions = new Map();
+    function findSubscription(connection, selector) {
+        for (const subscriptionId of connection.subscriptions) {
+            const subscription = subscriptions.get(subscriptionId);
+            if (subscription && matchesSubscription(subscription, selector)) {
+                return subscription;
+            }
+        }
+        return undefined;
+    }
+    function matchesSubscription(subscription, selector) {
+        return ((selector.subscriptionId === undefined ||
+            subscription.id === selector.subscriptionId) &&
+            (selector.topic === undefined || subscription.topic === selector.topic));
+    }
+    function addTopicSubscription(subscription) {
+        const previousCount = service.subscriptionCount(subscription.topic);
+        const topicSubscriptions = subscriptionsByTopic.get(subscription.topic) ?? new Set();
+        topicSubscriptions.add(subscription.id);
+        subscriptionsByTopic.set(subscription.topic, topicSubscriptions);
+        notifyTopicSubscriptionChange(subscription.topic, previousCount);
+    }
+    function removeConnectionSubscriptions(connection) {
+        for (const subscriptionId of Array.from(connection.subscriptions)) {
+            const subscription = subscriptions.get(subscriptionId);
+            if (subscription) {
+                removeSubscription(connection, subscription);
+            }
+        }
+    }
+    function removeTopicSubscriptions(topic) {
+        const subscriptionIds = subscriptionsByTopic.get(topic);
+        if (!subscriptionIds) {
+            return;
+        }
+        for (const subscriptionId of Array.from(subscriptionIds)) {
+            const subscription = subscriptions.get(subscriptionId);
+            const connection = subscription
+                ? connections.get(subscription.connectionId)
+                : undefined;
+            if (subscription && connection) {
+                removeSubscription(connection, subscription);
+            }
+        }
+    }
+    function resolveSubscriptionUserId(connection, topic) {
+        if ((topicOptions.get(topic)?.audience ?? 'public') !== 'user') {
+            return undefined;
+        }
+        const userId = connection.principal?.userId;
+        if (!userId) {
+            throw new RealtimeProtocolError('AUTHENTICATION_REQUIRED', `Realtime topic "${topic}" requires authentication.`);
+        }
+        return userId;
+    }
+    function removeSubscription(connection, subscription) {
+        const previousCount = service.subscriptionCount(subscription.topic);
+        connection.subscriptions.delete(subscription.id);
+        subscriptions.delete(subscription.id);
+        const topicSubscriptions = subscriptionsByTopic.get(subscription.topic);
+        topicSubscriptions?.delete(subscription.id);
+        if (topicSubscriptions?.size === 0) {
+            subscriptionsByTopic.delete(subscription.topic);
+        }
+        notifyTopicSubscriptionChange(subscription.topic, previousCount);
+    }
+    function notifyTopicSubscriptionChange(topic, previousCount) {
+        const count = service.subscriptionCount(topic);
+        if (count === previousCount) {
+            return;
+        }
+        const listeners = listenersByTopic.get(topic);
+        if (!listeners) {
+            return;
+        }
+        for (const listener of Array.from(listeners)) {
+            try {
+                listener(count);
+            }
+            catch (error) {
+                if (options.logger)
+                    options.logger.error({ err: error }, 'Realtime subscription listener failed');
+                else
+                    console.error(error);
+            }
+        }
+    }
+    function sendProtocolError(connection, error) {
+        const code = error instanceof RealtimeProtocolError ? error.code : 'REALTIME_ERROR';
+        const message = error instanceof Error ? error.message : 'Realtime message failed.';
+        send(connection, {
+            type: 'error',
+            code,
+            message,
+        });
+    }
+    function send(connection, message) {
+        if (connection.ws.readyState !== WEB_SOCKET_OPEN) {
+            service.disconnect(connection);
+            return;
+        }
+        connection.ws.send(encodeRealtimeServerMessage(message));
+    }
+    return service;
+}
+//# sourceMappingURL=service.js.map

package/dist/realtime/service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"service.js","sourceRoot":"","sources":["../../src/realtime/service.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAWzC,OAAO,EACL,2BAA2B,EAC3B,0BAA0B,EAC1B,qBAAqB,EACrB,qBAAqB,GAEtB,MAAM,6BAA6B,CAAC;AAErC,MAAM,eAAe,GAAG,CAAC,CAAC;AAC1B,MAAM,gBAAgB,GAAG,IAAI,CAAC;AAe9B,MAAM,UAAU,qBAAqB,CACnC,UAAkC,EAAE;IAEpC,MAAM,6BAA6B,GACjC,OAAO,CAAC,6BAA6B,IAAI,EAAE,CAAC;IAC9C,MAAM,WAAW,GAAG,IAAI,GAAG,EAA8B,CAAC;IAC1D,MAAM,aAAa,GAAG,IAAI,GAAG,EAAgC,CAAC;IAC9D,MAAM,oBAAoB,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC5D,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAwC,CAAC;IAEzE,MAAM,OAAO,GAAoB;QAC/B,OAAO,CAAC,EAAE,EAAE,OAAO,GAAG,EAAE;YACtB,MAAM,UAAU,GAAuB;gBACrC,EAAE,EAAE,UAAU,EAAE;gBAChB,EAAE;gBACF,WAAW,EAAE,IAAI,IAAI,EAAE;gBACvB,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,aAAa,EAAE,IAAI,GAAG,EAAE;aACzB,CAAC;YACF,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YAC3C,OAAO,UAAU,CAAC;QACpB,CAAC;QAED,UAAU,CAAC,UAAU;YACnB,MAAM,MAAM,GAAG,WAAW,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YAC9C,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,OAAO;YACT,CAAC;YAED,6BAA6B,CAAC,MAAM,CAAC,CAAC;YACtC,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAChC,CAAC;QAED,cAAc,CAAC,MAAM;YACnB,IAAI,YAAY,GAAG,CAAC,CAAC;YACrB,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;gBAC1D,IAAI,UAAU,CAAC,SAAS,EAAE,MAAM,KAAK,MAAM;oBAAE,SAAS;gBACtD,IAAI,UAAU,CAAC,EAAE,CAAC,UAAU,KAAK,eAAe,EAAE,CAAC;oBACjD,UAAU,CAAC,EAAE,CAAC,KAAK,CAAC,gBAAgB,EAAE,yBAAyB,CAAC,CAAC;gBACnE,CAAC;gBACD,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;gBAC/B,YAAY,IAAI,CAAC,CAAC;YACpB,CAAC;YACD,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,SAAS,CAAC,UAAU,EAAE,KAAK;YACzB,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAE7B,MAAM,QAAQ,GAAG,gBAAgB,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YACzD,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,QAAQ,CAAC;YAClB,CAAC;YAED,IAAI,UAAU,CAAC,aAAa,CAAC,IAAI,IAAI,6BAA6B,EAAE,CAAC;gBACnE,MAAM,IAAI,qBAAqB,CAC7B,wBAAwB,EACxB,iDAAiD,CAClD,CAAC;YACJ,CAAC;YAED,MAAM,YAAY,GAAyB;gBACzC,EAAE,EAAE,UAAU,EAAE;gBAChB,YAAY,EAAE,UAAU,CAAC,EAAE;gBAC3B,KAAK;gBACL,MAAM,EAAE,yBAAyB,CAAC,UAAU,EAAE,KAAK,CAAC;gBACpD,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB,CAAC;YACF,aAAa,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,EAAE,YAAY,CAAC,CAAC;YACjD,UAAU,CAAC,aAAa,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;YAC9C,oBAAoB,CAAC,YAAY,CAAC,CAAC;YACnC,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,WAAW,CAAC,UAAU,EAAE,QAAQ;YAC9B,MAAM,OAAO,GAA2B,EAAE,CAAC;YAE3C,KAAK,MAAM,cAAc,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;gBAClE,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;gBACvD,IAAI,CAAC,YAAY,IAAI,CAAC,mBAAmB,CAAC,YAAY,EAAE,QAAQ,CAAC,EAAE,CAAC;oBAClE,SAAS;gBACX,CAAC;gBAED,kBAAkB,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;gBAC7C,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC7B,CAAC;YAED,OAAO,OAAO,CAAC;QACjB,CAAC;QAED,mBAAmB,CAAC,UAAU,EAAE,IAAI;YAClC,IAAI,OAAO,CAAC;YACZ,IAAI,CAAC;gBACH,OAAO,GAAG,0BAA0B,CAAC,IAAI,CAAC,CAAC;YAC7C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,iBAAiB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBACrC,OAAO;YACT,CAAC;YAED,IAAI,CAAC;gBACH,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;oBACrB,KAAK,WAAW,CAAC,CAAC,CAAC;wBACjB,MAAM,YAAY,GAAG,OAAO,CAAC,SAAS,CAAC,UAAU,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;wBAClE,IAAI,CAAC,UAAU,EAAE;4BACf,IAAI,EAAE,YAAY;4BAClB,EAAE,EAAE,OAAO,CAAC,EAAE;4BACd,KAAK,EAAE,YAAY,CAAC,KAAK;4BACzB,cAAc,EAAE,YAAY,CAAC,EAAE;yBAChC,CAAC,CAAC;wBACH,OAAO;oBACT,CAAC;oBACD,KAAK,aAAa,CAAC,CAAC,CAAC;wBACnB,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC,UAAU,EAAE;4BAC9C,cAAc,EAAE,OAAO,CAAC,cAAc;4BACtC,KAAK,EAAE,OAAO,CAAC,KAAK;yBACrB,CAAC,CAAC;wBACH,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;wBACzB,IAAI,CAAC,UAAU,EAAE;4BACf,IAAI,EAAE,cAAc;4BACpB,EAAE,EAAE,OAAO,CAAC,EAAE;4BACd,cAAc,EAAE,OAAO,CAAC,cAAc,IAAI,KAAK,EAAE,EAAE;4BACnD,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,KAAK,EAAE,KAAK;yBACrC,CAAC,CAAC;wBACH,OAAO;oBACT,CAAC;oBACD,KAAK,SAAS;wBACZ,IAAI,CAAC,UAAU,EAAE;4BACf,IAAI,EAAE,OAAO;4BACb,EAAE,EAAE,OAAO,CAAC,EAAE;4BACd,IAAI,EAAE,mBAAmB;4BACzB,OAAO,EAAE,gCAAgC;yBAC1C,CAAC,CAAC;wBACH,OAAO;oBACT,KAAK,MAAM;wBACT,IAAI,CAAC,UAAU,EAAE;4BACf,IAAI,EAAE,MAAM;4BACZ,EAAE,EAAE,OAAO,CAAC,EAAE;yBACf,CAAC,CAAC;wBACH,OAAO;gBACX,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,iBAAiB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YACvC,CAAC;QACH,CAAC;QAED,aAAa,CAAC,KAAK,EAAE,eAAe;YAClC,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAC7B,IAAI,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC5B,MAAM,IAAI,qBAAqB,CAC7B,0BAA0B,EAC1B,mBAAmB,KAAK,0BAA0B,CACnD,CAAC;YACJ,CAAC;YAED,YAAY,CAAC,GAAG,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;YACzC,OAAO,GAAG,EAAE;gBACV,IAAI,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,eAAe,EAAE,CAAC;oBAChD,OAAO;gBACT,CAAC;gBAED,wBAAwB,CAAC,KAAK,CAAC,CAAC;gBAChC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC7B,CAAC,CAAC;QACJ,CAAC;QAED,WAAW,CACT,KAAa,EACb,eAAgD;YAEhD,MAAM,KAAK,GAAG,OAAO,CAAC,aAAa,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;YAC5D,MAAM,YAAY,GAChB,eAAe,CAAC,QAAQ,KAAK,MAAM;gBACjC,CAAC,CAAC;oBACE,IAAI,EAAE,KAAK;oBACX,QAAQ,EAAE,MAAe;oBACzB,UAAU,CAAC,MAAc,EAAE,OAAgB;wBACzC,OAAO,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;oBACrD,CAAC;oBACD,KAAK;iBACN;gBACH,CAAC,CAAC;oBACE,IAAI,EAAE,KAAK;oBACX,QAAQ,EAAE,QAAiB;oBAC3B,OAAO,CAAC,OAAgB;wBACtB,OAAO,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;oBACzC,CAAC;oBACD,KAAK;iBACN,CAAC;YAER,OAAO,YAAuD,CAAC;QACjE,CAAC;QAED,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,GAAG,EAAE;YAClC,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAC7B,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,QAAQ,IAAI,QAAQ,CAAC;YAC/D,IAAI,QAAQ,KAAK,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;gBAC3C,MAAM,IAAI,qBAAqB,CAC7B,wBAAwB,EACxB,mBAAmB,KAAK,6BAA6B,CACtD,CAAC;YACJ,CAAC;YACD,IAAI,QAAQ,KAAK,QAAQ,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBAC5C,MAAM,IAAI,qBAAqB,CAC7B,kBAAkB,EAClB,mBAAmB,KAAK,oCAAoC,CAC7D,CAAC;YACJ,CAAC;YAED,MAAM,eAAe,GAAG,oBAAoB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACxD,IAAI,CAAC,eAAe,EAAE,IAAI,EAAE,CAAC;gBAC3B,OAAO;oBACL,KAAK;oBACL,eAAe,EAAE,CAAC;iBACnB,CAAC;YACJ,CAAC;YAED,MAAM,OAAO,GAA0B;gBACrC,IAAI,EAAE,OAAO;gBACb,KAAK;gBACL,OAAO;gBACP,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;aACtC,CAAC;YACF,MAAM,eAAe,GAAG,IAAI,GAAG,EAAU,CAAC;YAE1C,KAAK,MAAM,cAAc,IAAI,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;gBACzD,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;gBACvD,MAAM,UAAU,GAAG,YAAY;oBAC7B,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,YAAY,CAAC;oBAC5C,CAAC,CAAC,SAAS,CAAC;gBACd,IACE,CAAC,YAAY;oBACb,CAAC,UAAU;oBACX,CAAC,QAAQ,KAAK,MAAM,IAAI,YAAY,CAAC,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC;oBAC/D,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,EAClC,CAAC;oBACD,SAAS;gBACX,CAAC;gBAED,IAAI,UAAU,CAAC,EAAE,CAAC,UAAU,KAAK,eAAe,EAAE,CAAC;oBACjD,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;oBAC/B,SAAS;gBACX,CAAC;gBAED,eAAe,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;gBACnC,IAAI,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YAC5B,CAAC;YAED,OAAO;gBACL,KAAK;gBACL,eAAe,EAAE,eAAe,CAAC,IAAI;aACtC,CAAC;QACJ,CAAC;QAED,iBAAiB,CAAC,KAAK;YACrB,OAAO,oBAAoB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,IAAI,IAAI,CAAC,CAAC;QACpD,CAAC;QAED,yBAAyB,CAAC,KAAK,EAAE,QAAQ;YACvC,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAE7B,MAAM,SAAS,GACb,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,GAAG,EAA2B,CAAC;YACpE,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACxB,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;YAEvC,OAAO,GAAG,EAAE;gBACV,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBAC3B,IAAI,SAAS,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;oBACzB,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACjC,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;QAED,KAAK;YACH,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;gBAC1D,IAAI,UAAU,CAAC,EAAE,CAAC,UAAU,KAAK,eAAe,EAAE,CAAC;oBACjD,UAAU,CAAC,EAAE,CAAC,KAAK,CAAC,gBAAgB,EAAE,yBAAyB,CAAC,CAAC;gBACnE,CAAC;gBAED,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;YACjC,CAAC;YAED,aAAa,CAAC,KAAK,EAAE,CAAC;YACtB,oBAAoB,CAAC,KAAK,EAAE,CAAC;YAC7B,gBAAgB,CAAC,KAAK,EAAE,CAAC;YACzB,YAAY,CAAC,KAAK,EAAE,CAAC;QACvB,CAAC;KACF,CAAC;IAEF,MAAM,YAAY,GAAG,IAAI,GAAG,EAAgC,CAAC;IAE7D,SAAS,gBAAgB,CACvB,UAA8B,EAC9B,QAAqD;QAErD,KAAK,MAAM,cAAc,IAAI,UAAU,CAAC,aAAa,EAAE,CAAC;YACtD,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YACvD,IAAI,YAAY,IAAI,mBAAmB,CAAC,YAAY,EAAE,QAAQ,CAAC,EAAE,CAAC;gBAChE,OAAO,YAAY,CAAC;YACtB,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,SAAS,mBAAmB,CAC1B,YAAkC,EAClC,QAAqD;QAErD,OAAO,CACL,CAAC,QAAQ,CAAC,cAAc,KAAK,SAAS;YACpC,YAAY,CAAC,EAAE,KAAK,QAAQ,CAAC,cAAc,CAAC;YAC9C,CAAC,QAAQ,CAAC,KAAK,KAAK,SAAS,IAAI,YAAY,CAAC,KAAK,KAAK,QAAQ,CAAC,KAAK,CAAC,CACxE,CAAC;IACJ,CAAC;IAED,SAAS,oBAAoB,CAAC,YAAkC;QAC9D,MAAM,aAAa,GAAG,OAAO,CAAC,iBAAiB,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;QACpE,MAAM,kBAAkB,GACtB,oBAAoB,CAAC,GAAG,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,IAAI,GAAG,EAAU,CAAC;QACpE,kBAAkB,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QACxC,oBAAoB,CAAC,GAAG,CAAC,YAAY,CAAC,KAAK,EAAE,kBAAkB,CAAC,CAAC;QACjE,6BAA6B,CAAC,YAAY,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;IACnE,CAAC;IAED,SAAS,6BAA6B,CAAC,UAA8B;QACnE,KAAK,MAAM,cAAc,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;YAClE,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YACvD,IAAI,YAAY,EAAE,CAAC;gBACjB,kBAAkB,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;IACH,CAAC;IAED,SAAS,wBAAwB,CAAC,KAAa;QAC7C,MAAM,eAAe,GAAG,oBAAoB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACxD,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,OAAO;QACT,CAAC;QAED,KAAK,MAAM,cAAc,IAAI,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC;YACzD,MAAM,YAAY,GAAG,aAAa,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;YACvD,MAAM,UAAU,GAAG,YAAY;gBAC7B,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,YAAY,CAAC;gBAC5C,CAAC,CAAC,SAAS,CAAC;YACd,IAAI,YAAY,IAAI,UAAU,EAAE,CAAC;gBAC/B,kBAAkB,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;IACH,CAAC;IAED,SAAS,yBAAyB,CAChC,UAA8B,EAC9B,KAAa;QAEb,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,QAAQ,IAAI,QAAQ,CAAC,KAAK,MAAM,EAAE,CAAC;YAC/D,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,MAAM,GAAG,UAAU,CAAC,SAAS,EAAE,MAAM,CAAC;QAC5C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,qBAAqB,CAC7B,yBAAyB,EACzB,mBAAmB,KAAK,4BAA4B,CACrD,CAAC;QACJ,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,SAAS,kBAAkB,CACzB,UAA8B,EAC9B,YAAkC;QAElC,MAAM,aAAa,GAAG,OAAO,CAAC,iBAAiB,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;QACpE,UAAU,CAAC,aAAa,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QACjD,aAAa,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QAEtC,MAAM,kBAAkB,GAAG,oBAAoB,CAAC,GAAG,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;QACxE,kBAAkB,EAAE,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QAC5C,IAAI,kBAAkB,EAAE,IAAI,KAAK,CAAC,EAAE,CAAC;YACnC,oBAAoB,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;QAClD,CAAC;QACD,6BAA6B,CAAC,YAAY,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;IACnE,CAAC;IAED,SAAS,6BAA6B,CACpC,KAAa,EACb,aAAqB;QAErB,MAAM,KAAK,GAAG,OAAO,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAC/C,IAAI,KAAK,KAAK,aAAa,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,MAAM,SAAS,GAAG,gBAAgB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC9C,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QAED,KAAK,MAAM,QAAQ,IAAI,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;YAC7C,IAAI,CAAC;gBACH,QAAQ,CAAC,KAAK,CAAC,CAAC;YAClB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,OAAO,CAAC,MAAM;oBAChB,OAAO,CAAC,MAAM,CAAC,KAAK,CAClB,EAAE,GAAG,EAAE,KAAK,EAAE,EACd,uCAAuC,CACxC,CAAC;;oBACC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;IACH,CAAC;IAED,SAAS,iBAAiB,CACxB,UAA8B,EAC9B,KAAc;QAEd,MAAM,IAAI,GACR,KAAK,YAAY,qBAAqB,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,gBAAgB,CAAC;QACzE,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,0BAA0B,CAAC;QACtE,IAAI,CAAC,UAAU,EAAE;YACf,IAAI,EAAE,OAAO;YACb,IAAI;YACJ,OAAO;SACR,CAAC,CAAC;IACL,CAAC;IAED,SAAS,IAAI,CACX,UAA8B,EAC9B,OAA8B;QAE9B,IAAI,UAAU,CAAC,EAAE,CAAC,UAAU,KAAK,eAAe,EAAE,CAAC;YACjD,OAAO,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;YAC/B,OAAO;QACT,CAAC;QAED,UAAU,CAAC,EAAE,CAAC,IAAI,CAAC,2BAA2B,CAAC,OAAO,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/realtime/types.d.ts

@@ -0,0 +1,74 @@
+import type { AppWebSocket, AppWebSocketMessageData } from '@nocobase/app-websocket';
+import { type ServiceToken } from '@nocobase/service-provider';
+export interface RealtimeConnectionContext {
+    request?: Request;
+    principal?: RealtimePrincipal;
+}
+export interface RealtimePrincipal {
+    readonly userId: string;
+}
+export interface RealtimePrincipalResolver {
+    resolve(request: Request): Promise<RealtimePrincipal | undefined>;
+}
+export interface RealtimeConnection {
+    readonly id: string;
+    readonly ws: AppWebSocket;
+    readonly connectedAt: Date;
+    readonly request?: Request;
+    readonly principal?: RealtimePrincipal;
+    readonly subscriptions: Set<string>;
+}
+export interface RealtimeSubscription {
+    readonly id: string;
+    readonly connectionId: string;
+    readonly topic: string;
+    readonly userId?: string;
+    readonly createdAt: Date;
+}
+export interface RealtimePublishResult {
+    readonly topic: string;
+    readonly subscriberCount: number;
+}
+export type RealtimeTopicAudience = 'public' | 'user';
+export interface RealtimeTopicOptions {
+    readonly audience: RealtimeTopicAudience;
+}
+export interface RealtimePublishOptions {
+    readonly userId?: string;
+}
+export interface RealtimeTopicBase {
+    readonly name: string;
+    close(): void;
+}
+export interface RealtimePublicTopic<Payload> extends RealtimeTopicBase {
+    readonly audience: 'public';
+    publish(payload: Payload): RealtimePublishResult;
+}
+export interface RealtimeUserTopic<Payload> extends RealtimeTopicBase {
+    readonly audience: 'user';
+    publishFor(userId: string, payload: Payload): RealtimePublishResult;
+}
+export type DefinedRealtimeTopic<Payload, Audience extends RealtimeTopicAudience> = Audience extends 'user' ? RealtimeUserTopic<Payload> : RealtimePublicTopic<Payload>;
+export interface RealtimeService {
+    connect(ws: AppWebSocket, context?: RealtimeConnectionContext): RealtimeConnection;
+    disconnect(connection: RealtimeConnection): void;
+    /** Closes every active connection authenticated as the given user. */
+    disconnectUser(userId: string): number;
+    subscribe(connection: RealtimeConnection, topic: string): RealtimeSubscription;
+    unsubscribe(connection: RealtimeConnection, selector: {
+        subscriptionId?: string;
+        topic?: string;
+    }): RealtimeSubscription[];
+    handleClientMessage(connection: RealtimeConnection, data: AppWebSocketMessageData): void;
+    registerTopic(topic: string, options: RealtimeTopicOptions): () => void;
+    defineTopic<Payload, Audience extends RealtimeTopicAudience>(topic: string, options: {
+        readonly audience: Audience;
+    }): DefinedRealtimeTopic<Payload, Audience>;
+    publish(topic: string, payload: unknown, options?: RealtimePublishOptions): RealtimePublishResult;
+    subscriptionCount(topic: string): number;
+    onTopicSubscriptionChange(topic: string, listener: (count: number) => void): () => void;
+    close(): void;
+}
+export declare const realtimeServiceToken: ServiceToken<RealtimeService>;
+export declare const realtimePrincipalResolverToken: ServiceToken<RealtimePrincipalResolver>;
+//# sourceMappingURL=types.d.ts.map

package/dist/realtime/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/realtime/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,YAAY,EACZ,uBAAuB,EACxB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAEL,KAAK,YAAY,EAClB,MAAM,4BAA4B,CAAC;AAEpC,MAAM,WAAW,yBAAyB;IACxC,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,SAAS,CAAC,EAAE,iBAAiB,CAAC;CAC/B;AAED,MAAM,WAAW,iBAAiB;IAChC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,yBAAyB;IACxC,OAAO,CAAC,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC,iBAAiB,GAAG,SAAS,CAAC,CAAC;CACnE;AAED,MAAM,WAAW,kBAAkB;IACjC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,EAAE,YAAY,CAAC;IAC1B,QAAQ,CAAC,WAAW,EAAE,IAAI,CAAC;IAC3B,QAAQ,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC;IAC3B,QAAQ,CAAC,SAAS,CAAC,EAAE,iBAAiB,CAAC;IACvC,QAAQ,CAAC,aAAa,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;CACrC;AAED,MAAM,WAAW,oBAAoB;IACnC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,SAAS,EAAE,IAAI,CAAC;CAC1B;AAED,MAAM,WAAW,qBAAqB;IACpC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,eAAe,EAAE,MAAM,CAAC;CAClC;AAED,MAAM,MAAM,qBAAqB,GAAG,QAAQ,GAAG,MAAM,CAAC;AAEtD,MAAM,WAAW,oBAAoB;IACnC,QAAQ,CAAC,QAAQ,EAAE,qBAAqB,CAAC;CAC1C;AAED,MAAM,WAAW,sBAAsB;IACrC,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,iBAAiB;IAChC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,KAAK,IAAI,IAAI,CAAC;CACf;AAED,MAAM,WAAW,mBAAmB,CAAC,OAAO,CAAE,SAAQ,iBAAiB;IACrE,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAC;IAC5B,OAAO,CAAC,OAAO,EAAE,OAAO,GAAG,qBAAqB,CAAC;CAClD;AAED,MAAM,WAAW,iBAAiB,CAAC,OAAO,CAAE,SAAQ,iBAAiB;IACnE,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,qBAAqB,CAAC;CACrE;AAED,MAAM,MAAM,oBAAoB,CAC9B,OAAO,EACP,QAAQ,SAAS,qBAAqB,IACpC,QAAQ,SAAS,MAAM,GACvB,iBAAiB,CAAC,OAAO,CAAC,GAC1B,mBAAmB,CAAC,OAAO,CAAC,CAAC;AAEjC,MAAM,WAAW,eAAe;IAC9B,OAAO,CACL,EAAE,EAAE,YAAY,EAChB,OAAO,CAAC,EAAE,yBAAyB,GAClC,kBAAkB,CAAC;IACtB,UAAU,CAAC,UAAU,EAAE,kBAAkB,GAAG,IAAI,CAAC;IACjD,sEAAsE;IACtE,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAAC;IACvC,SAAS,CACP,UAAU,EAAE,kBAAkB,EAC9B,KAAK,EAAE,MAAM,GACZ,oBAAoB,CAAC;IACxB,WAAW,CACT,UAAU,EAAE,kBAAkB,EAC9B,QAAQ,EAAE;QAAE,cAAc,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,GACpD,oBAAoB,EAAE,CAAC;IAC1B,mBAAmB,CACjB,UAAU,EAAE,kBAAkB,EAC9B,IAAI,EAAE,uBAAuB,GAC5B,IAAI,CAAC;IACR,aAAa,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,oBAAoB,GAAG,MAAM,IAAI,CAAC;IACxE,WAAW,CAAC,OAAO,EAAE,QAAQ,SAAS,qBAAqB,EACzD,KAAK,EAAE,MAAM,EACb,OAAO,EAAE;QAAE,QAAQ,CAAC,QAAQ,EAAE,QAAQ,CAAA;KAAE,GACvC,oBAAoB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC3C,OAAO,CACL,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,OAAO,EAChB,OAAO,CAAC,EAAE,sBAAsB,GAC/B,qBAAqB,CAAC;IACzB,iBAAiB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAAC;IACzC,yBAAyB,CACvB,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,GAChC,MAAM,IAAI,CAAC;IACd,KAAK,IAAI,IAAI,CAAC;CACf;AAED,eAAO,MAAM,oBAAoB,EAAE,YAAY,CAAC,eAAe,CACQ,CAAC;AAExE,eAAO,MAAM,8BAA8B,EAAE,YAAY,CAAC,yBAAyB,CAGhF,CAAC"}

package/dist/realtime/types.js

@@ -0,0 +1,4 @@
+import { createServiceToken, } from '@nocobase/service-provider';
+export const realtimeServiceToken = createServiceToken('@nocobase/app/realtime-service');
+export const realtimePrincipalResolverToken = createServiceToken('@nocobase/app/realtime-principal-resolver');
+//# sourceMappingURL=types.js.map

package/dist/realtime/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/realtime/types.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,kBAAkB,GAEnB,MAAM,4BAA4B,CAAC;AA2GpC,MAAM,CAAC,MAAM,oBAAoB,GAC/B,kBAAkB,CAAkB,gCAAgC,CAAC,CAAC;AAExE,MAAM,CAAC,MAAM,8BAA8B,GACzC,kBAAkB,CAChB,2CAA2C,CAC5C,CAAC"}

package/dist/realtime/websocket.d.ts

@@ -0,0 +1,7 @@
+import type { Hono } from 'hono';
+import type { ServiceResolver } from '@nocobase/service-provider';
+import type { AppWebSocketHandler } from '../websocket.js';
+export declare const REALTIME_WEBSOCKET_PATH: string;
+export declare function registerRealtimeWebSocketRoutes(router: Hono): void;
+export declare function createRealtimeWebSocketHandler(container: ServiceResolver): AppWebSocketHandler;
+//# sourceMappingURL=websocket.d.ts.map

package/dist/realtime/websocket.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket.d.ts","sourceRoot":"","sources":["../../src/realtime/websocket.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAEjC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,4BAA4B,CAAC;AAClE,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAO3D,eAAO,MAAM,uBAAuB,EAAE,MAAc,CAAC;AAErD,wBAAgB,+BAA+B,CAAC,MAAM,EAAE,IAAI,GAAG,IAAI,CAclE;AAED,wBAAgB,8BAA8B,CAC5C,SAAS,EAAE,eAAe,GACzB,mBAAmB,CAmDrB"}

package/dist/realtime/websocket.js

@@ -0,0 +1,57 @@
+import { realtimePrincipalResolverToken, realtimeServiceToken, } from './types.js';
+export const REALTIME_WEBSOCKET_PATH = '/ws';
+export function registerRealtimeWebSocketRoutes(router) {
+    router.get(REALTIME_WEBSOCKET_PATH, (context) => context.json({
+        error: 'WebSocket upgrade required',
+    }, {
+        status: 426,
+        headers: {
+            upgrade: 'websocket',
+        },
+    }));
+}
+export function createRealtimeWebSocketHandler(container) {
+    return async (request) => {
+        const url = new URL(request.url);
+        if (url.pathname !== REALTIME_WEBSOCKET_PATH) {
+            return null;
+        }
+        const origin = request.headers.get('origin');
+        // Browsers always send Origin on WebSocket handshakes and must be
+        // same-origin. Origin-less handshakes remain available to non-browser
+        // clients, which must provide credentials explicitly.
+        if (origin && origin !== url.origin) {
+            return Response.json({ error: 'WebSocket origin is not allowed.' }, { status: 403 });
+        }
+        const realtime = container.resolve(realtimeServiceToken);
+        const principal = container.has(realtimePrincipalResolverToken)
+            ? await container.resolve(realtimePrincipalResolverToken).resolve(request)
+            : undefined;
+        let connection;
+        const disconnect = () => {
+            if (!connection) {
+                return;
+            }
+            realtime.disconnect(connection);
+            connection = undefined;
+        };
+        return {
+            onOpen(_event, ws) {
+                connection = realtime.connect(ws, { request, principal });
+            },
+            onMessage(event, ws) {
+                if (!connection) {
+                    connection = realtime.connect(ws, { request, principal });
+                }
+                realtime.handleClientMessage(connection, event.data);
+            },
+            onClose() {
+                disconnect();
+            },
+            onError() {
+                disconnect();
+            },
+        };
+    };
+}
+//# sourceMappingURL=websocket.js.map

package/dist/realtime/websocket.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"websocket.js","sourceRoot":"","sources":["../../src/realtime/websocket.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,8BAA8B,EAC9B,oBAAoB,GAErB,MAAM,YAAY,CAAC;AAEpB,MAAM,CAAC,MAAM,uBAAuB,GAAW,KAAK,CAAC;AAErD,MAAM,UAAU,+BAA+B,CAAC,MAAY;IAC1D,MAAM,CAAC,GAAG,CAAC,uBAAuB,EAAE,CAAC,OAAO,EAAE,EAAE,CAC9C,OAAO,CAAC,IAAI,CACV;QACE,KAAK,EAAE,4BAA4B;KACpC,EACD;QACE,MAAM,EAAE,GAAG;QACX,OAAO,EAAE;YACP,OAAO,EAAE,WAAW;SACrB;KACF,CACF,CACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,8BAA8B,CAC5C,SAA0B;IAE1B,OAAO,KAAK,EAAE,OAAO,EAAE,EAAE;QACvB,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,GAAG,CAAC,QAAQ,KAAK,uBAAuB,EAAE,CAAC;YAC7C,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7C,kEAAkE;QAClE,sEAAsE;QACtE,sDAAsD;QACtD,IAAI,MAAM,IAAI,MAAM,KAAK,GAAG,CAAC,MAAM,EAAE,CAAC;YACpC,OAAO,QAAQ,CAAC,IAAI,CAClB,EAAE,KAAK,EAAE,kCAAkC,EAAE,EAC7C,EAAE,MAAM,EAAE,GAAG,EAAE,CAChB,CAAC;QACJ,CAAC;QAED,MAAM,QAAQ,GAAG,SAAS,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;QACzD,MAAM,SAAS,GAAG,SAAS,CAAC,GAAG,CAAC,8BAA8B,CAAC;YAC7D,CAAC,CAAC,MAAM,SAAS,CAAC,OAAO,CAAC,8BAA8B,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;YAC1E,CAAC,CAAC,SAAS,CAAC;QACd,IAAI,UAA0C,CAAC;QAC/C,MAAM,UAAU,GAAG,GAAS,EAAE;YAC5B,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,OAAO;YACT,CAAC;YAED,QAAQ,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;YAChC,UAAU,GAAG,SAAS,CAAC;QACzB,CAAC,CAAC;QAEF,OAAO;YACL,MAAM,CAAC,MAAM,EAAE,EAAE;gBACf,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;YAC5D,CAAC;YACD,SAAS,CAAC,KAAK,EAAE,EAAE;gBACjB,IAAI,CAAC,UAAU,EAAE,CAAC;oBAChB,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC,CAAC;gBAC5D,CAAC;gBAED,QAAQ,CAAC,mBAAmB,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YACvD,CAAC;YACD,OAAO;gBACL,UAAU,EAAE,CAAC;YACf,CAAC;YACD,OAAO;gBACL,UAAU,EAAE,CAAC;YACf,CAAC;SACF,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC"}