Add callback support

README.md

@@ -31,6 +31,8 @@ import {
   onConnect,
   onDisconnect,
   onMessage,
+  payload,
+  messageCallback
 } from "inversify-socket-utils";
 
 @injectable()
@@ -51,6 +53,23 @@ export class MessageController {
     console.log("message received");
     socket.emit("message", "hello!");
   }
+
+  @onMessage("custom-event")
+  customEvent(@payload() payload: any[]) {
+    console.log("payload", payload);
+  }
+
+  @onMessage("get-info")
+  getInfo(@payload() payload: any[]) {
+    console.log("payload", payload);
+    return "info";
+  }
+
+  @onMessage("get-more-info")
+  getMoreInfo(@payload() payload: any[], @messageCallback() cb: function) {
+    console.log("payload", payload);
+    cb("info 1","info 2", "info 3");
+  }
 }
 ```
 

src/constants.ts

@@ -22,6 +22,7 @@ export enum PARAMETER_TYPE {
   SOCKET_ID,
   SOCKET_REQUEST,
   SOCKET_ROOMS,
+  SOCKET_CALLBACK
 }
 
 export const NO_CONTROLLERS_FOUND =

src/decorators.ts

@@ -166,6 +166,9 @@ export const connectedSocket: () => ParameterDecorator = paramDecoratorFactory(
 export const payload: () => ParameterDecorator = paramDecoratorFactory(
   PARAMETER_TYPE.SOCKET_BODY,
 );
+export const messageCallback: () => ParameterDecorator = paramDecoratorFactory(
+  PARAMETER_TYPE.SOCKET_CALLBACK,
+);
 
 export const socketQueryParam: (name: string) => ParameterDecorator =
   paramDecoratorFactory(PARAMETER_TYPE.SOCKET_QUERY_PARAM);

src/server.ts

@@ -79,7 +79,7 @@ export class InversifySocketServer {
       }
 
       if (metadata.type === ACTION_TYPE.MESSAGE) {
-        socket.on(metadata.name, (payload) => {
+        socket.on(metadata.name, (...payload) => {
           this.handleAction(
             socket,
             controllerMetadata,
@@ -97,26 +97,36 @@ export class InversifySocketServer {
     controller: interfaces.ControllerMetadata,
     action: interfaces.ControllerActionMetadata,
     parameters: interfaces.ControllerParameterMetadata,
-    payload?: unknown,
+    payload: unknown[],
   ) {
     let paramList: interfaces.ParameterMetadata[] = [];
     if (parameters) {
       paramList = parameters[action.key] || [];
     }
+
+    const cb = (typeof payload[payload.length-1] == "function") ? payload.splice(-1,1)[0] : undefined;
 
-    const args = this.extractParams(socket, payload, paramList);
-    (
+    const args = this.extractParams(socket, payload, paramList, cb);
+    const result = (
       this.container.getNamed<interfaces.Controller>(
         TYPE.Controller,
         (controller.target as { name: string }).name,
       ) as Record<string, (...a: Array<unknown>) => unknown>
     )[action.key](...args);
+    if(cb) {
+      if(result instanceof Promise){
+        result.then(cb)
+      } else {
+        cb(result);
+      }
+    }
   }
 
   private extractParams(
     socket: SocketIO.Socket,
     payload: unknown,
     params: Array<interfaces.ParameterMetadata>,
+    cb: function
   ) {
     const args: Array<unknown> = [];
 
@@ -140,8 +150,14 @@ export class InversifySocketServer {
         case PARAMETER_TYPE.SOCKET_ROOMS:
           args[index] = socket.rooms;
           return;
-        default:
+        case PARAMETER_TYPE.SOCKET_BODY:
           args[index] = payload;
+          return;
+        case PARAMETER_TYPE.SOCKET_CALLBACK:
+          args[index] = cb;
+          return;
+        default:
+          args[index] = undefined;
       }
     });