Merge pull request #2608 from BC-M/feat/support-string-mappings

feat: add support for string mapping types
nestjs · Sep 11, 2023 · 27eb8e2 · 27eb8e2
2 parents 5f3caa1 + 195874f
commit 27eb8e2
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 3 deletions.
diff --git a/lib/plugin/utils/ast-utils.ts b/lib/plugin/utils/ast-utils.ts
@@ -46,6 +46,10 @@ export function isStringLiteral(type: Type) {
   return hasFlag(type, TypeFlags.StringLiteral) && !type.isUnion();
 }
 
+export function isStringMapping(type: Type) {
+  return hasFlag(type, TypeFlags.StringMapping);
+}
+
 export function isNumber(type: Type) {
   return hasFlag(type, TypeFlags.Number);
 }

diff --git a/lib/plugin/utils/plugin-utils.ts b/lib/plugin/utils/plugin-utils.ts
@@ -13,7 +13,8 @@ import {
   isInterface,
   isNumber,
   isString,
-  isStringLiteral
+  isStringLiteral,
+  isStringMapping
 } from './ast-utils';
 
 export function getDecoratorOrUndefinedByNames(
@@ -65,7 +66,7 @@ export function getTypeReferenceAsString(
   if (isBigInt(type)) {
     return { typeName: BigInt.name, arrayDepth };
   }
-  if (isString(type) || isStringLiteral(type)) {
+  if (isString(type) || isStringLiteral(type) || isStringMapping(type)) {
     return { typeName: String.name, arrayDepth };
   }
   if (isPromiseOrObservable(getText(type, typeChecker))) {

diff --git a/test/plugin/fixtures/project/cats/classes/cat.class.ts b/test/plugin/fixtures/project/cats/classes/cat.class.ts
@@ -25,4 +25,12 @@ export class Cat {
   enum: LettersEnum;
 
   enumArr: LettersEnum;
+
+  uppercaseString: Uppercase<string>;
+
+  lowercaseString: Lowercase<string>;
+
+  capitalizeString: Capitalize<string>;
+
+  uncapitalizeString: Uncapitalize<string>;
 }
diff --git a/test/plugin/fixtures/serialized-meta.fixture.ts b/test/plugin/fixtures/serialized-meta.fixture.ts
@@ -64,7 +64,11 @@ export default async () => {
               enumArr: {
                 required: true,
                 enum: t['./cats/dto/pagination-query.dto'].LettersEnum
-              }
+              },
+              uppercaseString: { required: true, type: () => String },
+              lowercaseString: { required: true, type: () => String },
+              capitalizeString: { required: true, type: () => String },
+              uncapitalizeString: { required: true, type: () => String }
             }
           }
         ],