Merge pull request #11549 from dammy001/patch-1

chore(validation): add ability to pass custom message in file size validator
nestjs · May 16, 2023 · 927525e · 927525e
2 parents 37f332c + d46db25
commit 927525e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/packages/common/pipes/file/max-file-size.validator.ts b/packages/common/pipes/file/max-file-size.validator.ts
@@ -2,6 +2,7 @@ import { FileValidator } from './file-validator.interface';
 
 export type MaxFileSizeValidatorOptions = {
   maxSize: number;
+  message?: string | ((maxSize: number) => string);
 };
 
 /**
@@ -13,6 +14,14 @@ export type MaxFileSizeValidatorOptions = {
  */
 export class MaxFileSizeValidator extends FileValidator<MaxFileSizeValidatorOptions> {
   buildErrorMessage(): string {
+    if ('message' in this.validationOptions) {
+      if (typeof this.validationOptions.message === 'function') {
+        return this.validationOptions.message(this.validationOptions.maxSize);
+      }
+
+      return this.validationOptions.message;
+    }
+
     return `Validation failed (expected size is less than ${this.validationOptions.maxSize})`;
   }