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

package/src/generates/internal/SdkWebSocketRouteProgrammer.ts

@@ -1,323 +1,304 @@
-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 {
+  type Expression,
+  NodeFlags,
+  type Statement,
+  SyntaxKind,
+  type TypeNode,
+  factory,
+} from "@ttsc/factory";
+import { IJsDocTagInfo } from "typia";
+
+import { IdentifierFactory } from "../../factories/IdentifierFactory";
+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): Statement[] => [
+      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): Statement => {
+      const connection: string = StringUtil.escapeDuplicate([route.name])(
+        "connection",
+      );
+      return factory.createFunctionDeclaration(
+        [
+          factory.createModifier(SyntaxKind.ExportKeyword),
+          factory.createModifier(SyntaxKind.AsyncKeyword),
+        ],
+        undefined,
+        route.name,
+        undefined,
+        [
+          IdentifierFactory.parameter(
+            connection,
+            factory.createTypeReferenceNode(
+              SdkImportWizard.IConnection(importer),
+              [factory.createTypeReferenceNode(`${route.name}.Header`)],
+            ),
+          ),
+          ...SdkWebSocketParameterProgrammer.getParameterDeclarations({
+            project,
+            route,
+            provider: true,
+            prefix: true,
+          }),
+        ],
+        factory.createTypeReferenceNode("Promise", [
+          factory.createTypeReferenceNode(`${route.name}.Output`),
+        ]),
+        factory.createBlock(
+          writeFunctionBody(project)(importer)(route)(connection),
+          true,
+        ),
+      );
+    };
+
+  const writeFunctionBody =
+    (project: INestiaProject) =>
+    (importer: ImportDictionary) =>
+    (route: ITypedWebSocketRoute) =>
+    (connection: string): Statement[] => {
+      const access = (key: string) =>
+        project.config.keyword === true
+          ? factory.createPropertyAccessExpression(
+              factory.createIdentifier("props"),
+              key,
+            )
+          : factory.createIdentifier(key);
+      return [
+        local("url")(factory.createTypeReferenceNode("string"))(
+          joinPath(
+            connection,
+            factory.createCallExpression(
+              factory.createPropertyAccessExpression(
+                factory.createIdentifier(route.name),
+                "path",
+              ),
+              [],
+              project.config.keyword === true &&
+                SdkWebSocketParameterProgrammer.isPathEmpty(route) === false
+                ? [factory.createIdentifier("props")]
+                : SdkWebSocketParameterProgrammer.getEntries({
+                    project,
+                    route,
+                    provider: false,
+                    prefix: true,
+                  }).map((p) => factory.createIdentifier(p.key)),
+            ),
+          ),
+        ),
+        local("connector")(
+          factory.createTypeReferenceNode(
+            importer.external({
+              declaration: false,
+              file: "tgrid",
+              type: "element",
+              name: "WebSocketConnector",
+            }),
+            [
+              factory.createTypeReferenceNode(`${route.name}.Header`),
+              factory.createTypeReferenceNode(`${route.name}.Provider`),
+              factory.createTypeReferenceNode(`${route.name}.Listener`),
+            ],
+          ),
+        )(
+          factory.createNewExpression(
+            factory.createIdentifier(
+              importer.external({
+                declaration: false,
+                file: "tgrid",
+                type: "element",
+                name: "WebSocketConnector",
+              }),
+            ),
+            undefined,
+            [
+              factory.createAsExpression(
+                factory.createBinaryExpression(
+                  factory.createPropertyAccessExpression(
+                    factory.createIdentifier(connection),
+                    "headers",
+                  ),
+                  factory.createToken(SyntaxKind.QuestionQuestionToken),
+                  factory.createObjectLiteralExpression([], false),
+                ),
+                factory.createKeywordTypeNode(SyntaxKind.AnyKeyword),
+              ),
+              access("provider"),
+            ],
+          ),
+        ),
+        factory.createExpressionStatement(
+          factory.createAwaitExpression(
+            factory.createCallExpression(
+              factory.createPropertyAccessExpression(
+                factory.createIdentifier("connector"),
+                "connect",
+              ),
+              undefined,
+              [factory.createIdentifier("url")],
+            ),
+          ),
+        ),
+        local("driver")(
+          factory.createTypeReferenceNode(
+            importer.external({
+              declaration: true,
+              file: "tgrid",
+              type: "element",
+              name: "Driver",
+            }),
+            [factory.createTypeReferenceNode(`${route.name}.Listener`)],
+          ),
+        )(
+          factory.createCallExpression(
+            factory.createPropertyAccessExpression(
+              factory.createIdentifier("connector"),
+              "getDriver",
+            ),
+            undefined,
+            undefined,
+          ),
+        ),
+        factory.createReturnStatement(
+          factory.createObjectLiteralExpression(
+            [
+              factory.createShorthandPropertyAssignment("connector"),
+              factory.createShorthandPropertyAssignment("driver"),
+              factory.createPropertyAssignment(
+                factory.createIdentifier("reconnect"),
+                factory.createArrowFunction(
+                  [factory.createToken(SyntaxKind.AsyncKeyword)],
+                  undefined,
+                  [],
+                  undefined,
+                  factory.createToken(SyntaxKind.EqualsGreaterThanToken),
+                  factory.createAwaitExpression(
+                    factory.createCallExpression(
+                      factory.createPropertyAccessExpression(
+                        factory.createIdentifier("connector"),
+                        factory.createIdentifier("connect"),
+                      ),
+                      undefined,
+                      [factory.createIdentifier("url")],
+                    ),
+                  ),
+                ),
+              ),
+            ],
+            true,
+          ),
+        ),
+      ];
+    };
+}
+
+const local = (name: string) => (type: TypeNode) => (expression: Expression) =>
+  factory.createVariableStatement(
+    [],
+    factory.createVariableDeclarationList(
+      [factory.createVariableDeclaration(name, undefined, type, expression)],
+      NodeFlags.Const,
+    ),
+  );
+
+const joinPath = (connection: string, caller: Expression) =>
+  factory.createTemplateExpression(factory.createTemplateHead("", ""), [
+    factory.createTemplateSpan(
+      factory.createConditionalExpression(
+        factory.createCallExpression(
+          factory.createPropertyAccessExpression(
+            factory.createPropertyAccessExpression(
+              factory.createIdentifier(connection),
+              factory.createIdentifier("host"),
+            ),
+            factory.createIdentifier("endsWith"),
+          ),
+          undefined,
+          [factory.createStringLiteral("/")],
+        ),
+        factory.createToken(SyntaxKind.QuestionToken),
+        factory.createCallExpression(
+          factory.createPropertyAccessExpression(
+            factory.createPropertyAccessExpression(
+              factory.createIdentifier(connection),
+              factory.createIdentifier("host"),
+            ),
+            factory.createIdentifier("substring"),
+          ),
+          undefined,
+          [
+            factory.createNumericLiteral("0"),
+            factory.createBinaryExpression(
+              factory.createPropertyAccessExpression(
+                factory.createPropertyAccessExpression(
+                  factory.createIdentifier(connection),
+                  factory.createIdentifier("host"),
+                ),
+                factory.createIdentifier("length"),
+              ),
+              factory.createToken(SyntaxKind.MinusToken),
+              factory.createNumericLiteral("1"),
+            ),
+          ],
+        ),
+        factory.createToken(SyntaxKind.ColonToken),
+        factory.createPropertyAccessExpression(
+          factory.createIdentifier(connection),
+          factory.createIdentifier("host"),
+        ),
+      ),
+      factory.createTemplateMiddle(""),
+    ),
+    factory.createTemplateSpan(caller, factory.createTemplateTail("")),
+  ]);