@eleven-am/pondsocket-nest 0.0.26 → 0.0.29

package/package.json

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

package/services/pondSocket.js

@@ -123,7 +123,7 @@ class PondSocketService {
                         channels: [],
                     });
                 }
-                modules.get(module).endpoints.push(endpoint.discoveredClass.instance);
+                modules.get(module).endpoints.push(endpoint.discoveredClass);
             });
             channels.forEach((channel) => {
                 const module = channel.discoveredClass.parentModule.injectType;
@@ -133,39 +133,25 @@ class PondSocketService {
                         channels: [],
                     });
                 }
-                if (Array.isArray(channel.discoveredClass.instance)) {
-                    modules.get(module).channels.push(...channel.discoveredClass.instance);
-                }
-                else {
-                    modules.get(module).channels.push(channel.discoveredClass.instance);
-                }
+                modules.get(module).channels.push(channel.discoveredClass);
             });
             const instances = [...modules.values()];
             const channelsWithNoEndpoints = instances.filter((instance) => instance.channels.length > 0 && instance.endpoints.length === 0);
             const channelsWithEndpoints = instances.filter((instance) => instance.channels.length > 0 && instance.endpoints.length > 0);
-            const endpointsWithNoChannels = instances.filter((instance) => instance.endpoints.length > 0 && instance.channels.length === 0);
             const groupedInstances = channelsWithEndpoints.map((instance) => instance.endpoints.map((endpoint) => ({
                 endpoint,
                 channels: instance.channels,
             }))).flat();
             const baseEndpoint = endpoints.find((endpoint) => endpoint.discoveredClass.parentModule.name === 'AppModule');
-            const baseEndpointInstance = groupedInstances.find((groupedInstance) => groupedInstance.endpoint.instance === (baseEndpoint === null || baseEndpoint === void 0 ? void 0 : baseEndpoint.discoveredClass.instance));
-            if (channelsWithNoEndpoints.length > 0 && baseEndpointInstance) {
+            if (channelsWithNoEndpoints.length > 0 && baseEndpoint) {
                 const channels = channelsWithNoEndpoints.map((instance) => instance.channels).flat();
                 groupedInstances.push({
-                    endpoint: baseEndpointInstance.endpoint,
+                    endpoint: baseEndpoint.discoveredClass,
                     channels,
                 });
             }
-            else if (channelsWithNoEndpoints.length > 0) {
-                const endpoints = endpointsWithNoChannels.map((instance) => instance.endpoints).flat();
-                const channels = channelsWithNoEndpoints.map((instance) => instance.channels).flat();
-                endpoints.forEach((endpoint) => {
-                    groupedInstances.push({
-                        endpoint,
-                        channels,
-                    });
-                });
+            else {
+                throw new Error('No base endpoint found');
             }
             return groupedInstances;
         });