package com.estate.expert.admin.service.impl;

import com.estate.expert.admin.domain.common.request.PageReq;
import com.estate.expert.admin.domain.common.response.PageResp;
import com.estate.expert.admin.domain.exam.record.warning.WarningRecord;
import com.estate.expert.admin.domain.exam.record.warning.WarningRecordDto;
import com.estate.expert.admin.domain.exam.record.warning.WarningRecordRepository;
import com.estate.expert.admin.domain.user.User;
import com.estate.expert.admin.enums.BizResultCode;
import com.estate.expert.admin.service.CheckTaskService;
import com.estate.expert.admin.service.ProblemService;
import com.estate.expert.admin.service.WarningRecordService;
import com.estate.expert.admin.service.common.CrudServiceBaseImpl;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import java.time.LocalDate;
import java.util.List;
import java.util.UUID;

/**
 * 正确做法记录服务实现
 */
@Service
public class WarningRecordServiceImpl extends CrudServiceBaseImpl<WarningRecord, WarningRecordDto> implements WarningRecordService {

    @Resource
    private WarningRecordRepository warningRecordRepository;

    @Resource
    private CheckTaskService checkTaskService;

    @Resource
    private ProblemService problemService;

    @Override
    public List<WarningRecord> list(Integer checkTaskId, Integer creatorId) {
        return warningRecordRepository.list(checkTaskId, creatorId, PageReq.maxPageReq().toPageRequest()).toList();
    }

    @Override
    public PageResp<WarningRecordDto> list(Integer checkTaskId, Integer creatorId, PageReq pageReq) {
        var page = warningRecordRepository.list(checkTaskId, creatorId, pageReq.toPageRequest());
        return PageResp.of(page, toDto(page.getContent()));
    }

    @Override
    public PageResp<WarningRecordDto> list(Integer clientId, Integer planId, Integer taskId, LocalDate startDate, LocalDate endDate, PageReq pageReq) {
        var page = warningRecordRepository.list(clientId, planId, taskId,
                startDate != null ? startDate.atTime(0,0,0) : null,
                endDate != null ? endDate.atTime(23, 59, 59) : null,
                pageReq.toPageRequest());
        return PageResp.of(page, toDto(page.getContent()));
    }

    @Override
    public List<WarningRecord> list(Integer clientId, Integer planId, Integer taskId, LocalDate startDate, LocalDate endDate) {
        return warningRecordRepository.list(clientId, planId, taskId,
                startDate != null ? startDate.atTime(0,0,0) : null,
                endDate != null ? endDate.atTime(23, 59, 59) : null);
    }

    @Override
    public WarningRecord add(User creator, WarningRecord entity) {
        var checkTask = checkTaskService.get(entity.getCheckTask().getId());
        BizResultCode.TASK_NOT_FOUND.check(checkTask == null);

        var hierarchyNames = problemService.getHierarchyNames(entity.getProblem());
        entity.setHierarchyNames(hierarchyNames);
        if(!hierarchyNames.isEmpty())
            entity.setHierarchyName(hierarchyNames.get(hierarchyNames.size()-1));
        entity.setName(UUID.randomUUID().toString());
        return super.add(creator, entity);
    }

    @Override
    public WarningRecord edit(WarningRecord entity) {
        var checkTask = checkTaskService.get(entity.getCheckTask().getId());
        BizResultCode.TASK_NOT_FOUND.check(checkTask == null);

        return super.edit(entity);
    }
}