import Database from "@Common/databases/database";
import BaseRepository from "@Repositories/BaseRepository";
import { Service } from "typedi";
import { Roles, Prisma } from "@prisma/client";
import { Role } from "le-coffre-resources/dist/SuperAdmin";

@Service()
export default class RolesRepository extends BaseRepository {
	constructor(private database: Database) {
		super();
	}
	protected get model() {
		return this.database.getClient().roles;
	}
	protected get instanceDb() {
		return this.database.getClient();
	}

	/**
	 * @description : Find many roles
	 */
	public async findMany(query: any): Promise<Roles[]> {
		query.take = Math.min(query.take || this.defaultFetchRows, this.maxFetchRows);
		return this.model.findMany(query);
	}

	/**
	 * @description : Create new role with rules
	 */
	public async create(role: Role): Promise<Roles> {
		const createArgs: Prisma.RolesCreateArgs = {
			data: {
				name: role.name,
				role_has_rules: {
					createMany: {
						data: role.role_has_rules!.rules.map((relation) => ({
							rule_uid: relation.uid!,
						})),
						skipDuplicates: true,
					},
				},
			},
		};
		if (role.office_role_has_rules) {
			createArgs.data.office_role_has_rules = {
					createMany: {
						data: role.office_role_has_rules.rules.map((relation) => ({
							office_uid: role.office_role_has_rules!.office.uid!,
							rule_uid: relation.uid!,
						})),
						skipDuplicates: true,
					},
			};
		}

		return this.model.create(createArgs);
	}

	/**
	 * @description : Update data of a role with rules
	 */
	public async update(role: Role): Promise<Roles> {
		const updateArgs: Prisma.RolesUpdateArgs = {
			where: {
				uid: role.uid,
			},
			data: {
				name: role.name,
				role_has_rules: {
					deleteMany: { role_uid: role.uid },
					createMany: {
						data: role.role_has_rules!.rules.map((relation) => ({
							rule_uid: relation.uid!,
						})),
						skipDuplicates: true,
					},
				},
			},
		};
		if (role.office_role_has_rules) {
			updateArgs.data.office_role_has_rules = {
				deleteMany: { role_uid: role.uid },
				createMany: {
					data: role.office_role_has_rules.rules.map((relation) => ({
						office_uid: role.office_role_has_rules!.office.uid!,
						rule_uid: relation.uid!,
					})),
					skipDuplicates: true,
				},
			};
		}

		return this.model.update(updateArgs);
	}

	/**
	 * @description : Find one role
	 */
	public async findOneByUid(uid: string, query?: any): Promise<Roles | null> {
		const findOneArgs: Prisma.RolesFindUniqueArgs = {
			where: {
				uid: uid,
			},
		};
		if (query) {
			findOneArgs.include = query;
		}
		const roleEntity = await this.model.findUnique(findOneArgs);

		return roleEntity;
	}

	/**
	 * @description : Delete a role
	 */
	public async delete(uid: string): Promise<Roles> {
		return this.model.delete({
			where: {
				uid: uid,
			},
		});
	}
}