lecoffre-back/src/common/repositories/UsersRepository.ts

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

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

	/**
	 * @description : Find many users
	 */
	public async findMany(query: Prisma.UsersFindManyArgs) {
		query.take = Math.min(query.take || this.defaultFetchRows, this.maxFetchRows);
		return this.model.findMany(query);
	}

	/**
	 * @description : Create a user
	 */
	public async create(user: User): Promise<Users> {
		const createArgs: Prisma.UsersCreateArgs = {
			data: {
				idNot: user.idNot,
				office_membership: {
					connectOrCreate: {
						where: {
							idNot: user.office_membership!.idNot,
						},
						create: {
							idNot: user.office_membership!.idNot,
							name: user.office_membership!.name,
							crpcen: user.office_membership!.crpcen,
							address: {
								create: {
									address: user.office_membership!.address!.address,
									zip_code: user.office_membership!.address!.zip_code,
									city: user.office_membership!.address!.city,
								},
							},
						},
					},
				},
				contact: {
					create: {
						first_name: user.contact!.first_name,
						last_name: user.contact!.last_name,
						email: user.contact!.email,
						phone_number: user.contact?.phone_number,
						cell_phone_number: user.contact!.cell_phone_number,
						civility: ECivility[user.contact?.civility as keyof typeof ECivility],
					},
				},
				role: {
					connect: {
						uid: user.role!.uid,
					},
				},
			},
		};
		if (user.contact!.address) {
			createArgs.data.contact!.create!.address!.create = {
				address: user.contact!.address.address,
				zip_code: user.contact!.address.zip_code,
				city: user.contact!.address.city,
			};
		}
		if (user.office_role) {
			createArgs.data.office_role = {
				connect: {
					uid: user.office_role.uid,
				},
			};
		}
		return this.model.create(createArgs);
	}

	/**
	 * @description : Update data from a user
	 */
	public async update(uid: string, user: User): Promise<Users> {
		const updateArgs: Prisma.UsersUpdateArgs = {
			where: {
				uid: uid,
			},
			data: {
				idNot: user.idNot,
				contact: {
					update: {
						first_name: user.contact?.first_name,
						last_name: user.contact?.last_name,
						email: user.contact?.email,
						phone_number: user.contact?.phone_number,
						cell_phone_number: user.contact?.cell_phone_number,
						civility: ECivility[user.contact?.civility as keyof typeof ECivility],
					},
				},
			},
		};

		if(user.office_membership) {
			updateArgs.data.office_membership = {
				connect: {
					idNot: user.office_membership?.idNot,
				},
			};
		}

		if (user.contact?.address) {
			updateArgs.data.contact!.update!.address!.update = {
				address: user.contact?.address?.address,
				zip_code: user.contact?.address?.zip_code,
				city: user.contact?.address?.city,
			};
		}

		if(user.office_role) {
			updateArgs.data.office_role = {
				connect: {
					uid: user.office_role.uid,
				},
			};
		}

		if(user.role) {
			updateArgs.data.role = {
				connect: {
					uid: user.role.uid,
				},
			};
		}

		return this.model.update({ ...updateArgs, include: { contact: true, office_membership: { include: { address: true } } } });
	}

	/**
	 * @description : Update role from a user
	 */
	public async updateRole(uid: string, user: User): Promise<Users> {
		const updateArgs: Prisma.UsersUpdateArgs = {
			where: {
				uid: uid,
			},
			data: {}
		};

		if(user.office_role) {
			updateArgs.data.office_role = {
				connect: {
					uid: user.office_role.uid,
				},
			};
		}

		if(user.role) {
			updateArgs.data.role = {
				connect: {
					uid: user.role.uid,
				},
			};
		}

		return this.model.update({ ...updateArgs, include: { contact: true, office_membership: { include: { address: true } } } });
	}

	/**
	 * @description : Find one user
	 */
	public async findOneByUid(uid: string, query?: Prisma.UsersInclude): Promise<Users | null> {
		return this.model.findUnique({
			where: {
				uid: uid,
			},
			include: query,
		});
	}

	/**
	 * @description : Find one user with office
	 */
	public async findOneByUidWithOffice(uid: string) {
		return this.model.findUnique({
			where: {
				uid: uid,
			},
			include: { office_membership: true },
		});
	}

	/**
	 * @description : Find one user with office
	 */
	public async findOneByUidWithRole(uid: string): Promise<((Users & {role: Role} )| null)> {
		return this.model.findUnique({
			where: {
				uid: uid,
			},
			include: { role: true },
		});
	}

	/**
	 * @description : Find one user
	 */
	public async findOneByProvider(providerName: string, id: string) {
		return this.model.findUnique({
			where: { [providerName]: id },
			include: {
				role: { include: { rules: true } },
				office_role: { include: { rules: true } },
				office_membership: true,
			},
		});
	}
}