mirror of https://github.com/fantasticit/think.git
26 lines
888 B
TypeScript
26 lines
888 B
TypeScript
import { ArgumentMetadata, BadRequestException, Injectable, PipeTransform } from '@nestjs/common';
|
|
import { plainToClass } from 'class-transformer';
|
|
import { validate } from 'class-validator';
|
|
|
|
@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);
|
|
}
|
|
}
|