| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598 | "use strict";var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {    if (k2 === undefined) k2 = k;    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });}) : (function(o, m, k, k2) {    if (k2 === undefined) k2 = k;    o[k2] = m[k];}));var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {    Object.defineProperty(o, "default", { enumerable: true, value: v });}) : function(o, v) {    o["default"] = v;});var __importStar = (this && this.__importStar) || function (mod) {    if (mod && mod.__esModule) return mod;    var result = {};    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);    __setModuleDefault(result, mod);    return result;};var __importDefault = (this && this.__importDefault) || function (mod) {    return (mod && mod.__esModule) ? mod : { "default": mod };};Object.defineProperty(exports, "__esModule", { value: true });exports.Namespace = exports.Socket = exports.Server = void 0;const http = require("http");const fs_1 = require("fs");const zlib_1 = require("zlib");const accepts = require("accepts");const stream_1 = require("stream");const path = require("path");const engine_io_1 = require("engine.io");const client_1 = require("./client");const events_1 = require("events");const namespace_1 = require("./namespace");Object.defineProperty(exports, "Namespace", { enumerable: true, get: function () { return namespace_1.Namespace; } });const parent_namespace_1 = require("./parent-namespace");const socket_io_adapter_1 = require("socket.io-adapter");const parser = __importStar(require("socket.io-parser"));const debug_1 = __importDefault(require("debug"));const socket_1 = require("./socket");Object.defineProperty(exports, "Socket", { enumerable: true, get: function () { return socket_1.Socket; } });const typed_events_1 = require("./typed-events");const uws_1 = require("./uws");const debug = (0, debug_1.default)("socket.io:server");const clientVersion = require("../package.json").version;const dotMapRegex = /\.map/;class Server extends typed_events_1.StrictEventEmitter {    constructor(srv, opts = {}) {        super();        /**         * @private         */        this._nsps = new Map();        this.parentNsps = new Map();        if ("object" === typeof srv &&            srv instanceof Object &&            !srv.listen) {            opts = srv;            srv = undefined;        }        this.path(opts.path || "/socket.io");        this.connectTimeout(opts.connectTimeout || 45000);        this.serveClient(false !== opts.serveClient);        this._parser = opts.parser || parser;        this.encoder = new this._parser.Encoder();        this.adapter(opts.adapter || socket_io_adapter_1.Adapter);        this.sockets = this.of("/");        this.opts = opts;        if (srv || typeof srv == "number")            this.attach(srv);    }    serveClient(v) {        if (!arguments.length)            return this._serveClient;        this._serveClient = v;        return this;    }    /**     * Executes the middleware for an incoming namespace not already created on the server.     *     * @param name - name of incoming namespace     * @param auth - the auth parameters     * @param fn - callback     *     * @private     */    _checkNamespace(name, auth, fn) {        if (this.parentNsps.size === 0)            return fn(false);        const keysIterator = this.parentNsps.keys();        const run = () => {            const nextFn = keysIterator.next();            if (nextFn.done) {                return fn(false);            }            nextFn.value(name, auth, (err, allow) => {                if (err || !allow) {                    return run();                }                if (this._nsps.has(name)) {                    // the namespace was created in the meantime                    debug("dynamic namespace %s already exists", name);                    return fn(this._nsps.get(name));                }                const namespace = this.parentNsps.get(nextFn.value).createChild(name);                debug("dynamic namespace %s was created", name);                // @ts-ignore                this.sockets.emitReserved("new_namespace", namespace);                fn(namespace);            });        };        run();    }    path(v) {        if (!arguments.length)            return this._path;        this._path = v.replace(/\/$/, "");        const escapedPath = this._path.replace(/[-\/\\^$*+?.()|[\]{}]/g, "\\$&");        this.clientPathRegex = new RegExp("^" +            escapedPath +            "/socket\\.io(\\.msgpack|\\.esm)?(\\.min)?\\.js(\\.map)?(?:\\?|$)");        return this;    }    connectTimeout(v) {        if (v === undefined)            return this._connectTimeout;        this._connectTimeout = v;        return this;    }    adapter(v) {        if (!arguments.length)            return this._adapter;        this._adapter = v;        for (const nsp of this._nsps.values()) {            nsp._initAdapter();        }        return this;    }    /**     * Attaches socket.io to a server or port.     *     * @param srv - server or port     * @param opts - options passed to engine.io     * @return self     * @public     */    listen(srv, opts = {}) {        return this.attach(srv, opts);    }    /**     * Attaches socket.io to a server or port.     *     * @param srv - server or port     * @param opts - options passed to engine.io     * @return self     * @public     */    attach(srv, opts = {}) {        if ("function" == typeof srv) {            const msg = "You are trying to attach socket.io to an express " +                "request handler function. Please pass a http.Server instance.";            throw new Error(msg);        }        // handle a port as a string        if (Number(srv) == srv) {            srv = Number(srv);        }        if ("number" == typeof srv) {            debug("creating http server and binding to %d", srv);            const port = srv;            srv = http.createServer((req, res) => {                res.writeHead(404);                res.end();            });            srv.listen(port);        }        // merge the options passed to the Socket.IO server        Object.assign(opts, this.opts);        // set engine.io path to `/socket.io`        opts.path = opts.path || this._path;        this.initEngine(srv, opts);        return this;    }    attachApp(app /*: TemplatedApp */, opts = {}) {        // merge the options passed to the Socket.IO server        Object.assign(opts, this.opts);        // set engine.io path to `/socket.io`        opts.path = opts.path || this._path;        // initialize engine        debug("creating uWebSockets.js-based engine with opts %j", opts);        const engine = new engine_io_1.uServer(opts);        engine.attach(app, opts);        // bind to engine events        this.bind(engine);        if (this._serveClient) {            // attach static file serving            app.get(`${this._path}/*`, (res, req) => {                if (!this.clientPathRegex.test(req.getUrl())) {                    req.setYield(true);                    return;                }                const filename = req                    .getUrl()                    .replace(this._path, "")                    .replace(/\?.*$/, "")                    .replace(/^\//, "");                const isMap = dotMapRegex.test(filename);                const type = isMap ? "map" : "source";                // Per the standard, ETags must be quoted:                // https://tools.ietf.org/html/rfc7232#section-2.3                const expectedEtag = '"' + clientVersion + '"';                const weakEtag = "W/" + expectedEtag;                const etag = req.getHeader("if-none-match");                if (etag) {                    if (expectedEtag === etag || weakEtag === etag) {                        debug("serve client %s 304", type);                        res.writeStatus("304 Not Modified");                        res.end();                        return;                    }                }                debug("serve client %s", type);                res.writeHeader("cache-control", "public, max-age=0");                res.writeHeader("content-type", "application/" + (isMap ? "json" : "javascript"));                res.writeHeader("etag", expectedEtag);                const filepath = path.join(__dirname, "../client-dist/", filename);                (0, uws_1.serveFile)(res, filepath);            });        }        (0, uws_1.patchAdapter)(app);    }    /**     * Initialize engine     *     * @param srv - the server to attach to     * @param opts - options passed to engine.io     * @private     */    initEngine(srv, opts) {        // initialize engine        debug("creating engine.io instance with opts %j", opts);        this.eio = (0, engine_io_1.attach)(srv, opts);        // attach static file serving        if (this._serveClient)            this.attachServe(srv);        // Export http server        this.httpServer = srv;        // bind to engine events        this.bind(this.eio);    }    /**     * Attaches the static file serving.     *     * @param srv http server     * @private     */    attachServe(srv) {        debug("attaching client serving req handler");        const evs = srv.listeners("request").slice(0);        srv.removeAllListeners("request");        srv.on("request", (req, res) => {            if (this.clientPathRegex.test(req.url)) {                this.serve(req, res);            }            else {                for (let i = 0; i < evs.length; i++) {                    evs[i].call(srv, req, res);                }            }        });    }    /**     * Handles a request serving of client source and map     *     * @param req     * @param res     * @private     */    serve(req, res) {        const filename = req.url.replace(this._path, "").replace(/\?.*$/, "");        const isMap = dotMapRegex.test(filename);        const type = isMap ? "map" : "source";        // Per the standard, ETags must be quoted:        // https://tools.ietf.org/html/rfc7232#section-2.3        const expectedEtag = '"' + clientVersion + '"';        const weakEtag = "W/" + expectedEtag;        const etag = req.headers["if-none-match"];        if (etag) {            if (expectedEtag === etag || weakEtag === etag) {                debug("serve client %s 304", type);                res.writeHead(304);                res.end();                return;            }        }        debug("serve client %s", type);        res.setHeader("Cache-Control", "public, max-age=0");        res.setHeader("Content-Type", "application/" + (isMap ? "json" : "javascript"));        res.setHeader("ETag", expectedEtag);        Server.sendFile(filename, req, res);    }    /**     * @param filename     * @param req     * @param res     * @private     */    static sendFile(filename, req, res) {        const readStream = (0, fs_1.createReadStream)(path.join(__dirname, "../client-dist/", filename));        const encoding = accepts(req).encodings(["br", "gzip", "deflate"]);        const onError = (err) => {            if (err) {                res.end();            }        };        switch (encoding) {            case "br":                res.writeHead(200, { "content-encoding": "br" });                readStream.pipe((0, zlib_1.createBrotliCompress)()).pipe(res);                (0, stream_1.pipeline)(readStream, (0, zlib_1.createBrotliCompress)(), res, onError);                break;            case "gzip":                res.writeHead(200, { "content-encoding": "gzip" });                (0, stream_1.pipeline)(readStream, (0, zlib_1.createGzip)(), res, onError);                break;            case "deflate":                res.writeHead(200, { "content-encoding": "deflate" });                (0, stream_1.pipeline)(readStream, (0, zlib_1.createDeflate)(), res, onError);                break;            default:                res.writeHead(200);                (0, stream_1.pipeline)(readStream, res, onError);        }    }    /**     * Binds socket.io to an engine.io instance.     *     * @param {engine.Server} engine engine.io (or compatible) server     * @return self     * @public     */    bind(engine) {        this.engine = engine;        this.engine.on("connection", this.onconnection.bind(this));        return this;    }    /**     * Called with each incoming transport connection.     *     * @param {engine.Socket} conn     * @return self     * @private     */    onconnection(conn) {        debug("incoming connection with id %s", conn.id);        const client = new client_1.Client(this, conn);        if (conn.protocol === 3) {            // @ts-ignore            client.connect("/");        }        return this;    }    /**     * Looks up a namespace.     *     * @param {String|RegExp|Function} name nsp name     * @param fn optional, nsp `connection` ev handler     * @public     */    of(name, fn) {        if (typeof name === "function" || name instanceof RegExp) {            const parentNsp = new parent_namespace_1.ParentNamespace(this);            debug("initializing parent namespace %s", parentNsp.name);            if (typeof name === "function") {                this.parentNsps.set(name, parentNsp);            }            else {                this.parentNsps.set((nsp, conn, next) => next(null, name.test(nsp)), parentNsp);            }            if (fn) {                // @ts-ignore                parentNsp.on("connect", fn);            }            return parentNsp;        }        if (String(name)[0] !== "/")            name = "/" + name;        let nsp = this._nsps.get(name);        if (!nsp) {            debug("initializing namespace %s", name);            nsp = new namespace_1.Namespace(this, name);            this._nsps.set(name, nsp);            if (name !== "/") {                // @ts-ignore                this.sockets.emitReserved("new_namespace", nsp);            }        }        if (fn)            nsp.on("connect", fn);        return nsp;    }    /**     * Closes server connection     *     * @param [fn] optional, called as `fn([err])` on error OR all conns closed     * @public     */    close(fn) {        for (const socket of this.sockets.sockets.values()) {            socket._onclose("server shutting down");        }        this.engine.close();        // restore the Adapter prototype        (0, uws_1.restoreAdapter)();        if (this.httpServer) {            this.httpServer.close(fn);        }        else {            fn && fn();        }    }    /**     * Sets up namespace middleware.     *     * @return self     * @public     */    use(fn) {        this.sockets.use(fn);        return this;    }    /**     * Targets a room when emitting.     *     * @param room     * @return self     * @public     */    to(room) {        return this.sockets.to(room);    }    /**     * Targets a room when emitting.     *     * @param room     * @return self     * @public     */    in(room) {        return this.sockets.in(room);    }    /**     * Excludes a room when emitting.     *     * @param name     * @return self     * @public     */    except(name) {        return this.sockets.except(name);    }    /**     * Sends a `message` event to all clients.     *     * @return self     * @public     */    send(...args) {        this.sockets.emit("message", ...args);        return this;    }    /**     * Sends a `message` event to all clients.     *     * @return self     * @public     */    write(...args) {        this.sockets.emit("message", ...args);        return this;    }    /**     * Emit a packet to other Socket.IO servers     *     * @param ev - the event name     * @param args - an array of arguments, which may include an acknowledgement callback at the end     * @public     */    serverSideEmit(ev, ...args) {        return this.sockets.serverSideEmit(ev, ...args);    }    /**     * Gets a list of socket ids.     *     * @public     */    allSockets() {        return this.sockets.allSockets();    }    /**     * Sets the compress flag.     *     * @param compress - if `true`, compresses the sending data     * @return self     * @public     */    compress(compress) {        return this.sockets.compress(compress);    }    /**     * Sets a modifier for a subsequent event emission that the event data may be lost if the client is not ready to     * receive messages (because of network slowness or other issues, or because they’re connected through long polling     * and is in the middle of a request-response cycle).     *     * @return self     * @public     */    get volatile() {        return this.sockets.volatile;    }    /**     * Sets a modifier for a subsequent event emission that the event data will only be broadcast to the current node.     *     * @return self     * @public     */    get local() {        return this.sockets.local;    }    /**     * Adds a timeout in milliseconds for the next operation     *     * <pre><code>     *     * io.timeout(1000).emit("some-event", (err, responses) => {     *   // ...     * });     *     * </pre></code>     *     * @param timeout     */    timeout(timeout) {        return this.sockets.timeout(timeout);    }    /**     * Returns the matching socket instances     *     * @public     */    fetchSockets() {        return this.sockets.fetchSockets();    }    /**     * Makes the matching socket instances join the specified rooms     *     * @param room     * @public     */    socketsJoin(room) {        return this.sockets.socketsJoin(room);    }    /**     * Makes the matching socket instances leave the specified rooms     *     * @param room     * @public     */    socketsLeave(room) {        return this.sockets.socketsLeave(room);    }    /**     * Makes the matching socket instances disconnect     *     * @param close - whether to close the underlying connection     * @public     */    disconnectSockets(close = false) {        return this.sockets.disconnectSockets(close);    }}exports.Server = Server;/** * Expose main namespace (/). */const emitterMethods = Object.keys(events_1.EventEmitter.prototype).filter(function (key) {    return typeof events_1.EventEmitter.prototype[key] === "function";});emitterMethods.forEach(function (fn) {    Server.prototype[fn] = function () {        return this.sockets[fn].apply(this.sockets, arguments);    };});module.exports = (srv, opts) => new Server(srv, opts);module.exports.Server = Server;module.exports.Namespace = namespace_1.Namespace;module.exports.Socket = socket_1.Socket;
 |