@avleon/core 0.0.28 → 0.0.29

package/dist/event-subscriber.js

@@ -0,0 +1,96 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EventSubscriberRegistry = void 0;
+exports.Private = Private;
+exports.isPrivate = isPrivate;
+exports.getPrivateChannelResolver = getPrivateChannelResolver;
+exports.registerSocketSubscriber = registerSocketSubscriber;
+exports.getSocketSubscribers = getSocketSubscribers;
+exports.Subscribe = Subscribe;
+exports.PrivateSubscribe = PrivateSubscribe;
+const typedi_1 = require("typedi");
+const event_dispatcher_1 = require("./event-dispatcher");
+require("reflect-metadata");
+const PRIVATE_META_KEY = 'avleon:private';
+function Private(channelResolver) {
+    return function (target, propertyKey) {
+        Reflect.defineMetadata(PRIVATE_META_KEY, true, target, propertyKey);
+        Reflect.defineMetadata(`private:channel:${propertyKey}`, channelResolver, target);
+    };
+}
+function isPrivate(target, propertyKey) {
+    return Reflect.getMetadata(PRIVATE_META_KEY, target, propertyKey) || false;
+}
+function getPrivateChannelResolver(target, propertyKey) {
+    return Reflect.getMetadata(`private:channel:${propertyKey}`, target);
+}
+const socketSubscriberClasses = new Set();
+function registerSocketSubscriber(target) {
+    socketSubscriberClasses.add(target);
+}
+function getSocketSubscribers() {
+    return Array.from(socketSubscriberClasses);
+}
+// decorators/Subscribe.ts
+function Subscribe(event) {
+    return (target, propertyKey) => {
+        Reflect.defineMetadata('socket:event', event, target, propertyKey);
+        registerSocketSubscriber(target.constructor);
+    };
+}
+// decorators/Private.ts
+function PrivateSubscribe(resolver) {
+    return function (target, propertyKey) {
+        Reflect.defineMetadata('avleon:private', true, target, propertyKey);
+        Reflect.defineMetadata(`private:channel:${propertyKey}`, resolver, target);
+    };
+}
+let EventSubscriberRegistry = class EventSubscriberRegistry {
+    constructor(socketContext) {
+        this.socketContext = socketContext;
+    }
+    register(socket) {
+        const subscriberClasses = getSocketSubscribers();
+        for (const SubscriberClass of subscriberClasses) {
+            const instance = typedi_1.Container.get(SubscriberClass);
+            const prototype = Object.getPrototypeOf(instance);
+            const methodNames = Object.getOwnPropertyNames(prototype)
+                .filter(name => typeof prototype[name] === 'function');
+            for (const methodName of methodNames) {
+                const event = Reflect.getMetadata('socket:event', prototype, methodName);
+                const isPrivateListener = isPrivate(instance, methodName);
+                const channelResolver = getPrivateChannelResolver(instance, methodName);
+                //const event = Reflect.getMetadata(`subscribe:event:${method}`, instance);
+                if (event) {
+                    const channel = isPrivateListener && channelResolver ? channelResolver(socket) : event;
+                    console.log('Channel', channel);
+                    socket.on(channel, (payload) => {
+                        this.socketContext.run(socket, async () => {
+                            if (isPrivateListener) {
+                                const user = socket.data.user;
+                                if (!user)
+                                    return; // unauthorized
+                                // optionally add more validation here
+                            }
+                            await instance[methodName](payload, socket.data);
+                        });
+                    });
+                }
+            }
+        }
+    }
+};
+exports.EventSubscriberRegistry = EventSubscriberRegistry;
+exports.EventSubscriberRegistry = EventSubscriberRegistry = __decorate([
+    (0, typedi_1.Service)(),
+    __metadata("design:paramtypes", [event_dispatcher_1.SocketContextService])
+], EventSubscriberRegistry);

package/dist/exceptions/http-exceptions.js

@@ -79,5 +79,5 @@ exports.HttpExceptions = {
     badRequest: (message = "") => new BadRequestException(message),
     unauthorized: (message = "") => new UnauthorizedException(message),
     forbidden: (message = "") => new ForbiddenException(message),
-    internalError: (message = "") => new InternalErrorException(message)
+    internalError: (message = "") => new InternalErrorException(message),
 };

package/dist/exceptions/index.d.ts

@@ -1 +1 @@
-export * from './http-exceptions';
+export * from "./http-exceptions";

package/dist/exceptions/system-exception.js

@@ -11,7 +11,9 @@ class DuplicateRouteException extends Error {
     constructor(params) {
         let sameController = params.controller == params.inverseController;
         let message = `Duplicate route found for method ${params.method.toUpperCase()}:${params.path == "" ? "'/'" : params.path} `;
-        message += sameController ? `in ${params.controller}` : `both in ${params.controller} and ${params.inverseController}`;
+        message += sameController
+            ? `in ${params.controller}`
+            : `both in ${params.controller} and ${params.inverseController}`;
         super(message);
     }
 }

package/dist/file-storage.js

@@ -90,7 +90,7 @@ let FileStorage = class FileStorage {
                 overwrite: options && options.overwrite ? options.overwrite : true,
             };
             for (let f of files) {
-                let uploadPath = `public`;
+                let uploadPath = "public";
                 if (options === null || options === void 0 ? void 0 : options.to) {
                     uploadPath = `public/${options.to}`;
                 }

package/dist/helpers.js

@@ -40,7 +40,7 @@ function inject(cls) {
         return container_1.default.get(cls);
     }
     catch (error) {
-        throw new system_exception_1.SystemUseError(`Not a project class. Maybe you wanna register it first.`);
+        throw new system_exception_1.SystemUseError("Not a project class. Maybe you wanna register it first.");
     }
 }
 function isConstructor(func) {

package/dist/icore.d.ts

@@ -15,6 +15,7 @@ import { IConfig } from "./config";
 import { FastifyCorsOptions } from "@fastify/cors";
 import { FastifyMultipartOptions } from "@fastify/multipart";
 import { MultipartFile } from "./multipart";
+import { ServerOptions } from 'socket.io';
 export type FuncRoute = {
     handler: any;
     middlewares?: any[];
@@ -124,6 +125,7 @@ export declare class AvleonApplication {
     private middlewares;
     private rMap;
     private hasSwagger;
+    private _hasWebsocket;
     private globalSwaggerOptions;
     private dataSourceOptions?;
     private controllers;
@@ -142,6 +144,8 @@ export declare class AvleonApplication {
     private initSwagger;
     private _isConfigClass;
     useCors<T = FastifyCorsOptions>(corsOptions?: ConfigInput<T>): void;
+    useWebSocket<T = Partial<ServerOptions>>(socketOptions: ConfigInput<T>): void;
+    private _initWebSocket;
     useOpenApi<T = OpenApiUiOptions>(configOrClass: OpenApiConfigInput<T>): void;
     useMultipart<T extends MultipartOptions>(options: ConfigInput<T>): void;
     useDataSource<T extends DataSourceOptions>(options: ConfigInput<T>): void;
@@ -197,6 +201,7 @@ export declare class AvleonApplication {
     };
     private _mapFeatures;
     initializeDatabase(): Promise<void>;
+    handleSocket(socket: any): void;
     run(port?: number, fn?: CallableFunction): Promise<void>;
     getTestApp(buildOptions?: any): TestApplication;
 }

package/dist/icore.js

@@ -58,10 +58,13 @@ const cors_1 = __importDefault(require("@fastify/cors"));
 const multipart_1 = __importDefault(require("@fastify/multipart"));
 const validation_1 = require("./validation");
 const utils_1 = require("./utils");
+const socket_io_1 = require("socket.io");
+const event_subscriber_1 = require("./event-subscriber");
 const isTsNode = process.env.TS_NODE_DEV ||
     process.env.TS_NODE_PROJECT ||
     process[Symbol.for("ts-node.register.instance")];
 const controllerDir = path_1.default.join(process.cwd(), isTsNode ? "./src/controllers" : "./dist/cotrollers");
+const subscriberRegistry = typedi_1.default.get(event_subscriber_1.EventSubscriberRegistry);
 // InternalApplication
 class AvleonApplication {
     constructor() {
@@ -71,6 +74,7 @@ class AvleonApplication {
         this.middlewares = new Map();
         this.rMap = new Map();
         this.hasSwagger = false;
+        this._hasWebsocket = false;
         this.globalSwaggerOptions = {};
         this.dataSourceOptions = undefined;
         this.controllers = [];
@@ -78,7 +82,7 @@ class AvleonApplication {
         this.dataSource = undefined;
         this.isMapFeatures = false;
         this.registerControllerAuto = false;
-        this.registerControllerPath = './src';
+        this.registerControllerPath = "./src";
         this.metaCache = new Map();
         this.app = (0, fastify_1.default)();
         this.appConfig = new config_1.AppConfig();
@@ -122,7 +126,7 @@ class AvleonApplication {
         if (options.ui && options.ui == "scalar") {
             const scalarPlugin = (0, utils_1.optionalRequire)("@scalar/fastify-api-reference", {
                 failOnMissing: true,
-                customMessage: 'Install "@scalar/fastify-api-reference" to enable API docs.\n\n  npm install @scalar/fastify-api-reference',
+                customMessage: "Install \"@scalar/fastify-api-reference\" to enable API docs.\n\n  npm install @scalar/fastify-api-reference",
             });
             await this.app.register(scalarPlugin, {
                 routePrefix: rPrefix,
@@ -141,7 +145,7 @@ class AvleonApplication {
         else {
             const fastifySwaggerUi = (0, utils_1.optionalRequire)("@fastify/swagger-ui", {
                 failOnMissing: true,
-                customMessage: 'Install "@fastify/swagger-ui" to enable API docs.\n\n  npm install @fastify/swagger-ui',
+                customMessage: "Install \"@fastify/swagger-ui\" to enable API docs.\n\n  npm install @fastify/swagger-ui",
             });
             await this.app.register(fastifySwaggerUi, {
                 logo: logo ? logo : null,
@@ -152,8 +156,8 @@ class AvleonApplication {
     }
     _isConfigClass(input) {
         var _a;
-        return (typeof input === 'function' &&
-            typeof input.prototype === 'object' &&
+        return (typeof input === "function" &&
+            typeof input.prototype === "object" &&
             ((_a = input.prototype) === null || _a === void 0 ? void 0 : _a.constructor) === input);
     }
     useCors(corsOptions) {
@@ -168,6 +172,19 @@ class AvleonApplication {
         }
         this.app.register(cors_1.default, coptions);
     }
+    useWebSocket(socketOptions) {
+        this._hasWebsocket = true;
+        this._initWebSocket(socketOptions);
+    }
+    async _initWebSocket(options) {
+        const fsSocketIO = (0, utils_1.optionalRequire)("fastify-socket.io", {
+            failOnMissing: true,
+            customMessage: "Install \"fastify-socket.io\" to enable socket.io.\n\n run pnpm install fastify-socket.io",
+        });
+        await this.app.register(fsSocketIO, options);
+        const socketIO = await this.app.io;
+        typedi_1.default.set(socket_io_1.Server, socketIO);
+    }
     useOpenApi(configOrClass) {
         let openApiConfig;
         if (this._isConfigClass(configOrClass)) {
@@ -211,8 +228,7 @@ class AvleonApplication {
         this.dataSource = datasource;
         typedi_1.default.set("idatasource", datasource);
     }
-    _useCache(options) {
-    }
+    _useCache(options) { }
     useMiddlewares(mclasses) {
         for (const mclass of mclasses) {
             const cls = typedi_1.default.get(mclass);
@@ -317,7 +333,7 @@ class AvleonApplication {
                     const args = await this._mapArgs(reqClone, allMeta);
                     for (let paramMeta of allMeta.params) {
                         if (paramMeta.required) {
-                            (0, validation_1.validateOrThrow)({ [paramMeta.key]: args[paramMeta.index] }, { [paramMeta.key]: { type: paramMeta.dataType } }, { location: 'param' });
+                            (0, validation_1.validateOrThrow)({ [paramMeta.key]: args[paramMeta.index] }, { [paramMeta.key]: { type: paramMeta.dataType } }, { location: "param" });
                         }
                     }
                     for (let queryMeta of allMeta.query) {
@@ -333,7 +349,7 @@ class AvleonApplication {
                             }
                         }
                         if (queryMeta.required) {
-                            (0, validation_1.validateOrThrow)({ [queryMeta.key]: args[queryMeta.index] }, { [queryMeta.key]: { type: queryMeta.dataType } }, { location: 'queryparam' });
+                            (0, validation_1.validateOrThrow)({ [queryMeta.key]: args[queryMeta.index] }, { [queryMeta.key]: { type: queryMeta.dataType } }, { location: "queryparam" });
                         }
                     }
                     for (let bodyMeta of allMeta.body) {
@@ -443,7 +459,7 @@ class AvleonApplication {
             process.env.TS_NODE_PROJECT ||
             process[Symbol.for("ts-node.register.instance")];
         const controllerDir = path_1.default.join(process.cwd(), this.registerControllerPath);
-        return isTsNode ? controllerDir : controllerDir.replace('src', 'dist');
+        return isTsNode ? controllerDir : controllerDir.replace("src", "dist");
     }
     async autoControllers(controllersPath) {
         const conDir = this._resolveControllerDir(controllersPath);
@@ -457,8 +473,8 @@ class AvleonApplication {
                 const module = await Promise.resolve(`${filePath}`).then(s => __importStar(require(s)));
                 for (const exported of Object.values(module)) {
                     if (typeof exported === "function" && (0, container_1.isApiController)(exported)) {
-                        console.log('adding', exported.name);
-                        if (!this.controllers.some(con => exported.name == con.name)) {
+                        console.log("adding", exported.name);
+                        if (!this.controllers.some((con) => exported.name == con.name)) {
                             this.controllers.push(exported);
                         }
                         //this.buildController(exported);
@@ -470,7 +486,7 @@ class AvleonApplication {
     useControllers(controllers) {
         if (Array.isArray(controllers)) {
             this.controllers = controllers;
-            controllers.forEach(controller => {
+            controllers.forEach((controller) => {
                 if (!this.controllers.includes(controller)) {
                     this.controllers.push(controller);
                 }
@@ -593,14 +609,17 @@ class AvleonApplication {
         return this._routeHandler(path, "DELETE", fn);
     }
     _mapFeatures() {
-        const features = typedi_1.default.get('features');
-        console.log('Features', features);
+        const features = typedi_1.default.get("features");
+        console.log("Features", features);
     }
     async initializeDatabase() {
         if (this.dataSourceOptions && this.dataSource) {
             await this.dataSource.initialize();
         }
     }
+    handleSocket(socket) {
+        subscriberRegistry.register(socket);
+    }
     async run(port = 4000, fn) {
         if (this.alreadyRun)
             throw new system_exception_1.SystemUseError("App already running");
@@ -631,7 +650,8 @@ class AvleonApplication {
         });
         this.app.setErrorHandler(async (error, req, res) => {
             const handledErr = this._handleError(error);
-            if (error instanceof exceptions_1.ValidationErrorException || error instanceof exceptions_1.BadRequestException) {
+            if (error instanceof exceptions_1.ValidationErrorException ||
+                error instanceof exceptions_1.BadRequestException) {
                 return res.status(handledErr.code).send({
                     code: handledErr.code,
                     error: handledErr.error,
@@ -641,6 +661,21 @@ class AvleonApplication {
             return res.status(handledErr.code).send(handledErr);
         });
         await this.app.ready();
+        if (this._hasWebsocket) {
+            await this.app.io.on('connection', this.handleSocket);
+            await this.app.io.use((socket, next) => {
+                const token = socket.handshake.auth.token;
+                try {
+                    console.log('token', token);
+                    const user = { id: 1, name: "tareq" };
+                    socket.data.user = user; // this powers @AuthUser()
+                    next();
+                }
+                catch (_a) {
+                    next(new Error('Unauthorized'));
+                }
+            });
+        }
         await this.app.listen({ port });
         console.log(`Application running on http://127.0.0.1:${port}`);
     }
@@ -683,7 +718,7 @@ class AvleonApplication {
                 delete: async (url, options) => this.app.inject({ method: "DELETE", url, ...options }),
                 options: async (url, options) => this.app.inject({ method: "OPTIONS", url, ...options }),
                 getController: (controller, deps = []) => {
-                    const paramTypes = Reflect.getMetadata('design:paramtypes', controller) || [];
+                    const paramTypes = Reflect.getMetadata("design:paramtypes", controller) || [];
                     deps.forEach((dep, i) => {
                         typedi_1.default.set(paramTypes[i], dep);
                     });
@@ -704,7 +739,7 @@ class AvleonTest {
         process.env.NODE_ENV = "test";
     }
     static getController(controller, deps = []) {
-        const paramTypes = Reflect.getMetadata('design:paramtypes', controller) || [];
+        const paramTypes = Reflect.getMetadata("design:paramtypes", controller) || [];
         deps.forEach((dep, i) => {
             typedi_1.default.set(paramTypes[i], dep);
         });

package/dist/index.d.ts

@@ -24,5 +24,7 @@ export * from "./utils/hash";
 export * from "./multipart";
 export * from "./file-storage";
 export * from "./logger";
+export * from "./event-dispatcher";
+export { Subscribe, Private, PrivateSubscribe } from "./event-subscriber";
 export declare const GetSchema: typeof sw.generateSwaggerSchema;
 export { default as Container } from "./container";

package/dist/index.js

@@ -39,7 +39,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Container = exports.GetSchema = exports.exclude = exports.pick = exports.validateRequestBody = exports.inject = void 0;
+exports.Container = exports.GetSchema = exports.PrivateSubscribe = exports.Private = exports.Subscribe = exports.exclude = exports.pick = exports.validateRequestBody = exports.inject = void 0;
 /**
  * @copyright 2024
  * @author Tareq Hossain
@@ -70,6 +70,11 @@ __exportStar(require("./utils/hash"), exports);
 __exportStar(require("./multipart"), exports);
 __exportStar(require("./file-storage"), exports);
 __exportStar(require("./logger"), exports);
+__exportStar(require("./event-dispatcher"), exports);
+var event_subscriber_1 = require("./event-subscriber");
+Object.defineProperty(exports, "Subscribe", { enumerable: true, get: function () { return event_subscriber_1.Subscribe; } });
+Object.defineProperty(exports, "Private", { enumerable: true, get: function () { return event_subscriber_1.Private; } });
+Object.defineProperty(exports, "PrivateSubscribe", { enumerable: true, get: function () { return event_subscriber_1.PrivateSubscribe; } });
 exports.GetSchema = sw.generateSwaggerSchema;
 var container_1 = require("./container");
 Object.defineProperty(exports, "Container", { enumerable: true, get: function () { return __importDefault(container_1).default; } });

package/dist/utils/index.d.ts

@@ -1,2 +1,2 @@
-export * from './hash';
-export * from './optional-require';
+export * from "./hash";
+export * from "./optional-require";

package/dist/utils/optional-require.js

@@ -37,7 +37,6 @@ exports.optionalRequire = optionalRequire;
 exports.optionalImport = optionalImport;
 function optionalRequire(moduleName, options = {}) {
     try {
-        // eslint-disable-next-line @typescript-eslint/no-var-requires
         return require(moduleName);
     }
     catch (err) {
@@ -57,7 +56,8 @@ async function optionalImport(moduleName, options = {}) {
         return mod;
     }
     catch (err) {
-        if ((err.code === "ERR_MODULE_NOT_FOUND" || err.code === "MODULE_NOT_FOUND") &&
+        if ((err.code === "ERR_MODULE_NOT_FOUND" ||
+            err.code === "MODULE_NOT_FOUND") &&
             err.message.includes(moduleName)) {
             if (options.failOnMissing) {
                 throw new Error(options.customMessage ||

package/dist/validation.d.ts

@@ -26,7 +26,7 @@ export type ValidationProps = {
     [key: string]: ValidationRule;
 };
 export type ValidateOptons = {
-    location?: 'header' | 'queryparam' | 'param' | 'body' | 'custom';
+    location?: "header" | "queryparam" | "param" | "body" | "custom";
 };
 export declare function validateOrThrow<T extends {}>(obj: T, rules: ValidationProps, options?: ValidateOptons): any;
 export {};

package/dist/websocket.d.ts

@@ -0,0 +1,7 @@
+export declare class AvleonSocketIo {
+    private io?;
+    sendToAll(): void;
+    sendOnly(): void;
+    sendRoom(): void;
+    receive(channel: string): void;
+}

package/dist/websocket.js

@@ -0,0 +1,20 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AvleonSocketIo = void 0;
+const typedi_1 = require("typedi");
+let AvleonSocketIo = class AvleonSocketIo {
+    sendToAll() { }
+    sendOnly() { }
+    sendRoom() { }
+    receive(channel) { }
+};
+exports.AvleonSocketIo = AvleonSocketIo;
+exports.AvleonSocketIo = AvleonSocketIo = __decorate([
+    (0, typedi_1.Service)()
+], AvleonSocketIo);

package/dist/websocket.test.js

@@ -0,0 +1 @@
+"use strict";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@avleon/core",
-  "version": "0.0.28",
+  "version": "0.0.29",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
   "scripts": {
@@ -56,10 +56,12 @@
     "cross-env": "^7.0.3",
     "dotenv": "^16.4.7",
     "fastify": "^5.1.0",
+    "fastify-socket.io": "^5.1.0",
     "highlight.js": "^11.9.0",
     "pino": "^9.6.0",
     "pino-pretty": "^13.0.0",
     "reflect-metadata": "^0.2.2",
+    "socket.io": "^4.8.1",
     "typedi": "^0.10.0"
   },
   "peerDependencies": {

package/src/config.test.ts

@@ -1,5 +1,5 @@
 import "reflect-metadata";
-import { Config, CreateConfig, GetConfig, IConfig } from "./config";
+import { AppConfig, CreateConfig, GetConfig, IConfig } from "./config";
 import { Environment } from "./environment-variables";
 
 type AppConfig = { name: string; os: string };
@@ -7,7 +7,7 @@ type AppConfig = { name: string; os: string };
 describe("Config", () => {
   describe("class", () => {
     it("should be call by get config", () => {
-      @Config
+      @AppConfig
       class MyConfig {
         config(env: Environment) {
           return {

package/src/config.ts

@@ -12,11 +12,11 @@ export interface IConfig<T = any> {
   config(env: Environment): T;
 }
 
-export function Config<T extends IConfig>(target: Constructable<T>) {
+export function AppConfig<T extends IConfig>(target: Constructable<T>) {
   Container.set({ id: target, type: target });
 }
 
-export class AppConfig {
+export class AvleonConfig {
   get<T extends IConfig<R>, R>(configClass: Constructable<T>): R {
     const instance = Container.get(configClass);
     if (!instance) {

package/src/event-dispatcher.ts

@@ -0,0 +1,100 @@
+import { Service, Container } from "typedi";
+import { Server as SocketIOServer, Socket, ServerOptions } from "socket.io";
+import { AsyncLocalStorage } from "node:async_hooks";
+
+
+@Service()
+export class SocketContextService {
+  private readonly storage = new AsyncLocalStorage<{ socket: Socket }>();
+
+  run(socket: Socket, fn: () => void | Promise<void>) {
+    this.storage.run({ socket }, fn);
+  }
+
+  getSocket(): Socket | undefined {
+    return this.storage.getStore()?.socket;
+  }
+}
+
+export type DispatchOptions = {
+  room?: string;
+  broadcast?: boolean;
+  transports?: ("socket" | "kafka" | "rabbitmq")[];
+  retry?: number;
+  retryDelay?: number;
+};
+
+export function sleep(ms: number) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+
+@Service()
+export class EventDispatcher {
+  constructor(private readonly _context: SocketContextService) {}
+
+  async dispatch<T = any>(
+    event: string,
+    data: T,
+    options: DispatchOptions = {}
+  ) {
+    const retryCount = options.retry ?? 0;
+    const delay = options.retryDelay ?? 300;
+
+    for (let attempt = 0; attempt <= retryCount; attempt++) {
+      try {
+        await this.dispatchToTransports(event, data, options);
+        break;
+      } catch (err) {
+        if (attempt === retryCount) throw err;
+        await sleep(delay * (attempt + 1));
+      }
+    }
+  }
+
+  private async dispatchToTransports(event: string, data: any, options: DispatchOptions) {
+    const transports = options.transports ?? ["socket"];
+
+    for (const transport of transports) {
+      if (transport === "socket") {
+        const io = Container.get(SocketIOServer);
+
+        //console.log('SOckert', Container.get(SocketContextService));
+
+        const context = Container.get(SocketContextService);
+        const socket = context.getSocket();
+
+      
+
+        if (options.broadcast && socket) {
+          if (options.room) {
+            socket.broadcast.to(options.room).emit(event, data);
+          } else {
+            socket.broadcast.emit(event, data);
+          }
+        } else {
+          if (options.room) {
+            io.to(options.room).emit(event, data);
+          } else {
+            io.emit(event, data);
+          }
+        }
+      }
+    }
+  }
+}
+
+
+export function Dispatch(event: string, options?: Omit<DispatchOptions, "transports"> & { transports?: DispatchOptions["transports"] }) {
+  return function (target: any, propertyKey: string, descriptor: PropertyDescriptor) {
+    const original = descriptor.value;
+
+    descriptor.value = async function (...args: any[]) {
+      const result = await original.apply(this, args);
+
+      const dispatcher = Container.get(EventDispatcher);
+      await dispatcher.dispatch(event, result, options);
+
+      return result;
+    };
+  };
+}