@eleven-am/pondsocket 0.1.110 → 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');
@@ -43,10 +48,6 @@ function isNotEmpty(value) {
         value !== '' &&
         Object.keys(value).length !== 0;
 }
-function createClassDecorator(key, value) {
-    // eslint-disable-next-line new-cap
-    return (0, common_1.applyDecorators)((0, common_1.Injectable)(), (0, common_1.SetMetadata)(key, value));
-}
 function manageClassData(key, target) {
     return {
         get() {
@@ -58,6 +59,12 @@ function manageClassData(key, target) {
         },
     };
 }
+function createClassDecorator(key, path) {
+    return (target) => {
+        const { set } = manageClassData(key, target);
+        set(path);
+    };
+}
 function manageMethodData(key, target, propertyKey) {
     function getter() {
         var _a;
@@ -238,6 +245,12 @@ class Context {
     retrieveMethodData(key) {
         return manageMethodData(key, this.instance, this.propertyKey).get();
     }
+    getClass() {
+        return this.instance.constructor;
+    }
+    getHandler() {
+        return this.instance[this.propertyKey];
+    }
     addData(key, value) {
         this.data[key] = value;
     }
@@ -269,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) {
@@ -500,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) {
@@ -519,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);
@@ -540,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);
@@ -561,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.110",
+  "version": "0.1.112",
   "description": "PondSocket is a fast simple socket server",
   "keywords": [
     "socket",

package/types.d.ts

@@ -1,6 +1,6 @@
 import { Server as HTTPServer, IncomingHttpHeaders } from 'http';
 
-import type { ModuleMetadata, DynamicModule } from '@nestjs/common';
+import type { ModuleMetadata, DynamicModule, Type } from '@nestjs/common';
 import type { Express } from 'express';
 import { WebSocketServer } from 'ws';
 
@@ -170,16 +170,15 @@ declare class Context<Path extends string = string> {
     event: PondEvent<Path> | null;
 
     /**
-     * @desc Retrieves metadata associated with the class
-     * @param key - the key to retrieve
+     * @desc Returns the *type* of the controller class which the current handler belongs to.
      */
-    public retrieveClassData<A = unknown>(key: symbol): A | null;
+    getClass<T = any>(): Type<T>;
 
     /**
-     * @desc Retrieves metadata associated with the method
-     * @param key - the key to retrieve
+     * @desc Returns a reference to the handler (method) that will be invoked next in the
+     * request pipeline.
      */
-    public retrieveMethodData<A = unknown>(key: symbol): A | null;
+    getHandler(): Function;
 
     /**
      * @desc Adds request data to the context