@based/server 3.4.6 → 3.4.7
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/api/index.d.ts +7 -0
- package/dist/api/index.js +173 -0
- package/dist/api/index.js.map +1 -0
- package/dist/auth/dummyAuth.d.ts +2 -0
- package/dist/auth/dummyAuth.js +8 -0
- package/dist/auth/dummyAuth.js.map +1 -0
- package/dist/auth/index.d.ts +15 -0
- package/dist/auth/index.js +49 -0
- package/dist/auth/index.js.map +1 -0
- package/dist/auth/parseAuthState.d.ts +2 -0
- package/dist/auth/parseAuthState.js +16 -0
- package/dist/auth/parseAuthState.js.map +1 -0
- package/dist/auth/types.d.ts +10 -0
- package/dist/auth/types.js +3 -0
- package/dist/auth/types.js.map +1 -0
- package/dist/compress.d.ts +5 -0
- package/dist/compress.js +52 -0
- package/dist/compress.js.map +1 -0
- package/dist/context.d.ts +54 -0
- package/dist/context.js +18 -0
- package/dist/context.js.map +1 -0
- package/dist/error/errorTypeHandlers.d.ts +6 -0
- package/dist/error/errorTypeHandlers.js +130 -0
- package/dist/error/errorTypeHandlers.js.map +1 -0
- package/dist/error/index.d.ts +5 -0
- package/dist/error/index.js +61 -0
- package/dist/error/index.js.map +1 -0
- package/dist/error/types.d.ts +105 -0
- package/dist/error/types.js +30 -0
- package/dist/error/types.js.map +1 -0
- package/dist/functions/index.d.ts +34 -0
- package/dist/functions/index.js +210 -0
- package/dist/functions/index.js.map +1 -0
- package/dist/functions/timeout.d.ts +3 -0
- package/dist/functions/timeout.js +23 -0
- package/dist/functions/timeout.js.map +1 -0
- package/dist/functions/types.d.ts +69 -0
- package/dist/functions/types.js +15 -0
- package/dist/functions/types.js.map +1 -0
- package/dist/handlers/copy/index.js +1 -1
- package/dist/handlers/copy/index.js.map +1 -1
- package/dist/incoming/http/authorize.d.ts +4 -0
- package/dist/incoming/http/authorize.js +37 -0
- package/dist/incoming/http/authorize.js.map +1 -0
- package/dist/incoming/http/function.d.ts +4 -0
- package/dist/incoming/http/function.js +47 -0
- package/dist/incoming/http/function.js.map +1 -0
- package/dist/incoming/http/get.d.ts +4 -0
- package/dist/incoming/http/get.js +155 -0
- package/dist/incoming/http/get.js.map +1 -0
- package/dist/incoming/http/index.d.ts +3 -0
- package/dist/incoming/http/index.js +134 -0
- package/dist/incoming/http/index.js.map +1 -0
- package/dist/incoming/http/parseQuery.d.ts +4 -0
- package/dist/incoming/http/parseQuery.js +25 -0
- package/dist/incoming/http/parseQuery.js.map +1 -0
- package/dist/incoming/http/readBody.d.ts +4 -0
- package/dist/incoming/http/readBody.js +107 -0
- package/dist/incoming/http/readBody.js.map +1 -0
- package/dist/incoming/http/streamFunction/DataStream.d.ts +7 -0
- package/dist/incoming/http/streamFunction/DataStream.js +19 -0
- package/dist/incoming/http/streamFunction/DataStream.js.map +1 -0
- package/dist/incoming/http/streamFunction/getExtension.d.ts +2 -0
- package/dist/incoming/http/streamFunction/getExtension.js +29 -0
- package/dist/incoming/http/streamFunction/getExtension.js.map +1 -0
- package/dist/incoming/http/streamFunction/index.d.ts +4 -0
- package/dist/incoming/http/streamFunction/index.js +106 -0
- package/dist/incoming/http/streamFunction/index.js.map +1 -0
- package/dist/incoming/http/streamFunction/multipartStream.d.ts +13 -0
- package/dist/incoming/http/streamFunction/multipartStream.js +208 -0
- package/dist/incoming/http/streamFunction/multipartStream.js.map +1 -0
- package/dist/incoming/http/streamFunction/stream.d.ts +6 -0
- package/dist/incoming/http/streamFunction/stream.js +108 -0
- package/dist/incoming/http/streamFunction/stream.js.map +1 -0
- package/dist/incoming/index.d.ts +3 -0
- package/dist/incoming/index.js +84 -0
- package/dist/incoming/index.js.map +1 -0
- package/dist/incoming/upgrade.d.ts +4 -0
- package/dist/incoming/upgrade.js +64 -0
- package/dist/incoming/upgrade.js.map +1 -0
- package/dist/incoming/ws/auth.d.ts +5 -0
- package/dist/incoming/ws/auth.js +40 -0
- package/dist/incoming/ws/auth.js.map +1 -0
- package/dist/incoming/ws/function.d.ts +3 -0
- package/dist/incoming/ws/function.js +107 -0
- package/dist/incoming/ws/function.js.map +1 -0
- package/dist/incoming/ws/get.d.ts +3 -0
- package/dist/incoming/ws/get.js +136 -0
- package/dist/incoming/ws/get.js.map +1 -0
- package/dist/incoming/ws/index.d.ts +3 -0
- package/dist/incoming/ws/index.js +61 -0
- package/dist/incoming/ws/index.js.map +1 -0
- package/dist/incoming/ws/observable.d.ts +6 -0
- package/dist/incoming/ws/observable.js +112 -0
- package/dist/incoming/ws/observable.js.map +1 -0
- package/dist/ip.d.ts +2 -0
- package/dist/ip.js +8 -0
- package/dist/ip.js.map +1 -0
- package/dist/observable/create.d.ts +3 -0
- package/dist/observable/create.js +30 -0
- package/dist/observable/create.js.map +1 -0
- package/dist/observable/destroy.d.ts +2 -0
- package/dist/observable/destroy.js +51 -0
- package/dist/observable/destroy.js.map +1 -0
- package/dist/observable/error.d.ts +4 -0
- package/dist/observable/error.js +17 -0
- package/dist/observable/error.js.map +1 -0
- package/dist/observable/extendCache.d.ts +2 -0
- package/dist/observable/extendCache.js +11 -0
- package/dist/observable/extendCache.js.map +1 -0
- package/dist/observable/genObservableId.d.ts +1 -0
- package/dist/observable/genObservableId.js +9 -0
- package/dist/observable/genObservableId.js.map +1 -0
- package/dist/observable/get.d.ts +4 -0
- package/dist/observable/get.js +15 -0
- package/dist/observable/get.js.map +1 -0
- package/dist/observable/index.d.ts +11 -0
- package/dist/observable/index.js +28 -0
- package/dist/observable/index.js.map +1 -0
- package/dist/observable/send.d.ts +6 -0
- package/dist/observable/send.js +32 -0
- package/dist/observable/send.js.map +1 -0
- package/dist/observable/start/error.d.ts +3 -0
- package/dist/observable/start/error.js +36 -0
- package/dist/observable/start/error.js.map +1 -0
- package/dist/observable/start/index.d.ts +2 -0
- package/dist/observable/start/index.js +46 -0
- package/dist/observable/start/index.js.map +1 -0
- package/dist/observable/start/update.d.ts +6 -0
- package/dist/observable/start/update.js +104 -0
- package/dist/observable/start/update.js.map +1 -0
- package/dist/observable/subscribe.d.ts +7 -0
- package/dist/observable/subscribe.js +54 -0
- package/dist/observable/subscribe.js.map +1 -0
- package/dist/observable/types.d.ts +26 -0
- package/dist/observable/types.js +3 -0
- package/dist/observable/types.js.map +1 -0
- package/dist/observable/unsub.d.ts +6 -0
- package/dist/observable/unsub.js +43 -0
- package/dist/observable/unsub.js.map +1 -0
- package/dist/observable/verify.d.ts +4 -0
- package/dist/observable/verify.js +18 -0
- package/dist/observable/verify.js.map +1 -0
- package/dist/protocol.d.ts +21 -0
- package/dist/protocol.js +233 -0
- package/dist/protocol.js.map +1 -0
- package/dist/security.d.ts +6 -0
- package/dist/security.js +95 -0
- package/dist/security.js.map +1 -0
- package/dist/sendError.d.ts +5 -0
- package/dist/sendError.js +42 -0
- package/dist/sendError.js.map +1 -0
- package/dist/sendHttpResponse.d.ts +4 -0
- package/dist/sendHttpResponse.js +53 -0
- package/dist/sendHttpResponse.js.map +1 -0
- package/dist/server.d.ts +69 -0
- package/dist/server.js +113 -0
- package/dist/server.js.map +1 -0
- package/package.json +1 -1
|
@@ -0,0 +1,106 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.httpStreamFunction = void 0;
|
|
7
|
+
const stream_1 = __importDefault(require("./stream"));
|
|
8
|
+
const sendError_1 = require("../../../sendError");
|
|
9
|
+
const functions_1 = require("../../../functions");
|
|
10
|
+
const authorize_1 = require("../authorize");
|
|
11
|
+
const error_1 = require("../../../error");
|
|
12
|
+
const multipartStream_1 = __importDefault(require("./multipartStream"));
|
|
13
|
+
const sendHttpResponse_1 = require("../../../sendHttpResponse");
|
|
14
|
+
const httpStreamFunction = (server, ctx, payload, route) => {
|
|
15
|
+
if (!ctx.session) {
|
|
16
|
+
return;
|
|
17
|
+
}
|
|
18
|
+
const size = ctx.session.headers['content-length'];
|
|
19
|
+
if (route.maxPayloadSize > -1 && route.maxPayloadSize < size) {
|
|
20
|
+
(0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.PayloadTooLarge, route);
|
|
21
|
+
return;
|
|
22
|
+
}
|
|
23
|
+
const type = ctx.session.headers['content-type'];
|
|
24
|
+
// replace this with transder encoding 'chunked'
|
|
25
|
+
if (type && type.startsWith('multipart/form-data')) {
|
|
26
|
+
const files = [];
|
|
27
|
+
let thisIsFn;
|
|
28
|
+
(0, multipartStream_1.default)(ctx, server, payload, route, (p) => {
|
|
29
|
+
return new Promise((resolve) => {
|
|
30
|
+
(0, authorize_1.authorizeRequest)(server, ctx, p.payload, route, () => {
|
|
31
|
+
if (!thisIsFn) {
|
|
32
|
+
files.push({ p, resolve });
|
|
33
|
+
}
|
|
34
|
+
else {
|
|
35
|
+
resolve(thisIsFn(p, ctx));
|
|
36
|
+
}
|
|
37
|
+
}, () => {
|
|
38
|
+
resolve(undefined);
|
|
39
|
+
});
|
|
40
|
+
});
|
|
41
|
+
});
|
|
42
|
+
server.functions
|
|
43
|
+
.install(route.name)
|
|
44
|
+
.then((spec) => {
|
|
45
|
+
if (spec && !(0, functions_1.isObservableFunctionSpec)(spec) && spec.stream) {
|
|
46
|
+
thisIsFn = spec.function;
|
|
47
|
+
if (files.length) {
|
|
48
|
+
for (const file of files) {
|
|
49
|
+
console.info('File parsed before fn / auth');
|
|
50
|
+
file.resolve(thisIsFn(file.p, ctx));
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
else {
|
|
55
|
+
(0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.FunctionNotFound, route);
|
|
56
|
+
}
|
|
57
|
+
})
|
|
58
|
+
.catch(() => {
|
|
59
|
+
(0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.FunctionNotFound, route);
|
|
60
|
+
});
|
|
61
|
+
return;
|
|
62
|
+
}
|
|
63
|
+
const stream = (0, stream_1.default)(server, route, ctx, size);
|
|
64
|
+
// destroy stream from context
|
|
65
|
+
(0, authorize_1.authorizeRequest)(server, ctx, payload, route, (payload) => {
|
|
66
|
+
server.functions
|
|
67
|
+
.install(route.name)
|
|
68
|
+
.then((spec) => {
|
|
69
|
+
if (spec && !(0, functions_1.isObservableFunctionSpec)(spec) && spec.stream) {
|
|
70
|
+
// const stream = createDataStream(server, route, client, size)
|
|
71
|
+
const streamPayload = { payload, stream };
|
|
72
|
+
const fn = spec.function;
|
|
73
|
+
fn(streamPayload, ctx)
|
|
74
|
+
.catch((err) => {
|
|
75
|
+
stream.destroy();
|
|
76
|
+
(0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.FunctionError, {
|
|
77
|
+
err,
|
|
78
|
+
route,
|
|
79
|
+
});
|
|
80
|
+
})
|
|
81
|
+
.then((r) => {
|
|
82
|
+
if (stream.readableEnded ||
|
|
83
|
+
stream.listenerCount('data') === 0) {
|
|
84
|
+
(0, sendHttpResponse_1.sendHttpResponse)(ctx, r);
|
|
85
|
+
}
|
|
86
|
+
else {
|
|
87
|
+
stream.on('end', () => {
|
|
88
|
+
(0, sendHttpResponse_1.sendHttpResponse)(ctx, r);
|
|
89
|
+
});
|
|
90
|
+
}
|
|
91
|
+
});
|
|
92
|
+
}
|
|
93
|
+
else {
|
|
94
|
+
(0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.FunctionNotFound, route);
|
|
95
|
+
}
|
|
96
|
+
})
|
|
97
|
+
.catch((err) => {
|
|
98
|
+
console.error(err);
|
|
99
|
+
(0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.FunctionNotFound, route);
|
|
100
|
+
});
|
|
101
|
+
}, () => {
|
|
102
|
+
stream.destroy();
|
|
103
|
+
});
|
|
104
|
+
};
|
|
105
|
+
exports.httpStreamFunction = httpStreamFunction;
|
|
106
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/incoming/http/streamFunction/index.ts"],"names":[],"mappings":";;;;;;AAAA,sDAAuC;AAEvC,kDAA8C;AAC9C,kDAG2B;AAE3B,4CAA+C;AAC/C,0CAA+C;AAC/C,wEAA+C;AAC/C,gEAA4D;AAErD,MAAM,kBAAkB,GAAG,CAChC,MAAmB,EACnB,GAAyB,EACzB,OAAY,EACZ,KAAyB,EACzB,EAAE;IACF,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE;QAChB,OAAM;KACP;IAED,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAA;IAElD,IAAI,KAAK,CAAC,cAAc,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,cAAc,GAAG,IAAI,EAAE;QAC5D,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,eAAe,EAAE,KAAK,CAAC,CAAA;QAC7D,OAAM;KACP;IAED,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,cAAc,CAAC,CAAA;IAEhD,gDAAgD;IAChD,IAAI,IAAI,IAAI,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,EAAE;QAClD,MAAM,KAAK,GAAU,EAAE,CAAA;QACvB,IAAI,QAAkB,CAAA;QAEtB,IAAA,yBAAe,EAAC,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,EAAE;YACjD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC7B,IAAA,4BAAgB,EACd,MAAM,EACN,GAAG,EACH,CAAC,CAAC,OAAO,EACT,KAAK,EACL,GAAG,EAAE;oBACH,IAAI,CAAC,QAAQ,EAAE;wBACb,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,CAAA;qBAC3B;yBAAM;wBACL,OAAO,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAA;qBAC1B;gBACH,CAAC,EACD,GAAG,EAAE;oBACH,OAAO,CAAC,SAAS,CAAC,CAAA;gBACpB,CAAC,CACF,CAAA;YACH,CAAC,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,MAAM,CAAC,SAAS;aACb,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC;aACnB,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;YACb,IAAI,IAAI,IAAI,CAAC,IAAA,oCAAwB,EAAC,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE;gBAC1D,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAA;gBACxB,IAAI,KAAK,CAAC,MAAM,EAAE;oBAChB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;wBACxB,OAAO,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAA;wBAC5C,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAA;qBACpC;iBACF;aACF;iBAAM;gBACL,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAA;aAC/D;QACH,CAAC,CAAC;aACD,KAAK,CAAC,GAAG,EAAE;YACV,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAA;QAChE,CAAC,CAAC,CAAA;QAEJ,OAAM;KACP;IAED,MAAM,MAAM,GAAG,IAAA,gBAAgB,EAAC,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,CAAC,CAAA;IAEzD,8BAA8B;IAC9B,IAAA,4BAAgB,EACd,MAAM,EACN,GAAG,EACH,OAAO,EACP,KAAK,EACL,CAAC,OAAO,EAAE,EAAE;QACV,MAAM,CAAC,SAAS;aACb,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC;aACnB,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE;YACb,IAAI,IAAI,IAAI,CAAC,IAAA,oCAAwB,EAAC,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE;gBAC1D,+DAA+D;gBAC/D,MAAM,aAAa,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,CAAA;gBAEzC,MAAM,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAA;gBAExB,EAAE,CAAC,aAAa,EAAE,GAAG,CAAC;qBACnB,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;oBACb,MAAM,CAAC,OAAO,EAAE,CAAA;oBAChB,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,aAAa,EAAE;wBACnD,GAAG;wBACH,KAAK;qBACN,CAAC,CAAA;gBACJ,CAAC,CAAC;qBACD,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE;oBACV,IACE,MAAM,CAAC,aAAa;wBACpB,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,EAClC;wBACA,IAAA,mCAAgB,EAAC,GAAG,EAAE,CAAC,CAAC,CAAA;qBACzB;yBAAM;wBACL,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;4BACpB,IAAA,mCAAgB,EAAC,GAAG,EAAE,CAAC,CAAC,CAAA;wBAC1B,CAAC,CAAC,CAAA;qBACH;gBACH,CAAC,CAAC,CAAA;aACL;iBAAM;gBACL,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAA;aAC/D;QACH,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YACb,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;YAClB,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAA;QAChE,CAAC,CAAC,CAAA;IACN,CAAC,EACD,GAAG,EAAE;QACH,MAAM,CAAC,OAAO,EAAE,CAAA;IAClB,CAAC,CACF,CAAA;AACH,CAAC,CAAA;AAtHY,QAAA,kBAAkB,sBAsH9B"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import { HttpSession, Context } from '../../../context';
|
|
2
|
+
import { BasedServer } from '../../../server';
|
|
3
|
+
import { BasedFunctionRoute } from '../../../functions';
|
|
4
|
+
export declare type FileOptions = {
|
|
5
|
+
name?: string;
|
|
6
|
+
size?: number;
|
|
7
|
+
type: string;
|
|
8
|
+
extension: string;
|
|
9
|
+
} & {
|
|
10
|
+
[key: string]: string;
|
|
11
|
+
};
|
|
12
|
+
declare const _default: (ctx: Context<HttpSession>, server: BasedServer, payload: any, route: BasedFunctionRoute, fn: (payload: any) => Promise<any>) => void;
|
|
13
|
+
export default _default;
|
|
@@ -0,0 +1,208 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
const DataStream_1 = require("./DataStream");
|
|
7
|
+
const error_1 = require("../../../error");
|
|
8
|
+
const sendError_1 = require("../../../sendError");
|
|
9
|
+
const sendHttpResponse_1 = require("../../../sendHttpResponse");
|
|
10
|
+
const getExtension_1 = __importDefault(require("./getExtension"));
|
|
11
|
+
const MAX_CHUNK_SIZE = 1024 * 1024 * 5;
|
|
12
|
+
const streamProgress = (stream, size) => {
|
|
13
|
+
stream.emit('progress', 0);
|
|
14
|
+
if (size < 200000) {
|
|
15
|
+
stream.on('end', () => {
|
|
16
|
+
stream.emit('progress', 1);
|
|
17
|
+
});
|
|
18
|
+
}
|
|
19
|
+
else {
|
|
20
|
+
let progress = 0;
|
|
21
|
+
let total = 0;
|
|
22
|
+
let setInProgress = false;
|
|
23
|
+
const updateProgress = () => {
|
|
24
|
+
if (!setInProgress) {
|
|
25
|
+
setInProgress = true;
|
|
26
|
+
setTimeout(() => {
|
|
27
|
+
stream.emit('progress', progress);
|
|
28
|
+
setInProgress = false;
|
|
29
|
+
}, 250);
|
|
30
|
+
}
|
|
31
|
+
};
|
|
32
|
+
stream.on('end', () => {
|
|
33
|
+
progress = 1;
|
|
34
|
+
updateProgress();
|
|
35
|
+
});
|
|
36
|
+
stream.on('data', (chunk) => {
|
|
37
|
+
total += chunk.byteLength;
|
|
38
|
+
progress = total / size;
|
|
39
|
+
updateProgress();
|
|
40
|
+
});
|
|
41
|
+
}
|
|
42
|
+
};
|
|
43
|
+
// only use this if you have individual file else its just all
|
|
44
|
+
const setHeader = (file) => {
|
|
45
|
+
file.headersSet++;
|
|
46
|
+
if (file.headersSet === 2) {
|
|
47
|
+
return true;
|
|
48
|
+
}
|
|
49
|
+
return false;
|
|
50
|
+
};
|
|
51
|
+
const toBuffer = (str, firstWritten) => {
|
|
52
|
+
return Buffer.from(firstWritten ? str + '\r\n' : str, 'binary');
|
|
53
|
+
};
|
|
54
|
+
exports.default = (ctx, server, payload, route, fn) => {
|
|
55
|
+
if (!payload || (!payload && typeof payload !== 'object')) {
|
|
56
|
+
payload = {};
|
|
57
|
+
}
|
|
58
|
+
const files = [];
|
|
59
|
+
const contentLength = ctx.session.headers['content-length'];
|
|
60
|
+
const promiseQ = [];
|
|
61
|
+
let setInProgress = false;
|
|
62
|
+
let boundary = null;
|
|
63
|
+
let prevLine;
|
|
64
|
+
let isWriting = false;
|
|
65
|
+
let total = 0;
|
|
66
|
+
let progress = 0;
|
|
67
|
+
ctx.session.res.onData((chunk, isLast) => {
|
|
68
|
+
// see if this goes ok... (clearing mem etc)
|
|
69
|
+
if (chunk.byteLength > MAX_CHUNK_SIZE) {
|
|
70
|
+
(0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.ChunkTooLarge, route);
|
|
71
|
+
for (const file of files) {
|
|
72
|
+
file.stream.destroy();
|
|
73
|
+
}
|
|
74
|
+
return;
|
|
75
|
+
}
|
|
76
|
+
let firstWritten = false;
|
|
77
|
+
const blocks = Buffer.from(chunk).toString('binary').split('\r\n');
|
|
78
|
+
total += chunk.byteLength;
|
|
79
|
+
progress = total / contentLength;
|
|
80
|
+
for (const file of files) {
|
|
81
|
+
if (file.headersSet > 1 && !file.isDone && !file.opts.size) {
|
|
82
|
+
if (contentLength > 200000) {
|
|
83
|
+
if (!setInProgress) {
|
|
84
|
+
setInProgress = true;
|
|
85
|
+
setTimeout(() => {
|
|
86
|
+
for (const file of files) {
|
|
87
|
+
if (file.headersSet > 1 && !file.isDone && !file.opts.size) {
|
|
88
|
+
if (progress === 1) {
|
|
89
|
+
file.isDone = true;
|
|
90
|
+
}
|
|
91
|
+
file.stream.emit('progress', progress);
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
setInProgress = false;
|
|
95
|
+
}, 250);
|
|
96
|
+
}
|
|
97
|
+
break;
|
|
98
|
+
}
|
|
99
|
+
}
|
|
100
|
+
}
|
|
101
|
+
if (!boundary) {
|
|
102
|
+
boundary = blocks[0];
|
|
103
|
+
}
|
|
104
|
+
for (let i = 0; i < blocks.length; i++) {
|
|
105
|
+
const line = blocks[i];
|
|
106
|
+
if (!boundary) {
|
|
107
|
+
continue;
|
|
108
|
+
}
|
|
109
|
+
if (isWriting && (line === boundary || line === boundary + '--')) {
|
|
110
|
+
isWriting = false;
|
|
111
|
+
const file = files[files.length - 1];
|
|
112
|
+
if (prevLine) {
|
|
113
|
+
file.stream.end(toBuffer(prevLine, firstWritten));
|
|
114
|
+
}
|
|
115
|
+
else {
|
|
116
|
+
file.stream.end();
|
|
117
|
+
}
|
|
118
|
+
if (!file.opts.size && !file.isDone) {
|
|
119
|
+
file.stream.emit('progress', 1);
|
|
120
|
+
}
|
|
121
|
+
file.isDone = true;
|
|
122
|
+
prevLine = null;
|
|
123
|
+
if (line === boundary + '--') {
|
|
124
|
+
continue;
|
|
125
|
+
}
|
|
126
|
+
}
|
|
127
|
+
if (line === boundary && !isWriting) {
|
|
128
|
+
const file = {
|
|
129
|
+
stream: new DataStream_1.DataStream(),
|
|
130
|
+
headersSet: 0,
|
|
131
|
+
opts: {},
|
|
132
|
+
isDone: false,
|
|
133
|
+
};
|
|
134
|
+
files.push(file);
|
|
135
|
+
continue;
|
|
136
|
+
}
|
|
137
|
+
const file = files[files.length - 1];
|
|
138
|
+
if (!file) {
|
|
139
|
+
// TODO: invalid file
|
|
140
|
+
return (0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.InvalidPayload, route);
|
|
141
|
+
}
|
|
142
|
+
if (!isWriting && line.includes('Content-Disposition')) {
|
|
143
|
+
const meta = line.match(/name="(.*?)"/)?.[1];
|
|
144
|
+
if (!meta) {
|
|
145
|
+
// TODO: invalid file
|
|
146
|
+
return (0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.InvalidPayload, route);
|
|
147
|
+
}
|
|
148
|
+
const opts = file.opts;
|
|
149
|
+
opts.name = line.match(/filename="(.*?)"/)?.[1] || 'untitled';
|
|
150
|
+
const disposition = meta.split('|');
|
|
151
|
+
for (const seg of disposition) {
|
|
152
|
+
if (/=/.test(seg)) {
|
|
153
|
+
const [k, v] = seg.split('=');
|
|
154
|
+
if (k === 'size') {
|
|
155
|
+
opts[k] = Number(v);
|
|
156
|
+
}
|
|
157
|
+
else {
|
|
158
|
+
opts[k] = v;
|
|
159
|
+
}
|
|
160
|
+
}
|
|
161
|
+
}
|
|
162
|
+
if (opts.size) {
|
|
163
|
+
streamProgress(file.stream, opts.size);
|
|
164
|
+
}
|
|
165
|
+
isWriting = setHeader(file);
|
|
166
|
+
if (isWriting) {
|
|
167
|
+
promiseQ.push(fn({ payload: { ...payload, ...file.opts }, stream: file.stream }));
|
|
168
|
+
}
|
|
169
|
+
continue;
|
|
170
|
+
}
|
|
171
|
+
if (!isWriting && line.includes('Content-Type')) {
|
|
172
|
+
const mimeType = line.match(/Content-Type: ([a-zA-Z0-9].+\/[a-zA-Z0-9].+)/)?.[1];
|
|
173
|
+
if (!mimeType) {
|
|
174
|
+
// TODO: invalid file (can speficy in route potentialy...)
|
|
175
|
+
return (0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.InvalidPayload, route);
|
|
176
|
+
}
|
|
177
|
+
file.opts.type = mimeType;
|
|
178
|
+
file.opts.extension = (0, getExtension_1.default)(mimeType);
|
|
179
|
+
isWriting = setHeader(file);
|
|
180
|
+
if (isWriting) {
|
|
181
|
+
promiseQ.push(fn({ payload: { ...payload, ...file.opts }, stream: file.stream }));
|
|
182
|
+
}
|
|
183
|
+
continue;
|
|
184
|
+
}
|
|
185
|
+
if (isWriting) {
|
|
186
|
+
if (prevLine) {
|
|
187
|
+
file.stream.write(toBuffer(prevLine, firstWritten));
|
|
188
|
+
}
|
|
189
|
+
prevLine = line;
|
|
190
|
+
firstWritten = true;
|
|
191
|
+
}
|
|
192
|
+
}
|
|
193
|
+
if (isLast) {
|
|
194
|
+
Promise.allSettled(promiseQ).then((results) => {
|
|
195
|
+
const r = results.map((v) => {
|
|
196
|
+
if (v.status === 'rejected') {
|
|
197
|
+
return { err: v.reason };
|
|
198
|
+
}
|
|
199
|
+
else {
|
|
200
|
+
return v.value;
|
|
201
|
+
}
|
|
202
|
+
});
|
|
203
|
+
(0, sendHttpResponse_1.sendHttpResponse)(ctx, r);
|
|
204
|
+
});
|
|
205
|
+
}
|
|
206
|
+
});
|
|
207
|
+
};
|
|
208
|
+
//# sourceMappingURL=multipartStream.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"multipartStream.js","sourceRoot":"","sources":["../../../../src/incoming/http/streamFunction/multipartStream.ts"],"names":[],"mappings":";;;;;AAAA,6CAAyC;AAEzC,0CAA+C;AAC/C,kDAA8C;AAC9C,gEAA4D;AAC5D,kEAAyC;AAIzC,MAAM,cAAc,GAAG,IAAI,GAAG,IAAI,GAAG,CAAC,CAAA;AAgBtC,MAAM,cAAc,GAAG,CAAC,MAAkB,EAAE,IAAY,EAAE,EAAE;IAC1D,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAA;IAC1B,IAAI,IAAI,GAAG,MAAM,EAAE;QACjB,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;YACpB,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAA;QAC5B,CAAC,CAAC,CAAA;KACH;SAAM;QACL,IAAI,QAAQ,GAAG,CAAC,CAAA;QAChB,IAAI,KAAK,GAAG,CAAC,CAAA;QACb,IAAI,aAAa,GAAG,KAAK,CAAA;QACzB,MAAM,cAAc,GAAG,GAAG,EAAE;YAC1B,IAAI,CAAC,aAAa,EAAE;gBAClB,aAAa,GAAG,IAAI,CAAA;gBACpB,UAAU,CAAC,GAAG,EAAE;oBACd,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;oBACjC,aAAa,GAAG,KAAK,CAAA;gBACvB,CAAC,EAAE,GAAG,CAAC,CAAA;aACR;QACH,CAAC,CAAA;QACD,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;YACpB,QAAQ,GAAG,CAAC,CAAA;YACZ,cAAc,EAAE,CAAA;QAClB,CAAC,CAAC,CAAA;QACF,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;YAC1B,KAAK,IAAI,KAAK,CAAC,UAAU,CAAA;YACzB,QAAQ,GAAG,KAAK,GAAG,IAAI,CAAA;YACvB,cAAc,EAAE,CAAA;QAClB,CAAC,CAAC,CAAA;KACH;AACH,CAAC,CAAA;AAED,8DAA8D;AAC9D,MAAM,SAAS,GAAG,CAAC,IAAoB,EAAW,EAAE;IAClD,IAAI,CAAC,UAAU,EAAE,CAAA;IACjB,IAAI,IAAI,CAAC,UAAU,KAAK,CAAC,EAAE;QACzB,OAAO,IAAI,CAAA;KACZ;IACD,OAAO,KAAK,CAAA;AACd,CAAC,CAAA;AAED,MAAM,QAAQ,GAAG,CAAC,GAAW,EAAE,YAAqB,EAAU,EAAE;IAC9D,OAAO,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,GAAG,GAAG,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAA;AACjE,CAAC,CAAA;AAED,kBAAe,CACb,GAAyB,EACzB,MAAmB,EACnB,OAAY,EACZ,KAAyB,EACzB,EAAkC,EAClC,EAAE;IACF,IAAI,CAAC,OAAO,IAAI,CAAC,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,CAAC,EAAE;QACzD,OAAO,GAAG,EAAE,CAAA;KACb;IAED,MAAM,KAAK,GAAqB,EAAE,CAAA;IAElC,MAAM,aAAa,GAAG,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAA;IAE3D,MAAM,QAAQ,GAAmB,EAAE,CAAA;IAEnC,IAAI,aAAa,GAAG,KAAK,CAAA;IACzB,IAAI,QAAQ,GAAG,IAAI,CAAA;IACnB,IAAI,QAAgB,CAAA;IACpB,IAAI,SAAS,GAAG,KAAK,CAAA;IACrB,IAAI,KAAK,GAAG,CAAC,CAAA;IACb,IAAI,QAAQ,GAAG,CAAC,CAAA;IAEhB,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE;QACvC,4CAA4C;QAC5C,IAAI,KAAK,CAAC,UAAU,GAAG,cAAc,EAAE;YACrC,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,aAAa,EAAE,KAAK,CAAC,CAAA;YAC3D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;gBACxB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAA;aACtB;YACD,OAAM;SACP;QAED,IAAI,YAAY,GAAG,KAAK,CAAA;QACxB,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;QAClE,KAAK,IAAI,KAAK,CAAC,UAAU,CAAA;QAEzB,QAAQ,GAAG,KAAK,GAAG,aAAa,CAAA;QAEhC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;YACxB,IAAI,IAAI,CAAC,UAAU,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;gBAC1D,IAAI,aAAa,GAAG,MAAM,EAAE;oBAC1B,IAAI,CAAC,aAAa,EAAE;wBAClB,aAAa,GAAG,IAAI,CAAA;wBACpB,UAAU,CAAC,GAAG,EAAE;4BACd,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;gCACxB,IAAI,IAAI,CAAC,UAAU,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;oCAC1D,IAAI,QAAQ,KAAK,CAAC,EAAE;wCAClB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAA;qCACnB;oCACD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;iCACvC;6BACF;4BACD,aAAa,GAAG,KAAK,CAAA;wBACvB,CAAC,EAAE,GAAG,CAAC,CAAA;qBACR;oBACD,MAAK;iBACN;aACF;SACF;QAED,IAAI,CAAC,QAAQ,EAAE;YACb,QAAQ,GAAG,MAAM,CAAC,CAAC,CAAC,CAAA;SACrB;QAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YACtC,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,CAAA;YAEtB,IAAI,CAAC,QAAQ,EAAE;gBACb,SAAQ;aACT;YAED,IAAI,SAAS,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,QAAQ,GAAG,IAAI,CAAC,EAAE;gBAChE,SAAS,GAAG,KAAK,CAAA;gBACjB,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;gBACpC,IAAI,QAAQ,EAAE;oBACZ,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAA;iBAClD;qBAAM;oBACL,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAA;iBAClB;gBACD,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;oBACnC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAA;iBAChC;gBACD,IAAI,CAAC,MAAM,GAAG,IAAI,CAAA;gBAElB,QAAQ,GAAG,IAAI,CAAA;gBACf,IAAI,IAAI,KAAK,QAAQ,GAAG,IAAI,EAAE;oBAC5B,SAAQ;iBACT;aACF;YAED,IAAI,IAAI,KAAK,QAAQ,IAAI,CAAC,SAAS,EAAE;gBACnC,MAAM,IAAI,GAAG;oBACX,MAAM,EAAE,IAAI,uBAAU,EAAE;oBACxB,UAAU,EAAE,CAAC;oBACb,IAAI,EAAE,EAAE;oBACR,MAAM,EAAE,KAAK;iBACd,CAAA;gBACD,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;gBAChB,SAAQ;aACT;YAED,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;YAEpC,IAAI,CAAC,IAAI,EAAE;gBACT,qBAAqB;gBACrB,OAAO,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,cAAc,EAAE,KAAK,CAAC,CAAA;aACpE;YAED,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,QAAQ,CAAC,qBAAqB,CAAC,EAAE;gBACtD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;gBAC5C,IAAI,CAAC,IAAI,EAAE;oBACT,qBAAqB;oBACrB,OAAO,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,cAAc,EAAE,KAAK,CAAC,CAAA;iBACpE;gBACD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAA;gBACtB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,UAAU,CAAA;gBAC7D,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;gBACnC,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE;oBAC7B,IAAI,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;wBACjB,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;wBAC7B,IAAI,CAAC,KAAK,MAAM,EAAE;4BAChB,IAAI,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAA;yBACpB;6BAAM;4BACL,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAA;yBACZ;qBACF;iBACF;gBACD,IAAI,IAAI,CAAC,IAAI,EAAE;oBACb,cAAc,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,CAAA;iBACvC;gBACD,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,CAAA;gBAC3B,IAAI,SAAS,EAAE;oBACb,QAAQ,CAAC,IAAI,CACX,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CACnE,CAAA;iBACF;gBACD,SAAQ;aACT;YAED,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE;gBAC/C,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CACzB,8CAA8C,CAC/C,EAAE,CAAC,CAAC,CAAC,CAAA;gBACN,IAAI,CAAC,QAAQ,EAAE;oBACb,0DAA0D;oBAC1D,OAAO,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,cAAc,EAAE,KAAK,CAAC,CAAA;iBACpE;gBACD,IAAI,CAAC,IAAI,CAAC,IAAI,GAAG,QAAQ,CAAA;gBACzB,IAAI,CAAC,IAAI,CAAC,SAAS,GAAG,IAAA,sBAAY,EAAC,QAAQ,CAAC,CAAA;gBAC5C,SAAS,GAAG,SAAS,CAAC,IAAI,CAAC,CAAA;gBAC3B,IAAI,SAAS,EAAE;oBACb,QAAQ,CAAC,IAAI,CACX,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CACnE,CAAA;iBACF;gBACD,SAAQ;aACT;YAED,IAAI,SAAS,EAAE;gBACb,IAAI,QAAQ,EAAE;oBACZ,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC,CAAA;iBACpD;gBACD,QAAQ,GAAG,IAAI,CAAA;gBACf,YAAY,GAAG,IAAI,CAAA;aACpB;SACF;QAED,IAAI,MAAM,EAAE;YACV,OAAO,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC5C,MAAM,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;oBAC1B,IAAI,CAAC,CAAC,MAAM,KAAK,UAAU,EAAE;wBAC3B,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAA;qBACzB;yBAAM;wBACL,OAAO,CAAC,CAAC,KAAK,CAAA;qBACf;gBACH,CAAC,CAAC,CAAA;gBACF,IAAA,mCAAgB,EAAC,GAAG,EAAE,CAAC,CAAC,CAAA;YAC1B,CAAC,CAAC,CAAA;SACH;IACH,CAAC,CAAC,CAAA;AACJ,CAAC,CAAA"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
import { DataStream } from './DataStream';
|
|
2
|
+
import { HttpSession, Context } from '../../../context';
|
|
3
|
+
import { BasedFunctionRoute } from '../../../functions';
|
|
4
|
+
import { BasedServer } from '../../../server';
|
|
5
|
+
declare const _default: (server: BasedServer, route: BasedFunctionRoute, ctx: Context<HttpSession>, size: number) => DataStream;
|
|
6
|
+
export default _default;
|
|
@@ -0,0 +1,108 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
const DataStream_1 = require("./DataStream");
|
|
7
|
+
const sendError_1 = require("../../../sendError");
|
|
8
|
+
const node_zlib_1 = __importDefault(require("node:zlib"));
|
|
9
|
+
const error_1 = require("../../../error");
|
|
10
|
+
const MAX_CHUNK_SIZE = 1024 * 1024 * 5;
|
|
11
|
+
const UNCOMPRESS_OPTS = {
|
|
12
|
+
chunkSize: 1024 * 1024 * 5,
|
|
13
|
+
};
|
|
14
|
+
exports.default = (server, route, ctx, size) => {
|
|
15
|
+
const stream = new DataStream_1.DataStream();
|
|
16
|
+
let total = 0;
|
|
17
|
+
let progress = 0;
|
|
18
|
+
let setInProgress = false;
|
|
19
|
+
stream.emit('progress', progress);
|
|
20
|
+
const emitProgress = size > 200000;
|
|
21
|
+
const contentEncoding = ctx.session.headers['content-encoding'];
|
|
22
|
+
if (contentEncoding) {
|
|
23
|
+
let uncompressStream;
|
|
24
|
+
if (contentEncoding === 'deflate') {
|
|
25
|
+
uncompressStream = node_zlib_1.default.createInflate(UNCOMPRESS_OPTS);
|
|
26
|
+
}
|
|
27
|
+
else if (contentEncoding === 'gzip') {
|
|
28
|
+
uncompressStream = node_zlib_1.default.createGunzip(UNCOMPRESS_OPTS);
|
|
29
|
+
}
|
|
30
|
+
else if (contentEncoding === 'br') {
|
|
31
|
+
uncompressStream = node_zlib_1.default.createBrotliDecompress(UNCOMPRESS_OPTS);
|
|
32
|
+
}
|
|
33
|
+
if (uncompressStream) {
|
|
34
|
+
ctx.session.res.onData((c, isLast) => {
|
|
35
|
+
total += c.byteLength;
|
|
36
|
+
if (c.byteLength > MAX_CHUNK_SIZE) {
|
|
37
|
+
(0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.ChunkTooLarge, route);
|
|
38
|
+
uncompressStream.destroy();
|
|
39
|
+
stream.destroy();
|
|
40
|
+
return;
|
|
41
|
+
}
|
|
42
|
+
const buf = Buffer.alloc(c.byteLength, Buffer.from(c));
|
|
43
|
+
if (emitProgress) {
|
|
44
|
+
progress = total / size;
|
|
45
|
+
if (!setInProgress) {
|
|
46
|
+
setInProgress = true;
|
|
47
|
+
setTimeout(() => {
|
|
48
|
+
stream.emit('progress', progress);
|
|
49
|
+
setInProgress = false;
|
|
50
|
+
}, 250);
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
if (isLast) {
|
|
54
|
+
if (!emitProgress) {
|
|
55
|
+
stream.emit('progress', 1);
|
|
56
|
+
}
|
|
57
|
+
uncompressStream.end(buf);
|
|
58
|
+
}
|
|
59
|
+
else {
|
|
60
|
+
if (!uncompressStream.write(buf)) {
|
|
61
|
+
// console.info('BACKPRESSURE')
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
});
|
|
65
|
+
uncompressStream.on('error', (err) => {
|
|
66
|
+
console.warn('Uncompress error', route, contentEncoding, err);
|
|
67
|
+
(0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.ChunkTooLarge, route);
|
|
68
|
+
uncompressStream.destroy();
|
|
69
|
+
stream.destroy();
|
|
70
|
+
});
|
|
71
|
+
uncompressStream.pipe(stream);
|
|
72
|
+
}
|
|
73
|
+
else {
|
|
74
|
+
(0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.UnsupportedContentEncoding, route);
|
|
75
|
+
}
|
|
76
|
+
}
|
|
77
|
+
else {
|
|
78
|
+
ctx.session.res.onData((c, isLast) => {
|
|
79
|
+
total += c.byteLength;
|
|
80
|
+
if (c.byteLength > MAX_CHUNK_SIZE) {
|
|
81
|
+
(0, sendError_1.sendError)(server, ctx, error_1.BasedErrorCode.ChunkTooLarge, route);
|
|
82
|
+
stream.destroy();
|
|
83
|
+
return;
|
|
84
|
+
}
|
|
85
|
+
if (emitProgress) {
|
|
86
|
+
progress = total / size;
|
|
87
|
+
if (!setInProgress) {
|
|
88
|
+
setInProgress = true;
|
|
89
|
+
setTimeout(() => {
|
|
90
|
+
stream.emit('progress', progress);
|
|
91
|
+
setInProgress = false;
|
|
92
|
+
}, 250);
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
if (isLast) {
|
|
96
|
+
if (!emitProgress) {
|
|
97
|
+
stream.emit('progress', 1);
|
|
98
|
+
}
|
|
99
|
+
stream.end(Buffer.from(c));
|
|
100
|
+
}
|
|
101
|
+
else {
|
|
102
|
+
stream.write(Buffer.from(c));
|
|
103
|
+
}
|
|
104
|
+
});
|
|
105
|
+
}
|
|
106
|
+
return stream;
|
|
107
|
+
};
|
|
108
|
+
//# sourceMappingURL=stream.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../../../src/incoming/http/streamFunction/stream.ts"],"names":[],"mappings":";;;;;AAAA,6CAAyC;AAGzC,kDAA8C;AAC9C,0DAA4B;AAC5B,0CAA+C;AAG/C,MAAM,cAAc,GAAG,IAAI,GAAG,IAAI,GAAG,CAAC,CAAA;AAEtC,MAAM,eAAe,GAAG;IACtB,SAAS,EAAE,IAAI,GAAG,IAAI,GAAG,CAAC;CAC3B,CAAA;AAED,kBAAe,CACb,MAAmB,EACnB,KAAyB,EACzB,GAAyB,EACzB,IAAY,EACA,EAAE;IACd,MAAM,MAAM,GAAG,IAAI,uBAAU,EAAE,CAAA;IAC/B,IAAI,KAAK,GAAG,CAAC,CAAA;IACb,IAAI,QAAQ,GAAG,CAAC,CAAA;IAChB,IAAI,aAAa,GAAG,KAAK,CAAA;IACzB,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;IACjC,MAAM,YAAY,GAAG,IAAI,GAAG,MAAM,CAAA;IAElC,MAAM,eAAe,GAAG,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAA;IAE/D,IAAI,eAAe,EAAE;QACnB,IAAI,gBAAoE,CAAA;QACxE,IAAI,eAAe,KAAK,SAAS,EAAE;YACjC,gBAAgB,GAAG,mBAAI,CAAC,aAAa,CAAC,eAAe,CAAC,CAAA;SACvD;aAAM,IAAI,eAAe,KAAK,MAAM,EAAE;YACrC,gBAAgB,GAAG,mBAAI,CAAC,YAAY,CAAC,eAAe,CAAC,CAAA;SACtD;aAAM,IAAI,eAAe,KAAK,IAAI,EAAE;YACnC,gBAAgB,GAAG,mBAAI,CAAC,sBAAsB,CAAC,eAAe,CAAC,CAAA;SAChE;QACD,IAAI,gBAAgB,EAAE;YACpB,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gBACnC,KAAK,IAAI,CAAC,CAAC,UAAU,CAAA;gBACrB,IAAI,CAAC,CAAC,UAAU,GAAG,cAAc,EAAE;oBACjC,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,aAAa,EAAE,KAAK,CAAC,CAAA;oBAC3D,gBAAgB,CAAC,OAAO,EAAE,CAAA;oBAC1B,MAAM,CAAC,OAAO,EAAE,CAAA;oBAChB,OAAM;iBACP;gBAED,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAA;gBAEtD,IAAI,YAAY,EAAE;oBAChB,QAAQ,GAAG,KAAK,GAAG,IAAI,CAAA;oBACvB,IAAI,CAAC,aAAa,EAAE;wBAClB,aAAa,GAAG,IAAI,CAAA;wBACpB,UAAU,CAAC,GAAG,EAAE;4BACd,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;4BACjC,aAAa,GAAG,KAAK,CAAA;wBACvB,CAAC,EAAE,GAAG,CAAC,CAAA;qBACR;iBACF;gBAED,IAAI,MAAM,EAAE;oBACV,IAAI,CAAC,YAAY,EAAE;wBACjB,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAA;qBAC3B;oBACD,gBAAgB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;iBAC1B;qBAAM;oBACL,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE;wBAChC,+BAA+B;qBAChC;iBACF;YACH,CAAC,CAAC,CAAA;YACF,gBAAgB,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;gBACnC,OAAO,CAAC,IAAI,CAAC,kBAAkB,EAAE,KAAK,EAAE,eAAe,EAAE,GAAG,CAAC,CAAA;gBAC7D,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,aAAa,EAAE,KAAK,CAAC,CAAA;gBAC3D,gBAAgB,CAAC,OAAO,EAAE,CAAA;gBAC1B,MAAM,CAAC,OAAO,EAAE,CAAA;YAClB,CAAC,CAAC,CAAA;YACF,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;SAC9B;aAAM;YACL,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAA;SACzE;KACF;SAAM;QACL,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;YACnC,KAAK,IAAI,CAAC,CAAC,UAAU,CAAA;YACrB,IAAI,CAAC,CAAC,UAAU,GAAG,cAAc,EAAE;gBACjC,IAAA,qBAAS,EAAC,MAAM,EAAE,GAAG,EAAE,sBAAc,CAAC,aAAa,EAAE,KAAK,CAAC,CAAA;gBAC3D,MAAM,CAAC,OAAO,EAAE,CAAA;gBAChB,OAAM;aACP;YACD,IAAI,YAAY,EAAE;gBAChB,QAAQ,GAAG,KAAK,GAAG,IAAI,CAAA;gBACvB,IAAI,CAAC,aAAa,EAAE;oBAClB,aAAa,GAAG,IAAI,CAAA;oBACpB,UAAU,CAAC,GAAG,EAAE;wBACd,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAA;wBACjC,aAAa,GAAG,KAAK,CAAA;oBACvB,CAAC,EAAE,GAAG,CAAC,CAAA;iBACR;aACF;YACD,IAAI,MAAM,EAAE;gBACV,IAAI,CAAC,YAAY,EAAE;oBACjB,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,CAAA;iBAC3B;gBACD,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAA;aAC3B;iBAAM;gBACL,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAA;aAC7B;QACH,CAAC,CAAC,CAAA;KACH;IAED,OAAO,MAAM,CAAA;AACf,CAAC,CAAA"}
|
|
@@ -0,0 +1,84 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
const uws_1 = __importDefault(require("@based/uws"));
|
|
7
|
+
const upgrade_1 = require("./upgrade");
|
|
8
|
+
const ws_1 = require("./ws");
|
|
9
|
+
const observable_1 = require("../observable");
|
|
10
|
+
const http_1 = require("./http");
|
|
11
|
+
const auth_1 = require("./ws/auth");
|
|
12
|
+
exports.default = (server, { key, cert, port, ws: wsListeners }) => {
|
|
13
|
+
const app = key && cert
|
|
14
|
+
? uws_1.default.SSLApp({
|
|
15
|
+
key_file_name: key,
|
|
16
|
+
cert_file_name: cert,
|
|
17
|
+
ssl_prefer_low_memory_usage: true,
|
|
18
|
+
})
|
|
19
|
+
: uws_1.default.App();
|
|
20
|
+
if (port) {
|
|
21
|
+
server.port = port;
|
|
22
|
+
}
|
|
23
|
+
if (!wsListeners) {
|
|
24
|
+
wsListeners = {
|
|
25
|
+
open: () => undefined,
|
|
26
|
+
close: () => undefined,
|
|
27
|
+
};
|
|
28
|
+
}
|
|
29
|
+
app
|
|
30
|
+
.ws('/*', {
|
|
31
|
+
maxPayloadLength: 1024 * 1024 * 10,
|
|
32
|
+
idleTimeout: 100,
|
|
33
|
+
maxBackpressure: 1024,
|
|
34
|
+
// compression: uws.SHARED_COMPRESSOR,
|
|
35
|
+
upgrade: server.auth?.authorizeConnection
|
|
36
|
+
? (res, req, ctx) => {
|
|
37
|
+
(0, upgrade_1.upgradeAuthorize)(server, res, req, ctx);
|
|
38
|
+
}
|
|
39
|
+
: (res, req, ctx) => {
|
|
40
|
+
(0, upgrade_1.upgrade)(server, res, req, ctx);
|
|
41
|
+
},
|
|
42
|
+
message: (ws, data, isBinary) => {
|
|
43
|
+
(0, ws_1.message)(server, ws.c, data, isBinary);
|
|
44
|
+
},
|
|
45
|
+
open: (ws) => {
|
|
46
|
+
if (ws) {
|
|
47
|
+
const ctx = {
|
|
48
|
+
session: ws,
|
|
49
|
+
};
|
|
50
|
+
ws.c = ctx;
|
|
51
|
+
wsListeners.open(ctx);
|
|
52
|
+
if (ctx.session.authState) {
|
|
53
|
+
(0, auth_1.sendAuthMessage)(ctx);
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
},
|
|
57
|
+
close: (ws) => {
|
|
58
|
+
console.info('CLOSE!');
|
|
59
|
+
// cl--
|
|
60
|
+
ws.obs.forEach((id) => {
|
|
61
|
+
(0, observable_1.unsubscribeWsIgnoreClient)(server, id, ws.c);
|
|
62
|
+
});
|
|
63
|
+
wsListeners.close(ws.c);
|
|
64
|
+
// Looks really ugly but same impact on memory and GC as using the ws directly
|
|
65
|
+
// and better for dc's when functions etc are in progress
|
|
66
|
+
ws.c.session = null;
|
|
67
|
+
ws.c = null;
|
|
68
|
+
},
|
|
69
|
+
drain: () => {
|
|
70
|
+
console.info('drain');
|
|
71
|
+
// lets handle drain efficiently (or more efficiently at least)
|
|
72
|
+
// call client.drain can be much more efficient
|
|
73
|
+
// if (ws.client && ws.client.backpressureQueue) {
|
|
74
|
+
// ws.client.drain()
|
|
75
|
+
// }
|
|
76
|
+
},
|
|
77
|
+
})
|
|
78
|
+
// REST
|
|
79
|
+
.get('/*', (res, req) => (0, http_1.httpHandler)(server, req, res))
|
|
80
|
+
.post('/*', (res, req) => (0, http_1.httpHandler)(server, req, res))
|
|
81
|
+
.options('/*', (res, req) => (0, http_1.httpHandler)(server, req, res));
|
|
82
|
+
server.uwsApp = app;
|
|
83
|
+
};
|
|
84
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/incoming/index.ts"],"names":[],"mappings":";;;;;AACA,qDAA4B;AAC5B,uCAAqD;AACrD,6BAA8B;AAC9B,8CAAyD;AACzD,iCAAoC;AAEpC,oCAA2C;AAE3C,kBAAe,CACb,MAAmB,EACnB,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,EAAE,WAAW,EAAiB,EACnD,EAAE;IACF,MAAM,GAAG,GACP,GAAG,IAAI,IAAI;QACT,CAAC,CAAC,aAAG,CAAC,MAAM,CAAC;YACT,aAAa,EAAE,GAAG;YAClB,cAAc,EAAE,IAAI;YACpB,2BAA2B,EAAE,IAAI;SAClC,CAAC;QACJ,CAAC,CAAC,aAAG,CAAC,GAAG,EAAE,CAAA;IAEf,IAAI,IAAI,EAAE;QACR,MAAM,CAAC,IAAI,GAAG,IAAI,CAAA;KACnB;IAED,IAAI,CAAC,WAAW,EAAE;QAChB,WAAW,GAAG;YACZ,IAAI,EAAE,GAAG,EAAE,CAAC,SAAS;YACrB,KAAK,EAAE,GAAG,EAAE,CAAC,SAAS;SACvB,CAAA;KACF;IAED,GAAG;SACA,EAAE,CAAC,IAAI,EAAE;QACR,gBAAgB,EAAE,IAAI,GAAG,IAAI,GAAG,EAAE;QAClC,WAAW,EAAE,GAAG;QAChB,eAAe,EAAE,IAAI;QACrB,sCAAsC;QACtC,OAAO,EAAE,MAAM,CAAC,IAAI,EAAE,mBAAmB;YACvC,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;gBAChB,IAAA,0BAAgB,EAAC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;YACzC,CAAC;YACH,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE;gBAChB,IAAA,iBAAO,EAAC,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAA;YAChC,CAAC;QACL,OAAO,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;YAC9B,IAAA,YAAO,EAAC,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAA;QACvC,CAAC;QACD,IAAI,EAAE,CAAC,EAAoB,EAAE,EAAE;YAC7B,IAAI,EAAE,EAAE;gBACN,MAAM,GAAG,GAA8B;oBACrC,OAAO,EAAE,EAAE;iBACZ,CAAA;gBACD,EAAE,CAAC,CAAC,GAAG,GAAG,CAAA;gBACV,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;gBACrB,IAAI,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE;oBACzB,IAAA,sBAAe,EAAC,GAAG,CAAC,CAAA;iBACrB;aACF;QACH,CAAC;QACD,KAAK,EAAE,CAAC,EAAoB,EAAE,EAAE;YAC9B,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;YACtB,OAAO;YACP,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE;gBACpB,IAAA,sCAAyB,EAAC,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAA;YAC7C,CAAC,CAAC,CAAA;YACF,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAA;YAEvB,8EAA8E;YAC9E,yDAAyD;YACzD,EAAE,CAAC,CAAC,CAAC,OAAO,GAAG,IAAI,CAAA;YACnB,EAAE,CAAC,CAAC,GAAG,IAAI,CAAA;QACb,CAAC;QACD,KAAK,EAAE,GAAG,EAAE;YACV,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;YACrB,+DAA+D;YAC/D,+CAA+C;YAC/C,kDAAkD;YAClD,sBAAsB;YACtB,IAAI;QACN,CAAC;KACF,CAAC;QACF,OAAO;SACN,GAAG,CAAC,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAA,kBAAW,EAAC,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;SACtD,IAAI,CAAC,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAA,kBAAW,EAAC,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;SACvD,OAAO,CAAC,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAA,kBAAW,EAAC,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAA;IAE7D,MAAM,CAAC,MAAM,GAAG,GAAG,CAAA;AACrB,CAAC,CAAA"}
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
import uws from '@based/uws';
|
|
2
|
+
import { BasedServer } from '../server';
|
|
3
|
+
export declare const upgrade: (server: BasedServer, res: uws.HttpResponse, req: uws.HttpRequest, ctx: uws.us_socket_context_t) => void;
|
|
4
|
+
export declare const upgradeAuthorize: (server: BasedServer, res: uws.HttpResponse, req: uws.HttpRequest, ctx: uws.us_socket_context_t) => void;
|
|
@@ -0,0 +1,64 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.upgradeAuthorize = exports.upgrade = void 0;
|
|
4
|
+
const auth_1 = require("../auth");
|
|
5
|
+
const security_1 = require("../security");
|
|
6
|
+
const ip_1 = require("../ip");
|
|
7
|
+
let clientId = 0;
|
|
8
|
+
const upgradeInternal = (res, req,
|
|
9
|
+
// eslint-disable-next-line
|
|
10
|
+
ctx, ip) => {
|
|
11
|
+
const query = req.getQuery(); // encode
|
|
12
|
+
const ua = req.getHeader('user-agent');
|
|
13
|
+
const secWebSocketKey = req.getHeader('sec-websocket-key');
|
|
14
|
+
const secWebSocketProtocol = req.getHeader('sec-websocket-protocol');
|
|
15
|
+
const secWebSocketExtensions = req.getHeader('sec-websocket-extensions');
|
|
16
|
+
res.writeStatus('101 Switching Protocols');
|
|
17
|
+
res.upgrade({
|
|
18
|
+
query,
|
|
19
|
+
ua,
|
|
20
|
+
ip,
|
|
21
|
+
id: ++clientId,
|
|
22
|
+
authState: secWebSocketProtocol
|
|
23
|
+
? (0, auth_1.parseAuthState)(decodeURI(secWebSocketProtocol))
|
|
24
|
+
: undefined,
|
|
25
|
+
obs: new Set(),
|
|
26
|
+
unauthorizedObs: new Set(),
|
|
27
|
+
}, secWebSocketKey, secWebSocketProtocol, secWebSocketExtensions, ctx);
|
|
28
|
+
};
|
|
29
|
+
const upgrade = (server, res, req,
|
|
30
|
+
// eslint-disable-next-line
|
|
31
|
+
ctx) => {
|
|
32
|
+
const ip = (0, ip_1.getIp)(res);
|
|
33
|
+
if ((0, security_1.blockIncomingRequest)(server, ip, res, req, server.rateLimit.ws, 10)) {
|
|
34
|
+
return;
|
|
35
|
+
}
|
|
36
|
+
upgradeInternal(res, req, ctx, ip);
|
|
37
|
+
};
|
|
38
|
+
exports.upgrade = upgrade;
|
|
39
|
+
const upgradeAuthorize = (server, res, req,
|
|
40
|
+
// eslint-disable-next-line
|
|
41
|
+
ctx) => {
|
|
42
|
+
let aborted = false;
|
|
43
|
+
res.onAborted(() => {
|
|
44
|
+
aborted = true;
|
|
45
|
+
});
|
|
46
|
+
const ip = (0, ip_1.getIp)(res);
|
|
47
|
+
if ((0, security_1.blockIncomingRequest)(server, ip, res, req, server.rateLimit.ws, 10)) {
|
|
48
|
+
return;
|
|
49
|
+
}
|
|
50
|
+
server.auth.authorizeConnection(req).then((authorized) => {
|
|
51
|
+
if (aborted) {
|
|
52
|
+
return;
|
|
53
|
+
}
|
|
54
|
+
if (authorized) {
|
|
55
|
+
upgradeInternal(res, req, ctx, ip);
|
|
56
|
+
}
|
|
57
|
+
else {
|
|
58
|
+
res.writeStatus('401 Unauthorized');
|
|
59
|
+
res.end();
|
|
60
|
+
}
|
|
61
|
+
});
|
|
62
|
+
};
|
|
63
|
+
exports.upgradeAuthorize = upgradeAuthorize;
|
|
64
|
+
//# sourceMappingURL=upgrade.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"upgrade.js","sourceRoot":"","sources":["../../src/incoming/upgrade.ts"],"names":[],"mappings":";;;AACA,kCAAwC;AAExC,0CAAkD;AAElD,8BAA6B;AAE7B,IAAI,QAAQ,GAAG,CAAC,CAAA;AAEhB,MAAM,eAAe,GAAG,CACtB,GAAqB,EACrB,GAAoB;AACpB,2BAA2B;AAC3B,GAA4B,EAC5B,EAAU,EACV,EAAE;IACF,MAAM,KAAK,GAAG,GAAG,CAAC,QAAQ,EAAE,CAAA,CAAC,SAAS;IACtC,MAAM,EAAE,GAAG,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,CAAA;IACtC,MAAM,eAAe,GAAG,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAA;IAC1D,MAAM,oBAAoB,GAAG,GAAG,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAA;IACpE,MAAM,sBAAsB,GAAG,GAAG,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAA;IACxE,GAAG,CAAC,WAAW,CAAC,yBAAyB,CAAC,CAAA;IAC1C,GAAG,CAAC,OAAO,CACS;QAChB,KAAK;QACL,EAAE;QACF,EAAE;QACF,EAAE,EAAE,EAAE,QAAQ;QACd,SAAS,EAAE,oBAAoB;YAC7B,CAAC,CAAC,IAAA,qBAAc,EAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;YACjD,CAAC,CAAC,SAAS;QACb,GAAG,EAAE,IAAI,GAAG,EAAE;QACd,eAAe,EAAE,IAAI,GAAG,EAAE;KAC3B,EACD,eAAe,EACf,oBAAoB,EACpB,sBAAsB,EACtB,GAAG,CACJ,CAAA;AACH,CAAC,CAAA;AAEM,MAAM,OAAO,GAAG,CACrB,MAAmB,EACnB,GAAqB,EACrB,GAAoB;AACpB,2BAA2B;AAC3B,GAA4B,EAC5B,EAAE;IACF,MAAM,EAAE,GAAG,IAAA,UAAK,EAAC,GAAG,CAAC,CAAA;IACrB,IAAI,IAAA,+BAAoB,EAAC,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE;QACvE,OAAM;KACP;IACD,eAAe,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,CAAA;AACpC,CAAC,CAAA;AAZY,QAAA,OAAO,WAYnB;AAEM,MAAM,gBAAgB,GAAG,CAC9B,MAAmB,EACnB,GAAqB,EACrB,GAAoB;AACpB,2BAA2B;AAC3B,GAA4B,EAC5B,EAAE;IACF,IAAI,OAAO,GAAG,KAAK,CAAA;IACnB,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE;QACjB,OAAO,GAAG,IAAI,CAAA;IAChB,CAAC,CAAC,CAAA;IACF,MAAM,EAAE,GAAG,IAAA,UAAK,EAAC,GAAG,CAAC,CAAA;IACrB,IAAI,IAAA,+BAAoB,EAAC,MAAM,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE;QACvE,OAAM;KACP;IACD,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,EAAE;QACvD,IAAI,OAAO,EAAE;YACX,OAAM;SACP;QACD,IAAI,UAAU,EAAE;YACd,eAAe,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,CAAA;SACnC;aAAM;YACL,GAAG,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAA;YACnC,GAAG,CAAC,GAAG,EAAE,CAAA;SACV;IACH,CAAC,CAAC,CAAA;AACJ,CAAC,CAAA;AA1BY,QAAA,gBAAgB,oBA0B5B"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
import { BasedServer } from '../../server';
|
|
2
|
+
import { WebSocketSession, Context } from '../../context';
|
|
3
|
+
export declare type AuthState = any;
|
|
4
|
+
export declare const authMessage: (arr: Uint8Array, start: number, len: number, isDeflate: boolean, ctx: Context<WebSocketSession>, server: BasedServer) => boolean;
|
|
5
|
+
export declare const sendAuthMessage: (ctx: Context<WebSocketSession>) => void;
|