@wabot-dev/framework 0.1.0-beta.6 → 0.1.0-beta.61

package/dist/src/feature/socket-controller/metadata/@socketEvent.js

@@ -0,0 +1,18 @@
+import { container } from '../../../core/injection/index.js';
+import { SocketControllerMetadataStore } from './SocketControllerMetadataStore.js';
+
+function socketEvent(config) {
+    return function (target, propertyKey) {
+        const functionName = propertyKey.toString();
+        const paramsTypes = Reflect.getMetadata('design:paramtypes', target, functionName);
+        const store = container.resolve(SocketControllerMetadataStore);
+        store.saveSocketEventMetadata({
+            controllerConstructor: target.constructor,
+            config: typeof config === 'string' ? { event: config } : config,
+            functionName,
+            paramsTypes,
+        });
+    };
+}
+
+export { socketEvent };

package/dist/src/feature/socket-controller/metadata/SocketControllerMetadataStore.js

@@ -0,0 +1,65 @@
+import { __decorate } from 'tslib';
+import { singleton } from '../../../core/injection/index.js';
+
+let SocketControllerMetadataStore = class SocketControllerMetadataStore {
+    socketControllers = new Map();
+    socketConnections = new Map();
+    socketEvents = new Map();
+    connectionMiddlewares = new Map();
+    saveControllerMetadata(controllerMetadata) {
+        this.socketControllers.set(controllerMetadata.controllerConstructor, controllerMetadata);
+    }
+    saveSocketConnectionMetadata(socketConnectionMetadata) {
+        let controllerConnections = this.socketConnections.get(socketConnectionMetadata.controllerConstructor);
+        if (!controllerConnections) {
+            this.socketConnections.set(socketConnectionMetadata.controllerConstructor, (controllerConnections = new Map()));
+        }
+        controllerConnections.set(socketConnectionMetadata.functionName, socketConnectionMetadata);
+    }
+    saveSocketEventMetadata(socketEventMetadata) {
+        let controllerEvents = this.socketEvents.get(socketEventMetadata.controllerConstructor);
+        if (!controllerEvents) {
+            this.socketEvents.set(socketEventMetadata.controllerConstructor, (controllerEvents = new Map()));
+        }
+        controllerEvents.set(socketEventMetadata.functionName, socketEventMetadata);
+    }
+    saveConnectionMiddlewareMetadata(middlewareMetadata) {
+        let controllerMiddlewares = this.connectionMiddlewares.get(middlewareMetadata.controllerConstructor);
+        if (!controllerMiddlewares) {
+            this.connectionMiddlewares.set(middlewareMetadata.controllerConstructor, (controllerMiddlewares = new Map()));
+        }
+        let methodMiddlewares = controllerMiddlewares.get(middlewareMetadata.functionName);
+        if (!methodMiddlewares) {
+            controllerMiddlewares.set(middlewareMetadata.functionName, (methodMiddlewares = []));
+        }
+        methodMiddlewares.unshift(middlewareMetadata);
+    }
+    getControllerSockerConnectionsInfo(controllerConstructor) {
+        const controller = this.socketControllers.get(controllerConstructor);
+        if (!controller) {
+            throw new Error(`${controllerConstructor.name} should be decorated with @socketController`);
+        }
+        const connections = this.socketConnections.get(controllerConstructor);
+        const events = this.socketEvents.get(controllerConstructor) ?? new Map();
+        if (!connections?.size) {
+            // TODO: Warning
+            return [];
+        }
+        return [...connections.values()].map((connection) => ({
+            ...connection,
+            events: (() => {
+                const connectionNamespace = connection.config?.namespace;
+                return [...events.values()].filter((x) => x.config?.namespace === connectionNamespace);
+            })(),
+            connectionMiddlewares: this.connectionMiddlewares
+                .get(connection.controllerConstructor)
+                ?.get(connection.functionName) ?? [],
+            controller: this.socketControllers.get(connection.controllerConstructor),
+        }));
+    }
+};
+SocketControllerMetadataStore = __decorate([
+    singleton()
+], SocketControllerMetadataStore);
+
+export { SocketControllerMetadataStore };

package/dist/src/feature/socket-controller/runSocketControllers.js

@@ -0,0 +1,99 @@
+import { SocketControllerMetadataStore } from './metadata/SocketControllerMetadataStore.js';
+import { container } from '../../core/injection/index.js';
+import path from 'path';
+import { Logger } from '../../core/logger/Logger.js';
+import { SocketServerProvider } from '../socket/SocketServerProvider.js';
+import { CustomError } from '../../core/error/CustomError.js';
+import '../../core/validation/metadata/ValidationMetadataStore.js';
+import { validate } from '../../core/validation/validate.js';
+
+function runSocketControllers(controllers) {
+    const logger = new Logger('wabot:socket');
+    const metadataStore = container.resolve(SocketControllerMetadataStore);
+    const socketServerProvider = container.resolve(SocketServerProvider);
+    const socketServer = socketServerProvider.getSocketServer();
+    controllers.forEach((controller) => {
+        const connections = metadataStore.getControllerSockerConnectionsInfo(controller);
+        connections.forEach((connection) => {
+            const namespace = path
+                .join(connection.controller.config?.namespace ?? '/', connection.config?.namespace ?? '')
+                .replaceAll('\\', '/');
+            logger.info(`config connection to ${namespace}`);
+            const namespaceServer = socketServer.of(namespace);
+            namespaceServer.use(async (socket, next) => {
+                const connectionContainer = container.createChildContainer();
+                try {
+                    const middlewares = connection.connectionMiddlewares.map((x) => connectionContainer.resolve(x.middlewareConstructor));
+                    for (const middleware of middlewares) {
+                        await middleware.handle(socket, connectionContainer);
+                    }
+                    socket.data.connectionContainer = connectionContainer;
+                    next();
+                }
+                catch (err) {
+                    next(err);
+                    connectionContainer.dispose();
+                }
+            });
+            namespaceServer.on('connection', async (socket) => {
+                logger.trace(`connection on '${namespace}'`);
+                const connectionContainer = socket.data.connectionContainer;
+                try {
+                    const controllerInstance = connectionContainer.resolve(connection.controllerConstructor);
+                    connection.events.forEach((event) => {
+                        logger.trace(`config listener to '${event.config.event}' event on '${namespace}'`);
+                        socket.on(event.config.event, async (req, callback) => {
+                            logger.trace(`received '${event.config.event}' event on '${namespace}'`);
+                            try {
+                                const reqType = event.paramsTypes[0];
+                                if (typeof reqType !== 'function') {
+                                    throw new CustomError({
+                                        httpCode: 400,
+                                        message: 'Unable to validate request',
+                                    });
+                                }
+                                const { value, error } = validate(req, reqType);
+                                if (error) {
+                                    throw new CustomError({
+                                        httpCode: 400,
+                                        message: error.description,
+                                        info: error,
+                                    });
+                                }
+                                const out = await controllerInstance[event.functionName].apply(controllerInstance, [value, socket]);
+                                callback(out);
+                            }
+                            catch (err) {
+                                logger.error(err);
+                                if (err instanceof Error) {
+                                    const keys = Object.keys(err).filter((key) => !['message', 'stack'].includes(key));
+                                    const { httpCode, ...info } = keys.reduce((acc, key) => {
+                                        acc[key] = err[key];
+                                        return acc;
+                                    }, {});
+                                    if (typeof callback === 'function') {
+                                        callback({ error: { ...info, message: err.message, stack: err.stack } });
+                                    }
+                                }
+                                else {
+                                    if (typeof callback === 'function') {
+                                        callback({ error: { message: 'Unspected error' } });
+                                    }
+                                }
+                            }
+                        });
+                    });
+                    await controllerInstance[connection.functionName].apply(controllerInstance, [socket]);
+                }
+                catch (err) {
+                    logger.error(err);
+                    socket.disconnect();
+                    connectionContainer.dispose();
+                }
+            });
+        });
+    });
+    socketServerProvider.listen();
+}
+
+export { runSocketControllers };