@based/server 3.4.6 → 3.4.7

package/dist/protocol.js

@@ -0,0 +1,233 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.decode = exports.encodeErrorResponse = exports.encodeAuthResponse = exports.encodeObservableDiffResponse = exports.encodeObservableResponse = exports.updateId = exports.encodeGetResponse = exports.encodeFunctionResponse = exports.decodeName = exports.decodePayload = exports.valueToBuffer = exports.encodeHeader = exports.readUint8 = exports.storeUint8 = exports.decodeHeader = exports.COMPRESS_FROM_BYTES = void 0;
+const node_zlib_1 = __importDefault(require("node:zlib"));
+const textDecoder = new TextDecoder();
+exports.COMPRESS_FROM_BYTES = 150;
+const decodeHeader = (nr) => {
+    // 4 bytes
+    // type (3 bits)
+    //   0 = function
+    //   1 = subscribe
+    //   2 = unsubscribe
+    //   3 = get from observable
+    //   4 = auth
+    //   5 = error
+    // isDeflate (1 bit)
+    // len (28 bits)
+    const len = nr >> 4;
+    const meta = nr & 15;
+    const type = meta >> 1;
+    const isDeflate = meta & 1;
+    return {
+        type,
+        isDeflate: isDeflate === 1,
+        len,
+    };
+};
+exports.decodeHeader = decodeHeader;
+const storeUint8 = (buff, n, start, len) => {
+    for (let index = start; index < start + len; index++) {
+        const byte = n & 0xff;
+        buff[index] = byte;
+        n = (n - byte) / 256;
+    }
+};
+exports.storeUint8 = storeUint8;
+const readUint8 = (buff, start, len) => {
+    let n = 0;
+    const s = len - 1 + start;
+    for (let i = s; i >= start; i--) {
+        n = n * 256 + buff[i];
+    }
+    return n;
+};
+exports.readUint8 = readUint8;
+const encodeHeader = (type, isDeflate, len) => {
+    // 4 bytes
+    // type (3 bits)
+    //   0 = functionData
+    //   1 = subscriptionData
+    //   2 = subscriptionDiffData
+    // isDeflate (1 bit)
+    // len (28 bits)
+    const encodedMeta = (type << 1) + (Number(isDeflate) | 0);
+    const nr = (len << 4) + encodedMeta;
+    return nr;
+};
+exports.encodeHeader = encodeHeader;
+const valueToBuffer = (payload) => {
+    // can use a more elloborate typed response e.g. number etc in there
+    if (payload === undefined) {
+        return Buffer.from([]);
+    }
+    // only stringify if not string...
+    return Buffer.from(JSON.stringify(payload));
+};
+exports.valueToBuffer = valueToBuffer;
+const decodePayload = (payload, isDeflate) => {
+    if (!isDeflate) {
+        return textDecoder.decode(payload);
+    }
+    const buffer = node_zlib_1.default.inflateRawSync(payload);
+    return textDecoder.decode(buffer);
+};
+exports.decodePayload = decodePayload;
+const decodeName = (arr, start, end) => {
+    const name = new Uint8Array(arr.slice(start, end));
+    return textDecoder.decode(name);
+};
+exports.decodeName = decodeName;
+const encodeFunctionResponse = (id, buffer) => {
+    // Type 0
+    // | 4 header | 3 id | * payload |
+    let isDeflate = false;
+    // implement later
+    const chunks = 1;
+    if (buffer.length > exports.COMPRESS_FROM_BYTES) {
+        isDeflate = true;
+        buffer = node_zlib_1.default.deflateRawSync(buffer, {});
+    }
+    if (chunks === 1) {
+        const headerSize = 4;
+        const idSize = 3;
+        const msgSize = idSize + buffer.length;
+        const header = (0, exports.encodeHeader)(0, isDeflate, msgSize);
+        // not very nessecary but ok
+        const array = new Uint8Array(headerSize + msgSize);
+        (0, exports.storeUint8)(array, header, 0, 4);
+        (0, exports.storeUint8)(array, id, 4, 3);
+        if (buffer.length) {
+            array.set(buffer, 7);
+        }
+        return array;
+    }
+    else {
+        console.warn('chunk not implemented yet');
+        return new Uint8Array(0);
+    }
+};
+exports.encodeFunctionResponse = encodeFunctionResponse;
+const encodeGetResponse = (id) => {
+    // Type 4
+    // | 4 header | 8 id |
+    const header = (0, exports.encodeHeader)(3, false, 8);
+    const array = new Uint8Array(12);
+    (0, exports.storeUint8)(array, header, 0, 4);
+    (0, exports.storeUint8)(array, id, 4, 8);
+    return array;
+};
+exports.encodeGetResponse = encodeGetResponse;
+const updateId = (payload, id) => {
+    const prevId = payload.slice(4, 12);
+    // if (readUint8(prevId, 0, 8) !== readUint8(payload, 4, 8)) {
+    // throw new Error('INCORRECT PARSING')
+    // }
+    (0, exports.storeUint8)(payload, id, 4, 8);
+    return prevId;
+};
+exports.updateId = updateId;
+const encodeObservableResponse = (id, checksum, buffer) => {
+    // Type 1 (full data) // TODO: include crc32 4 Bytes
+    // | 4 header | 8 id | 8 checksum | * payload |
+    let isDeflate = false;
+    // implement later
+    const chunks = 1;
+    if (buffer.length > exports.COMPRESS_FROM_BYTES) {
+        isDeflate = true;
+        buffer = node_zlib_1.default.deflateRawSync(buffer, {});
+    }
+    if (chunks === 1) {
+        const msgSize = 16 + buffer.length;
+        const header = (0, exports.encodeHeader)(1, isDeflate, msgSize);
+        const array = new Uint8Array(4 + msgSize);
+        (0, exports.storeUint8)(array, header, 0, 4);
+        (0, exports.storeUint8)(array, id, 4, 8);
+        (0, exports.storeUint8)(array, checksum, 12, 8);
+        if (buffer.length) {
+            array.set(buffer, 20);
+        }
+        return [array, isDeflate];
+    }
+    else {
+        console.warn('chunk not implemented yet');
+        return [new Uint8Array(0), false];
+    }
+};
+exports.encodeObservableResponse = encodeObservableResponse;
+const encodeObservableDiffResponse = (id, checksum, previousChecksum, buffer) => {
+    // Type 2 (diff data) // TODO: include crc32 4 Bytes
+    // | 4 header | 8 id | 8 checksum | 8 previousChecksum | * diff |
+    let isDeflate = false;
+    // implement later
+    const chunks = 1;
+    if (buffer.length > exports.COMPRESS_FROM_BYTES) {
+        isDeflate = true;
+        buffer = node_zlib_1.default.deflateRawSync(buffer, {});
+    }
+    if (chunks === 1) {
+        const msgSize = 24 + buffer.length;
+        const header = (0, exports.encodeHeader)(2, isDeflate, msgSize);
+        const array = new Uint8Array(4 + msgSize);
+        (0, exports.storeUint8)(array, header, 0, 4);
+        (0, exports.storeUint8)(array, id, 4, 8);
+        (0, exports.storeUint8)(array, checksum, 12, 8);
+        (0, exports.storeUint8)(array, previousChecksum, 20, 8);
+        if (buffer.length) {
+            array.set(buffer, 28);
+        }
+        return array;
+    }
+    else {
+        console.warn('chunk not implemented yet');
+        return new Uint8Array(0);
+    }
+};
+exports.encodeObservableDiffResponse = encodeObservableDiffResponse;
+const encodeSimpleResponse = (type, buffer) => {
+    // | 4 header | * payload |
+    let isDeflate = false;
+    if (buffer.length > exports.COMPRESS_FROM_BYTES) {
+        isDeflate = true;
+        buffer = node_zlib_1.default.deflateRawSync(buffer, {});
+    }
+    const headerSize = 4;
+    const msgSize = buffer.length;
+    const header = (0, exports.encodeHeader)(type, isDeflate, msgSize);
+    const array = new Uint8Array(headerSize + msgSize);
+    (0, exports.storeUint8)(array, header, 0, 4);
+    if (buffer.length) {
+        array.set(buffer, 4);
+    }
+    return array;
+};
+const encodeAuthResponse = (buffer) => {
+    // Type 4
+    return encodeSimpleResponse(4, buffer);
+};
+exports.encodeAuthResponse = encodeAuthResponse;
+const encodeErrorResponse = (buffer) => {
+    // Type 5
+    return encodeSimpleResponse(5, buffer);
+};
+exports.encodeErrorResponse = encodeErrorResponse;
+// decode cached data
+const decode = (buffer) => {
+    const header = (0, exports.readUint8)(buffer, 0, 4);
+    const { isDeflate, len, type } = (0, exports.decodeHeader)(header);
+    if (type === 1) {
+        // | 4 header | 8 id | 8 checksum | * payload |
+        if (len === 16) {
+            return;
+        }
+        const start = 20;
+        const end = len + 4;
+        return (0, exports.decodePayload)(buffer.slice(start, end), isDeflate);
+    }
+    // decode diff as well
+};
+exports.decode = decode;
+//# sourceMappingURL=protocol.js.map

package/dist/protocol.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"protocol.js","sourceRoot":"","sources":["../src/protocol.ts"],"names":[],"mappings":";;;;;;AAAA,0DAA4B;AAE5B,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAA;AAExB,QAAA,mBAAmB,GAAG,GAAG,CAAA;AAE/B,MAAM,YAAY,GAAG,CAC1B,EAAU,EACyC,EAAE;IACrD,UAAU;IACV,gBAAgB;IAChB,iBAAiB;IACjB,kBAAkB;IAClB,oBAAoB;IACpB,4BAA4B;IAC5B,aAAa;IACb,cAAc;IACd,oBAAoB;IACpB,gBAAgB;IAChB,MAAM,GAAG,GAAG,EAAE,IAAI,CAAC,CAAA;IACnB,MAAM,IAAI,GAAG,EAAE,GAAG,EAAE,CAAA;IACpB,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,CAAA;IACtB,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,CAAA;IAC1B,OAAO;QACL,IAAI;QACJ,SAAS,EAAE,SAAS,KAAK,CAAC;QAC1B,GAAG;KACJ,CAAA;AACH,CAAC,CAAA;AAtBY,QAAA,YAAY,gBAsBxB;AAEM,MAAM,UAAU,GAAG,CACxB,IAAgB,EAChB,CAAS,EACT,KAAa,EACb,GAAW,EACX,EAAE;IACF,KAAK,IAAI,KAAK,GAAG,KAAK,EAAE,KAAK,GAAG,KAAK,GAAG,GAAG,EAAE,KAAK,EAAE,EAAE;QACpD,MAAM,IAAI,GAAG,CAAC,GAAG,IAAI,CAAA;QACrB,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAA;QAClB,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,GAAG,CAAA;KACrB;AACH,CAAC,CAAA;AAXY,QAAA,UAAU,cAWtB;AAEM,MAAM,SAAS,GAAG,CACvB,IAAgB,EAChB,KAAa,EACb,GAAW,EACH,EAAE;IACV,IAAI,CAAC,GAAG,CAAC,CAAA;IACT,MAAM,CAAC,GAAG,GAAG,GAAG,CAAC,GAAG,KAAK,CAAA;IACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC,EAAE,EAAE;QAC/B,CAAC,GAAG,CAAC,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAA;KACtB;IACD,OAAO,CAAC,CAAA;AACV,CAAC,CAAA;AAXY,QAAA,SAAS,aAWrB;AAEM,MAAM,YAAY,GAAG,CAC1B,IAAY,EACZ,SAAkB,EAClB,GAAW,EACH,EAAE;IACV,UAAU;IACV,gBAAgB;IAChB,qBAAqB;IACrB,yBAAyB;IACzB,6BAA6B;IAC7B,oBAAoB;IACpB,gBAAgB;IAChB,MAAM,WAAW,GAAG,CAAC,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAA;IACzD,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC,GAAG,WAAW,CAAA;IACnC,OAAO,EAAE,CAAA;AACX,CAAC,CAAA;AAfY,QAAA,YAAY,gBAexB;AAEM,MAAM,aAAa,GAAG,CAAC,OAAY,EAAU,EAAE;IACpD,oEAAoE;IACpE,IAAI,OAAO,KAAK,SAAS,EAAE;QACzB,OAAO,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;KACvB;IACD,kCAAkC;IAClC,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAA;AAC7C,CAAC,CAAA;AAPY,QAAA,aAAa,iBAOzB;AAEM,MAAM,aAAa,GAAG,CAAC,OAAmB,EAAE,SAAkB,EAAO,EAAE;IAC5E,IAAI,CAAC,SAAS,EAAE;QACd,OAAO,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAA;KACnC;IAED,MAAM,MAAM,GAAG,mBAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAA;IAC3C,OAAO,WAAW,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;AACnC,CAAC,CAAA;AAPY,QAAA,aAAa,iBAOzB;AAEM,MAAM,UAAU,GAAG,CACxB,GAAe,EACf,KAAa,EACb,GAAW,EACH,EAAE;IACV,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAA;IAClD,OAAO,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAA;AACjC,CAAC,CAAA;AAPY,QAAA,UAAU,cAOtB;AAEM,MAAM,sBAAsB,GAAG,CACpC,EAAU,EACV,MAAc,EACF,EAAE;IACd,SAAS;IACT,kCAAkC;IAElC,IAAI,SAAS,GAAG,KAAK,CAAA;IACrB,kBAAkB;IAClB,MAAM,MAAM,GAAG,CAAC,CAAA;IAEhB,IAAI,MAAM,CAAC,MAAM,GAAG,2BAAmB,EAAE;QACvC,SAAS,GAAG,IAAI,CAAA;QAChB,MAAM,GAAG,mBAAI,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;KACzC;IAED,IAAI,MAAM,KAAK,CAAC,EAAE;QAChB,MAAM,UAAU,GAAG,CAAC,CAAA;QACpB,MAAM,MAAM,GAAG,CAAC,CAAA;QAChB,MAAM,OAAO,GAAG,MAAM,GAAG,MAAM,CAAC,MAAM,CAAA;QACtC,MAAM,MAAM,GAAG,IAAA,oBAAY,EAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,CAAA;QAElD,4BAA4B;QAC5B,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,UAAU,GAAG,OAAO,CAAC,CAAA;QAClD,IAAA,kBAAU,EAAC,KAAK,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;QAC/B,IAAA,kBAAU,EAAC,KAAK,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;QAC3B,IAAI,MAAM,CAAC,MAAM,EAAE;YACjB,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;SACrB;QACD,OAAO,KAAK,CAAA;KACb;SAAM;QACL,OAAO,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAA;QACzC,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,CAAA;KACzB;AACH,CAAC,CAAA;AAlCY,QAAA,sBAAsB,0BAkClC;AAEM,MAAM,iBAAiB,GAAG,CAAC,EAAU,EAAc,EAAE;IAC1D,SAAS;IACT,sBAAsB;IACtB,MAAM,MAAM,GAAG,IAAA,oBAAY,EAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAA;IACxC,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,EAAE,CAAC,CAAA;IAChC,IAAA,kBAAU,EAAC,KAAK,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;IAC/B,IAAA,kBAAU,EAAC,KAAK,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;IAC3B,OAAO,KAAK,CAAA;AACd,CAAC,CAAA;AARY,QAAA,iBAAiB,qBAQ7B;AAEM,MAAM,QAAQ,GAAG,CAAC,OAAmB,EAAE,EAAU,EAAc,EAAE;IACtE,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;IACnC,8DAA8D;IAC9D,uCAAuC;IACvC,IAAI;IACJ,IAAA,kBAAU,EAAC,OAAO,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;IAC7B,OAAO,MAAM,CAAA;AACf,CAAC,CAAA;AAPY,QAAA,QAAQ,YAOpB;AAEM,MAAM,wBAAwB,GAAG,CACtC,EAAU,EACV,QAAgB,EAChB,MAAc,EACS,EAAE;IACzB,oDAAoD;IACpD,+CAA+C;IAE/C,IAAI,SAAS,GAAG,KAAK,CAAA;IACrB,kBAAkB;IAClB,MAAM,MAAM,GAAG,CAAC,CAAA;IAEhB,IAAI,MAAM,CAAC,MAAM,GAAG,2BAAmB,EAAE;QACvC,SAAS,GAAG,IAAI,CAAA;QAChB,MAAM,GAAG,mBAAI,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;KACzC;IAED,IAAI,MAAM,KAAK,CAAC,EAAE;QAChB,MAAM,OAAO,GAAG,EAAE,GAAG,MAAM,CAAC,MAAM,CAAA;QAClC,MAAM,MAAM,GAAG,IAAA,oBAAY,EAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,CAAA;QAClD,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,CAAC,GAAG,OAAO,CAAC,CAAA;QACzC,IAAA,kBAAU,EAAC,KAAK,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;QAC/B,IAAA,kBAAU,EAAC,KAAK,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;QAC3B,IAAA,kBAAU,EAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAA;QAClC,IAAI,MAAM,CAAC,MAAM,EAAE;YACjB,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;SACtB;QACD,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,CAAA;KAC1B;SAAM;QACL,OAAO,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAA;QACzC,OAAO,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAA;KAClC;AACH,CAAC,CAAA;AAhCY,QAAA,wBAAwB,4BAgCpC;AAEM,MAAM,4BAA4B,GAAG,CAC1C,EAAU,EACV,QAAgB,EAChB,gBAAwB,EACxB,MAAc,EACF,EAAE;IACd,oDAAoD;IACpD,iEAAiE;IAEjE,IAAI,SAAS,GAAG,KAAK,CAAA;IACrB,kBAAkB;IAClB,MAAM,MAAM,GAAG,CAAC,CAAA;IAEhB,IAAI,MAAM,CAAC,MAAM,GAAG,2BAAmB,EAAE;QACvC,SAAS,GAAG,IAAI,CAAA;QAChB,MAAM,GAAG,mBAAI,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;KACzC;IAED,IAAI,MAAM,KAAK,CAAC,EAAE;QAChB,MAAM,OAAO,GAAG,EAAE,GAAG,MAAM,CAAC,MAAM,CAAA;QAClC,MAAM,MAAM,GAAG,IAAA,oBAAY,EAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,CAAA;QAClD,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,CAAC,GAAG,OAAO,CAAC,CAAA;QACzC,IAAA,kBAAU,EAAC,KAAK,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;QAC/B,IAAA,kBAAU,EAAC,KAAK,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;QAC3B,IAAA,kBAAU,EAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAA;QAClC,IAAA,kBAAU,EAAC,KAAK,EAAE,gBAAgB,EAAE,EAAE,EAAE,CAAC,CAAC,CAAA;QAC1C,IAAI,MAAM,CAAC,MAAM,EAAE;YACjB,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;SACtB;QACD,OAAO,KAAK,CAAA;KACb;SAAM;QACL,OAAO,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAA;QACzC,OAAO,IAAI,UAAU,CAAC,CAAC,CAAC,CAAA;KACzB;AACH,CAAC,CAAA;AAlCY,QAAA,4BAA4B,gCAkCxC;AAED,MAAM,oBAAoB,GAAG,CAAC,IAAY,EAAE,MAAc,EAAc,EAAE;IACxE,2BAA2B;IAC3B,IAAI,SAAS,GAAG,KAAK,CAAA;IAErB,IAAI,MAAM,CAAC,MAAM,GAAG,2BAAmB,EAAE;QACvC,SAAS,GAAG,IAAI,CAAA;QAChB,MAAM,GAAG,mBAAI,CAAC,cAAc,CAAC,MAAM,EAAE,EAAE,CAAC,CAAA;KACzC;IAED,MAAM,UAAU,GAAG,CAAC,CAAA;IACpB,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAA;IAC7B,MAAM,MAAM,GAAG,IAAA,oBAAY,EAAC,IAAI,EAAE,SAAS,EAAE,OAAO,CAAC,CAAA;IACrD,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,UAAU,GAAG,OAAO,CAAC,CAAA;IAClD,IAAA,kBAAU,EAAC,KAAK,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;IAC/B,IAAI,MAAM,CAAC,MAAM,EAAE;QACjB,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC,CAAA;KACrB;IACD,OAAO,KAAK,CAAA;AACd,CAAC,CAAA;AAEM,MAAM,kBAAkB,GAAG,CAAC,MAAc,EAAc,EAAE;IAC/D,SAAS;IAET,OAAO,oBAAoB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAA;AACxC,CAAC,CAAA;AAJY,QAAA,kBAAkB,sBAI9B;AAEM,MAAM,mBAAmB,GAAG,CAAC,MAAc,EAAc,EAAE;IAChE,SAAS;IAET,OAAO,oBAAoB,CAAC,CAAC,EAAE,MAAM,CAAC,CAAA;AACxC,CAAC,CAAA;AAJY,QAAA,mBAAmB,uBAI/B;AAED,qBAAqB;AACd,MAAM,MAAM,GAAG,CAAC,MAAkB,EAAO,EAAE;IAChD,MAAM,MAAM,GAAG,IAAA,iBAAS,EAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAA;IACtC,MAAM,EAAE,SAAS,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,IAAA,oBAAY,EAAC,MAAM,CAAC,CAAA;IACrD,IAAI,IAAI,KAAK,CAAC,EAAE;QACd,+CAA+C;QAC/C,IAAI,GAAG,KAAK,EAAE,EAAE;YACd,OAAM;SACP;QACD,MAAM,KAAK,GAAG,EAAE,CAAA;QAChB,MAAM,GAAG,GAAG,GAAG,GAAG,CAAC,CAAA;QACnB,OAAO,IAAA,qBAAa,EAAC,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,SAAS,CAAC,CAAA;KAC1D;IACD,sBAAsB;AACxB,CAAC,CAAA;AAbY,QAAA,MAAM,UAalB"}

package/dist/security.d.ts

@@ -0,0 +1,6 @@
+import type { BasedServer } from './server';
+import uws from '@based/uws';
+import { HttpSession, WebSocketSession, Context } from './context';
+export declare const rateLimitRequest: (server: BasedServer, ctx: Context<WebSocketSession | HttpSession>, tokens: number, max: number) => boolean;
+export declare const endRateLimitHttp: (res: uws.HttpResponse) => void;
+export declare const blockIncomingRequest: (server: BasedServer, ip: string, res: uws.HttpResponse, req: uws.HttpRequest, max: number, tokens: number) => boolean;

package/dist/security.js

@@ -0,0 +1,95 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.blockIncomingRequest = exports.endRateLimitHttp = exports.rateLimitRequest = void 0;
+const error_1 = require("./error");
+var IsBlocked;
+(function (IsBlocked) {
+    IsBlocked[IsBlocked["notBlocked"] = 0] = "notBlocked";
+    IsBlocked[IsBlocked["blocked"] = 1] = "blocked";
+    IsBlocked[IsBlocked["firstBlocked"] = 2] = "firstBlocked";
+})(IsBlocked || (IsBlocked = {}));
+const drainRequestCounter = (server) => {
+    server.requestsCounterInProgress = true;
+    server.requestsCounterTimeout = setTimeout(() => {
+        server.requestsCounterInProgress = false;
+        server.rateLimitCounter.forEach((value, ip) => {
+            if (value.requests <= 0) {
+                server.rateLimitCounter.delete(ip);
+                return;
+            }
+            value.requests -= server.rateLimit.drain;
+        });
+        if (server.rateLimitCounter.size) {
+            drainRequestCounter(server);
+        }
+    }, 30e3);
+};
+const incomingRequestCounter = (server, ip, tokens, max) => {
+    let ipReqCounter = server.rateLimitCounter.get(ip);
+    if (!ipReqCounter) {
+        ipReqCounter = {
+            requests: tokens,
+        };
+        server.rateLimitCounter.set(ip, ipReqCounter);
+    }
+    else {
+        ipReqCounter.requests += tokens;
+    }
+    if (ipReqCounter.requests === max) {
+        return 2;
+    }
+    if (ipReqCounter.requests > max) {
+        return 1;
+    }
+    if (!server.requestsCounterInProgress) {
+        drainRequestCounter(server);
+    }
+    return 0;
+};
+const rateLimitRequest = (server, ctx, tokens, max) => {
+    if (!ctx.session) {
+        return false;
+    }
+    const ip = ctx.session.ip;
+    const code = incomingRequestCounter(server, ip, tokens, max);
+    if (code === 0) {
+        return false;
+    }
+    if (code === 2) {
+        server.emit('error', ctx, { code: error_1.BasedErrorCode.RateLimit });
+    }
+    return true;
+};
+exports.rateLimitRequest = rateLimitRequest;
+const endRateLimitHttp = (res) => {
+    res.cork(() => {
+        res.writeStatus('429 Too Many Requests');
+        res.close();
+    });
+};
+exports.endRateLimitHttp = endRateLimitHttp;
+const blockIncomingRequest = (server, ip, res, req, max, tokens) => {
+    if (server.allowedIps.has(ip)) {
+        return false;
+    }
+    if (server.blockedIps.has(ip)) {
+        res.close();
+        return true;
+    }
+    const code = incomingRequestCounter(server, ip, tokens, max);
+    if (code === 0) {
+        return false;
+    }
+    if (code === 2) {
+        server.emit('error', {
+            session: {
+                ua: req.getHeader('user-agent'),
+                ip,
+            },
+        }, { code: error_1.BasedErrorCode.RateLimit });
+    }
+    (0, exports.endRateLimitHttp)(res);
+    return true;
+};
+exports.blockIncomingRequest = blockIncomingRequest;
+//# sourceMappingURL=security.js.map

package/dist/security.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"security.js","sourceRoot":"","sources":["../src/security.ts"],"names":[],"mappings":";;;AAAA,mCAAwC;AAKxC,IAAK,SAIJ;AAJD,WAAK,SAAS;IACZ,qDAAc,CAAA;IACd,+CAAW,CAAA;IACX,yDAAgB,CAAA;AAClB,CAAC,EAJI,SAAS,KAAT,SAAS,QAIb;AAED,MAAM,mBAAmB,GAAG,CAAC,MAAmB,EAAE,EAAE;IAClD,MAAM,CAAC,yBAAyB,GAAG,IAAI,CAAA;IACvC,MAAM,CAAC,sBAAsB,GAAG,UAAU,CAAC,GAAG,EAAE;QAC9C,MAAM,CAAC,yBAAyB,GAAG,KAAK,CAAA;QACxC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE;YAC5C,IAAI,KAAK,CAAC,QAAQ,IAAI,CAAC,EAAE;gBACvB,MAAM,CAAC,gBAAgB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAA;gBAClC,OAAM;aACP;YACD,KAAK,CAAC,QAAQ,IAAI,MAAM,CAAC,SAAS,CAAC,KAAK,CAAA;QAC1C,CAAC,CAAC,CAAA;QACF,IAAI,MAAM,CAAC,gBAAgB,CAAC,IAAI,EAAE;YAChC,mBAAmB,CAAC,MAAM,CAAC,CAAA;SAC5B;IACH,CAAC,EAAE,IAAI,CAAC,CAAA;AACV,CAAC,CAAA;AAED,MAAM,sBAAsB,GAAG,CAC7B,MAAmB,EACnB,EAAU,EACV,MAAc,EACd,GAAW,EACA,EAAE;IACb,IAAI,YAAY,GAAG,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;IAClD,IAAI,CAAC,YAAY,EAAE;QACjB,YAAY,GAAG;YACb,QAAQ,EAAE,MAAM;SACjB,CAAA;QACD,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,EAAE,YAAY,CAAC,CAAA;KAC9C;SAAM;QACL,YAAY,CAAC,QAAQ,IAAI,MAAM,CAAA;KAChC;IACD,IAAI,YAAY,CAAC,QAAQ,KAAK,GAAG,EAAE;QACjC,OAAO,CAAC,CAAA;KACT;IACD,IAAI,YAAY,CAAC,QAAQ,GAAG,GAAG,EAAE;QAC/B,OAAO,CAAC,CAAA;KACT;IACD,IAAI,CAAC,MAAM,CAAC,yBAAyB,EAAE;QACrC,mBAAmB,CAAC,MAAM,CAAC,CAAA;KAC5B;IACD,OAAO,CAAC,CAAA;AACV,CAAC,CAAA;AAEM,MAAM,gBAAgB,GAAG,CAC9B,MAAmB,EACnB,GAA4C,EAC5C,MAAc,EACd,GAAW,EACF,EAAE;IACX,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE;QAChB,OAAO,KAAK,CAAA;KACb;IACD,MAAM,EAAE,GAAG,GAAG,CAAC,OAAO,CAAC,EAAE,CAAA;IACzB,MAAM,IAAI,GAAG,sBAAsB,CAAC,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,CAAA;IAC5D,IAAI,IAAI,KAAK,CAAC,EAAE;QACd,OAAO,KAAK,CAAA;KACb;IACD,IAAI,IAAI,KAAK,CAAC,EAAE;QACd,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,IAAI,EAAE,sBAAc,CAAC,SAAS,EAAE,CAAC,CAAA;KAC9D;IACD,OAAO,IAAI,CAAA;AACb,CAAC,CAAA;AAlBY,QAAA,gBAAgB,oBAkB5B;AAEM,MAAM,gBAAgB,GAAG,CAAC,GAAqB,EAAE,EAAE;IACxD,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE;QACZ,GAAG,CAAC,WAAW,CAAC,uBAAuB,CAAC,CAAA;QACxC,GAAG,CAAC,KAAK,EAAE,CAAA;IACb,CAAC,CAAC,CAAA;AACJ,CAAC,CAAA;AALY,QAAA,gBAAgB,oBAK5B;AAEM,MAAM,oBAAoB,GAAG,CAClC,MAAmB,EACnB,EAAU,EACV,GAAqB,EACrB,GAAoB,EACpB,GAAW,EACX,MAAc,EACL,EAAE;IACX,IAAI,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;QAC7B,OAAO,KAAK,CAAA;KACb;IACD,IAAI,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;QAC7B,GAAG,CAAC,KAAK,EAAE,CAAA;QACX,OAAO,IAAI,CAAA;KACZ;IACD,MAAM,IAAI,GAAG,sBAAsB,CAAC,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,CAAA;IAC5D,IAAI,IAAI,KAAK,CAAC,EAAE;QACd,OAAO,KAAK,CAAA;KACb;IAED,IAAI,IAAI,KAAK,CAAC,EAAE;QACd,MAAM,CAAC,IAAI,CACT,OAAO,EACP;YACE,OAAO,EAAE;gBACP,EAAE,EAAE,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC;gBAC/B,EAAE;aACH;SACF,EACD,EAAE,IAAI,EAAE,sBAAc,CAAC,SAAS,EAAE,CACnC,CAAA;KACF;IAED,IAAA,wBAAgB,EAAC,GAAG,CAAC,CAAA;IAErB,OAAO,IAAI,CAAA;AACb,CAAC,CAAA;AApCY,QAAA,oBAAoB,wBAoChC"}

package/dist/sendError.d.ts

@@ -0,0 +1,5 @@
+import { BasedServer } from './server';
+import { HttpSession, WebSocketSession, Context } from './context';
+import { BasedErrorCode, ErrorPayload } from './error';
+export declare function sendHttpError<T extends BasedErrorCode>(server: BasedServer, ctx: Context<HttpSession>, basedCode: T, payload: ErrorPayload[T]): void;
+export declare function sendError<T extends BasedErrorCode>(server: BasedServer, ctx: Context<WebSocketSession | HttpSession>, basedCode: T, payload: ErrorPayload[T]): void;

package/dist/sendError.js

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sendError = exports.sendHttpError = void 0;
+const sendHttpResponse_1 = require("./sendHttpResponse");
+const context_1 = require("./context");
+const protocol_1 = require("./protocol");
+const error_1 = require("./error");
+const sendHttpErrorMessage = (res, error) => {
+    const { code, message, statusCode, statusMessage } = error;
+    res.writeStatus(`${statusCode} ${statusMessage}`);
+    res.writeHeader('Access-Control-Allow-Origin', '*');
+    res.writeHeader('Access-Control-Allow-Headers', 'content-type');
+    res.writeHeader('Content-Type', 'application/json');
+    return JSON.stringify({
+        error: message,
+        code,
+    });
+};
+function sendHttpError(server, ctx, basedCode, payload) {
+    if (!ctx.session) {
+        return;
+    }
+    ctx.session.res.cork(() => {
+        (0, sendHttpResponse_1.end)(ctx, sendHttpErrorMessage(ctx.session.res, (0, error_1.createError)(server, ctx, basedCode, payload)));
+    });
+}
+exports.sendHttpError = sendHttpError;
+function sendError(server, ctx, basedCode, payload) {
+    if (!ctx.session) {
+        return;
+    }
+    if ((0, context_1.isHttpContext)(ctx)) {
+        return sendHttpError(server, ctx, basedCode, payload);
+    }
+    else if ((0, context_1.isWsSession)(ctx.session)) {
+        const errorData = (0, error_1.createError)(server, ctx, basedCode, payload);
+        const ws = ctx.session;
+        ws.send((0, protocol_1.encodeErrorResponse)((0, protocol_1.valueToBuffer)(errorData)), true, false);
+    }
+}
+exports.sendError = sendError;
+//# sourceMappingURL=sendError.js.map

package/dist/sendError.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sendError.js","sourceRoot":"","sources":["../src/sendError.ts"],"names":[],"mappings":";;;AACA,yDAAwC;AAExC,uCAMkB;AAClB,yCAA+D;AAC/D,mCAKgB;AAEhB,MAAM,oBAAoB,GAAG,CAC3B,GAAqB,EACrB,KAAqB,EACb,EAAE;IACV,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,GAAG,KAAK,CAAA;IAC1D,GAAG,CAAC,WAAW,CAAC,GAAG,UAAU,IAAI,aAAa,EAAE,CAAC,CAAA;IACjD,GAAG,CAAC,WAAW,CAAC,6BAA6B,EAAE,GAAG,CAAC,CAAA;IACnD,GAAG,CAAC,WAAW,CAAC,8BAA8B,EAAE,cAAc,CAAC,CAAA;IAC/D,GAAG,CAAC,WAAW,CAAC,cAAc,EAAE,kBAAkB,CAAC,CAAA;IACnD,OAAO,IAAI,CAAC,SAAS,CAAC;QACpB,KAAK,EAAE,OAAO;QACd,IAAI;KACL,CAAC,CAAA;AACJ,CAAC,CAAA;AAED,SAAgB,aAAa,CAC3B,MAAmB,EACnB,GAAyB,EACzB,SAAY,EACZ,OAAwB;IAExB,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE;QAChB,OAAM;KACP;IACD,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE;QACxB,IAAA,sBAAG,EACD,GAAG,EACH,oBAAoB,CAClB,GAAG,CAAC,OAAO,CAAC,GAAG,EACf,IAAA,mBAAW,EAAC,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,OAAO,CAAC,CAC7C,CACF,CAAA;IACH,CAAC,CAAC,CAAA;AACJ,CAAC;AAlBD,sCAkBC;AAED,SAAgB,SAAS,CACvB,MAAmB,EACnB,GAA4C,EAC5C,SAAY,EACZ,OAAwB;IAExB,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE;QAChB,OAAM;KACP;IACD,IAAI,IAAA,uBAAa,EAAC,GAAG,CAAC,EAAE;QACtB,OAAO,aAAa,CAAC,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,OAAO,CAAC,CAAA;KACtD;SAAM,IAAI,IAAA,qBAAW,EAAC,GAAG,CAAC,OAAO,CAAC,EAAE;QACnC,MAAM,SAAS,GAAG,IAAA,mBAAW,EAAC,MAAM,EAAE,GAAG,EAAE,SAAS,EAAE,OAAO,CAAC,CAAA;QAC9D,MAAM,EAAE,GAAqB,GAAG,CAAC,OAAO,CAAA;QACxC,EAAE,CAAC,IAAI,CAAC,IAAA,8BAAmB,EAAC,IAAA,wBAAa,EAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,CAAA;KACpE;AACH,CAAC;AAhBD,8BAgBC"}

package/dist/sendHttpResponse.d.ts

@@ -0,0 +1,4 @@
+/// <reference types="node" />
+import { Context, HttpSession } from './context';
+export declare const end: (ctx: Context<HttpSession>, payload?: string | Buffer | Uint8Array) => void;
+export declare const sendHttpResponse: (ctx: Context<HttpSession>, result: any) => void;

package/dist/sendHttpResponse.js

@@ -0,0 +1,53 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sendHttpResponse = exports.end = void 0;
+const compress_1 = require("./compress");
+const end = (ctx, payload) => {
+    if (!ctx.session) {
+        return;
+    }
+    ctx.session.res.writeHeader('Access-Control-Allow-Origin', '*');
+    // only allowed headers
+    ctx.session.res.writeHeader('Access-Control-Allow-Headers', '*');
+    if (payload === undefined) {
+        ctx.session.res.end();
+    }
+    else {
+        ctx.session.res.end(payload);
+    }
+    ctx.session.res = null;
+    ctx.session.req = null;
+    ctx.session = null;
+};
+exports.end = end;
+const sendHttpResponse = (ctx, result) => {
+    if (!ctx.session) {
+        return;
+    }
+    let cType;
+    // for functions there is never cache (idea is they are used to execute - observable fns are for cache)
+    let parsed;
+    if (typeof result === 'string') {
+        cType = 'text/plain';
+        parsed = result;
+    }
+    else {
+        cType = 'application/json';
+        parsed = JSON.stringify(result);
+    }
+    (0, compress_1.compress)(parsed, ctx.session.headers.encoding).then(({ payload, encoding }) => {
+        if (ctx.session.res) {
+            ctx.session.res.cork(() => {
+                ctx.session.res.writeStatus('200 OK');
+                ctx.session.res.writeHeader('Cache-Control', 'max-age=0, must-revalidate');
+                ctx.session.res.writeHeader('Content-Type', cType);
+                if (encoding) {
+                    ctx.session.res.writeHeader('Content-Encoding', encoding);
+                }
+                (0, exports.end)(ctx, payload);
+            });
+        }
+    });
+};
+exports.sendHttpResponse = sendHttpResponse;
+//# sourceMappingURL=sendHttpResponse.js.map

package/dist/sendHttpResponse.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sendHttpResponse.js","sourceRoot":"","sources":["../src/sendHttpResponse.ts"],"names":[],"mappings":";;;AACA,yCAAqC;AAE9B,MAAM,GAAG,GAAG,CACjB,GAAyB,EACzB,OAAsC,EACtC,EAAE;IACF,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE;QAChB,OAAM;KACP;IACD,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,6BAA6B,EAAE,GAAG,CAAC,CAAA;IAC/D,uBAAuB;IACvB,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,8BAA8B,EAAE,GAAG,CAAC,CAAA;IAChE,IAAI,OAAO,KAAK,SAAS,EAAE;QACzB,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,CAAA;KACtB;SAAM;QACL,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAA;KAC7B;IACD,GAAG,CAAC,OAAO,CAAC,GAAG,GAAG,IAAI,CAAA;IACtB,GAAG,CAAC,OAAO,CAAC,GAAG,GAAG,IAAI,CAAA;IACtB,GAAG,CAAC,OAAO,GAAG,IAAI,CAAA;AACpB,CAAC,CAAA;AAlBY,QAAA,GAAG,OAkBf;AAEM,MAAM,gBAAgB,GAAG,CAAC,GAAyB,EAAE,MAAW,EAAE,EAAE;IACzE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE;QAChB,OAAM;KACP;IAED,IAAI,KAAa,CAAA;IAEjB,uGAAuG;IACvG,IAAI,MAAc,CAAA;IAClB,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE;QAC9B,KAAK,GAAG,YAAY,CAAA;QACpB,MAAM,GAAG,MAAM,CAAA;KAChB;SAAM;QACL,KAAK,GAAG,kBAAkB,CAAA;QAC1B,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;KAChC;IACD,IAAA,mBAAQ,EAAC,MAAM,EAAE,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,IAAI,CACjD,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE;QACxB,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,EAAE;YACnB,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE;gBACxB,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAA;gBACrC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CACzB,eAAe,EACf,4BAA4B,CAC7B,CAAA;gBACD,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,cAAc,EAAE,KAAK,CAAC,CAAA;gBAClD,IAAI,QAAQ,EAAE;oBACZ,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,kBAAkB,EAAE,QAAQ,CAAC,CAAA;iBAC1D;gBACD,IAAA,WAAG,EAAC,GAAG,EAAE,OAAO,CAAC,CAAA;YACnB,CAAC,CAAC,CAAA;SACH;IACH,CAAC,CACF,CAAA;AACH,CAAC,CAAA;AAlCY,QAAA,gBAAgB,oBAkC5B"}

package/dist/server.d.ts

@@ -0,0 +1,69 @@
+/// <reference types="node" />
+import type { Context } from './context';
+import type { ActiveObservable } from './observable';
+import uws from '@based/uws';
+import { BasedFunctions, FunctionConfig } from './functions';
+import { BasedAuth, AuthConfig } from './auth';
+import { BasedErrorCode, BasedErrorData } from './error';
+declare type EventMap = {
+    error: BasedErrorData;
+    ratelimit: void;
+    log: any;
+};
+declare type Event = keyof EventMap;
+declare type Listener<T> = (context: Context, data?: T, err?: Error) => void;
+declare type RateLimit = {
+    ws: number;
+    http: number;
+    drain: number;
+};
+export declare type ServerOptions = {
+    port?: number;
+    key?: string;
+    cert?: string;
+    functions?: FunctionConfig;
+    rateLimit?: RateLimit;
+    auth?: AuthConfig;
+    workerRequest?: (type: string, payload?: any) => void | Promise<any>;
+    ws?: {
+        open: (client: Context) => void;
+        close: (client: Context) => void;
+    };
+    http?: {
+        open: (client: Context) => void;
+        close: (client: Context) => void;
+    };
+};
+export declare class BasedServer {
+    functions: BasedFunctions;
+    auth: BasedAuth;
+    port: number;
+    uwsApp: uws.TemplatedApp;
+    rateLimit: RateLimit;
+    listenSocket: any;
+    blockedIps: Set<string>;
+    allowedIps: Set<string>;
+    rateLimitCounter: Map<string, {
+        requests: number;
+        errors?: Map<BasedErrorCode, number>;
+    }>;
+    requestsCounterInProgress: boolean;
+    requestsCounterTimeout: NodeJS.Timeout;
+    activeObservables: {
+        [name: string]: Map<number, ActiveObservable>;
+    };
+    activeObservablesById: Map<number, ActiveObservable>;
+    listeners: {
+        [E in Event]?: Listener<EventMap[E]>[];
+    };
+    workerRequest: (type: string, payload?: any) => void | Promise<any>;
+    constructor(opts: ServerOptions);
+    emit(type: Event, client: Context, val: EventMap[Event], err?: Error): void;
+    on(type: Event, fn: Listener<EventMap[Event]>): void;
+    removeAllListeners(): void;
+    once(type: Event, fn: Listener<EventMap[Event]>): void;
+    off(type: Event, fn: Listener<EventMap[Event]>): void;
+    start(port?: number, sharedSocket?: boolean): Promise<BasedServer>;
+    destroy(): Promise<void>;
+}
+export {};

package/dist/server.js

@@ -0,0 +1,113 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BasedServer = void 0;
+const uws_1 = __importDefault(require("@based/uws"));
+const incoming_1 = __importDefault(require("./incoming"));
+const functions_1 = require("./functions");
+const auth_1 = require("./auth");
+const utils_1 = require("@saulx/utils");
+// extend emitter
+class BasedServer {
+    constructor(opts) {
+        this.rateLimit = {
+            ws: 2e3,
+            http: 1e3,
+            drain: 500,
+        };
+        this.blockedIps = new Set();
+        // opposite of blockedIps can never get blocked
+        this.allowedIps = new Set();
+        // per ip so consitent unfortanetly
+        // check how large it is and make a loop to downgrade it
+        this.rateLimitCounter = new Map();
+        this.requestsCounterInProgress = false;
+        this.activeObservables = {};
+        this.activeObservablesById = new Map();
+        this.listeners = {};
+        this.functions = new functions_1.BasedFunctions(this, opts.functions);
+        this.auth = new auth_1.BasedAuth(this, opts.auth);
+        if (opts.workerRequest) {
+            this.workerRequest = opts.workerRequest;
+        }
+        if (opts.rateLimit) {
+            this.rateLimit = opts.rateLimit;
+        }
+        (0, incoming_1.default)(this, opts);
+    }
+    emit(type, client, val, err) {
+        if (this.listeners[type]) {
+            this.listeners[type].forEach((fn) => fn(client, val, err));
+        }
+    }
+    on(type, fn) {
+        if (!this.listeners[type]) {
+            this.listeners[type] = [];
+        }
+        this.listeners[type].push(fn);
+    }
+    removeAllListeners() {
+        this.listeners = {};
+    }
+    once(type, fn) {
+        this.on(type, (v) => {
+            fn(v);
+            this.off(type, fn);
+        });
+    }
+    off(type, fn) {
+        const listeners = this.listeners[type];
+        if (listeners) {
+            if (!fn) {
+                delete this.listeners[type];
+            }
+            else {
+                for (let i = 0, len = listeners.length; i < len; i++) {
+                    if (listeners[i] === fn) {
+                        listeners.splice(i, 1);
+                        break;
+                    }
+                }
+                if (listeners.length === 0) {
+                    delete this.listeners[type];
+                }
+            }
+        }
+    }
+    async start(port, sharedSocket) {
+        if (!port) {
+            port = this.port;
+        }
+        else {
+            this.port = port;
+        }
+        await (0, utils_1.wait)(10);
+        return new Promise((resolve, reject) => {
+            this.uwsApp.listen(this.port, sharedSocket ? 0 : 1, (listenSocket) => {
+                if (listenSocket) {
+                    console.info('💫  Based-edge-server listening on port:', this.port);
+                    // do this better wrap a nice thing arround it
+                    this.listenSocket = listenSocket;
+                    resolve(this);
+                }
+                else {
+                    console.info('🤮  Based-edge-server error on port:', this.port);
+                    reject(new Error('Cannot start based-server on port: ' + this.port));
+                }
+            });
+        });
+    }
+    async destroy() {
+        console.info('🔥 Destroy Based-edge-server');
+        if (this.listenSocket) {
+            uws_1.default.us_listen_socket_close(this.listenSocket);
+            this.listenSocket = null;
+        }
+        this.listenSocket = null;
+        this.uwsApp = null;
+    }
+}
+exports.BasedServer = BasedServer;
+//# sourceMappingURL=server.js.map

package/dist/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../src/server.ts"],"names":[],"mappings":";;;;;;AAEA,qDAA4B;AAC5B,0DAAoC;AACpC,2CAA4D;AAC5D,iCAA8C;AAE9C,wCAAmC;AAoCnC,iBAAiB;AACjB,MAAa,WAAW;IAgDtB,YAAY,IAAmB;QAvCxB,cAAS,GAAc;YAC5B,EAAE,EAAE,GAAG;YACP,IAAI,EAAE,GAAG;YACT,KAAK,EAAE,GAAG;SACX,CAAA;QAIM,eAAU,GAAgB,IAAI,GAAG,EAAE,CAAA;QAE1C,+CAA+C;QACxC,eAAU,GAAgB,IAAI,GAAG,EAAE,CAAA;QAE1C,mCAAmC;QACnC,wDAAwD;QACjD,qBAAgB,GAMnB,IAAI,GAAG,EAAE,CAAA;QAEN,8BAAyB,GAAY,KAAK,CAAA;QAI1C,sBAAiB,GAEpB,EAAE,CAAA;QAEC,0BAAqB,GAAkC,IAAI,GAAG,EAAE,CAAA;QAEhE,cAAS,GAEZ,EAAE,CAAA;QAKJ,IAAI,CAAC,SAAS,GAAG,IAAI,0BAAc,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAA;QACzD,IAAI,CAAC,IAAI,GAAG,IAAI,gBAAS,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAA;QAC1C,IAAI,IAAI,CAAC,aAAa,EAAE;YACtB,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAA;SACxC;QACD,IAAI,IAAI,CAAC,SAAS,EAAE;YAClB,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAA;SAChC;QACD,IAAA,kBAAW,EAAC,IAAI,EAAE,IAAI,CAAC,CAAA;IACzB,CAAC;IAED,IAAI,CAAC,IAAW,EAAE,MAAe,EAAE,GAAoB,EAAE,GAAW;QAClE,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;YACxB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAA;SAC3D;IACH,CAAC;IAED,EAAE,CAAC,IAAW,EAAE,EAA6B;QAC3C,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE;YACzB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,EAAE,CAAA;SAC1B;QACD,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;IAC/B,CAAC;IAED,kBAAkB;QAChB,IAAI,CAAC,SAAS,GAAG,EAAE,CAAA;IACrB,CAAC;IAED,IAAI,CAAC,IAAW,EAAE,EAA6B;QAC7C,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,EAAE;YAClB,EAAE,CAAC,CAAC,CAAC,CAAA;YACL,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAA;QACpB,CAAC,CAAC,CAAA;IACJ,CAAC;IAED,GAAG,CAAC,IAAW,EAAE,EAA6B;QAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA;QACtC,IAAI,SAAS,EAAE;YACb,IAAI,CAAC,EAAE,EAAE;gBACP,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA;aAC5B;iBAAM;gBACL,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,GAAG,GAAG,EAAE,CAAC,EAAE,EAAE;oBACpD,IAAI,SAAS,CAAC,CAAC,CAAC,KAAK,EAAE,EAAE;wBACvB,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;wBACtB,MAAK;qBACN;iBACF;gBACD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;oBAC1B,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA;iBAC5B;aACF;SACF;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAa,EAAE,YAAsB;QAC/C,IAAI,CAAC,IAAI,EAAE;YACT,IAAI,GAAG,IAAI,CAAC,IAAI,CAAA;SACjB;aAAM;YACL,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;SACjB;QACD,MAAM,IAAA,YAAI,EAAC,EAAE,CAAC,CAAA;QACd,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,EAAE,EAAE;gBACnE,IAAI,YAAY,EAAE;oBAChB,OAAO,CAAC,IAAI,CAAC,0CAA0C,EAAE,IAAI,CAAC,IAAI,CAAC,CAAA;oBACnE,8CAA8C;oBAC9C,IAAI,CAAC,YAAY,GAAG,YAAY,CAAA;oBAChC,OAAO,CAAC,IAAI,CAAC,CAAA;iBACd;qBAAM;oBACL,OAAO,CAAC,IAAI,CAAC,sCAAsC,EAAE,IAAI,CAAC,IAAI,CAAC,CAAA;oBAC/D,MAAM,CAAC,IAAI,KAAK,CAAC,qCAAqC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;iBACrE;YACH,CAAC,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC;IAED,KAAK,CAAC,OAAO;QACX,OAAO,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAA;QAC5C,IAAI,IAAI,CAAC,YAAY,EAAE;YACrB,aAAG,CAAC,sBAAsB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;YAC7C,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;SACzB;QACD,IAAI,CAAC,YAAY,GAAG,IAAI,CAAA;QACxB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAA;IACpB,CAAC;CACF;AAtID,kCAsIC"}