@based/server 3.4.6 → 4.0.1

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.
Files changed (269) hide show
  1. package/dist/api/index.d.ts +7 -0
  2. package/dist/api/index.js +173 -0
  3. package/dist/api/index.js.map +1 -0
  4. package/dist/auth/authorize.d.ts +2 -0
  5. package/dist/auth/authorize.js +8 -0
  6. package/dist/auth/authorize.js.map +1 -0
  7. package/dist/auth/dummyAuth.d.ts +2 -0
  8. package/dist/auth/dummyAuth.js +8 -0
  9. package/dist/auth/dummyAuth.js.map +1 -0
  10. package/dist/auth/index.d.ts +15 -0
  11. package/dist/auth/index.js +49 -0
  12. package/dist/auth/index.js.map +1 -0
  13. package/dist/auth/parseAuthState.d.ts +2 -0
  14. package/dist/auth/parseAuthState.js +16 -0
  15. package/dist/auth/parseAuthState.js.map +1 -0
  16. package/dist/auth/types.d.ts +10 -0
  17. package/dist/auth/types.js +3 -0
  18. package/dist/auth/types.js.map +1 -0
  19. package/dist/clientContext.d.ts +0 -0
  20. package/dist/clientContext.js +83 -0
  21. package/dist/clientContext.js.map +1 -0
  22. package/dist/compress.d.ts +5 -0
  23. package/dist/compress.js +52 -0
  24. package/dist/compress.js.map +1 -0
  25. package/dist/context.d.ts +54 -0
  26. package/dist/context.js +18 -0
  27. package/dist/context.js.map +1 -0
  28. package/dist/createSimpleServer.d.ts +22 -0
  29. package/dist/createSimpleServer.js +120 -0
  30. package/dist/createSimpleServer.js.map +1 -0
  31. package/dist/error/errorTypeHandlers.d.ts +6 -0
  32. package/dist/error/errorTypeHandlers.js +130 -0
  33. package/dist/error/errorTypeHandlers.js.map +1 -0
  34. package/dist/error/index.d.ts +5 -0
  35. package/dist/error/index.js +61 -0
  36. package/dist/error/index.js.map +1 -0
  37. package/dist/error/types.d.ts +105 -0
  38. package/dist/error/types.js +30 -0
  39. package/dist/error/types.js.map +1 -0
  40. package/dist/error.d.ts +66 -0
  41. package/dist/error.js +149 -0
  42. package/dist/error.js.map +1 -0
  43. package/dist/functions/index.d.ts +34 -0
  44. package/dist/functions/index.js +210 -0
  45. package/dist/functions/index.js.map +1 -0
  46. package/dist/functions/timeout.d.ts +3 -0
  47. package/dist/functions/timeout.js +23 -0
  48. package/dist/functions/timeout.js.map +1 -0
  49. package/dist/functions/types.d.ts +67 -0
  50. package/dist/functions/types.js +15 -0
  51. package/dist/functions/types.js.map +1 -0
  52. package/dist/functions/workerPool.d.ts +0 -0
  53. package/dist/functions/workerPool.js +5 -0
  54. package/dist/functions/workerPool.js.map +1 -0
  55. package/dist/handlers/configuration/index.d.ts +1 -1
  56. package/dist/incoming/http/authorize.d.ts +4 -0
  57. package/dist/incoming/http/authorize.js +37 -0
  58. package/dist/incoming/http/authorize.js.map +1 -0
  59. package/dist/incoming/http/function.d.ts +4 -0
  60. package/dist/incoming/http/function.js +47 -0
  61. package/dist/incoming/http/function.js.map +1 -0
  62. package/dist/incoming/http/get.d.ts +4 -0
  63. package/dist/incoming/http/get.js +155 -0
  64. package/dist/incoming/http/get.js.map +1 -0
  65. package/dist/incoming/http/index.d.ts +3 -0
  66. package/dist/incoming/http/index.js +134 -0
  67. package/dist/incoming/http/index.js.map +1 -0
  68. package/dist/incoming/http/parseQuery.d.ts +4 -0
  69. package/dist/incoming/http/parseQuery.js +25 -0
  70. package/dist/incoming/http/parseQuery.js.map +1 -0
  71. package/dist/incoming/http/readBody.d.ts +4 -0
  72. package/dist/incoming/http/readBody.js +107 -0
  73. package/dist/incoming/http/readBody.js.map +1 -0
  74. package/dist/incoming/http/streamFunction/DataStream.d.ts +7 -0
  75. package/dist/incoming/http/streamFunction/DataStream.js +19 -0
  76. package/dist/incoming/http/streamFunction/DataStream.js.map +1 -0
  77. package/dist/incoming/http/streamFunction/getExtension.d.ts +2 -0
  78. package/dist/incoming/http/streamFunction/getExtension.js +29 -0
  79. package/dist/incoming/http/streamFunction/getExtension.js.map +1 -0
  80. package/dist/incoming/http/streamFunction/index.d.ts +4 -0
  81. package/dist/incoming/http/streamFunction/index.js +106 -0
  82. package/dist/incoming/http/streamFunction/index.js.map +1 -0
  83. package/dist/incoming/http/streamFunction/multipartStream.d.ts +13 -0
  84. package/dist/incoming/http/streamFunction/multipartStream.js +208 -0
  85. package/dist/incoming/http/streamFunction/multipartStream.js.map +1 -0
  86. package/dist/incoming/http/streamFunction/stream.d.ts +6 -0
  87. package/dist/incoming/http/streamFunction/stream.js +108 -0
  88. package/dist/incoming/http/streamFunction/stream.js.map +1 -0
  89. package/dist/incoming/index.d.ts +3 -0
  90. package/dist/incoming/index.js +84 -0
  91. package/dist/incoming/index.js.map +1 -0
  92. package/dist/incoming/upgrade.d.ts +4 -0
  93. package/dist/incoming/upgrade.js +64 -0
  94. package/dist/incoming/upgrade.js.map +1 -0
  95. package/dist/incoming/ws/auth.d.ts +5 -0
  96. package/dist/incoming/ws/auth.js +40 -0
  97. package/dist/incoming/ws/auth.js.map +1 -0
  98. package/dist/incoming/ws/function.d.ts +3 -0
  99. package/dist/incoming/ws/function.js +107 -0
  100. package/dist/incoming/ws/function.js.map +1 -0
  101. package/dist/incoming/ws/get.d.ts +3 -0
  102. package/dist/incoming/ws/get.js +136 -0
  103. package/dist/incoming/ws/get.js.map +1 -0
  104. package/dist/incoming/ws/index.d.ts +3 -0
  105. package/dist/incoming/ws/index.js +61 -0
  106. package/dist/incoming/ws/index.js.map +1 -0
  107. package/dist/incoming/ws/observable.d.ts +6 -0
  108. package/dist/incoming/ws/observable.js +112 -0
  109. package/dist/incoming/ws/observable.js.map +1 -0
  110. package/dist/index.d.ts +11 -35
  111. package/dist/index.js +19 -196
  112. package/dist/index.js.map +1 -1
  113. package/dist/ip.d.ts +2 -0
  114. package/dist/ip.js +8 -0
  115. package/dist/ip.js.map +1 -0
  116. package/dist/network/http/authorize.d.ts +3 -0
  117. package/dist/network/http/authorize.js +28 -0
  118. package/dist/network/http/authorize.js.map +1 -0
  119. package/dist/network/http/compress.d.ts +6 -0
  120. package/dist/network/http/compress.js +56 -0
  121. package/dist/network/http/compress.js.map +1 -0
  122. package/dist/network/http/end.d.ts +4 -0
  123. package/dist/network/http/end.js +19 -0
  124. package/dist/network/http/end.js.map +1 -0
  125. package/dist/network/http/function.d.ts +3 -0
  126. package/dist/network/http/function.js +47 -0
  127. package/dist/network/http/function.js.map +1 -0
  128. package/dist/network/http/get.d.ts +3 -0
  129. package/dist/network/http/get.js +129 -0
  130. package/dist/network/http/get.js.map +1 -0
  131. package/dist/network/http/index.d.ts +3 -0
  132. package/dist/network/http/index.js +122 -0
  133. package/dist/network/http/index.js.map +1 -0
  134. package/dist/network/http/readBody.d.ts +3 -0
  135. package/dist/network/http/readBody.js +112 -0
  136. package/dist/network/http/readBody.js.map +1 -0
  137. package/dist/network/http/send.d.ts +5 -0
  138. package/dist/network/http/send.js +60 -0
  139. package/dist/network/http/send.js.map +1 -0
  140. package/dist/network/http/streamFunction/DataStream.d.ts +7 -0
  141. package/dist/network/http/streamFunction/DataStream.js +19 -0
  142. package/dist/network/http/streamFunction/DataStream.js.map +1 -0
  143. package/dist/network/http/streamFunction/getExtension.d.ts +2 -0
  144. package/dist/network/http/streamFunction/getExtension.js +29 -0
  145. package/dist/network/http/streamFunction/getExtension.js.map +1 -0
  146. package/dist/network/http/streamFunction/index.d.ts +3 -0
  147. package/dist/network/http/streamFunction/index.js +81 -0
  148. package/dist/network/http/streamFunction/index.js.map +1 -0
  149. package/dist/network/http/streamFunction/multipartStream.d.ts +12 -0
  150. package/dist/network/http/streamFunction/multipartStream.js +208 -0
  151. package/dist/network/http/streamFunction/multipartStream.js.map +1 -0
  152. package/dist/network/http/streamFunction/stream.d.ts +5 -0
  153. package/dist/network/http/streamFunction/stream.js +103 -0
  154. package/dist/network/http/streamFunction/stream.js.map +1 -0
  155. package/dist/network/index.d.ts +4 -0
  156. package/dist/network/index.js +84 -0
  157. package/dist/network/index.js.map +1 -0
  158. package/dist/network/message/auth.d.ts +4 -0
  159. package/dist/network/message/auth.js +32 -0
  160. package/dist/network/message/auth.js.map +1 -0
  161. package/dist/network/message/function.d.ts +3 -0
  162. package/dist/network/message/function.js +67 -0
  163. package/dist/network/message/function.js.map +1 -0
  164. package/dist/network/message/get.d.ts +3 -0
  165. package/dist/network/message/get.js +109 -0
  166. package/dist/network/message/get.js.map +1 -0
  167. package/dist/network/message/index.d.ts +3 -0
  168. package/dist/network/message/index.js +62 -0
  169. package/dist/network/message/index.js.map +1 -0
  170. package/dist/network/message/observable.d.ts +5 -0
  171. package/dist/network/message/observable.js +103 -0
  172. package/dist/network/message/observable.js.map +1 -0
  173. package/dist/network/message/send.d.ts +4 -0
  174. package/dist/network/message/send.js +11 -0
  175. package/dist/network/message/send.js.map +1 -0
  176. package/dist/network/upgrade.d.ts +4 -0
  177. package/dist/network/upgrade.js +49 -0
  178. package/dist/network/upgrade.js.map +1 -0
  179. package/dist/network/worker/get.d.ts +0 -0
  180. package/dist/network/worker/get.js +122 -0
  181. package/dist/network/worker/get.js.map +1 -0
  182. package/dist/network/worker/observable.d.ts +0 -0
  183. package/dist/network/worker/observable.js +120 -0
  184. package/dist/network/worker/observable.js.map +1 -0
  185. package/dist/observable/create.d.ts +3 -0
  186. package/dist/observable/create.js +30 -0
  187. package/dist/observable/create.js.map +1 -0
  188. package/dist/observable/destroy.d.ts +2 -0
  189. package/dist/observable/destroy.js +51 -0
  190. package/dist/observable/destroy.js.map +1 -0
  191. package/dist/observable/error.d.ts +4 -0
  192. package/dist/observable/error.js +17 -0
  193. package/dist/observable/error.js.map +1 -0
  194. package/dist/observable/extendCache.d.ts +2 -0
  195. package/dist/observable/extendCache.js +11 -0
  196. package/dist/observable/extendCache.js.map +1 -0
  197. package/dist/observable/genObservableId.d.ts +1 -0
  198. package/dist/observable/genObservableId.js +9 -0
  199. package/dist/observable/genObservableId.js.map +1 -0
  200. package/dist/observable/get.d.ts +4 -0
  201. package/dist/observable/get.js +15 -0
  202. package/dist/observable/get.js.map +1 -0
  203. package/dist/observable/index.d.ts +11 -0
  204. package/dist/observable/index.js +28 -0
  205. package/dist/observable/index.js.map +1 -0
  206. package/dist/observable/send.d.ts +6 -0
  207. package/dist/observable/send.js +32 -0
  208. package/dist/observable/send.js.map +1 -0
  209. package/dist/observable/start/error.d.ts +3 -0
  210. package/dist/observable/start/error.js +36 -0
  211. package/dist/observable/start/error.js.map +1 -0
  212. package/dist/observable/start/index.d.ts +2 -0
  213. package/dist/observable/start/index.js +46 -0
  214. package/dist/observable/start/index.js.map +1 -0
  215. package/dist/observable/start/update.d.ts +6 -0
  216. package/dist/observable/start/update.js +104 -0
  217. package/dist/observable/start/update.js.map +1 -0
  218. package/dist/observable/subscribe.d.ts +7 -0
  219. package/dist/observable/subscribe.js +54 -0
  220. package/dist/observable/subscribe.js.map +1 -0
  221. package/dist/observable/types.d.ts +26 -0
  222. package/dist/observable/types.js +3 -0
  223. package/dist/observable/types.js.map +1 -0
  224. package/dist/observable/unsub.d.ts +6 -0
  225. package/dist/observable/unsub.js +43 -0
  226. package/dist/observable/unsub.js.map +1 -0
  227. package/dist/observable/verify.d.ts +4 -0
  228. package/dist/observable/verify.js +18 -0
  229. package/dist/observable/verify.js.map +1 -0
  230. package/dist/protocol.d.ts +21 -0
  231. package/dist/protocol.js +233 -0
  232. package/dist/protocol.js.map +1 -0
  233. package/dist/security/index.d.ts +3 -0
  234. package/dist/security/index.js +66 -0
  235. package/dist/security/index.js.map +1 -0
  236. package/dist/security.d.ts +6 -0
  237. package/dist/security.js +95 -0
  238. package/dist/security.js.map +1 -0
  239. package/dist/sendError.d.ts +5 -0
  240. package/dist/sendError.js +42 -0
  241. package/dist/sendError.js.map +1 -0
  242. package/dist/sendHttpResponse.d.ts +4 -0
  243. package/dist/sendHttpResponse.js +53 -0
  244. package/dist/sendHttpResponse.js.map +1 -0
  245. package/dist/server.d.ts +69 -0
  246. package/dist/server.js +113 -0
  247. package/dist/server.js.map +1 -0
  248. package/dist/worker/authorize.d.ts +5 -0
  249. package/dist/worker/authorize.js +15 -0
  250. package/dist/worker/authorize.js.map +1 -0
  251. package/dist/worker/fnMap.d.ts +2 -0
  252. package/dist/worker/fnMap.js +5 -0
  253. package/dist/worker/fnMap.js.map +1 -0
  254. package/dist/worker/functions.d.ts +3 -0
  255. package/dist/worker/functions.js +8 -0
  256. package/dist/worker/functions.js.map +1 -0
  257. package/dist/worker/http/function.d.ts +4 -0
  258. package/dist/worker/http/function.js +75 -0
  259. package/dist/worker/http/function.js.map +1 -0
  260. package/dist/worker/index.d.ts +1 -0
  261. package/dist/worker/index.js +101 -0
  262. package/dist/worker/index.js.map +1 -0
  263. package/dist/worker/observable.d.ts +15 -0
  264. package/dist/worker/observable.js +110 -0
  265. package/dist/worker/observable.js.map +1 -0
  266. package/dist/worker/ws/function.d.ts +3 -0
  267. package/dist/worker/ws/function.js +42 -0
  268. package/dist/worker/ws/function.js.map +1 -0
  269. package/package.json +16 -14
@@ -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,3 @@
1
+ import type { BasedServer, ServerOptions } from '../server';
2
+ declare const _default: (server: BasedServer, { key, cert, port, ws: wsListeners }: ServerOptions) => void;
3
+ export default _default;
@@ -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;