add constraint

src/modules/database/constraints/data.exist.constraint.ts

@@ -0,0 +1,69 @@
+import { Injectable } from '@nestjs/common';
+import {
+    ValidationArguments,
+    ValidatorConstraint,
+    ValidatorConstraintInterface,
+    ValidationOptions,
+    registerDecorator,
+} from 'class-validator';
+import { ObjectType, Repository, DataSource } from 'typeorm';
+
+type Condition = {
+    entity: ObjectType<any>;
+    map?: string;
+};
+
+@ValidatorConstraint({ name: 'dataExist', async: true })
+@Injectable()
+export class DataExistConstraint implements ValidatorConstraintInterface {
+    constructor(private dataSource: DataSource) {}
+
+    async validate(value: any, validationArguments?: ValidationArguments) {
+        let repo: Repository<any>;
+        if (!value) {
+            return true;
+        }
+        let map = 'id';
+        if ('entity' in validationArguments.constraints[0]) {
+            map = validationArguments.constraints[0].map ?? 'id';
+            repo = this.dataSource.getRepository(validationArguments.constraints[0].entitiy);
+        } else {
+            repo = this.dataSource.getRepository(validationArguments.constraints[0]);
+        }
+        const item = await repo.findOne({ where: { [map]: value } });
+        return !!item;
+    }
+    defaultMessage?(validationArguments?: ValidationArguments): string {
+        if (!validationArguments.constraints[0]) {
+            return 'Model not been specified!';
+        }
+        return `All instance of ${validationArguments.constraints[0].name} must been exists in databse!`;
+    }
+}
+
+function IsDataExist(
+    entity: ObjectType<any>,
+    validationOptions?: ValidationOptions,
+): (object: RecordAny, propertyName: string) => void;
+
+function IsDataExist(
+    condition: Condition,
+    validationOptions?: ValidationOptions,
+): (object: RecordAny, propertyName: string) => void;
+
+function IsDataExist(
+    condition: Condition | ObjectType<any>,
+    validationOptions?: ValidationOptions,
+): (object: RecordAny, propertyName: string) => void {
+    return (object: RecordAny, propertyName: string) => {
+        registerDecorator({
+            target: object.constructor,
+            propertyName,
+            options: validationOptions,
+            constraints: [condition],
+            validator: DataExistConstraint,
+        });
+    };
+}
+
+export { IsDataExist };