@nestia/core 12.0.0-dev.20260601.1 → 12.0.0-dev.20260612.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 (116) hide show
  1. package/LICENSE +21 -21
  2. package/MIGRATION.md +169 -169
  3. package/README.md +93 -93
  4. package/lib/adaptors/McpAdaptor.d.ts +75 -0
  5. package/lib/adaptors/McpAdaptor.js +257 -0
  6. package/lib/adaptors/McpAdaptor.js.map +1 -0
  7. package/lib/adaptors/WebSocketAdaptor.js +4 -4
  8. package/lib/adaptors/WebSocketAdaptor.js.map +1 -1
  9. package/lib/decorators/McpRoute.d.ts +69 -0
  10. package/lib/decorators/McpRoute.js +58 -0
  11. package/lib/decorators/McpRoute.js.map +1 -0
  12. package/lib/decorators/TypedParam.js +4 -4
  13. package/lib/decorators/TypedParam.js.map +1 -1
  14. package/lib/decorators/TypedRoute.js +1 -1
  15. package/lib/decorators/TypedRoute.js.map +1 -1
  16. package/lib/decorators/internal/IMcpRouteReflect.d.ts +2 -0
  17. package/lib/decorators/internal/IMcpRouteReflect.js +3 -0
  18. package/lib/decorators/internal/IMcpRouteReflect.js.map +1 -0
  19. package/lib/decorators/internal/get_path_and_querify.js +4 -4
  20. package/lib/decorators/internal/get_path_and_querify.js.map +1 -1
  21. package/lib/decorators/internal/get_path_and_stringify.js +4 -4
  22. package/lib/decorators/internal/get_path_and_stringify.js.map +1 -1
  23. package/lib/decorators/internal/load_controller.js +34 -65
  24. package/lib/decorators/internal/load_controller.js.map +1 -1
  25. package/lib/decorators/internal/validate_request_body.js +4 -4
  26. package/lib/decorators/internal/validate_request_body.js.map +1 -1
  27. package/lib/decorators/internal/validate_request_form_data.js +4 -4
  28. package/lib/decorators/internal/validate_request_form_data.js.map +1 -1
  29. package/lib/decorators/internal/validate_request_headers.js +4 -4
  30. package/lib/decorators/internal/validate_request_headers.js.map +1 -1
  31. package/lib/decorators/internal/validate_request_query.js +4 -4
  32. package/lib/decorators/internal/validate_request_query.js.map +1 -1
  33. package/lib/module.d.ts +2 -0
  34. package/lib/module.js +2 -0
  35. package/lib/module.js.map +1 -1
  36. package/native/cmd/ttsc-nestia/main.go +11 -11
  37. package/native/go.mod +32 -32
  38. package/native/go.sum +54 -54
  39. package/native/plugin/plan.go +102 -102
  40. package/native/transform/ast.go +32 -32
  41. package/native/transform/build.go +380 -444
  42. package/native/transform/cleanup.go +408 -408
  43. package/native/transform/contributor.go +97 -68
  44. package/native/transform/core_querify.go +231 -227
  45. package/native/transform/core_transform.go +1996 -1713
  46. package/native/transform/core_websocket.go +115 -115
  47. package/native/transform/exports.go +13 -13
  48. package/native/transform/mcp_transform.go +414 -0
  49. package/native/transform/node_transform.go +357 -0
  50. package/native/transform/path_rewrite.go +285 -285
  51. package/native/transform/printer.go +244 -244
  52. package/native/transform/rewrite.go +668 -662
  53. package/native/transform/run.go +73 -73
  54. package/native/transform/transform.go +336 -403
  55. package/native/transform/typia_fast.go +352 -326
  56. package/native/transform/typia_replacement.go +24 -24
  57. package/native/transform.cjs +43 -43
  58. package/package.json +15 -8
  59. package/src/adaptors/McpAdaptor.ts +276 -0
  60. package/src/adaptors/WebSocketAdaptor.ts +429 -429
  61. package/src/decorators/DynamicModule.ts +44 -44
  62. package/src/decorators/EncryptedBody.ts +97 -97
  63. package/src/decorators/EncryptedController.ts +40 -40
  64. package/src/decorators/EncryptedModule.ts +98 -98
  65. package/src/decorators/EncryptedRoute.ts +213 -213
  66. package/src/decorators/HumanRoute.ts +21 -21
  67. package/src/decorators/McpRoute.ts +154 -0
  68. package/src/decorators/NoTransformConfigurationError.ts +40 -40
  69. package/src/decorators/PlainBody.ts +76 -76
  70. package/src/decorators/SwaggerCustomizer.ts +97 -97
  71. package/src/decorators/SwaggerExample.ts +180 -180
  72. package/src/decorators/TypedBody.ts +57 -57
  73. package/src/decorators/TypedException.ts +147 -147
  74. package/src/decorators/TypedFormData.ts +187 -187
  75. package/src/decorators/TypedHeaders.ts +66 -66
  76. package/src/decorators/TypedParam.ts +77 -77
  77. package/src/decorators/TypedQuery.ts +234 -234
  78. package/src/decorators/TypedRoute.ts +198 -196
  79. package/src/decorators/WebSocketRoute.ts +242 -242
  80. package/src/decorators/doNotThrowTransformError.ts +5 -5
  81. package/src/decorators/internal/EncryptedConstant.ts +2 -2
  82. package/src/decorators/internal/IMcpRouteReflect.ts +40 -0
  83. package/src/decorators/internal/IWebSocketRouteReflect.ts +23 -23
  84. package/src/decorators/internal/get_path_and_querify.ts +94 -94
  85. package/src/decorators/internal/get_path_and_stringify.ts +110 -110
  86. package/src/decorators/internal/get_text_body.ts +16 -16
  87. package/src/decorators/internal/headers_to_object.ts +11 -11
  88. package/src/decorators/internal/is_request_body_undefined.ts +12 -12
  89. package/src/decorators/internal/load_controller.ts +91 -76
  90. package/src/decorators/internal/route_error.ts +43 -43
  91. package/src/decorators/internal/validate_request_body.ts +64 -64
  92. package/src/decorators/internal/validate_request_form_data.ts +67 -67
  93. package/src/decorators/internal/validate_request_headers.ts +76 -76
  94. package/src/decorators/internal/validate_request_query.ts +83 -83
  95. package/src/index.ts +5 -5
  96. package/src/module.ts +25 -23
  97. package/src/options/IRequestBodyValidator.ts +20 -20
  98. package/src/options/IRequestFormDataProps.ts +27 -27
  99. package/src/options/IRequestHeadersValidator.ts +22 -22
  100. package/src/options/IRequestQueryValidator.ts +20 -20
  101. package/src/options/IResponseBodyQuerifier.ts +25 -25
  102. package/src/options/IResponseBodyStringifier.ts +30 -30
  103. package/src/transform.ts +101 -101
  104. package/src/typings/Creator.ts +3 -3
  105. package/src/typings/get-function-location.d.ts +7 -7
  106. package/src/utils/ArrayUtil.ts +7 -7
  107. package/src/utils/ExceptionManager.ts +115 -115
  108. package/src/utils/Singleton.ts +16 -16
  109. package/src/utils/SourceFinder.ts +54 -54
  110. package/src/utils/VersioningStrategy.ts +27 -27
  111. package/native/transform/cleanup_test.go +0 -76
  112. package/native/transform/commonjs_import_alias_test.go +0 -49
  113. package/native/transform/core_dispatch_test.go +0 -127
  114. package/native/transform/path_rewrite_test.go +0 -243
  115. package/native/transform/rewrite_test.go +0 -118
  116. package/native/transform/rewrite_unique_base_test.go +0 -48
@@ -1,429 +1,429 @@
1
- /// <reference path="../typings/get-function-location.d.ts" />
2
- import { INestApplication, VersioningType } from "@nestjs/common";
3
- import {
4
- HOST_METADATA,
5
- MODULE_PATH,
6
- PATH_METADATA,
7
- SCOPE_OPTIONS_METADATA,
8
- VERSION_METADATA,
9
- } from "@nestjs/common/constants";
10
- import { VERSION_NEUTRAL, VersionValue } from "@nestjs/common/interfaces";
11
- import { NestContainer } from "@nestjs/core";
12
- import { InstanceWrapper } from "@nestjs/core/injector/instance-wrapper";
13
- import { Module } from "@nestjs/core/injector/module";
14
- import getFunctionLocation from "get-function-location";
15
- import { IncomingMessage, Server } from "http";
16
- import path from "path";
17
- import { Path } from "path-parser";
18
- import { Duplex } from "stream";
19
- import { WebSocketAcceptor } from "tgrid";
20
- import typia from "typia";
21
- import WebSocket from "ws";
22
-
23
- import { IWebSocketRouteReflect } from "../decorators/internal/IWebSocketRouteReflect";
24
- import { ArrayUtil } from "../utils/ArrayUtil";
25
- import { VersioningStrategy } from "../utils/VersioningStrategy";
26
-
27
- export class WebSocketAdaptor {
28
- public static async upgrade(
29
- app: INestApplication,
30
- ): Promise<WebSocketAdaptor> {
31
- return new this(app, await visitApplication(app));
32
- }
33
-
34
- public readonly close = async (): Promise<void> =>
35
- new Promise((resolve) => {
36
- this.http.off("close", this.close);
37
- this.http.off("upgrade", this.handleUpgrade);
38
- this.ws.close(() => resolve());
39
- });
40
-
41
- private constructor(app: INestApplication, operations: IOperator[]) {
42
- this.operators = operations;
43
- this.ws = new WebSocket.Server({ noServer: true });
44
- this.http = app.getHttpServer();
45
- this.http.on("close", this.close);
46
- this.http.on("upgrade", this.handleUpgrade);
47
- }
48
-
49
- private readonly handleUpgrade = (
50
- request: IncomingMessage,
51
- duplex: Duplex,
52
- head: Buffer,
53
- ) => {
54
- this.ws.handleUpgrade(request, duplex, head, (client, request) =>
55
- WebSocketAcceptor.upgrade(
56
- request,
57
- client as any,
58
- async (acceptor): Promise<void> => {
59
- const path: string = (() => {
60
- const index: number = acceptor.path.indexOf("?");
61
- return index === -1 ? acceptor.path : acceptor.path.slice(0, index);
62
- })();
63
- for (const op of this.operators) {
64
- const params: Record<string, string> | null = op.parser.test(path);
65
- if (params !== null)
66
- try {
67
- await op.handler({ params, acceptor });
68
- } catch (error) {
69
- if (
70
- acceptor.state === WebSocketAcceptor.State.OPEN ||
71
- acceptor.state === WebSocketAcceptor.State.ACCEPTING
72
- )
73
- await acceptor.reject(
74
- 1008,
75
- error instanceof Error
76
- ? JSON.stringify({ ...error })
77
- : "unknown error",
78
- );
79
- } finally {
80
- return;
81
- }
82
- }
83
- await acceptor.reject(1002, `WebSocket API not found`);
84
- },
85
- ),
86
- );
87
- };
88
-
89
- private readonly http: Server;
90
- private readonly operators: IOperator[];
91
- private readonly ws: WebSocket.Server;
92
- }
93
-
94
- const visitApplication = async (
95
- app: INestApplication,
96
- ): Promise<IOperator[]> => {
97
- const operators: IOperator[] = [];
98
- const errors: IControllerError[] = [];
99
-
100
- const config: IConfig = {
101
- globalPrefix:
102
- typeof (app as any).config?.globalPrefix === "string"
103
- ? (app as any).config.globalPrefix
104
- : undefined,
105
- versioning: (() => {
106
- const versioning = (app as any).config?.versioningOptions;
107
- return versioning === undefined || versioning.type !== VersioningType.URI
108
- ? undefined
109
- : {
110
- prefix:
111
- versioning.prefix === undefined || versioning.prefix === false
112
- ? "v"
113
- : versioning.prefix,
114
- defaultVersion: versioning.defaultVersion,
115
- };
116
- })(),
117
- };
118
- const container: NestContainer = (app as any).container as NestContainer;
119
- const modules: Module[] = [...container.getModules().values()].filter(
120
- (m) => !!m.controllers?.size,
121
- );
122
- for (const m of modules) {
123
- const modulePrefix: string =
124
- Reflect.getMetadata(
125
- MODULE_PATH + container.getModules().applicationId,
126
- m.metatype,
127
- ) ??
128
- Reflect.getMetadata(MODULE_PATH, m.metatype) ??
129
- "";
130
- for (const controller of m.controllers.values())
131
- await visitController({
132
- config,
133
- errors,
134
- operators,
135
- controller,
136
- modulePrefix,
137
- });
138
- }
139
- if (errors.length)
140
- throw new Error(
141
- [
142
- `WebSocketAdaptor: ${errors.length} error(s) found:`,
143
- ``,
144
- ...errors.map((e) =>
145
- [
146
- ` - controller: ${e.name}`,
147
- ` methods:`,
148
- ...e.methods.map((m) =>
149
- [
150
- ` - name: ${m.name}`,
151
- ` file: ${m.source}:${m.line}:${m.column}`,
152
- ` reasons:`,
153
- ...m.messages.map(
154
- (msg) =>
155
- ` - ${msg
156
- .split("\n")
157
- .map((str, i) => (i == 0 ? str : ` ${str}`))
158
- .join("\n")}`,
159
- ),
160
- ].join("\n"),
161
- ),
162
- ].join("\n"),
163
- ),
164
- ].join("\n"),
165
- );
166
- return operators;
167
- };
168
-
169
- const visitController = async (props: {
170
- config: IConfig;
171
- errors: IControllerError[];
172
- operators: IOperator[];
173
- controller: InstanceWrapper<object>;
174
- modulePrefix: string;
175
- }): Promise<void> => {
176
- if (
177
- ArrayUtil.has(
178
- Reflect.getMetadataKeys(props.controller.metatype as Function),
179
- PATH_METADATA,
180
- HOST_METADATA,
181
- SCOPE_OPTIONS_METADATA,
182
- ) === false
183
- )
184
- return;
185
-
186
- const methodErrors: IMethodError[] = [];
187
- const controller: IController = {
188
- name: props.controller.name,
189
- instance: props.controller.instance,
190
- constructor: props.controller.metatype as Function,
191
- prototype: Object.getPrototypeOf(props.controller.instance),
192
- prefixes: (() => {
193
- const value: string | string[] = Reflect.getMetadata(
194
- PATH_METADATA,
195
- props.controller.metatype as object,
196
- );
197
- if (typeof value === "string") return [value];
198
- else if (value.length === 0) return [""];
199
- else return value;
200
- })(),
201
- versions: props.config.versioning
202
- ? VersioningStrategy.cast(
203
- Reflect.getMetadata(
204
- VERSION_METADATA,
205
- props.controller.metatype as Function,
206
- ),
207
- )
208
- : undefined,
209
- modulePrefix: props.modulePrefix,
210
- };
211
- for (const mk of getOwnPropertyNames(controller.prototype).filter(
212
- (key) =>
213
- key !== "constructor" && typeof controller.prototype[key] === "function",
214
- )) {
215
- const errorMessages: string[] = [];
216
- visitMethod({
217
- config: props.config,
218
- operators: props.operators,
219
- controller,
220
- method: {
221
- key: mk,
222
- value: controller.prototype[mk],
223
- },
224
- report: (msg) => errorMessages.push(msg),
225
- });
226
- if (errorMessages.length) {
227
- const file = await getFunctionLocation(controller.prototype[mk]);
228
- methodErrors.push({
229
- name: mk,
230
- messages: errorMessages,
231
- ...file,
232
- source: path.relative(
233
- process.cwd(),
234
- file.source.replace("file:///", ""),
235
- ),
236
- });
237
- }
238
- }
239
-
240
- if (methodErrors.length)
241
- props.errors.push({
242
- name: controller.name,
243
- methods: methodErrors,
244
- });
245
- };
246
-
247
- const visitMethod = (props: {
248
- config: IConfig;
249
- operators: IOperator[];
250
- controller: IController;
251
- method: Entry<Function>;
252
- report: (message: string) => void;
253
- }): void => {
254
- const route: IWebSocketRouteReflect | undefined = Reflect.getMetadata(
255
- "nestia/WebSocketRoute",
256
- props.method.value,
257
- );
258
- if (typia.is<IWebSocketRouteReflect>(route) === false) return;
259
-
260
- const parameters: IWebSocketRouteReflect.IArgument[] = (
261
- (Reflect.getMetadata(
262
- "nestia/WebSocketRoute/Parameters",
263
- props.controller.prototype,
264
- props.method.key,
265
- ) ?? []) as IWebSocketRouteReflect.IArgument[]
266
- ).sort((a, b) => a.index - b.index);
267
- // acceptor must be
268
- if (parameters.some((p) => p.category === "acceptor") === false)
269
- return props.report(
270
- "@WebSocketRoute.Acceptor() decorated parameter must be.",
271
- );
272
- // length of parameters must be fulfilled
273
- if (parameters.length !== props.method.value.length)
274
- return props.report(
275
- [
276
- "Every parameters must be one of below:",
277
- " - @WebSocketRoute.Acceptor()",
278
- " - @WebSocketRoute.Driver()",
279
- " - @WebSocketRoute.Header()",
280
- " - @WebSocketRoute.Param()",
281
- " - @WebSocketRoute.Query()",
282
- ].join("\n"),
283
- );
284
-
285
- const versions: string[] = VersioningStrategy.merge(props.config.versioning)([
286
- ...(props.controller.versions ?? []),
287
- ...VersioningStrategy.cast(
288
- Reflect.getMetadata(VERSION_METADATA, props.method.value),
289
- ),
290
- ]);
291
- for (const v of versions)
292
- for (const cp of wrapPaths(props.controller.prefixes))
293
- for (const mp of wrapPaths(route.paths)) {
294
- const parser: Path = new Path(
295
- "/" +
296
- [
297
- props.config.globalPrefix ?? "",
298
- v,
299
- props.controller.modulePrefix,
300
- cp,
301
- mp,
302
- ]
303
- .filter((str) => !!str.length)
304
- .join("/")
305
- .split("/")
306
- .filter((str) => str.length)
307
- .join("/"),
308
- );
309
- const pathParams: IWebSocketRouteReflect.IParam[] = parameters.filter(
310
- (p) => p.category === "param",
311
- ) as IWebSocketRouteReflect.IParam[];
312
- if (parser.params.length !== pathParams.length) {
313
- props.report(
314
- [
315
- `Path "${parser}" must have same number of parameters with @WebSocketRoute.Param()`,
316
- ` - path: ${JSON.stringify(parser.params)}`,
317
- ` - arguments: ${JSON.stringify(pathParams.map((p) => p.field))}`,
318
- ].join("\n"),
319
- );
320
- continue;
321
- }
322
- const meet: boolean = pathParams
323
- .map((p) => {
324
- const has: boolean = parser.params.includes(p.field);
325
- if (has === false)
326
- props.report(
327
- `Path "${parser}" must have parameter "${p.field}" with @WebSocketRoute.Param()`,
328
- );
329
- return has;
330
- })
331
- .every((b) => b);
332
- if (meet === false) continue;
333
-
334
- props.operators.push({
335
- parser,
336
- handler: async (input: {
337
- params: Record<string, string>;
338
- acceptor: WebSocketAcceptor<any, any, any>;
339
- }): Promise<void> => {
340
- const args: any[] = [];
341
- try {
342
- for (const p of parameters)
343
- if (p.category === "acceptor") args.push(input.acceptor);
344
- else if (p.category === "driver")
345
- args.push(input.acceptor.getDriver());
346
- else if (p.category === "header") {
347
- const error: Error | null = p.validate(input.acceptor.header);
348
- if (error !== null) throw error;
349
- args.push(input.acceptor.header);
350
- } else if (p.category === "param")
351
- args.push(p.assert(input.params[p.field]!));
352
- else if (p.category === "query") {
353
- const query: any | Error = p.validate(
354
- new URLSearchParams(
355
- input.acceptor.path.indexOf("?") !== -1
356
- ? input.acceptor.path.split("?")[1]
357
- : "",
358
- ),
359
- );
360
- if (query instanceof Error) throw query;
361
- args.push(query);
362
- }
363
- } catch (exp) {
364
- await input.acceptor.reject(
365
- 1003,
366
- exp instanceof Error
367
- ? JSON.stringify({ ...exp })
368
- : "unknown error",
369
- );
370
- return;
371
- }
372
- await props.method.value.call(props.controller.instance, ...args);
373
- },
374
- });
375
- }
376
- };
377
-
378
- const wrapPaths = (value: string[]) => (value.length === 0 ? [""] : value);
379
- const getOwnPropertyNames = (prototype: any): string[] => {
380
- const result: Set<string> = new Set();
381
- const iterate = (m: any) => {
382
- if (m === null) return;
383
- for (const k of Object.getOwnPropertyNames(m)) result.add(k);
384
- iterate(Object.getPrototypeOf(m));
385
- };
386
- iterate(prototype);
387
- return Array.from(result);
388
- };
389
-
390
- interface Entry<T> {
391
- key: string;
392
- value: T;
393
- }
394
-
395
- interface IController {
396
- name: string;
397
- versions: Array<string | typeof VERSION_NEUTRAL> | undefined;
398
- instance: object;
399
- constructor: Function;
400
- prototype: any;
401
- prefixes: string[];
402
- modulePrefix: string;
403
- }
404
- interface IOperator {
405
- parser: Path;
406
- handler: (props: {
407
- params: Record<string, string>;
408
- acceptor: WebSocketAcceptor<any, any, any>;
409
- }) => Promise<any>;
410
- }
411
- interface IConfig {
412
- globalPrefix?: string;
413
- versioning?: {
414
- prefix: string;
415
- defaultVersion?: VersionValue;
416
- };
417
- }
418
-
419
- interface IControllerError {
420
- name: string;
421
- methods: IMethodError[];
422
- }
423
- interface IMethodError {
424
- name: string;
425
- messages: string[];
426
- source: string;
427
- line: number;
428
- column: number;
429
- }
1
+ /// <reference path="../typings/get-function-location.d.ts" />
2
+ import { INestApplication, VersioningType } from "@nestjs/common";
3
+ import {
4
+ HOST_METADATA,
5
+ MODULE_PATH,
6
+ PATH_METADATA,
7
+ SCOPE_OPTIONS_METADATA,
8
+ VERSION_METADATA,
9
+ } from "@nestjs/common/constants";
10
+ import { VERSION_NEUTRAL, VersionValue } from "@nestjs/common/interfaces";
11
+ import { NestContainer } from "@nestjs/core";
12
+ import { InstanceWrapper } from "@nestjs/core/injector/instance-wrapper";
13
+ import { Module } from "@nestjs/core/injector/module";
14
+ import getFunctionLocation from "get-function-location";
15
+ import { IncomingMessage, Server } from "http";
16
+ import path from "path";
17
+ import { Path } from "path-parser";
18
+ import { Duplex } from "stream";
19
+ import { WebSocketAcceptor } from "tgrid";
20
+ import typia from "typia";
21
+ import WebSocket from "ws";
22
+
23
+ import { IWebSocketRouteReflect } from "../decorators/internal/IWebSocketRouteReflect";
24
+ import { ArrayUtil } from "../utils/ArrayUtil";
25
+ import { VersioningStrategy } from "../utils/VersioningStrategy";
26
+
27
+ export class WebSocketAdaptor {
28
+ public static async upgrade(
29
+ app: INestApplication,
30
+ ): Promise<WebSocketAdaptor> {
31
+ return new this(app, await visitApplication(app));
32
+ }
33
+
34
+ public readonly close = async (): Promise<void> =>
35
+ new Promise((resolve) => {
36
+ this.http.off("close", this.close);
37
+ this.http.off("upgrade", this.handleUpgrade);
38
+ this.ws.close(() => resolve());
39
+ });
40
+
41
+ private constructor(app: INestApplication, operations: IOperator[]) {
42
+ this.operators = operations;
43
+ this.ws = new WebSocket.Server({ noServer: true });
44
+ this.http = app.getHttpServer();
45
+ this.http.on("close", this.close);
46
+ this.http.on("upgrade", this.handleUpgrade);
47
+ }
48
+
49
+ private readonly handleUpgrade = (
50
+ request: IncomingMessage,
51
+ duplex: Duplex,
52
+ head: Buffer,
53
+ ) => {
54
+ this.ws.handleUpgrade(request, duplex, head, (client, request) =>
55
+ WebSocketAcceptor.upgrade(
56
+ request,
57
+ client as any,
58
+ async (acceptor): Promise<void> => {
59
+ const path: string = (() => {
60
+ const index: number = acceptor.path.indexOf("?");
61
+ return index === -1 ? acceptor.path : acceptor.path.slice(0, index);
62
+ })();
63
+ for (const op of this.operators) {
64
+ const params: Record<string, string> | null = op.parser.test(path);
65
+ if (params !== null)
66
+ try {
67
+ await op.handler({ params, acceptor });
68
+ } catch (error) {
69
+ if (
70
+ acceptor.state === WebSocketAcceptor.State.OPEN ||
71
+ acceptor.state === WebSocketAcceptor.State.ACCEPTING
72
+ )
73
+ await acceptor.reject(
74
+ 1008,
75
+ error instanceof Error
76
+ ? JSON.stringify({ ...error })
77
+ : "unknown error",
78
+ );
79
+ } finally {
80
+ return;
81
+ }
82
+ }
83
+ await acceptor.reject(1002, `WebSocket API not found`);
84
+ },
85
+ ),
86
+ );
87
+ };
88
+
89
+ private readonly http: Server;
90
+ private readonly operators: IOperator[];
91
+ private readonly ws: WebSocket.Server;
92
+ }
93
+
94
+ const visitApplication = async (
95
+ app: INestApplication,
96
+ ): Promise<IOperator[]> => {
97
+ const operators: IOperator[] = [];
98
+ const errors: IControllerError[] = [];
99
+
100
+ const config: IConfig = {
101
+ globalPrefix:
102
+ typeof (app as any).config?.globalPrefix === "string"
103
+ ? (app as any).config.globalPrefix
104
+ : undefined,
105
+ versioning: (() => {
106
+ const versioning = (app as any).config?.versioningOptions;
107
+ return versioning === undefined || versioning.type !== VersioningType.URI
108
+ ? undefined
109
+ : {
110
+ prefix:
111
+ versioning.prefix === undefined || versioning.prefix === false
112
+ ? "v"
113
+ : versioning.prefix,
114
+ defaultVersion: versioning.defaultVersion,
115
+ };
116
+ })(),
117
+ };
118
+ const container: NestContainer = (app as any).container as NestContainer;
119
+ const modules: Module[] = [...container.getModules().values()].filter(
120
+ (m) => !!m.controllers?.size,
121
+ );
122
+ for (const m of modules) {
123
+ const modulePrefix: string =
124
+ Reflect.getMetadata(
125
+ MODULE_PATH + container.getModules().applicationId,
126
+ m.metatype,
127
+ ) ??
128
+ Reflect.getMetadata(MODULE_PATH, m.metatype) ??
129
+ "";
130
+ for (const controller of m.controllers.values())
131
+ await visitController({
132
+ config,
133
+ errors,
134
+ operators,
135
+ controller,
136
+ modulePrefix,
137
+ });
138
+ }
139
+ if (errors.length)
140
+ throw new Error(
141
+ [
142
+ `WebSocketAdaptor: ${errors.length} error(s) found:`,
143
+ ``,
144
+ ...errors.map((e) =>
145
+ [
146
+ ` - controller: ${e.name}`,
147
+ ` methods:`,
148
+ ...e.methods.map((m) =>
149
+ [
150
+ ` - name: ${m.name}`,
151
+ ` file: ${m.source}:${m.line}:${m.column}`,
152
+ ` reasons:`,
153
+ ...m.messages.map(
154
+ (msg) =>
155
+ ` - ${msg
156
+ .split("\n")
157
+ .map((str, i) => (i == 0 ? str : ` ${str}`))
158
+ .join("\n")}`,
159
+ ),
160
+ ].join("\n"),
161
+ ),
162
+ ].join("\n"),
163
+ ),
164
+ ].join("\n"),
165
+ );
166
+ return operators;
167
+ };
168
+
169
+ const visitController = async (props: {
170
+ config: IConfig;
171
+ errors: IControllerError[];
172
+ operators: IOperator[];
173
+ controller: InstanceWrapper<object>;
174
+ modulePrefix: string;
175
+ }): Promise<void> => {
176
+ if (
177
+ ArrayUtil.has(
178
+ Reflect.getMetadataKeys(props.controller.metatype as Function),
179
+ PATH_METADATA,
180
+ HOST_METADATA,
181
+ SCOPE_OPTIONS_METADATA,
182
+ ) === false
183
+ )
184
+ return;
185
+
186
+ const methodErrors: IMethodError[] = [];
187
+ const controller: IController = {
188
+ name: props.controller.name,
189
+ instance: props.controller.instance,
190
+ constructor: props.controller.metatype as Function,
191
+ prototype: Object.getPrototypeOf(props.controller.instance),
192
+ prefixes: (() => {
193
+ const value: string | string[] = Reflect.getMetadata(
194
+ PATH_METADATA,
195
+ props.controller.metatype as object,
196
+ );
197
+ if (typeof value === "string") return [value];
198
+ else if (value.length === 0) return [""];
199
+ else return value;
200
+ })(),
201
+ versions: props.config.versioning
202
+ ? VersioningStrategy.cast(
203
+ Reflect.getMetadata(
204
+ VERSION_METADATA,
205
+ props.controller.metatype as Function,
206
+ ),
207
+ )
208
+ : undefined,
209
+ modulePrefix: props.modulePrefix,
210
+ };
211
+ for (const mk of getOwnPropertyNames(controller.prototype).filter(
212
+ (key) =>
213
+ key !== "constructor" && typeof controller.prototype[key] === "function",
214
+ )) {
215
+ const errorMessages: string[] = [];
216
+ visitMethod({
217
+ config: props.config,
218
+ operators: props.operators,
219
+ controller,
220
+ method: {
221
+ key: mk,
222
+ value: controller.prototype[mk],
223
+ },
224
+ report: (msg) => errorMessages.push(msg),
225
+ });
226
+ if (errorMessages.length) {
227
+ const file = await getFunctionLocation(controller.prototype[mk]);
228
+ methodErrors.push({
229
+ name: mk,
230
+ messages: errorMessages,
231
+ ...file,
232
+ source: path.relative(
233
+ process.cwd(),
234
+ file.source.replace("file:///", ""),
235
+ ),
236
+ });
237
+ }
238
+ }
239
+
240
+ if (methodErrors.length)
241
+ props.errors.push({
242
+ name: controller.name,
243
+ methods: methodErrors,
244
+ });
245
+ };
246
+
247
+ const visitMethod = (props: {
248
+ config: IConfig;
249
+ operators: IOperator[];
250
+ controller: IController;
251
+ method: Entry<Function>;
252
+ report: (message: string) => void;
253
+ }): void => {
254
+ const route: IWebSocketRouteReflect | undefined = Reflect.getMetadata(
255
+ "nestia/WebSocketRoute",
256
+ props.method.value,
257
+ );
258
+ if (typia.is<IWebSocketRouteReflect>(route) === false) return;
259
+
260
+ const parameters: IWebSocketRouteReflect.IArgument[] = (
261
+ (Reflect.getMetadata(
262
+ "nestia/WebSocketRoute/Parameters",
263
+ props.controller.prototype,
264
+ props.method.key,
265
+ ) ?? []) as IWebSocketRouteReflect.IArgument[]
266
+ ).sort((a, b) => a.index - b.index);
267
+ // acceptor must be
268
+ if (parameters.some((p) => p.category === "acceptor") === false)
269
+ return props.report(
270
+ "@WebSocketRoute.Acceptor() decorated parameter must be.",
271
+ );
272
+ // length of parameters must be fulfilled
273
+ if (parameters.length !== props.method.value.length)
274
+ return props.report(
275
+ [
276
+ "Every parameters must be one of below:",
277
+ " - @WebSocketRoute.Acceptor()",
278
+ " - @WebSocketRoute.Driver()",
279
+ " - @WebSocketRoute.Header()",
280
+ " - @WebSocketRoute.Param()",
281
+ " - @WebSocketRoute.Query()",
282
+ ].join("\n"),
283
+ );
284
+
285
+ const versions: string[] = VersioningStrategy.merge(props.config.versioning)([
286
+ ...(props.controller.versions ?? []),
287
+ ...VersioningStrategy.cast(
288
+ Reflect.getMetadata(VERSION_METADATA, props.method.value),
289
+ ),
290
+ ]);
291
+ for (const v of versions)
292
+ for (const cp of wrapPaths(props.controller.prefixes))
293
+ for (const mp of wrapPaths(route.paths)) {
294
+ const parser: Path = new Path(
295
+ "/" +
296
+ [
297
+ props.config.globalPrefix ?? "",
298
+ v,
299
+ props.controller.modulePrefix,
300
+ cp,
301
+ mp,
302
+ ]
303
+ .filter((str) => !!str.length)
304
+ .join("/")
305
+ .split("/")
306
+ .filter((str) => str.length)
307
+ .join("/"),
308
+ );
309
+ const pathParams: IWebSocketRouteReflect.IParam[] = parameters.filter(
310
+ (p) => p.category === "param",
311
+ ) as IWebSocketRouteReflect.IParam[];
312
+ if (parser.params.length !== pathParams.length) {
313
+ props.report(
314
+ [
315
+ `Path "${parser}" must have same number of parameters with @WebSocketRoute.Param()`,
316
+ ` - path: ${JSON.stringify(parser.params)}`,
317
+ ` - arguments: ${JSON.stringify(pathParams.map((p) => p.field))}`,
318
+ ].join("\n"),
319
+ );
320
+ continue;
321
+ }
322
+ const meet: boolean = pathParams
323
+ .map((p) => {
324
+ const has: boolean = parser.params.includes(p.field);
325
+ if (has === false)
326
+ props.report(
327
+ `Path "${parser}" must have parameter "${p.field}" with @WebSocketRoute.Param()`,
328
+ );
329
+ return has;
330
+ })
331
+ .every((b) => b);
332
+ if (meet === false) continue;
333
+
334
+ props.operators.push({
335
+ parser,
336
+ handler: async (input: {
337
+ params: Record<string, string>;
338
+ acceptor: WebSocketAcceptor<any, any, any>;
339
+ }): Promise<void> => {
340
+ const args: any[] = [];
341
+ try {
342
+ for (const p of parameters)
343
+ if (p.category === "acceptor") args.push(input.acceptor);
344
+ else if (p.category === "driver")
345
+ args.push(input.acceptor.getDriver());
346
+ else if (p.category === "header") {
347
+ const error: Error | null = p.validate(input.acceptor.header);
348
+ if (error !== null) throw error;
349
+ args.push(input.acceptor.header);
350
+ } else if (p.category === "param")
351
+ args.push(p.assert(input.params[p.field]!));
352
+ else if (p.category === "query") {
353
+ const query: any | Error = p.validate(
354
+ new URLSearchParams(
355
+ input.acceptor.path.indexOf("?") !== -1
356
+ ? input.acceptor.path.split("?")[1]
357
+ : "",
358
+ ),
359
+ );
360
+ if (query instanceof Error) throw query;
361
+ args.push(query);
362
+ }
363
+ } catch (exp) {
364
+ await input.acceptor.reject(
365
+ 1003,
366
+ exp instanceof Error
367
+ ? JSON.stringify({ ...exp })
368
+ : "unknown error",
369
+ );
370
+ return;
371
+ }
372
+ await props.method.value.call(props.controller.instance, ...args);
373
+ },
374
+ });
375
+ }
376
+ };
377
+
378
+ const wrapPaths = (value: string[]) => (value.length === 0 ? [""] : value);
379
+ const getOwnPropertyNames = (prototype: any): string[] => {
380
+ const result: Set<string> = new Set();
381
+ const iterate = (m: any) => {
382
+ if (m === null) return;
383
+ for (const k of Object.getOwnPropertyNames(m)) result.add(k);
384
+ iterate(Object.getPrototypeOf(m));
385
+ };
386
+ iterate(prototype);
387
+ return Array.from(result);
388
+ };
389
+
390
+ interface Entry<T> {
391
+ key: string;
392
+ value: T;
393
+ }
394
+
395
+ interface IController {
396
+ name: string;
397
+ versions: Array<string | typeof VERSION_NEUTRAL> | undefined;
398
+ instance: object;
399
+ constructor: Function;
400
+ prototype: any;
401
+ prefixes: string[];
402
+ modulePrefix: string;
403
+ }
404
+ interface IOperator {
405
+ parser: Path;
406
+ handler: (props: {
407
+ params: Record<string, string>;
408
+ acceptor: WebSocketAcceptor<any, any, any>;
409
+ }) => Promise<any>;
410
+ }
411
+ interface IConfig {
412
+ globalPrefix?: string;
413
+ versioning?: {
414
+ prefix: string;
415
+ defaultVersion?: VersionValue;
416
+ };
417
+ }
418
+
419
+ interface IControllerError {
420
+ name: string;
421
+ methods: IMethodError[];
422
+ }
423
+ interface IMethodError {
424
+ name: string;
425
+ messages: string[];
426
+ source: string;
427
+ line: number;
428
+ column: number;
429
+ }