import { Provide, Inject } from "@midwayjs/core";
import { InjectRepository } from "@midwayjs/sequelize";
import { Repository } from "sequelize-typescript";
import { Op, UniqueConstraintError } from "sequelize";
import { Context } from "@midwayjs/koa";
import Contract from "../entity/contract.entity";
import Customer from "../entity/customer.entity";
import SysUserRole from "../entity/sysUserRole.entity";
import SysRole from "../entity/sysRole.entity";
import { LogService } from "./log.service";

@Provide()
export class ContractService {
	@InjectRepository(Contract)
	contractModel!: Repository<Contract>;

	@InjectRepository(Customer)
	customerModel!: Repository<Customer>;

	@InjectRepository(SysUserRole)
	userRoleModel!: Repository<SysUserRole>;

	@InjectRepository(SysRole)
	roleModel!: Repository<SysRole>;

	@Inject()
	logService!: LogService;

	private async isAdmin(userId: number): Promise<boolean> {
		const userRoles = await this.userRoleModel.findAll({ where: { userId }, include: [this.roleModel] });
		return userRoles.some((ur) => (ur as any).role?.roleCode === "ADMIN");
	}

	private async isCustomerOwner(userId: number, customerId: number): Promise<boolean> {
		if (await this.isAdmin(userId)) return true;
		const customer = await this.customerModel.findByPk(customerId);
		return !!customer && customer.salesmanId === userId;
	}

	private async generateContractNo(retry: number = 0): Promise<string> {
		const now = new Date();
		const d = new Date(now.getTime() + 8 * 60 * 60 * 1000);
		const yyyy = d.getUTCFullYear();
		const yyyymmdd = `${yyyy}${String(d.getUTCMonth() + 1).padStart(2, "0")}${String(d.getUTCDate()).padStart(2, "0")}`;
		const prefix = `HT${yyyymmdd}`;
		const last = await this.contractModel.findOne({ where: { contractNo: { [Op.like]: `${prefix}%` } }, order: [["contract_no", "DESC"]] });
		let seq = 1;
		if (last) seq = (parseInt(last.contractNo.slice(-3), 10) || 0) + 1;
		const result = `${prefix}${String(seq).padStart(3, "0")}`;
		const exists = await this.contractModel.findOne({ where: { contractNo: result } });
		if (exists) {
			if (retry < 1) return this.generateContractNo(retry + 1);
			throw new Error("合同单号生成冲突，请重试");
		}
		return result;
	}

	private async generateXtContractNo(): Promise<string> {
		const now = new Date();
		const d = new Date(now.getTime() + 8 * 60 * 60 * 1000);
		const prefix = `XT${d.getUTCFullYear()}`;
		const last = await this.contractModel.findOne({ where: { xtContractNo: { [Op.like]: `${prefix}%` } }, order: [["xt_contract_no", "DESC"]] });
		let seq = 1;
		if (last) {
			const lastNo = last.xtContractNo ?? "";
			const seqStr = lastNo.slice(-4);
			seq = (parseInt(seqStr, 10) || 0) + 1;
		}
		return `${prefix}${String(seq).padStart(4, "0")}`;
	}

	async list(query: any, ctx: Context): Promise<any> {
		const page = Number(query.page) || 1;
		const pageSize = Number(query.pageSize) || 10;
		const userId = ctx.state.user.userId;
		const where: any = {};

		if (query.keyword) {
			where[Op.or] = [
				{ contractNo: { [Op.like]: `%${query.keyword}%` } },
				{ xtContractNo: { [Op.like]: `%${query.keyword}%` } },
				{ contractDesc: { [Op.like]: `%${query.keyword}%` } }
			];
		}
		if (query.customerId !== undefined) where.customerId = query.customerId;
		if (query.salesmanId) where.salesmanId = query.salesmanId;
		if (query.status !== undefined) where.status = query.status;
		if (query.signStartDate || query.signEndDate) {
			where.signDate = {};
			if (query.signStartDate) where.signDate[Op.gte] = query.signStartDate;
			if (query.signEndDate) where.signDate[Op.lte] = query.signEndDate;
		}

		if (!(await this.isAdmin(userId))) {
			const myCustomers = await this.customerModel.findAll({ where: { salesmanId: userId }, attributes: ["id"] });
			const ids = myCustomers.map((c) => c.id);
			if (ids.length === 0) return { code: 200, message: "获取项目合同列表成功", data: { list: [], total: 0, page: Number(page), pageSize: Number(pageSize) } };
			if (where.customerId !== undefined) {
				if (!ids.includes(where.customerId)) return { code: 200, message: "获取项目合同列表成功", data: { list: [], total: 0, page: Number(page), pageSize: Number(pageSize) } };
			} else {
				where.customerId = { [Op.in]: ids };
			}
			delete where.salesmanId;
		}

		const { count, rows } = await this.contractModel.findAndCountAll({ where, limit: pageSize, offset: (page - 1) * pageSize, order: [["created_at", "DESC"]] });
		return { code: 200, message: "获取项目合同列表成功", data: { list: rows.map((r) => r.toJSON()), total: count, page: Number(page), pageSize: Number(pageSize) } };
	}

	async detail(id: number, ctx: Context): Promise<any> {
		const record = await this.contractModel.findByPk(id);
		if (!record) return { code: 404, message: "项目合同不存在", data: null };
		if (!(await this.isCustomerOwner(ctx.state.user.userId, record.customerId))) return { code: 403, message: "无权限访问该合同", data: null };
		return { code: 200, message: "获取项目合同详情成功", data: record.toJSON() };
	}

	async create(data: any, ctx: Context): Promise<any> {
		const userId = ctx.state.user.userId;
		if (!(await this.isCustomerOwner(userId, data.customerId))) return { code: 403, message: "无权限为该客户创建合同", data: null };
		if (!(await this.isAdmin(userId))) data.salesmanId = userId;
		try {
			data.contractNo = await this.generateContractNo();
			if (!data.xtContractNo) data.xtContractNo = await this.generateXtContractNo();
			const created = await this.contractModel.create(data);
			await this.logService.logAction(ctx, "项目合同", "创建", data.contractNo, `创建项目合同 ${data.contractNo}`);
			return { code: 200, message: "创建项目合同成功", data: created.toJSON() };
		} catch (err: any) {
			if (err instanceof UniqueConstraintError) {
				data.contractNo = await this.generateContractNo();
				const created = await this.contractModel.create(data);
				return { code: 200, message: "创建项目合同成功", data: created.toJSON() };
			}
			throw err;
		}
	}

	async update(id: number, data: any, ctx: Context): Promise<any> {
		const record = await this.contractModel.findByPk(id);
		if (!record) return { code: 404, message: "项目合同不存在", data: null };
		const isAdmin = await this.isAdmin(ctx.state.user.userId);
		if (!isAdmin) {
			if (!(await this.isCustomerOwner(ctx.state.user.userId, record.customerId))) return { code: 403, message: "无权限修改该合同", data: null };
			delete data.salesmanId;
			delete data.customerId;
		}
		delete data.contractNo;
		await record.update(data);
		await this.logService.logAction(ctx, "项目合同", "更新", record.contractNo, `更新项目合同 ${record.contractNo}`);
		return { code: 200, message: "更新项目合同成功", data: record.toJSON() };
	}

	async delete(id: number, ctx: Context): Promise<any> {
		const record = await this.contractModel.findByPk(id);
		if (!record) return { code: 404, message: "项目合同不存在", data: null };
		if (!(await this.isCustomerOwner(ctx.state.user.userId, record.customerId))) return { code: 403, message: "无权限删除该合同", data: null };
		const contractNo = record.contractNo;
		await record.destroy();
		await this.logService.logAction(ctx, "项目合同", "删除", contractNo, `删除项目合同 ${contractNo}`);
		return { code: 200, message: "删除项目合同成功", data: null };
	}

	async findOptions(customerId: number): Promise<any> {
		const where: any = {};
		if (customerId) where.customerId = customerId;
		const records = await this.contractModel.findAll({ where, order: [["created_at", "DESC"]] });
		return { code: 200, message: "获取项目合同选项成功", data: records.map((r) => r.toJSON()) };
	}
}