@nestia/sdk 12.0.0-dev.20260601.1 → 12.0.0-dev.20260612.2

package/src/generates/internal/SdkWebSocketRouteProgrammer.ts

@@ -1,323 +1,331 @@
-import { Node, NodeFlags, SyntaxKind, TypeScriptFactory } from "@nestia/factory";
-import { IdentifierFactory } from "@nestia/factory";
-import { IJsDocTagInfo } from "typia";
-
-import { INestiaProject } from "../../structures/INestiaProject";
-import { ITypedWebSocketRoute } from "../../structures/ITypedWebSocketRoute";
-import { FilePrinter } from "./FilePrinter";
-import { ImportDictionary } from "./ImportDictionary";
-import { SdkImportWizard } from "./SdkImportWizard";
-import { SdkWebSocketNamespaceProgrammer } from "./SdkWebSocketNamespaceProgrammer";
-import { SdkWebSocketParameterProgrammer } from "./SdkWebSocketParameterProgrammer";
-
-export namespace SdkWebSocketRouteProgrammer {
-  export const write =
-    (project: INestiaProject) =>
-    (importer: ImportDictionary) =>
-    (route: ITypedWebSocketRoute): Node[] => [
-      FilePrinter.description(
-        writeFunction(project)(importer)(route),
-        writeDescription(route),
-      ),
-      SdkWebSocketNamespaceProgrammer.write(project)(importer)(route),
-    ];
-
-  const writeDescription = (route: ITypedWebSocketRoute): string => {
-    // MAIN DESCRIPTION
-    const comments: string[] = route.description
-      ? route.description.split("\n")
-      : [];
-
-    // COMMENT TAGS
-    const tags: IJsDocTagInfo[] = route.jsDocTags.filter(
-      (tag) =>
-        tag.name !== "param" ||
-        [...route.pathParameters, ...(route.query ? [route.query] : [])]
-          .filter((p) => p.category === "param" || p.category === "query")
-          .some((p) => p.name === tag.text?.[0]?.text),
-    );
-    if (tags.length !== 0) {
-      const content: string[] = tags.map((t) =>
-        t.text?.length
-          ? `@${t.name} ${t.text.map((e) => e.text).join("")}`
-          : `@${t.name}`,
-      );
-      comments.push("", ...new Set(content));
-    }
-
-    // POSTFIX
-    if (!!comments.length) comments.push("");
-    comments.push(
-      `@controller ${route.controller.class.name}.${route.name}`,
-      `@path ${route.path}`,
-      `@nestia Generated by Nestia - https://github.com/samchon/nestia`,
-    );
-    return comments.join("\n");
-  };
-
-  const writeFunction =
-    (project: INestiaProject) =>
-    (importer: ImportDictionary) =>
-    (route: ITypedWebSocketRoute): Node => {
-      return TypeScriptFactory.createFunctionDeclaration(
-        [
-          TypeScriptFactory.createModifier(SyntaxKind.ExportKeyword),
-          TypeScriptFactory.createModifier(SyntaxKind.AsyncKeyword),
-        ],
-        undefined,
-        route.name,
-        undefined,
-        [
-          IdentifierFactory.parameter(
-            "connection",
-            TypeScriptFactory.createTypeReferenceNode(
-              SdkImportWizard.IConnection(importer),
-              [
-                TypeScriptFactory.createTypeReferenceNode(
-                  `${route.name}.Header`,
-                ),
-              ],
-            ),
-          ),
-          ...SdkWebSocketParameterProgrammer.getParameterDeclarations({
-            project,
-            route,
-            provider: true,
-            prefix: true,
-          }),
-        ],
-        TypeScriptFactory.createTypeReferenceNode("Promise", [
-          TypeScriptFactory.createTypeReferenceNode(`${route.name}.Output`),
-        ]),
-        TypeScriptFactory.createBlock(
-          writeFunctionBody(project)(importer)(route),
-          true,
-        ),
-      );
-    };
-
-  const writeFunctionBody =
-    (project: INestiaProject) =>
-    (importer: ImportDictionary) =>
-    (route: ITypedWebSocketRoute): Node[] => {
-      const access = (key: string) =>
-        project.config.keyword === true
-          ? TypeScriptFactory.createPropertyAccessExpression(
-              TypeScriptFactory.createIdentifier("props"),
-              key,
-            )
-          : TypeScriptFactory.createIdentifier(key);
-      return [
-        local("url")(TypeScriptFactory.createTypeReferenceNode("string"))(
-          joinPath(
-            TypeScriptFactory.createCallExpression(
-              TypeScriptFactory.createPropertyAccessExpression(
-                TypeScriptFactory.createIdentifier(route.name),
-                "path",
-              ),
-              [],
-              project.config.keyword === true &&
-                SdkWebSocketParameterProgrammer.isPathEmpty(route) === false
-                ? [TypeScriptFactory.createIdentifier("props")]
-                : SdkWebSocketParameterProgrammer.getEntries({
-                    project,
-                    route,
-                    provider: false,
-                    prefix: true,
-                  }).map((p) => TypeScriptFactory.createIdentifier(p.key)),
-            ),
-          ),
-        ),
-        local("connector")(
-          TypeScriptFactory.createTypeReferenceNode(
-            importer.external({
-              declaration: false,
-              file: "tgrid",
-              type: "element",
-              name: "WebSocketConnector",
-            }),
-            [
-              TypeScriptFactory.createTypeReferenceNode(`${route.name}.Header`),
-              TypeScriptFactory.createTypeReferenceNode(
-                `${route.name}.Provider`,
-              ),
-              TypeScriptFactory.createTypeReferenceNode(
-                `${route.name}.Listener`,
-              ),
-            ],
-          ),
-        )(
-          TypeScriptFactory.createNewExpression(
-            TypeScriptFactory.createIdentifier(
-              importer.external({
-                declaration: false,
-                file: "tgrid",
-                type: "element",
-                name: "WebSocketConnector",
-              }),
-            ),
-            undefined,
-            [
-              TypeScriptFactory.createAsExpression(
-                TypeScriptFactory.createBinaryExpression(
-                  TypeScriptFactory.createPropertyAccessExpression(
-                    TypeScriptFactory.createIdentifier("connection"),
-                    "headers",
-                  ),
-                  TypeScriptFactory.createToken(
-                    SyntaxKind.QuestionQuestionToken,
-                  ),
-                  TypeScriptFactory.createObjectLiteralExpression([], false),
-                ),
-                TypeScriptFactory.createKeywordTypeNode(
-                  SyntaxKind.AnyKeyword,
-                ),
-              ),
-              access("provider"),
-            ],
-          ),
-        ),
-        TypeScriptFactory.createExpressionStatement(
-          TypeScriptFactory.createAwaitExpression(
-            TypeScriptFactory.createCallExpression(
-              TypeScriptFactory.createPropertyAccessExpression(
-                TypeScriptFactory.createIdentifier("connector"),
-                "connect",
-              ),
-              undefined,
-              [TypeScriptFactory.createIdentifier("url")],
-            ),
-          ),
-        ),
-        local("driver")(
-          TypeScriptFactory.createTypeReferenceNode(
-            importer.external({
-              declaration: true,
-              file: "tgrid",
-              type: "element",
-              name: "Driver",
-            }),
-            [
-              TypeScriptFactory.createTypeReferenceNode(
-                `${route.name}.Listener`,
-              ),
-            ],
-          ),
-        )(
-          TypeScriptFactory.createCallExpression(
-            TypeScriptFactory.createPropertyAccessExpression(
-              TypeScriptFactory.createIdentifier("connector"),
-              "getDriver",
-            ),
-            undefined,
-            undefined,
-          ),
-        ),
-        TypeScriptFactory.createReturnStatement(
-          TypeScriptFactory.createObjectLiteralExpression(
-            [
-              TypeScriptFactory.createShorthandPropertyAssignment("connector"),
-              TypeScriptFactory.createShorthandPropertyAssignment("driver"),
-              TypeScriptFactory.createPropertyAssignment(
-                TypeScriptFactory.createIdentifier("reconnect"),
-                TypeScriptFactory.createArrowFunction(
-                  [TypeScriptFactory.createToken(SyntaxKind.AsyncKeyword)],
-                  undefined,
-                  [],
-                  undefined,
-                  TypeScriptFactory.createToken(
-                    SyntaxKind.EqualsGreaterThanToken,
-                  ),
-                  TypeScriptFactory.createAwaitExpression(
-                    TypeScriptFactory.createCallExpression(
-                      TypeScriptFactory.createPropertyAccessExpression(
-                        TypeScriptFactory.createIdentifier("connector"),
-                        TypeScriptFactory.createIdentifier("connect"),
-                      ),
-                      undefined,
-                      [TypeScriptFactory.createIdentifier("url")],
-                    ),
-                  ),
-                ),
-              ),
-            ],
-            true,
-          ),
-        ),
-      ];
-    };
-}
-
-const local =
-  (name: string) => (type: Node) => (expression: Node) =>
-    TypeScriptFactory.createVariableStatement(
-      [],
-      TypeScriptFactory.createVariableDeclarationList(
-        [
-          TypeScriptFactory.createVariableDeclaration(
-            name,
-            undefined,
-            type,
-            expression,
-          ),
-        ],
-        NodeFlags.Const,
-      ),
-    );
-
-const joinPath = (caller: Node) =>
-  TypeScriptFactory.createTemplateExpression(
-    TypeScriptFactory.createTemplateHead("", ""),
-    [
-      TypeScriptFactory.createTemplateSpan(
-        TypeScriptFactory.createConditionalExpression(
-          TypeScriptFactory.createCallExpression(
-            TypeScriptFactory.createPropertyAccessExpression(
-              TypeScriptFactory.createPropertyAccessExpression(
-                TypeScriptFactory.createIdentifier("connection"),
-                TypeScriptFactory.createIdentifier("host"),
-              ),
-              TypeScriptFactory.createIdentifier("endsWith"),
-            ),
-            undefined,
-            [TypeScriptFactory.createStringLiteral("/")],
-          ),
-          TypeScriptFactory.createToken(SyntaxKind.QuestionToken),
-          TypeScriptFactory.createCallExpression(
-            TypeScriptFactory.createPropertyAccessExpression(
-              TypeScriptFactory.createPropertyAccessExpression(
-                TypeScriptFactory.createIdentifier("connection"),
-                TypeScriptFactory.createIdentifier("host"),
-              ),
-              TypeScriptFactory.createIdentifier("substring"),
-            ),
-            undefined,
-            [
-              TypeScriptFactory.createNumericLiteral("0"),
-              TypeScriptFactory.createBinaryExpression(
-                TypeScriptFactory.createPropertyAccessExpression(
-                  TypeScriptFactory.createPropertyAccessExpression(
-                    TypeScriptFactory.createIdentifier("connection"),
-                    TypeScriptFactory.createIdentifier("host"),
-                  ),
-                  TypeScriptFactory.createIdentifier("length"),
-                ),
-                TypeScriptFactory.createToken(SyntaxKind.MinusToken),
-                TypeScriptFactory.createNumericLiteral("1"),
-              ),
-            ],
-          ),
-          TypeScriptFactory.createToken(SyntaxKind.ColonToken),
-          TypeScriptFactory.createPropertyAccessExpression(
-            TypeScriptFactory.createIdentifier("connection"),
-            TypeScriptFactory.createIdentifier("host"),
-          ),
-        ),
-        TypeScriptFactory.createTemplateMiddle("", ""),
-      ),
-      TypeScriptFactory.createTemplateSpan(
-        caller,
-        TypeScriptFactory.createTemplateTail("", ""),
-      ),
-    ],
-  );
+import {
+  Node,
+  NodeFlags,
+  SyntaxKind,
+  TypeScriptFactory,
+} from "@nestia/factory";
+import { IdentifierFactory } from "@nestia/factory";
+import { IJsDocTagInfo } from "typia";
+
+import { INestiaProject } from "../../structures/INestiaProject";
+import { ITypedWebSocketRoute } from "../../structures/ITypedWebSocketRoute";
+import { StringUtil } from "../../utils/StringUtil";
+import { FilePrinter } from "./FilePrinter";
+import { ImportDictionary } from "./ImportDictionary";
+import { SdkImportWizard } from "./SdkImportWizard";
+import { SdkWebSocketNamespaceProgrammer } from "./SdkWebSocketNamespaceProgrammer";
+import { SdkWebSocketParameterProgrammer } from "./SdkWebSocketParameterProgrammer";
+
+export namespace SdkWebSocketRouteProgrammer {
+  export const write =
+    (project: INestiaProject) =>
+    (importer: ImportDictionary) =>
+    (route: ITypedWebSocketRoute): Node[] => [
+      FilePrinter.description(
+        writeFunction(project)(importer)(route),
+        writeDescription(route),
+      ),
+      SdkWebSocketNamespaceProgrammer.write(project)(importer)(route),
+    ];
+
+  const writeDescription = (route: ITypedWebSocketRoute): string => {
+    // MAIN DESCRIPTION
+    const comments: string[] = route.description
+      ? route.description.split("\n")
+      : [];
+
+    // COMMENT TAGS
+    const tags: IJsDocTagInfo[] = route.jsDocTags.filter(
+      (tag) =>
+        tag.name !== "param" ||
+        [...route.pathParameters, ...(route.query ? [route.query] : [])]
+          .filter((p) => p.category === "param" || p.category === "query")
+          .some((p) => p.name === tag.text?.[0]?.text),
+    );
+    if (tags.length !== 0) {
+      const content: string[] = tags.map((t) =>
+        t.text?.length
+          ? `@${t.name} ${t.text.map((e) => e.text).join("")}`
+          : `@${t.name}`,
+      );
+      comments.push("", ...new Set(content));
+    }
+
+    // POSTFIX
+    if (!!comments.length) comments.push("");
+    comments.push(
+      `@controller ${route.controller.class.name}.${route.name}`,
+      `@path ${route.path}`,
+      `@nestia Generated by Nestia - https://github.com/samchon/nestia`,
+    );
+    return comments.join("\n");
+  };
+
+  const writeFunction =
+    (project: INestiaProject) =>
+    (importer: ImportDictionary) =>
+    (route: ITypedWebSocketRoute): Node => {
+      const connection: string = StringUtil.escapeDuplicate([route.name])(
+        "connection",
+      );
+      return TypeScriptFactory.createFunctionDeclaration(
+        [
+          TypeScriptFactory.createModifier(SyntaxKind.ExportKeyword),
+          TypeScriptFactory.createModifier(SyntaxKind.AsyncKeyword),
+        ],
+        undefined,
+        route.name,
+        undefined,
+        [
+          IdentifierFactory.parameter(
+            connection,
+            TypeScriptFactory.createTypeReferenceNode(
+              SdkImportWizard.IConnection(importer),
+              [
+                TypeScriptFactory.createTypeReferenceNode(
+                  `${route.name}.Header`,
+                ),
+              ],
+            ),
+          ),
+          ...SdkWebSocketParameterProgrammer.getParameterDeclarations({
+            project,
+            route,
+            provider: true,
+            prefix: true,
+          }),
+        ],
+        TypeScriptFactory.createTypeReferenceNode("Promise", [
+          TypeScriptFactory.createTypeReferenceNode(`${route.name}.Output`),
+        ]),
+        TypeScriptFactory.createBlock(
+          writeFunctionBody(project)(importer)(route)(connection),
+          true,
+        ),
+      );
+    };
+
+  const writeFunctionBody =
+    (project: INestiaProject) =>
+    (importer: ImportDictionary) =>
+    (route: ITypedWebSocketRoute) =>
+    (connection: string): Node[] => {
+      const access = (key: string) =>
+        project.config.keyword === true
+          ? TypeScriptFactory.createPropertyAccessExpression(
+              TypeScriptFactory.createIdentifier("props"),
+              key,
+            )
+          : TypeScriptFactory.createIdentifier(key);
+      return [
+        local("url")(TypeScriptFactory.createTypeReferenceNode("string"))(
+          joinPath(
+            connection,
+            TypeScriptFactory.createCallExpression(
+              TypeScriptFactory.createPropertyAccessExpression(
+                TypeScriptFactory.createIdentifier(route.name),
+                "path",
+              ),
+              [],
+              project.config.keyword === true &&
+                SdkWebSocketParameterProgrammer.isPathEmpty(route) === false
+                ? [TypeScriptFactory.createIdentifier("props")]
+                : SdkWebSocketParameterProgrammer.getEntries({
+                    project,
+                    route,
+                    provider: false,
+                    prefix: true,
+                  }).map((p) => TypeScriptFactory.createIdentifier(p.key)),
+            ),
+          ),
+        ),
+        local("connector")(
+          TypeScriptFactory.createTypeReferenceNode(
+            importer.external({
+              declaration: false,
+              file: "tgrid",
+              type: "element",
+              name: "WebSocketConnector",
+            }),
+            [
+              TypeScriptFactory.createTypeReferenceNode(`${route.name}.Header`),
+              TypeScriptFactory.createTypeReferenceNode(
+                `${route.name}.Provider`,
+              ),
+              TypeScriptFactory.createTypeReferenceNode(
+                `${route.name}.Listener`,
+              ),
+            ],
+          ),
+        )(
+          TypeScriptFactory.createNewExpression(
+            TypeScriptFactory.createIdentifier(
+              importer.external({
+                declaration: false,
+                file: "tgrid",
+                type: "element",
+                name: "WebSocketConnector",
+              }),
+            ),
+            undefined,
+            [
+              TypeScriptFactory.createAsExpression(
+                TypeScriptFactory.createBinaryExpression(
+                  TypeScriptFactory.createPropertyAccessExpression(
+                    TypeScriptFactory.createIdentifier(connection),
+                    "headers",
+                  ),
+                  TypeScriptFactory.createToken(
+                    SyntaxKind.QuestionQuestionToken,
+                  ),
+                  TypeScriptFactory.createObjectLiteralExpression([], false),
+                ),
+                TypeScriptFactory.createKeywordTypeNode(SyntaxKind.AnyKeyword),
+              ),
+              access("provider"),
+            ],
+          ),
+        ),
+        TypeScriptFactory.createExpressionStatement(
+          TypeScriptFactory.createAwaitExpression(
+            TypeScriptFactory.createCallExpression(
+              TypeScriptFactory.createPropertyAccessExpression(
+                TypeScriptFactory.createIdentifier("connector"),
+                "connect",
+              ),
+              undefined,
+              [TypeScriptFactory.createIdentifier("url")],
+            ),
+          ),
+        ),
+        local("driver")(
+          TypeScriptFactory.createTypeReferenceNode(
+            importer.external({
+              declaration: true,
+              file: "tgrid",
+              type: "element",
+              name: "Driver",
+            }),
+            [
+              TypeScriptFactory.createTypeReferenceNode(
+                `${route.name}.Listener`,
+              ),
+            ],
+          ),
+        )(
+          TypeScriptFactory.createCallExpression(
+            TypeScriptFactory.createPropertyAccessExpression(
+              TypeScriptFactory.createIdentifier("connector"),
+              "getDriver",
+            ),
+            undefined,
+            undefined,
+          ),
+        ),
+        TypeScriptFactory.createReturnStatement(
+          TypeScriptFactory.createObjectLiteralExpression(
+            [
+              TypeScriptFactory.createShorthandPropertyAssignment("connector"),
+              TypeScriptFactory.createShorthandPropertyAssignment("driver"),
+              TypeScriptFactory.createPropertyAssignment(
+                TypeScriptFactory.createIdentifier("reconnect"),
+                TypeScriptFactory.createArrowFunction(
+                  [TypeScriptFactory.createToken(SyntaxKind.AsyncKeyword)],
+                  undefined,
+                  [],
+                  undefined,
+                  TypeScriptFactory.createToken(
+                    SyntaxKind.EqualsGreaterThanToken,
+                  ),
+                  TypeScriptFactory.createAwaitExpression(
+                    TypeScriptFactory.createCallExpression(
+                      TypeScriptFactory.createPropertyAccessExpression(
+                        TypeScriptFactory.createIdentifier("connector"),
+                        TypeScriptFactory.createIdentifier("connect"),
+                      ),
+                      undefined,
+                      [TypeScriptFactory.createIdentifier("url")],
+                    ),
+                  ),
+                ),
+              ),
+            ],
+            true,
+          ),
+        ),
+      ];
+    };
+}
+
+const local = (name: string) => (type: Node) => (expression: Node) =>
+  TypeScriptFactory.createVariableStatement(
+    [],
+    TypeScriptFactory.createVariableDeclarationList(
+      [
+        TypeScriptFactory.createVariableDeclaration(
+          name,
+          undefined,
+          type,
+          expression,
+        ),
+      ],
+      NodeFlags.Const,
+    ),
+  );
+
+const joinPath = (connection: string, caller: Node) =>
+  TypeScriptFactory.createTemplateExpression(
+    TypeScriptFactory.createTemplateHead("", ""),
+    [
+      TypeScriptFactory.createTemplateSpan(
+        TypeScriptFactory.createConditionalExpression(
+          TypeScriptFactory.createCallExpression(
+            TypeScriptFactory.createPropertyAccessExpression(
+              TypeScriptFactory.createPropertyAccessExpression(
+                TypeScriptFactory.createIdentifier(connection),
+                TypeScriptFactory.createIdentifier("host"),
+              ),
+              TypeScriptFactory.createIdentifier("endsWith"),
+            ),
+            undefined,
+            [TypeScriptFactory.createStringLiteral("/")],
+          ),
+          TypeScriptFactory.createToken(SyntaxKind.QuestionToken),
+          TypeScriptFactory.createCallExpression(
+            TypeScriptFactory.createPropertyAccessExpression(
+              TypeScriptFactory.createPropertyAccessExpression(
+                TypeScriptFactory.createIdentifier(connection),
+                TypeScriptFactory.createIdentifier("host"),
+              ),
+              TypeScriptFactory.createIdentifier("substring"),
+            ),
+            undefined,
+            [
+              TypeScriptFactory.createNumericLiteral("0"),
+              TypeScriptFactory.createBinaryExpression(
+                TypeScriptFactory.createPropertyAccessExpression(
+                  TypeScriptFactory.createPropertyAccessExpression(
+                    TypeScriptFactory.createIdentifier(connection),
+                    TypeScriptFactory.createIdentifier("host"),
+                  ),
+                  TypeScriptFactory.createIdentifier("length"),
+                ),
+                TypeScriptFactory.createToken(SyntaxKind.MinusToken),
+                TypeScriptFactory.createNumericLiteral("1"),
+              ),
+            ],
+          ),
+          TypeScriptFactory.createToken(SyntaxKind.ColonToken),
+          TypeScriptFactory.createPropertyAccessExpression(
+            TypeScriptFactory.createIdentifier(connection),
+            TypeScriptFactory.createIdentifier("host"),
+          ),
+        ),
+        TypeScriptFactory.createTemplateMiddle("", ""),
+      ),
+      TypeScriptFactory.createTemplateSpan(
+        caller,
+        TypeScriptFactory.createTemplateTail("", ""),
+      ),
+    ],
+  );