2022-03-12 03:27:56 +00:00
|
|
|
import { ArgumentMetadata, Injectable, PipeTransform, BadRequestException } from '@nestjs/common';
|
2022-02-20 11:51:55 +00:00
|
|
|
import { validate } from 'class-validator';
|
|
|
|
import { plainToClass } from 'class-transformer';
|
|
|
|
|
|
|
|
@Injectable()
|
|
|
|
export class ValidationPipe implements PipeTransform {
|
|
|
|
async transform(value: any, { metatype }: ArgumentMetadata) {
|
|
|
|
if (!metatype || !this.toValidate(metatype)) {
|
|
|
|
return value;
|
|
|
|
}
|
|
|
|
const object = plainToClass(metatype, value);
|
|
|
|
const errors = await validate(object);
|
|
|
|
if (errors.length > 0) {
|
|
|
|
const msg = Object.values(errors[0].constraints)[0];
|
|
|
|
console.error(`Validation failed: ${msg}`);
|
|
|
|
throw new BadRequestException(`Validation failed: ${msg}`);
|
|
|
|
}
|
|
|
|
return value;
|
|
|
|
}
|
|
|
|
|
|
|
|
private toValidate(metatype: any): boolean {
|
|
|
|
const types: any[] = [String, Boolean, Number, Array, Object];
|
|
|
|
return !types.includes(metatype);
|
|
|
|
}
|
|
|
|
}
|