update socket config to use functions and directory

Author: csim-sg

package.json

@@ -1,7 +1,7 @@
 {
   "name": "socket-controllers",
   "private": true,
-  "version": "0.0.4",
+  "version": "0.0.5",
   "description": "Use class-based controllers to handle websocket events",
   "license": "MIT",
   "readmeFilename": "README.md",

src/SocketControllerExecutor.ts

@@ -52,9 +52,9 @@ export class SocketControllerExecutor {
     // Public Methods
     // -------------------------------------------------------------------------
 
-    execute() {
-        this.registerControllers();
-        this.registerMiddlewares();
+    execute(controllerClasses?: Function[], middlewareClasses?: Function[]) {
+        this.registerControllers(controllerClasses);
+        this.registerMiddlewares(middlewareClasses);
     }
 
     // -------------------------------------------------------------------------

src/SocketControllersOptions.ts

@@ -8,7 +8,7 @@ export interface SocketControllersOptions {
     /**
      * List of directories from where to "require" all your controllers.
      */
-    controllers?: string[];
+    controllers?: Function[] | string[];
 
     /**
      * List of directories from where to "require" all your middlewares.

src/index.ts

@@ -31,11 +31,18 @@ function createExecutor(io: any, options: SocketControllersOptions): void {
     const executor = new SocketControllerExecutor(io);
 
     // second import all controllers and middlewares and error handlers
+    let controllerClasses: Function[];
     if (options && options.controllers && options.controllers.length)
-        importClassesFromDirectories(options.controllers);
+        controllerClasses = (options.controllers as any[]).filter(controller => controller instanceof Function);
+        const controllerDirs = (options.controllers as any[]).filter(controller => typeof controller === "string");
+        controllerClasses.push(...importClassesFromDirectories(controllerDirs));
 
-    if (options && options.middlewares && options.middlewares.length)
-        importClassesFromDirectories(options.middlewares);
+    let middlewareClasses: Function[];
+    if (options && options.middlewares && options.middlewares.length) {
+        middlewareClasses = (options.middlewares as any[]).filter(controller => controller instanceof Function);
+        const middlewareDirs = (options.middlewares as any[]).filter(controller => typeof controller === "string");
+        middlewareClasses.push(...importClassesFromDirectories(middlewareDirs));
+    }
 
     if (options.useClassTransformer !== undefined) {
         executor.useClassTransformer = options.useClassTransformer;