@eleven-am/pondsocket 0.1.111 → 0.1.112

package/nest.js

@@ -26,6 +26,11 @@ const common_1 = require("@nestjs/common");
 // eslint-disable-next-line import/no-unresolved
 const core_1 = require("@nestjs/core");
 require("reflect-metadata");
+const eventRequest_1 = require("./channel/eventRequest");
+const eventResponse_1 = require("./channel/eventResponse");
+const response_1 = require("./endpoint/response");
+const joinRequest_1 = require("./lobby/joinRequest");
+const joinResponse_1 = require("./lobby/joinResponse");
 const pondSocket_1 = require("./server/pondSocket");
 const onJoinHandlerKey = Symbol('onJoinHandlerKey');
 const onEventHandlerKey = Symbol('onEventHandlerKey');
@@ -277,13 +282,13 @@ function manageResponse(data, response) {
 }
 function manageError(error, response) {
     if (response.hasResponded) {
-        return;
+        return response;
     }
     if (error instanceof common_1.HttpException) {
-        response.reject(error.message, error.getStatus());
+        return response.reject(error.message, error.getStatus());
     }
     if (error instanceof Error) {
-        response.reject(error.message, 500);
+        return response.reject(error.message, 500);
     }
 }
 function manageConnectionHandlers(target) {
@@ -508,8 +513,22 @@ function GetLeaveEvent() {
     })(null);
 }
 exports.GetLeaveEvent = GetLeaveEvent;
-function manageAction(req, res, instance, moduleRef, originalMethod, propertyKey, response) {
+function manageAction(instance, moduleRef, originalMethod, propertyKey, request, response) {
     return __awaiter(this, void 0, void 0, function* () {
+        const req = {};
+        const res = {};
+        if (request instanceof joinRequest_1.JoinRequest && response instanceof joinResponse_1.JoinResponse) {
+            req.joinRequest = request;
+            res.joinResponse = response;
+        }
+        else if (request instanceof eventRequest_1.EventRequest && response instanceof eventResponse_1.EventResponse) {
+            req.eventRequest = request;
+            res.eventResponse = response;
+        }
+        else if ('headers' in request && response instanceof response_1.ConnectionResponse) {
+            req.connection = request;
+            res.connection = response;
+        }
         const context = new Context(req, res, instance, propertyKey);
         const canProceed = yield resolveGuards(moduleRef, context);
         if (canProceed) {
@@ -527,13 +546,7 @@ function OnConnectionRequest() {
         const { set } = manageConnectionHandlers(target);
         set('', (instance, moduleRef, request, response) => __awaiter(this, void 0, void 0, function* () {
             try {
-                const req = {
-                    connection: request,
-                };
-                const res = {
-                    connection: response,
-                };
-                yield manageAction(req, res, instance, moduleRef, originalMethod, propertyKey, response);
+                yield manageAction(instance, moduleRef, originalMethod, propertyKey, request, response);
             }
             catch (error) {
                 manageError(error, response);
@@ -548,13 +561,7 @@ function OnJoinRequest() {
         const { set } = manageJoinHandlers(target);
         set('', (instance, moduleRef, request, response) => __awaiter(this, void 0, void 0, function* () {
             try {
-                const req = {
-                    joinRequest: request,
-                };
-                const res = {
-                    joinResponse: response,
-                };
-                yield manageAction(req, res, instance, moduleRef, originalMethod, propertyKey, response);
+                yield manageAction(instance, moduleRef, originalMethod, propertyKey, request, response);
             }
             catch (error) {
                 manageError(error, response);
@@ -569,13 +576,7 @@ function OnEvent(event = '*') {
         const { set } = manageEventHandlers(target);
         set(event, (instance, moduleRef, request, response) => __awaiter(this, void 0, void 0, function* () {
             try {
-                const req = {
-                    eventRequest: request,
-                };
-                const res = {
-                    eventResponse: response,
-                };
-                yield manageAction(req, res, instance, moduleRef, originalMethod, propertyKey, response);
+                yield manageAction(instance, moduleRef, originalMethod, propertyKey, request, response);
             }
             catch (error) {
                 manageError(error, response);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@eleven-am/pondsocket",
-  "version": "0.1.111",
+  "version": "0.1.112",
   "description": "PondSocket is a fast simple socket server",
   "keywords": [
     "socket",