@azteam/express 1.2.470 → 1.2.472

package/src/SocketServer.js

@@ -1,183 +0,0 @@
-import {createAdapter} from '@socket.io/redis-adapter';
-import bodyParser from 'body-parser';
-import cookieParser from 'cookie-parser';
-import express from 'express';
-import fs from 'fs';
-import http from 'http';
-import _ from 'lodash';
-import socketIO from 'socket.io';
-
-const wrap = (middleware) => (socket, next) => middleware(socket.request, {}, next);
-
-class SocketServer {
-    constructor(currentDir = '', options = {}) {
-        this.options = {
-            redisConfig: null,
-            ...options,
-        };
-
-        this.middlewares = [];
-        this.controllers = [];
-        this.whiteList = [];
-        this.debug = process.env.NODE_ENV === 'development';
-        this.initController(currentDir);
-    }
-
-    setCallbackError(callback = null) {
-        this.callbackError = callback;
-        return this;
-    }
-
-    setWhiteList(whiteList) {
-        this.whiteList = whiteList;
-        return this;
-    }
-
-    setDebug(debug) {
-        this.debug = debug;
-        return this;
-    }
-
-    addController(name, version, controller) {
-        this.controllers.push({
-            name,
-            version,
-            controller,
-        });
-        return this;
-    }
-
-    initController(apiDir) {
-        if (apiDir) {
-            const controllerDirs = fs.readdirSync(apiDir);
-
-            for (const dirName of controllerDirs) {
-                if (fs.statSync(`${apiDir}/${dirName}`).isDirectory()) {
-                    const versionDirs = fs.readdirSync(`${apiDir}/${dirName}`);
-
-                    for (const versionName of versionDirs) {
-                        // eslint-disable-next-line import/no-dynamic-require,global-require
-                        const controller = require(`${apiDir}/${dirName}/${versionName}/controller`).default;
-                        this.addController(dirName, versionName, controller);
-                    }
-                }
-            }
-        }
-        return this;
-    }
-
-    addGlobalMiddleware(middleware) {
-        this.middlewares.push(middleware);
-        return this;
-    }
-
-    startPort(port) {
-        if (!_.isEmpty(this.controllers)) {
-            const WHITE_LIST = this.whiteList,
-                server = http.Server(express()),
-                io = socketIO(server, {
-                    wsEngine: 'eiows',
-                    perMessageDeflate: {
-                        threshold: 32768,
-                    },
-                    cors: {
-                        credentials: true,
-                        origin(origin, callback) {
-                            if (!origin || !WHITE_LIST.length || WHITE_LIST.some((re) => origin.endsWith(re))) {
-                                callback(null, true);
-                            } else {
-                                callback(new Error(`${origin} Not allowed by CORS`));
-                            }
-                        },
-                    },
-                });
-
-            if (this.options.redisConfig) {
-                io.adapter(createAdapter(this.options.redisConfig));
-            }
-
-            const msg = [];
-            _.map(this.controllers, (obj) => {
-                const {controller} = obj;
-
-                _.map(controller, (item, key) => {
-                    item.path = obj.version.startsWith('v') ? `/${obj.version}${item.path}` : item.path;
-
-                    const nsp = io.of(item.path),
-                        middlewares = [...this.middlewares, ...(item.middlewares || [])];
-
-                    nsp.use(wrap(bodyParser.urlencoded({limit: '5mb', extended: true})));
-                    nsp.use(wrap(bodyParser.json({limit: '5mb'})));
-                    nsp.use(wrap(cookieParser(process.env.SECRET_KEY)));
-
-                    _.map(middlewares, (middleware) => {
-                        nsp.use(wrap(middleware));
-                    });
-
-                    if (item.schedule) {
-                        item.schedule(io);
-                    }
-
-                    if (item.connection) {
-                        nsp.on('connection', (socket) => {
-                            item.connection(io, socket);
-                        });
-                    }
-
-                    msg.push({
-                        controller: obj.name,
-                        version: obj.version,
-                        path: item.path,
-                    });
-                });
-            });
-
-            console.table(msg);
-
-            server.on('listening', () => {
-                this._alert('listening', `Server start at http://localhost:${server.address().port}`);
-            });
-
-            server.on('error', (error) => {
-                if (error.syscall !== 'listen') {
-                    throw error;
-                }
-
-                let bind = typeof port === 'string' ? `Pipe ${port}` : `Port ${port}`;
-
-                switch (error.code) {
-                    case 'EACCES':
-                        this._alert('EACCES', `${bind} requires elevated privileges`);
-                        process.exit(1);
-                        break;
-                    case 'EADDRINUSE':
-                        this._alert('EACCES', `${bind} is already in use`);
-                        process.exit(1);
-                        break;
-                    default:
-                        throw error;
-                }
-            });
-
-            server.listen(port);
-
-            return server;
-        }
-        throw new Error('No controllers in use');
-    }
-
-    setAlertCallback(callback) {
-        this.alertCallback = callback;
-        return this;
-    }
-
-    _alert(status, msg) {
-        if (typeof this.alertCallback === 'function') {
-            this.alertCallback(status, msg);
-        } else {
-            console.log(status, msg);
-        }
-    }
-}
-
-export default SocketServer;