@nestia/core 4.3.0 → 4.4.0-dev.20241216-2

package/src/transformers/MethodTransformer.ts

@@ -1,103 +1,103 @@
-import ts from "typescript";
-
-import { INestiaTransformContext } from "../options/INestiaTransformProject";
-import { TypedRouteTransformer } from "./TypedRouteTransformer";
-import { WebSocketRouteTransformer } from "./WebSocketRouteTransformer";
-
-export namespace MethodTransformer {
-  export const transform = (props: {
-    context: INestiaTransformContext;
-    method: ts.MethodDeclaration;
-  }): ts.MethodDeclaration => {
-    const decorators: readonly ts.Decorator[] | undefined = ts.getDecorators
-      ? ts.getDecorators(props.method)
-      : (props.method as any).decorators;
-    if (!decorators?.length) return props.method;
-
-    const signature: ts.Signature | undefined =
-      props.context.checker.getSignatureFromDeclaration(props.method);
-    const original: ts.Type | undefined =
-      signature && props.context.checker.getReturnTypeOfSignature(signature);
-    const type: ts.Type | undefined =
-      original && get_escaped_type(props.context.checker)(original);
-
-    if (type === undefined) return props.method;
-
-    const operator = (decorator: ts.Decorator): ts.Decorator => {
-      decorator = TypedRouteTransformer.transform({
-        context: props.context,
-        decorator,
-        type,
-      });
-      decorator = WebSocketRouteTransformer.validate({
-        context: props.context,
-        method: props.method,
-        decorator,
-      });
-      return decorator;
-    };
-    if (ts.getDecorators !== undefined)
-      return ts.factory.updateMethodDeclaration(
-        props.method,
-        (props.method.modifiers || []).map((mod) =>
-          ts.isDecorator(mod) ? operator(mod) : mod,
-        ),
-        props.method.asteriskToken,
-        props.method.name,
-        props.method.questionToken,
-        props.method.typeParameters,
-        props.method.parameters,
-        props.method.type,
-        props.method.body,
-      );
-    // eslint-disable-next-line
-    return (ts.factory.updateMethodDeclaration as any)(
-      props.method,
-      decorators.map(operator),
-      (props.method as any).modifiers,
-      props.method.asteriskToken,
-      props.method.name,
-      props.method.questionToken,
-      props.method.typeParameters,
-      props.method.parameters,
-      props.method.type,
-      props.method.body,
-    );
-  };
-}
-
-const get_escaped_type =
-  (checker: ts.TypeChecker) =>
-  (type: ts.Type): ts.Type => {
-    const symbol: ts.Symbol | undefined = type.getSymbol() || type.aliasSymbol;
-    return symbol && get_name(symbol) === "Promise"
-      ? escape_promise(checker)(type)
-      : type;
-  };
-
-const escape_promise =
-  (checker: ts.TypeChecker) =>
-  (type: ts.Type): ts.Type => {
-    const generic: readonly ts.Type[] = checker.getTypeArguments(
-      type as ts.TypeReference,
-    );
-    if (generic.length !== 1)
-      throw new Error(
-        "Error on ImportAnalyzer.analyze(): invalid promise type.",
-      );
-    return generic[0];
-  };
-
-const get_name = (symbol: ts.Symbol): string =>
-  explore_name(symbol.getDeclarations()![0].parent)(
-    symbol.escapedName.toString(),
-  );
-
-const explore_name =
-  (decl: ts.Node) =>
-  (name: string): string =>
-    ts.isModuleBlock(decl)
-      ? explore_name(decl.parent.parent)(
-          `${decl.parent.name.getFullText().trim()}.${name}`,
-        )
-      : name;
+import ts from "typescript";
+
+import { INestiaTransformContext } from "../options/INestiaTransformProject";
+import { TypedRouteTransformer } from "./TypedRouteTransformer";
+import { WebSocketRouteTransformer } from "./WebSocketRouteTransformer";
+
+export namespace MethodTransformer {
+  export const transform = (props: {
+    context: INestiaTransformContext;
+    method: ts.MethodDeclaration;
+  }): ts.MethodDeclaration => {
+    const decorators: readonly ts.Decorator[] | undefined = ts.getDecorators
+      ? ts.getDecorators(props.method)
+      : (props.method as any).decorators;
+    if (!decorators?.length) return props.method;
+
+    const signature: ts.Signature | undefined =
+      props.context.checker.getSignatureFromDeclaration(props.method);
+    const original: ts.Type | undefined =
+      signature && props.context.checker.getReturnTypeOfSignature(signature);
+    const type: ts.Type | undefined =
+      original && get_escaped_type(props.context.checker)(original);
+
+    if (type === undefined) return props.method;
+
+    const operator = (decorator: ts.Decorator): ts.Decorator => {
+      decorator = TypedRouteTransformer.transform({
+        context: props.context,
+        decorator,
+        type,
+      });
+      decorator = WebSocketRouteTransformer.validate({
+        context: props.context,
+        method: props.method,
+        decorator,
+      });
+      return decorator;
+    };
+    if (ts.getDecorators !== undefined)
+      return ts.factory.updateMethodDeclaration(
+        props.method,
+        (props.method.modifiers || []).map((mod) =>
+          ts.isDecorator(mod) ? operator(mod) : mod,
+        ),
+        props.method.asteriskToken,
+        props.method.name,
+        props.method.questionToken,
+        props.method.typeParameters,
+        props.method.parameters,
+        props.method.type,
+        props.method.body,
+      );
+    // eslint-disable-next-line
+    return (ts.factory.updateMethodDeclaration as any)(
+      props.method,
+      decorators.map(operator),
+      (props.method as any).modifiers,
+      props.method.asteriskToken,
+      props.method.name,
+      props.method.questionToken,
+      props.method.typeParameters,
+      props.method.parameters,
+      props.method.type,
+      props.method.body,
+    );
+  };
+}
+
+const get_escaped_type =
+  (checker: ts.TypeChecker) =>
+  (type: ts.Type): ts.Type => {
+    const symbol: ts.Symbol | undefined = type.getSymbol() || type.aliasSymbol;
+    return symbol && get_name(symbol) === "Promise"
+      ? escape_promise(checker)(type)
+      : type;
+  };
+
+const escape_promise =
+  (checker: ts.TypeChecker) =>
+  (type: ts.Type): ts.Type => {
+    const generic: readonly ts.Type[] = checker.getTypeArguments(
+      type as ts.TypeReference,
+    );
+    if (generic.length !== 1)
+      throw new Error(
+        "Error on ImportAnalyzer.analyze(): invalid promise type.",
+      );
+    return generic[0];
+  };
+
+const get_name = (symbol: ts.Symbol): string =>
+  explore_name(symbol.getDeclarations()![0].parent)(
+    symbol.escapedName.toString(),
+  );
+
+const explore_name =
+  (decl: ts.Node) =>
+  (name: string): string =>
+    ts.isModuleBlock(decl)
+      ? explore_name(decl.parent.parent)(
+          `${decl.parent.name.getFullText().trim()}.${name}`,
+        )
+      : name;

package/src/transformers/NodeTransformer.ts

@@ -1,23 +1,23 @@
-import ts from "typescript";
-
-import { INestiaTransformContext } from "../options/INestiaTransformProject";
-import { MethodTransformer } from "./MethodTransformer";
-import { ParameterTransformer } from "./ParameterTransformer";
-
-export namespace NodeTransformer {
-  export const transform = (props: {
-    context: INestiaTransformContext;
-    node: ts.Node;
-  }): ts.Node =>
-    ts.isMethodDeclaration(props.node)
-      ? MethodTransformer.transform({
-          context: props.context,
-          method: props.node,
-        })
-      : ts.isParameter(props.node)
-        ? ParameterTransformer.transform({
-            context: props.context,
-            param: props.node,
-          })
-        : props.node;
-}
+import ts from "typescript";
+
+import { INestiaTransformContext } from "../options/INestiaTransformProject";
+import { MethodTransformer } from "./MethodTransformer";
+import { ParameterTransformer } from "./ParameterTransformer";
+
+export namespace NodeTransformer {
+  export const transform = (props: {
+    context: INestiaTransformContext;
+    node: ts.Node;
+  }): ts.Node =>
+    ts.isMethodDeclaration(props.node)
+      ? MethodTransformer.transform({
+          context: props.context,
+          method: props.node,
+        })
+      : ts.isParameter(props.node)
+        ? ParameterTransformer.transform({
+            context: props.context,
+            param: props.node,
+          })
+        : props.node;
+}

package/src/transformers/ParameterDecoratorTransformer.ts

@@ -1,143 +1,143 @@
-import path from "path";
-import ts from "typescript";
-
-import { INestiaTransformContext } from "../options/INestiaTransformProject";
-import { PlainBodyProgrammer } from "../programmers/PlainBodyProgrammer";
-import { TypedBodyProgrammer } from "../programmers/TypedBodyProgrammer";
-import { TypedFormDataBodyProgrammer } from "../programmers/TypedFormDataBodyProgrammer";
-import { TypedHeadersProgrammer } from "../programmers/TypedHeadersProgrammer";
-import { TypedParamProgrammer } from "../programmers/TypedParamProgrammer";
-import { TypedQueryBodyProgrammer } from "../programmers/TypedQueryBodyProgrammer";
-import { TypedQueryProgrammer } from "../programmers/TypedQueryProgrammer";
-
-export namespace ParameterDecoratorTransformer {
-  export const transform = (props: {
-    context: INestiaTransformContext;
-    type: ts.Type;
-    decorator: ts.Decorator;
-  }): ts.Decorator => {
-    //----
-    // VALIDATIONS
-    //----
-    // CHECK DECORATOR
-    if (!ts.isCallExpression(props.decorator.expression))
-      return props.decorator;
-
-    // SIGNATURE DECLARATION
-    const declaration: ts.Declaration | undefined =
-      props.context.checker.getResolvedSignature(
-        props.decorator.expression,
-      )?.declaration;
-    if (declaration === undefined) return props.decorator;
-
-    // FILE PATH
-    const file: string = path.resolve(declaration.getSourceFile().fileName);
-    if (file.indexOf(LIB_PATH) === -1 && file.indexOf(MONO_PATH) === -1)
-      return props.decorator;
-
-    //----
-    // TRANSFORMATION
-    //----
-    // FIND PROGRAMMER
-    const programmer: Programmer | undefined =
-      FUNCTORS[
-        getName(props.context.checker.getTypeAtLocation(declaration).symbol)
-      ];
-    if (programmer === undefined) return props.decorator;
-
-    // GET TYPE INFO
-    const typeNode: ts.TypeNode | undefined =
-      props.context.checker.typeToTypeNode(props.type, undefined, undefined);
-    if (typeNode === undefined) return props.decorator;
-
-    // DO TRANSFORM
-    return ts.factory.createDecorator(
-      ts.factory.updateCallExpression(
-        props.decorator.expression,
-        props.decorator.expression.expression,
-        props.decorator.expression.typeArguments,
-        programmer({
-          context: props.context,
-          modulo: props.decorator.expression.expression,
-          arguments: props.decorator.expression.arguments,
-          type: props.type,
-        }),
-      ),
-    );
-  };
-}
-
-type Programmer = (props: {
-  context: INestiaTransformContext;
-  modulo: ts.LeftHandSideExpression;
-  arguments: readonly ts.Expression[];
-  type: ts.Type;
-}) => readonly ts.Expression[];
-
-const FUNCTORS: Record<string, Programmer> = {
-  EncryptedBody: (props) =>
-    props.arguments.length
-      ? props.arguments
-      : [TypedBodyProgrammer.generate(props)],
-  TypedBody: (props) =>
-    props.arguments.length
-      ? props.arguments
-      : [TypedBodyProgrammer.generate(props)],
-  TypedHeaders: (props) =>
-    props.arguments.length
-      ? props.arguments
-      : [TypedHeadersProgrammer.generate(props)],
-  TypedParam: (props) =>
-    props.arguments.length !== 1
-      ? props.arguments
-      : TypedParamProgrammer.generate(props),
-  TypedQuery: (props) =>
-    props.arguments.length
-      ? props.arguments
-      : [TypedQueryProgrammer.generate(props)],
-  "TypedQuery.Body": (props) =>
-    props.arguments.length
-      ? props.arguments
-      : [TypedQueryBodyProgrammer.generate(props)],
-  "TypedFormData.Body": (props) =>
-    props.arguments.length === 0
-      ? [
-          ts.factory.createIdentifier("undefined"),
-          TypedFormDataBodyProgrammer.generate(props),
-        ]
-      : props.arguments.length === 1
-        ? [props.arguments[0], TypedFormDataBodyProgrammer.generate(props)]
-        : props.arguments,
-  PlainBody: (props) =>
-    props.arguments.length
-      ? props.arguments
-      : [PlainBodyProgrammer.generate(props)],
-  "WebSocketRoute.Header": (props) =>
-    props.arguments.length
-      ? props.arguments
-      : [TypedBodyProgrammer.generate(props)],
-  "WebSocketRoute.Param": (props) =>
-    props.arguments.length !== 1
-      ? props.arguments
-      : TypedParamProgrammer.generate(props),
-  "WebSocketRoute.Query": (props) =>
-    props.arguments.length
-      ? props.arguments
-      : [TypedQueryProgrammer.generate(props)],
-};
-
-const LIB_PATH = path.join("@nestia", "core", "lib", "decorators");
-const MONO_PATH = path.join("packages", "core", "lib", "decorators");
-
-const getName = (symbol: ts.Symbol): string => {
-  const parent = symbol.getDeclarations()?.[0]?.parent;
-  return parent ? exploreName(parent)(symbol.escapedName.toString()) : "__type";
-};
-const exploreName =
-  (decl: ts.Node) =>
-  (name: string): string =>
-    ts.isModuleBlock(decl)
-      ? exploreName(decl.parent.parent)(
-          `${decl.parent.name.getFullText().trim()}.${name}`,
-        )
-      : name;
+import path from "path";
+import ts from "typescript";
+
+import { INestiaTransformContext } from "../options/INestiaTransformProject";
+import { PlainBodyProgrammer } from "../programmers/PlainBodyProgrammer";
+import { TypedBodyProgrammer } from "../programmers/TypedBodyProgrammer";
+import { TypedFormDataBodyProgrammer } from "../programmers/TypedFormDataBodyProgrammer";
+import { TypedHeadersProgrammer } from "../programmers/TypedHeadersProgrammer";
+import { TypedParamProgrammer } from "../programmers/TypedParamProgrammer";
+import { TypedQueryBodyProgrammer } from "../programmers/TypedQueryBodyProgrammer";
+import { TypedQueryProgrammer } from "../programmers/TypedQueryProgrammer";
+
+export namespace ParameterDecoratorTransformer {
+  export const transform = (props: {
+    context: INestiaTransformContext;
+    type: ts.Type;
+    decorator: ts.Decorator;
+  }): ts.Decorator => {
+    //----
+    // VALIDATIONS
+    //----
+    // CHECK DECORATOR
+    if (!ts.isCallExpression(props.decorator.expression))
+      return props.decorator;
+
+    // SIGNATURE DECLARATION
+    const declaration: ts.Declaration | undefined =
+      props.context.checker.getResolvedSignature(
+        props.decorator.expression,
+      )?.declaration;
+    if (declaration === undefined) return props.decorator;
+
+    // FILE PATH
+    const file: string = path.resolve(declaration.getSourceFile().fileName);
+    if (file.indexOf(LIB_PATH) === -1 && file.indexOf(MONO_PATH) === -1)
+      return props.decorator;
+
+    //----
+    // TRANSFORMATION
+    //----
+    // FIND PROGRAMMER
+    const programmer: Programmer | undefined =
+      FUNCTORS[
+        getName(props.context.checker.getTypeAtLocation(declaration).symbol)
+      ];
+    if (programmer === undefined) return props.decorator;
+
+    // GET TYPE INFO
+    const typeNode: ts.TypeNode | undefined =
+      props.context.checker.typeToTypeNode(props.type, undefined, undefined);
+    if (typeNode === undefined) return props.decorator;
+
+    // DO TRANSFORM
+    return ts.factory.createDecorator(
+      ts.factory.updateCallExpression(
+        props.decorator.expression,
+        props.decorator.expression.expression,
+        props.decorator.expression.typeArguments,
+        programmer({
+          context: props.context,
+          modulo: props.decorator.expression.expression,
+          arguments: props.decorator.expression.arguments,
+          type: props.type,
+        }),
+      ),
+    );
+  };
+}
+
+type Programmer = (props: {
+  context: INestiaTransformContext;
+  modulo: ts.LeftHandSideExpression;
+  arguments: readonly ts.Expression[];
+  type: ts.Type;
+}) => readonly ts.Expression[];
+
+const FUNCTORS: Record<string, Programmer> = {
+  EncryptedBody: (props) =>
+    props.arguments.length
+      ? props.arguments
+      : [TypedBodyProgrammer.generate(props)],
+  TypedBody: (props) =>
+    props.arguments.length
+      ? props.arguments
+      : [TypedBodyProgrammer.generate(props)],
+  TypedHeaders: (props) =>
+    props.arguments.length
+      ? props.arguments
+      : [TypedHeadersProgrammer.generate(props)],
+  TypedParam: (props) =>
+    props.arguments.length !== 1
+      ? props.arguments
+      : TypedParamProgrammer.generate(props),
+  TypedQuery: (props) =>
+    props.arguments.length
+      ? props.arguments
+      : [TypedQueryProgrammer.generate(props)],
+  "TypedQuery.Body": (props) =>
+    props.arguments.length
+      ? props.arguments
+      : [TypedQueryBodyProgrammer.generate(props)],
+  "TypedFormData.Body": (props) =>
+    props.arguments.length === 0
+      ? [
+          ts.factory.createIdentifier("undefined"),
+          TypedFormDataBodyProgrammer.generate(props),
+        ]
+      : props.arguments.length === 1
+        ? [props.arguments[0], TypedFormDataBodyProgrammer.generate(props)]
+        : props.arguments,
+  PlainBody: (props) =>
+    props.arguments.length
+      ? props.arguments
+      : [PlainBodyProgrammer.generate(props)],
+  "WebSocketRoute.Header": (props) =>
+    props.arguments.length
+      ? props.arguments
+      : [TypedBodyProgrammer.generate(props)],
+  "WebSocketRoute.Param": (props) =>
+    props.arguments.length !== 1
+      ? props.arguments
+      : TypedParamProgrammer.generate(props),
+  "WebSocketRoute.Query": (props) =>
+    props.arguments.length
+      ? props.arguments
+      : [TypedQueryProgrammer.generate(props)],
+};
+
+const LIB_PATH = path.join("@nestia", "core", "lib", "decorators");
+const MONO_PATH = path.join("packages", "core", "lib", "decorators");
+
+const getName = (symbol: ts.Symbol): string => {
+  const parent = symbol.getDeclarations()?.[0]?.parent;
+  return parent ? exploreName(parent)(symbol.escapedName.toString()) : "__type";
+};
+const exploreName =
+  (decl: ts.Node) =>
+  (name: string): string =>
+    ts.isModuleBlock(decl)
+      ? exploreName(decl.parent.parent)(
+          `${decl.parent.name.getFullText().trim()}.${name}`,
+        )
+      : name;

package/src/transformers/ParameterTransformer.ts

@@ -1,57 +1,57 @@
-import ts from "typescript";
-
-import { INestiaTransformContext } from "../options/INestiaTransformProject";
-import { ParameterDecoratorTransformer } from "./ParameterDecoratorTransformer";
-
-export namespace ParameterTransformer {
-  export const transform = (props: {
-    context: INestiaTransformContext;
-    param: ts.ParameterDeclaration;
-  }): ts.ParameterDeclaration => {
-    // CHECK DECORATOR
-    const decorators: readonly ts.Decorator[] | undefined = ts.getDecorators
-      ? ts.getDecorators(props.param)
-      : (props.param as any).decorators;
-    if (!decorators?.length) return props.param;
-
-    // GET TYPE INFO
-    const type: ts.Type = props.context.checker.getTypeAtLocation(props.param);
-
-    // WHEN LATEST TS VERSION
-    if (ts.getDecorators !== undefined)
-      return ts.factory.updateParameterDeclaration(
-        props.param,
-        (props.param.modifiers || []).map((mod) =>
-          ts.isDecorator(mod)
-            ? ParameterDecoratorTransformer.transform({
-                context: props.context,
-                decorator: mod,
-                type,
-              })
-            : mod,
-        ),
-        props.param.dotDotDotToken,
-        props.param.name,
-        props.param.questionToken,
-        props.param.type,
-        props.param.initializer,
-      );
-    // eslint-disable-next-line
-    return (ts.factory.updateParameterDeclaration as any)(
-      props.param,
-      decorators.map((deco) =>
-        ParameterDecoratorTransformer.transform({
-          context: props.context,
-          decorator: deco,
-          type,
-        }),
-      ),
-      (props.param as any).modifiers,
-      props.param.dotDotDotToken,
-      props.param.name,
-      props.param.questionToken,
-      props.param.type,
-      props.param.initializer,
-    );
-  };
-}
+import ts from "typescript";
+
+import { INestiaTransformContext } from "../options/INestiaTransformProject";
+import { ParameterDecoratorTransformer } from "./ParameterDecoratorTransformer";
+
+export namespace ParameterTransformer {
+  export const transform = (props: {
+    context: INestiaTransformContext;
+    param: ts.ParameterDeclaration;
+  }): ts.ParameterDeclaration => {
+    // CHECK DECORATOR
+    const decorators: readonly ts.Decorator[] | undefined = ts.getDecorators
+      ? ts.getDecorators(props.param)
+      : (props.param as any).decorators;
+    if (!decorators?.length) return props.param;
+
+    // GET TYPE INFO
+    const type: ts.Type = props.context.checker.getTypeAtLocation(props.param);
+
+    // WHEN LATEST TS VERSION
+    if (ts.getDecorators !== undefined)
+      return ts.factory.updateParameterDeclaration(
+        props.param,
+        (props.param.modifiers || []).map((mod) =>
+          ts.isDecorator(mod)
+            ? ParameterDecoratorTransformer.transform({
+                context: props.context,
+                decorator: mod,
+                type,
+              })
+            : mod,
+        ),
+        props.param.dotDotDotToken,
+        props.param.name,
+        props.param.questionToken,
+        props.param.type,
+        props.param.initializer,
+      );
+    // eslint-disable-next-line
+    return (ts.factory.updateParameterDeclaration as any)(
+      props.param,
+      decorators.map((deco) =>
+        ParameterDecoratorTransformer.transform({
+          context: props.context,
+          decorator: deco,
+          type,
+        }),
+      ),
+      (props.param as any).modifiers,
+      props.param.dotDotDotToken,
+      props.param.name,
+      props.param.questionToken,
+      props.param.type,
+      props.param.initializer,
+    );
+  };
+}