StoreServiceImpl.java

package vion.service.impl;

import cn.dev33.satoken.stp.StpUtil;
import com.baomidou.mybatisplus.core.metadata.OrderItem;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.toolkit.Db;
import com.github.liaochong.myexcel.core.DefaultExcelBuilder;
import com.github.liaochong.myexcel.core.watermark.Watermark;
import com.github.liaochong.myexcel.utils.WatermarkUtil;
import com.github.yulichang.base.MPJBaseServiceImpl;
import com.github.yulichang.wrapper.MPJLambdaWrapper;
import io.github.linpeilie.Converter;
import jakarta.servlet.http.HttpServletResponse;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.dromara.hutool.core.array.ArrayUtil;
import org.dromara.hutool.core.collection.CollUtil;
import org.dromara.hutool.core.collection.ListUtil;
import org.dromara.hutool.core.date.DateUtil;
import org.dromara.hutool.core.io.file.FileNameUtil;
import org.dromara.hutool.core.io.file.FileUtil;
import org.dromara.hutool.core.lang.Opt;
import org.dromara.hutool.core.map.MapUtil;
import org.dromara.hutool.core.text.StrUtil;
import org.dromara.hutool.core.util.ObjUtil;
import org.dromara.hutool.cron.CronUtil;
import org.dromara.hutool.crypto.SecureUtil;
import org.dromara.hutool.http.html.HtmlUtil;
import org.dromara.hutool.json.JSONObject;
import org.dromara.hutool.json.JSONUtil;
import org.redisson.api.RedissonClient;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.transaction.interceptor.TransactionAspectSupport;
import org.springframework.web.multipart.MultipartFile;
import vion.constant.RedisKeyEnum;
import vion.dto.StatusDTO;
import vion.dto.StoreDTO;
import vion.mapper.StoreMapper;
import vion.mapper.TaskMapper;
import vion.model.*;
import vion.service.*;
import vion.third.DingMod;
import vion.utils.excel.AttachmentExportUtil;
import vion.vo.ContractVO;
import vion.vo.StoreVO;
import vion.vo.TagVO;
import vion.vo.UserVO;

import java.awt.*;
import java.io.File;
import java.io.IOException;
import java.util.List;
import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;

@Service
@RequiredArgsConstructor
@Slf4j
public class StoreServiceImpl extends MPJBaseServiceImpl<StoreMapper, Store> implements IStoreService {

    private final IAccountService accountService;
    private final IRContractStoreService contractStoreService;
    private final IContractPaymentService contractPaymentService;
    private final IContractService contractService;
    private final IFileService fileService;
    private final IFaultLogService faultLogService;
    private final IFormService formService;
    private final IInspectService inspectService;
    private final ITagService tagService;
    private final IRStoreTagService storeTagService;
    private final IRStoreUserService storeUserService;
    private final IStoreLogService storeLogService;
    private final ITaskTempService taskTempService;
    private final Converter converter;
    private final DingMod dingMod;
    // 引入 taskService 会循环依赖
    private final TaskMapper taskMapper;
    private final RedissonClient redissonClient;

    @Value("${fileUrl:}")
    private String fileUrl;

    @Override
    public Page<StoreVO> getStoreList(StoreDTO data, List<OrderItem> orderItemList) {
        var storeIdListByTagId = Optional.ofNullable(data.getTagId())
                .map(tagId -> storeTagService.lambdaQuery().eq(RStoreTag::getTagId, tagId).list())
                .map(storeTagList -> storeTagList.stream().map(RStoreTag::getStoreId).collect(Collectors.toList()))
                .filter(CollUtil::isNotEmpty)
                .orElse(List.of(-1L));

        var storeIdListByMainUser = Optional.ofNullable(data.getMainUser())
                .map(mainUser -> storeUserService.lambdaQuery().eq(RStoreUser::getUserId, mainUser).eq(RStoreUser::getIsMain, 1).list())
                .map(storeUserList -> storeUserList.stream().map(RStoreUser::getStoreId).collect(Collectors.toList()))
                .filter(CollUtil::isNotEmpty)
                .orElse(List.of(-1L));

        var store = converter.convert(data, Store.class);
        var page = Page.<Store>of(data.getPageNum(), data.getPageSize());
        Opt.ofEmptyAble(orderItemList).ifPresent(page::addOrder);

        // 单独处理 estDate 字段
        var estDate = store.getEstDate();
        store.setEstDate(null);

        var storeList = this.lambdaQuery(store)
                .in(ObjUtil.isNotNull(data.getTagId()), Store::getId, storeIdListByTagId)
                .in(ObjUtil.isNotNull(data.getMainUser()), Store::getId, storeIdListByMainUser)
                .gt(ObjUtil.isNotNull(estDate), Store::getEstDate, estDate)
                .in(ArrayUtil.isNotEmpty(data.getIds()), Store::getId, data.getIds())
                .page(page);
        // todo 缓存
        List<Account> accountList = accountService.list();

        var storeIdList = Opt.ofEmptyAble(storeList.getRecords())
                .map(recs -> recs.stream().map(Store::getId).collect(Collectors.toList()))
                .orElse(ListUtil.empty());

        // storeId -> 文件数量
        var store2CntMap = Opt.ofEmptyAble(storeIdList)
                .map(l -> fileService.lambdaQuery()
                        .in(FileInfo::getSourceId, l)
                        .in(FileInfo::getStoreId, l)
                        .in(FileInfo::getSourceType, 1, 24, 25, 26, 27, 28, 29, 30, 31)
                        .list())
                .map(fileList -> fileList.stream().collect(Collectors.groupingBy(FileInfo::getStoreId, Collectors.counting())))
                .orElse(MapUtil.empty());
        // storeId -> 项目绑定的合同的 list
        var storeIdMap = Opt.ofEmptyAble(storeIdList)
                .map(l -> contractStoreService.lambdaQuery().in(RContractStore::getStoreId, l).list())
                .map(contractStoreList -> contractStoreList.stream().collect(Collectors.groupingBy(RContractStore::getStoreId)))
                .orElse(MapUtil.empty());
        // 项目绑定的合同的id
        var contractIdList = Opt.ofEmptyAble(storeIdMap.values())
                .map(values -> values.stream()
                        .flatMap(List::stream)
                        .map(RContractStore::getContractId)
                        .collect(Collectors.toList()))
                .orElse(Collections.emptyList());
        // contractId -> contract
        var id2ContractMap = Opt.ofEmptyAble(contractIdList)
                .map(l -> contractService.lambdaQuery()
                        .select(Contract::getId, Contract::getName, Contract::getContractNo, Contract::getType, Contract::getSignDate, Contract::getWarrantyPeriod, Contract::getFinalDate, Contract::getStatus, Contract::getSaleName, Contract::getCustomerName, Contract::getMaintainSdate, Contract::getMaintainEdate)
                        .in(Contract::getId, l).list())
                .map(contractList -> contractList.stream().collect(Collectors.toMap(Contract::getId, Function.identity())))
                .orElse(MapUtil.empty());

        // contractId -> 合同付款约定
        var contractId2CpMap = Opt.ofEmptyAble(contractIdList)
                .map(l -> contractPaymentService.lambdaQuery().in(ContractPayment::getContractId, l).in(ContractPayment::getPaymentType, 3, 4).isNotNull(ContractPayment::getNodeDate).list())
                .map(contractPaymentList -> contractPaymentList.stream().collect(Collectors.groupingBy(ContractPayment::getContractId, Collectors.collectingAndThen(Collectors.toList(), v -> {
                    v.sort(Comparator.comparing(ContractPayment::getPaymentType).reversed());
                    return v;
                }))))
                .orElse(MapUtil.empty());

        // storeId -> taskIdList
        var store2TaskIdMap = Opt.ofEmptyAble(storeIdList)
                .map(l -> taskMapper.selectList(Wrappers.lambdaQuery(Task.class).in(Task::getStoreId, l)))
                .map(taskList -> taskList.stream().collect(Collectors.groupingBy(Task::getStoreId, Collectors.mapping(Task::getId, Collectors.toList()))))
                .orElse(MapUtil.empty());
        var taskIds = store2TaskIdMap.values().stream()
                .flatMap(List::stream)
                .collect(Collectors.toList());
        var taskServiceOrderMap = Opt.ofEmptyAble(taskIds)
                .map(t -> Db.list(Wrappers.lambdaQuery(ServiceOrder.class).in(ServiceOrder::getTaskId, taskIds)))
                .map(serviceOrderList -> serviceOrderList.stream().collect(Collectors.groupingBy(ServiceOrder::getTaskId)))
                .orElse(MapUtil.empty());

        // storeId -> storeLog
        var store2LogMap = Opt.ofEmptyAble(storeIdList)
                .map(l -> storeLogService.lambdaQuery().in(StoreLog::getStoreId, l).list())
                .map(storeLogList -> storeLogList.stream().collect(Collectors.groupingBy(StoreLog::getStoreId)))
                .orElse(MapUtil.empty());

        // storeId -> tagList
        var storeId2TagMap = Opt.ofEmptyAble(storeIdList)
                .map(l -> {
                    MPJLambdaWrapper<Tag> wrapper = new MPJLambdaWrapper<Tag>()
                            .select(Tag::getId, Tag::getName, Tag::getRemark)
                            .selectAs(RStoreTag::getStoreId, TagVO::getStoreId)
                            .leftJoin(RStoreTag.class, RStoreTag::getTagId, Tag::getId)
                            .in(RStoreTag::getStoreId, l);
                    return tagService.selectJoinList(TagVO.class, wrapper);
                })
                .filter(CollUtil::isNotEmpty)
                .map(tagVOS -> tagVOS.stream().collect(Collectors.groupingBy(TagVO::getStoreId)))
                .orElse(MapUtil.empty());


        // storeId -> userId(项目主要负责人 or 项目经理)
        var storeId2UserIdMap = Opt.ofEmptyAble(storeIdList)
                .map(l -> storeUserService.lambdaQuery().in(RStoreUser::getStoreId, l).eq(RStoreUser::getIsMain, 1).list())
                .map(suList -> suList.stream().collect(Collectors.toMap(RStoreUser::getStoreId, RStoreUser::getUserId)))
                .orElse(MapUtil.empty());

        List<StoreVO> storeVOList = new ArrayList<>();
        storeList.getRecords().forEach(item -> {
            var storeVO = converter.convert(item, StoreVO.class);
            storeVO.setAccountName(accountList.stream().filter(v -> v.getId().equals(item.getAccountId())).map(Account::getName).findFirst().orElse("--"));
            storeVO.setFileNum(store2CntMap.getOrDefault(item.getId(), 0L));

            Long storeId = item.getId();
            // 关联合同
            var contractStores = storeIdMap.get(storeId);
            storeVO.setContractCount(Opt.ofEmptyAble(contractStores).map(List::size).orElse(0));
            storeVO.setContractVOList(id2ContractMap.values().stream()
                    .filter(v -> CollUtil.isNotEmpty(contractStores) && contractStores.stream().map(RContractStore::getContractId).toList().contains(v.getId()))
                    .map(v -> converter.convert(v, ContractVO.class))
                    .collect(Collectors.toList()));

            var masterContract = id2ContractMap.getOrDefault(item.getMasterContract(), null);
            // 关联合同验收时间
            Opt.ofNullable(masterContract)
                    .map(c -> contractId2CpMap.getOrDefault(c.getId(), ListUtil.empty()))
                    .filter(CollUtil::isNotEmpty)
                    .map(l -> l.getFirst().getNodeDate())
                    .ifPresent(nodeDate -> masterContract.setFinalDate(nodeDate));
            storeVO.setMainContract(Opt.ofNullable(masterContract).map(c -> converter.convert(c, ContractVO.class)).orElse(null));

            storeVO.setTaskCount(store2TaskIdMap.getOrDefault(storeId, ListUtil.empty()).size());
            storeVO.setLogCount(store2LogMap.getOrDefault(storeId, ListUtil.empty()).size());

            // 补充服务单
            if (store2TaskIdMap.containsKey(item.getId())) {
                var cnt = store2TaskIdMap.get(item.getId())
                        .stream().mapToLong(taskId -> taskServiceOrderMap.getOrDefault(taskId, ListUtil.empty()).size()).sum();
                storeVO.setServiceOrderCount((int) cnt);
            }
            // 关联标签
            storeVO.setTagList(storeId2TagMap.getOrDefault(storeId, ListUtil.empty()));
            // 关联项目负责人 or 项目经理
            storeVO.setMainUser(storeId2UserIdMap.getOrDefault(storeId, null));
            storeVOList.add(storeVO);
        });
        return Page.<StoreVO>of(data.getPageNum(), data.getPageSize(), storeList.getTotal()).setRecords(storeVOList);
    }

    @Override
    public String updateStoreStage(StatusDTO statusDTO, String token) {
        this.update(Wrappers.<Store>lambdaUpdate()
                .set(Store::getProjectStage, statusDTO.getProjectStage())
                .eq(Store::getId, statusDTO.getStoreId()));
        UserVO userVO = (UserVO) StpUtil.getTokenSession().get("curLoginUser");

        Opt.ofNullable(statusDTO.getFiles()).ifPresent(tmpFiles -> {
            for (MultipartFile infile : tmpFiles) {
                //上传url地址
                String orgName = infile.getOriginalFilename();
                String mainName = FileNameUtil.mainName(orgName);
                String fileExt = FileNameUtil.extName(orgName);
                String filename = StrUtil.format("{}_{}.{}", mainName, DateUtil.format(new Date(), "yyyyMMdd_HHmmssSSS"), fileExt);
                String path = fileUrl + FileUtil.FILE_SEPARATOR + statusDTO.getStoreId() + FileUtil.FILE_SEPARATOR + statusDTO.getSourceId() + FileUtil.FILE_SEPARATOR + filename;
                File file = FileUtil.touch(path);
                try {
                    infile.transferTo(file);
                } catch (IOException e) {
                    log.error("保存文件出错", e);
                }
                // storeId = sourceId
                FileInfo tempFileInfo = new FileInfo();
                tempFileInfo.setStoreId(statusDTO.getStoreId());
                tempFileInfo.setSourceId(statusDTO.getSourceId());
                tempFileInfo.setSourceType(statusDTO.getSourceType());
                tempFileInfo.setName(filename);
                tempFileInfo.setUrl(path);
                tempFileInfo.setType(FileNameUtil.extName(file));
                tempFileInfo.setSha256(SecureUtil.sha256(file).toUpperCase());
                tempFileInfo.setUploader(userVO.getUsername());

                fileService.save(tempFileInfo);
            }
        });
        return "更新成功";
    }

    @Override
    public String calMaintainStatus(List<Long> storeIdList) {
        List<Store> storeList = this.lambdaQuery().in(CollUtil.isNotEmpty(storeIdList), Store::getId, storeIdList).list();
        Map<Long, List<Long>> storeId2ContractIdMap = Opt.ofEmptyAble(contractStoreService.list())
                .map(l -> l.stream().collect(Collectors.groupingBy(RContractStore::getStoreId, Collectors.mapping(RContractStore::getContractId, Collectors.toList()))))
                .orElse(MapUtil.empty());
        List<Contract> contractList = contractService.listByIds(storeId2ContractIdMap.values().stream().flatMap(List::stream).collect(Collectors.toList()));
        // storeId 为 key,List<Contract> 为 value 的 map
        Map<Long, List<Contract>> storeId2ContractListMap = storeList.stream()
                .collect(Collectors.toMap(Store::getId, store ->
                        storeId2ContractIdMap.getOrDefault(store.getId(), ListUtil.empty())
                                .stream()
                                .map(contractId -> contractList.stream().filter(contract -> contract.getId().equals(contractId)).findFirst().orElse(null))
                                .filter(Objects::nonNull)
                                .collect(Collectors.toList())));

        var contractId2CpMap = Opt.ofEmptyAble(contractPaymentService.lambdaQuery().in(ContractPayment::getPaymentType, 3, 4).isNotNull(ContractPayment::getNodeDate).list())
                .map(contractPaymentList -> contractPaymentList.stream().collect(Collectors.groupingBy(ContractPayment::getContractId, Collectors.collectingAndThen(Collectors.toList(), v -> {
                    v.sort(Comparator.comparing(ContractPayment::getPaymentType).reversed());
                    return v;
                }))))
                .orElse(MapUtil.empty());

        List<Store> updStoreList = new ArrayList<>();
        for (Store s : storeList) {
            s.setMaintainStatus("--");
            List<Contract> contracts = storeId2ContractListMap.getOrDefault(s.getId(), ListUtil.empty());
            if (CollUtil.isNotEmpty(contracts)) {
                // 1.项目关联的主合同在维保期内，那么这个项目就是在保
                Contract masterContract = contracts.stream().filter(contract -> ObjUtil.equals(contract.getId(), s.getMasterContract())).findFirst().orElse(null);
                if (ObjUtil.isNotNull(masterContract)) {
                    Integer warrantyPeriod = masterContract.getWarrantyPeriod();

                    Opt.ofNullable(contractId2CpMap.getOrDefault(masterContract.getId(), ListUtil.empty()))
                            .filter(CollUtil::isNotEmpty)
                            .map(l -> l.getFirst().getNodeDate())
                            .ifPresent(masterContract::setFinalDate);
                    Date finalDate = masterContract.getFinalDate();

                    if (ArrayUtil.isAllNotNull(finalDate, warrantyPeriod)) {
                        Date finalDateEnd = DateUtil.offsetMonth(finalDate, warrantyPeriod);
                        boolean isIn = DateUtil.isIn(new Date(), finalDate, finalDateEnd);
                        if (isIn) {
                            s.setMaintainStatus("在保");
                            Store updStore = new Store();
                            updStore.setId(s.getId());
                            updStore.setMaintainStatus(s.getMaintainStatus());
                            updStoreList.add(updStore);
                            continue;
                        } else {
                            s.setMaintainStatus("脱保");
                        }
                    }
                }

                // 2.主合同脱保，那就判断维保合同
                Set<Boolean> maintainStatusSet = new HashSet<>();
                contracts.stream()
                        .filter(contract -> ObjUtil.equals(contract.getType(), 1))
                        .map(Contract::getMaintainEdate)
                        .filter(Objects::nonNull)
                        .forEach(maintainEdate -> maintainStatusSet.add(DateUtil.compare(maintainEdate, new Date()) >= 0));
                s.setMaintainStatus(CollUtil.isEmpty(maintainStatusSet) ? s.getMaintainStatus() : maintainStatusSet.contains(true) ? "在保" : "脱保");
            }
            Store updStore = new Store();
            updStore.setId(s.getId());
            updStore.setMaintainStatus(s.getMaintainStatus());
            updStoreList.add(updStore);
        }
        this.updateBatchById(updStoreList);
        return "计算成功";
    }

    @Override
    public List<StoreVO> storeScreen() {
        MPJLambdaWrapper<Store> wrapper = new MPJLambdaWrapper<Store>()
                .select(Store::getId, Store::getName, Store::getTaskDetail, Store::getPlanFinishDate, Store::getPercentage, Store::getStuckPoint, Store::getIsImportant)
                .selectCollection(User.class, StoreVO::getUserNameList, map -> map.result(User::getUsername))
                .leftJoin(RStoreUser.class, on -> on.eq(RStoreUser::getStoreId, Store::getId).eq(RStoreUser::getIsMain, 1))
                .leftJoin(User.class, User::getId, RStoreUser::getUserId)
                .eq(Store::getIsImportant, 1)
                .orderByAsc(Store::getFinishDate);
        return this.selectJoinList(StoreVO.class, wrapper);
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public String mergeStore(StoreDTO dto) {
        // targetId 合并到 sourceId
        Long sourceId = dto.getSourceId();
        Store store = this.getById(sourceId);
        Long targetId = dto.getTargetId();
        // 项目关联合同合并
        Opt.ofEmptyAble(contractStoreService.lambdaQuery().select(RContractStore::getContractId).eq(RContractStore::getStoreId, targetId).list())
                .ifPresent(l -> {
                    List<RContractStore> existList = contractStoreService.lambdaQuery().select(RContractStore::getContractId).eq(RContractStore::getStoreId, sourceId).list();

                    List<RContractStore> insList = l.stream().filter(v -> !existList.contains(v)).map(v -> {
                        RContractStore contractStore = new RContractStore();
                        contractStore.setContractId(v.getContractId());
                        contractStore.setStoreId(sourceId);
                        return contractStore;
                    }).collect(Collectors.toList());
                    contractStoreService.saveBatch(insList);
                    contractStoreService.lambdaUpdate().eq(RContractStore::getStoreId, targetId).remove();
                });

        // 工单相关合并
        taskMapper.update(new Task(), Wrappers.lambdaUpdate(Task.class).set(Task::getStoreId, sourceId).set(Task::getAccountId, store.getAccountId()).eq(Task::getStoreId, targetId));
        taskTempService.lambdaUpdate().set(TaskTemp::getStoreId, sourceId).eq(TaskTemp::getStoreId, targetId).update();
        // 所有 targetId 的文件 转移到 sourceId
        fileService.lambdaUpdate().set(FileInfo::getStoreId, sourceId).eq(FileInfo::getStoreId, targetId).update();
        faultLogService.lambdaUpdate().set(FaultLog::getStoreId, sourceId).eq(FaultLog::getStoreId, targetId).update();
        formService.lambdaUpdate().set(Form::getStoreId, sourceId).set(Form::getAccountId, store.getAccountId()).eq(Form::getStoreId, targetId).update(new Form());
        inspectService.lambdaUpdate().set(Inspect::getStoreId, sourceId).set(Inspect::getAccountId, store.getAccountId()).eq(Inspect::getStoreId, targetId).update(new Inspect());
        // 删除 targetId 对应的项目
        this.removeById(targetId);
        return "合并成功";
    }

    @Override
    public String addTag(Long id, List<Long> tagIdList) {
        storeTagService.lambdaUpdate().eq(RStoreTag::getStoreId, id).remove();
        Opt.ofEmptyAble(tagIdList).ifPresent(l -> {
            List<RStoreTag> list = l.stream().map(tagId -> {
                RStoreTag storeTag = new RStoreTag();
                storeTag.setStoreId(id);
                storeTag.setTagId(tagId);
                return storeTag;
            }).collect(Collectors.toList());
            storeTagService.saveBatch(list);
        });
        return "成功";
    }

    @Override
    public List<Map<String, Number>> fileDetail(List<Long> storeIdList) {
        var fileList = fileService.lambdaQuery()
                .in(FileInfo::getSourceId, storeIdList)
                .in(FileInfo::getStoreId, storeIdList)
                .in(FileInfo::getSourceType, 1, 24, 25, 26, 27, 28)
                .list();
        var storeId2FileMap = fileList.stream().collect(Collectors.groupingBy(FileInfo::getStoreId, Collectors.groupingBy(FileInfo::getSourceType)));
        return storeIdList.stream().map(storeId -> {
            var fileMap = storeId2FileMap.getOrDefault(storeId, MapUtil.empty());
            Map<String, Number> map = new HashMap<>();
            map.put("storeId", storeId);
            map.put("1", fileMap.getOrDefault(1, List.of()).size());
            map.put("24", fileMap.getOrDefault(24, List.of()).size());
            map.put("25", fileMap.getOrDefault(25, List.of()).size());
            map.put("26", fileMap.getOrDefault(26, List.of()).size());
            map.put("27", fileMap.getOrDefault(27, List.of()).size());
            map.put("28", fileMap.getOrDefault(28, List.of()).size());
            return map;
        }).collect(Collectors.toList());
    }

    @Override
    public Boolean logNotify(Long id, String[] userIdStr) {
        MPJLambdaWrapper<Store> wrapper = new MPJLambdaWrapper<Store>()
                .select(Store::getName, Store::getBotToken)
                .selectCollection(StoreLog.class, StoreVO::getStoreLog)
                .leftJoin(StoreLog.class, StoreLog::getStoreId, Store::getId)
                .eq(StoreLog::getId, id);
        var storeVO = this.selectJoinOne(StoreVO.class, wrapper);
        var storeLog = storeVO.getStoreLog().getFirst();
        var user = (User) redissonClient.getBucket(RedisKeyEnum.DING_PREFIX.getVal() + RedisKeyEnum.USER_ID.getVal() + storeLog.getCreateUser()).get();

        var content = storeLog.getContent();
        var content1 = StrUtil.replace(content, "<br>", "\n");
        var logText = HtmlUtil.cleanHtmlTag(content1);
        var text = StrUtil.format("""
                ### 项目日志
                ### 项目：{}
                ### 提交人：{}
                ### 提交日期：{}
                ### 内容：
                > {}
                """, storeVO.getName(), user.getUsername(), DateUtil.formatDate(storeLog.getLogDate()), logText);
        var markdown = JSONUtil.ofObj()
                .set("title", "项目日志提醒")
                .set("text", text);

        JSONObject atObj = null;
        if (ArrayUtil.isNotEmpty(userIdStr)) {
            var arr = JSONUtil.ofArray();
            Arrays.stream(userIdStr).forEach(arr::put);
            var atUser = Arrays.stream(userIdStr).collect(Collectors.joining("@", "@", ""));
            System.out.println(atUser);
            atObj = JSONUtil.ofObj().set("atUserIds", arr);
            text = StrUtil.format("""
                    ### 项目日志
                    ### 查收人:{}
                    ### 项目：{}
                    ### 提交人：{}
                    ### 提交日期：{}
                    ### 内容：
                    > {}
                    """, atUser, storeVO.getName(), user.getUsername(), DateUtil.formatDate(storeLog.getLogDate()), logText);
            markdown.set("text", text);
        }

        var body = JSONUtil.ofObj()
                .set("msgtype", "markdown")
                .set("markdown", markdown)
                .setOpt("at", atObj);
        var pushRes = dingMod.robotPush(storeVO.getBotToken(), body.toString());
        return StrUtil.containsIgnoreCase(pushRes, "ok");
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public String addCron(Long storeId, String storeName, String cronStr, String userId) {
        String scheduleId = null;
        try {
            var update = this.lambdaUpdate().eq(Store::getId, storeId).set(Store::getCronStr, cronStr).update();

            if (update) {
                scheduleId = CronUtil.schedule(cronStr, (org.dromara.hutool.cron.task.Task) () -> {
                    if (StrUtil.isBlank(userId)) {
                        log.info("storeId:{},该项目未设置项目经理，不发送日志提醒", storeId);
                        return;
                    }
                    var user = (User) redissonClient.getBucket(RedisKeyEnum.DING_PREFIX.getVal() + RedisKeyEnum.USER_ID.getVal() + userId).get();
                    dingMod.workMsg(buildMsg(storeName, user.getUserid()));
                    log.info("storeId:{},项目发送日志提醒，提醒人:{}({})", storeId, user.getUsername(), user.getUserid());
                });
                if (!CronUtil.getScheduler().isStarted()) {
                    CronUtil.setMatchSecond(true);
                    CronUtil.start();
                }
                this.lambdaUpdate().eq(Store::getId, storeId).set(Store::getScheduleId, scheduleId).update();
                return "定时提醒任务添加成功";
            }
        } catch (Exception e) {
            TransactionAspectSupport.currentTransactionStatus().setRollbackOnly();
            return "定时提醒任务添加失败";
        } finally {
            if (StrUtil.isNotBlank(scheduleId)) {
                CronUtil.remove(scheduleId);
            }
        }
        return "定时提醒任务添加失败";
    }

    @Override
    public String stopCron(Long storeId) {
        var store = this.getById(storeId);
        var update = this.lambdaUpdate()
                .eq(Store::getId, storeId)
                .set(Store::getCronStr, null)
                .set(Store::getScheduleId, null)
                .update();
        if (update) {
            CronUtil.remove(store.getScheduleId());
            return "定时提醒任务移除成功";
        }
        return "定时提醒任务移除失败";
    }

    @Override
    public Map<Integer, Long> statState() {
        var storeList = this.lambdaQuery().eq(Store::getProjectState, 2).list();
        return storeList.stream().collect(Collectors.groupingBy(Store::getProjectStage, Collectors.counting()));
    }

    @Override
    public Boolean fileNotify(String storeName, String contractNo, String sourceName, String[] fileNameArr, String[] userIdStr) {
        var pushRes = dingMod.workMsg(buildMsg1(storeName, contractNo, sourceName, fileNameArr, userIdStr));
        return StrUtil.containsIgnoreCase(pushRes, "ok");
    }

    @Override
    public Map<String, Map<String, Map<String, Long>>> storeFileTable(StoreDTO data, List<OrderItem> orderItemList,
                                                                      HttpServletResponse response) {
        Page<StoreVO> voPage = this.getStoreList(data, orderItemList);
        var records = voPage.getRecords();
        var storeIds = records.stream().map(StoreVO::getId).toList();
        var id2NameMap = records.stream().collect(Collectors.toMap(StoreVO::getId, StoreVO::getName));

        var sourceTypeMap = redissonClient.<Integer, String>getMap(RedisKeyEnum.DICT_PREFIX.getVal() + RedisKeyEnum.SOURCE_TYPE.getVal());

        var files = Opt.ofEmptyAble(storeIds)
                .map(ids -> fileService.lambdaQuery()
                        .in(FileInfo::getSourceId, ids)
                        .in(FileInfo::getStoreId, ids)
                        .in(FileInfo::getSourceType, 1, 24, 25, 26, 27, 28, 29, 30, 31)
                        .list())
                .orElse(List.of());
        var contractId2NoMap = Opt.ofEmptyAble(files)
                .map(f -> f.stream().map(FileInfo::getContractId).toList())
                .map(contractService::listByIds)
                .filter(CollUtil::isNotEmpty)
                .map(contractList -> contractList.stream().collect(Collectors.toMap(Contract::getId, Contract::getContractNo)))
                .orElse(Map.of());
        // storeName -> <contractName -> <sourceTypeName -> cnt>>
        return Opt.ofEmptyAble(files)
                .map(fs -> fs.stream().collect(Collectors.groupingBy(f -> id2NameMap.getOrDefault(f.getStoreId(),
                                "未知项目"),
                        Collectors.groupingBy(f -> Opt.ofNullable(f.getContractId()).map(id -> contractId2NoMap.getOrDefault(id, "暂无关联合同")).orElse("暂无关联合同"),
                                Collectors.groupingBy(f -> sourceTypeMap.getOrDefault(f.getSourceType(), "未知文件来源"),
                                        Collectors.counting())))))
                .orElse(Map.of());
    }

    @Override
    public void storeFileExport(StoreDTO data, List<OrderItem> orderItemList, HttpServletResponse response) {
        UserVO user = (UserVO) StpUtil.getTokenSession().get("curLoginUser");
        Page<StoreVO> voPage = this.getStoreList(data, orderItemList);
        try (DefaultExcelBuilder<Map> defaultExcelBuilder = DefaultExcelBuilder.of(Map.class)) {
            var records = voPage.getRecords();
            var storeIds = records.stream().map(StoreVO::getId).toList();
            var titles = List.of("项目", "合同编号",
                    "前期勘察", "点位表",
                    "竣工图纸", "验收报告",
                    "项目总结", "巡检报告",
                    "结算资料", "启动会", "其它");

            var files = Opt.ofEmptyAble(storeIds)
                    .map(ids -> fileService.lambdaQuery()
                            .in(FileInfo::getSourceId, ids)
                            .in(FileInfo::getStoreId, ids)
                            .in(FileInfo::getSourceType, 1, 24, 25, 26, 27, 28, 29, 30, 31)
                            .list())
                    .orElse(List.of());
            var contractId2NoMap = Opt.ofEmptyAble(files)
                    .map(f -> f.stream().map(FileInfo::getContractId).toList())
                    .map(contractService::listByIds)
                    .filter(CollUtil::isNotEmpty)
                    .map(contractList -> contractList.stream().collect(Collectors.toMap(Contract::getId, Contract::getContractNo)))
                    .orElse(Map.of());

            // storeId -> <contractId -> <sourceType -> cnt>>
            var storeId2CntMap = Opt.ofEmptyAble(files)
                    .map(fs -> fs.stream().collect(Collectors.groupingBy(FileInfo::getStoreId,
                            Collectors.groupingBy(f -> Opt.ofNullable(f.getContractId()).orElse(-1L),
                                    Collectors.groupingBy(FileInfo::getSourceType,
                                            Collectors.counting())))))
                    .orElse(Map.of());

            List<Map> dataList = new ArrayList<>();
            storeId2CntMap.forEach((storeId, contractMap) -> {
                var store = records.stream().filter(v -> v.getId().equals(storeId)).findFirst().orElse(null);
                if (ObjUtil.isNull(store)) {
                    return;
                }

                contractMap.forEach((contractId, typeCntMap) -> {
                    var map = MapUtil.<String, Object>builder("a", store.getName())
                            .put("b", contractId2NoMap.getOrDefault(contractId, "暂无"))
                            .put("c", typeCntMap.getOrDefault(28, 0L))
                            .put("d", typeCntMap.getOrDefault(26, 0L))
                            .put("e", typeCntMap.getOrDefault(25, 0L))
                            .put("f", typeCntMap.getOrDefault(24, 0L))
                            .put("g", typeCntMap.getOrDefault(27, 0L))
                            .put("h", typeCntMap.getOrDefault(29, 0L))
                            .put("i", typeCntMap.getOrDefault(30, 0L))
                            .put("j", typeCntMap.getOrDefault(31, 0L))
                            .put("k", typeCntMap.getOrDefault(1, 0L))
                            .build();
                    dataList.add(map);
                });
            });

            var workbook = defaultExcelBuilder.sheetName("项目资料")
                    .titles(titles)
                    .widths(10, 20)
                    .fieldDisplayOrder(List.of("a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k"))
                    .build(dataList);
            // 水印添加指定字体，并在服务器上安装 SimSun 字体，解决中文字体变成方块的问题
            Watermark watermark = new Watermark();
            watermark.setText(user.getUsername() + "-" + user.getPhone());
            watermark.setFont(new Font("SimSun", Font.PLAIN, 16));
            WatermarkUtil.addWatermark(workbook, watermark);
            AttachmentExportUtil.export(workbook, StrUtil.format("项目资料_{}", DateUtil.formatDateTime(new Date())),
                    response);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    JSONObject buildMsg(String storeName, String userid) {
        var jsonObj = JSONUtil.ofObj()
                .set("agent_id", 2358374016L)
                .set("userid_list", userid);

        var msg = JSONUtil.ofObj();
        var content = JSONUtil.ofObj()
                .set("title", "项目日志提醒")
                .set("text", StrUtil.format("""
                        #### 项目：[{}]请及时提交日志
                        #### 发送时间：{}""", storeName, DateUtil.now()));

        msg.set("msgtype", "markdown").set("markdown", content);
        jsonObj.set("msg", msg);
        return jsonObj;
    }

    JSONObject buildMsg1(String storeName, String contractNo, String sourceName,
                         String[] fileNameArr, String[] userIdStr) {
        var jsonObj = JSONUtil.ofObj()
                .set("agent_id", 2358374016L)
                .set("userid_list", String.join(",", userIdStr));

        var msg = new JSONObject();
        var content = new JSONObject();

        String fileNameStr = Arrays.stream(fileNameArr).map(v -> StrUtil.format("{}. {}",
                Arrays.asList(fileNameArr).indexOf(v) + 1, v)).collect(Collectors.joining("\n"));

        content.set("title", "项目资料提交提醒")
                .set("text", StrUtil.format("""
                        ### 项目资料提交提醒
                        #### 项目名称：{}
                        #### 合同编号：{}
                        #### 资料类型：{}
                        #### 资料名称：
                        {}
                        #### 发送时间：{}""", storeName, contractNo, sourceName, fileNameStr, DateUtil.now()));

        msg.set("msgtype", "markdown").set("markdown", content);
        jsonObj.set("msg", msg);
        return jsonObj;
    }
}