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

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

	/**
	 * @description : Find many office folders
	 */
	public async findMany(query: Prisma.OfficeFoldersFindManyArgs) {
		query.take = 1000;
		return this.model.findMany(query);
	}

	/**
	 * @description : Create new office folder with stakeholders
	 */
	public async create(officeFolder: OfficeFolder): Promise<OfficeFolders> {
		const createArgs: Prisma.OfficeFoldersCreateArgs = {
			data: {
				folder_number: officeFolder.folder_number,
				name: officeFolder.name,
				description: officeFolder.description,
				status: EFolderStatus.LIVE,
				deed: {
					connect: {
						uid: officeFolder.deed!.uid,
					},
				},
				office: {
					connect: {
						uid: officeFolder.office!.uid,
					},
				},
				stakeholders: {
					connect: officeFolder.stakeholders?.map((stakeholder) => ({
						uid: stakeholder.uid!,
					})),
				},
			},
		};

		return this.model.create({ ...createArgs, include: { stakeholders: true } });
	}

	public async updateStatus(uid: string, status: EFolderStatus, archived_description: string | null) {
		return this.model.update({
			where: {
				uid: uid,
			},
			data: {
				status: status,
				archived_description: archived_description,
			},
		});
	}

	/**
	 * @description : Update data of an office folder
	 */
	public async update(officeFolderuid: string, officeFolder: OfficeFolder): Promise<OfficeFolders> {
		const updateArgs: Prisma.OfficeFoldersUpdateArgs = {
			where: {
				uid: officeFolderuid,
			},
			data: {
				folder_number: officeFolder.folder_number,
				name: officeFolder.name,
				description: officeFolder.description,
				stakeholders: {
					set: officeFolder.stakeholders?.map((stakeholder) => ({
						uid: stakeholder.uid!,
					})),
				},
				customers: {
					set: officeFolder.customers?.map((customer) => ({
						uid: customer.uid!,
					})),
				},
				documents: {
					set: officeFolder.documents?.map((document) => ({
						uid: document.uid!,
					})),
				},
				folder_anchor_uid: officeFolder.folder_anchor?.uid,
			},
		};

		return this.model.update({
			...updateArgs,
			include: {
				stakeholders: true,
				customers: true,
				documents: true,
				folder_anchor: true,
			},
		});
	}

	/**
	 * @description : Find one office folder
	 */
	public async findOneByUid(uid: string, query?: Prisma.OfficeFoldersInclude) {
		return this.model.findUnique({
			where: {
				uid: uid,
			},
			include: query,
		});
	}

	/**
	 * @description : Find one office folder
	 */
	public async findOneByUidWithCustomers(uid: string) {
		return this.model.findUnique({
			where: {
				uid: uid,
			},
			include: {
				customers: true,
			},
		});
	}

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

	/**
	 * @description : Find one office folder
	 */
	public async findOneByUidWithStakeholders(uid: string) {
		return this.model.findUnique({
			where: {
				uid: uid,
			},
			include: { stakeholders: true },
		});
	}

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