168 lines
7.6 KiB
TypeScript
168 lines
7.6 KiB
TypeScript
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()) };
|
|
}
|
|
} |