package com.estate.expert.admin.initdata;

import com.estate.expert.admin.domain.common.EntityRef;
import com.estate.expert.admin.domain.exam.record.error.ErrorLevel;
import com.estate.expert.admin.domain.exam.record.error.ErrorRecordDto;
import com.estate.expert.admin.domain.exam.record.review.ErrorReviewDto;
import com.estate.expert.admin.domain.exam.record.review.ErrorReviewState;
import com.estate.expert.admin.domain.exam.record.warning.WarningRecordDto;
import com.estate.expert.admin.domain.exam.task.check.CheckTaskDto;
import com.estate.expert.admin.domain.exam.task.pass.ByPassInfoDto;
import com.estate.expert.admin.domain.exam.task.review.ReviewTaskDto;
import com.estate.expert.admin.domain.project.info.ProjectDangerousType;
import com.estate.expert.admin.domain.project.info.ProjectInfoDto;
import com.estate.expert.admin.domain.project.info.ProjectSuperDangerousType;
import com.estate.expert.admin.domain.standard.hierachy.Hierarchy;
import com.estate.expert.admin.domain.standard.hierachy.HierarchyIds;
import com.estate.expert.admin.domain.standard.problem.Problem;
import com.estate.expert.admin.initdata.common.ImportUtil;
import com.estate.expert.admin.service.*;
import lombok.Builder;
import lombok.extern.slf4j.Slf4j;

import java.time.LocalDate;
import java.util.ArrayList;
import java.util.List;
import java.util.Set;

@Slf4j
@Builder
public class ImportTaskAndRecord {

    UserService userService;

    CheckTaskService checkTaskService;

    ErrorRecordService errorRecordService;

    WarningRecordService warningRecordService;

    ProblemService problemService;

    DeviceService deviceService;

    HierarchyService hierarchyService;

    private ProjectInfoService projectInfoService;

    public void importCheckTasks() {
        var systemUser = userService.getSystemUser();
        for(int i = 1; i <= 10; i++) {
            for(int j = 1; j <= 10; j++) {
                CheckTaskDto checkTaskDto = CheckTaskDto.builder()
                        .assemblePlace("集合点_" + i + "_" + j)
                        .date(LocalDate.now())
                        .project(EntityRef.builder().id(i).build())
                        .plan(EntityRef.builder().id(j).build())
                        .groupLeaders(Set.of(EntityRef.builder().id(1).build()))
                        .groupMembers(Set.of(EntityRef.builder().id(2).build()))
                        .drivers(List.of(EntityRef.builder().id(3).build()))
                        .cars(List.of(EntityRef.builder().id(1).build()))
                        .location("地点_" + i + "_" + j)
                        .gps(new Double[]{116.0, 40.0})
                        .build();
                checkTaskService.add(systemUser, checkTaskDto);
            }
        }
    }

    public void importProjectInfos()
    {
        log.info("Importing project infos...");
        var systemUser = userService.getSystemUser();
        var dtos = new ArrayList<ProjectInfoDto>();
        for(var checkTask : checkTaskService.findAll()) {
            var projectInfoDto = ProjectInfoDto.builder()
                    .checkTask(EntityRef.builder().id(checkTask.getId()).build())
                    .detail("项目详情_任务_" + checkTask.getId())
                    .progress("项目进展_任务_" + checkTask.getId())
                    .dangerousOptions(new ProjectDangerousType[] {ProjectDangerousType.DEMOLITION})
                    .superDangerousOptions(new ProjectSuperDangerousType[] {ProjectSuperDangerousType.DEMOLITION})
                    .progressThisSeason("近一季度施工进展情况_任务_" + checkTask.getId())
                    .updateDate(LocalDate.now())
                    .build();
            dtos.add(projectInfoDto);
        }
        projectInfoService.addDtos(systemUser, dtos);
    }

    // 从Id列表中随机选择n个Id
    private List<Problem> randomSelect(List<Problem> ids, int n) {
        if(ids.isEmpty())
            return List.of();
        var cloneIds = new ArrayList<>(ids);
        var result = new ArrayList<Problem>();
        var count = Math.min(n, 30);
        for(int i = 0; i < count; i++) {
            int index = (int)(Math.random() * cloneIds.size());
            result.add(ids.get(index));
            cloneIds.remove(index);
        }
        return result;
    }

    private String getProblemHierarchyName(Problem problem, Hierarchy hierarchy) {
        return hierarchy.getContent().get(problem.getHierarchyPath()).getName();
    }

    private List<String> getProblemHierarchyNames(Problem problem, Hierarchy hierarchy) {
        return hierarchy.getContent().get(problem.getHierarchyPath()).getNames();
    }

    public void importRecords() {
        var hierarchy = hierarchyService.get(HierarchyIds.SYSTEM);

        for(var checkTask : checkTaskService.findAll()) {
            var errorRecords = new ArrayList<ErrorRecordDto>();
            var warningRecords = new ArrayList<WarningRecordDto>();

            final int checkTaskId = checkTask.getId();
            var problems = checkTaskService.getSystemProblems(checkTaskId, null);

            randomSelect(problems, checkTaskId).forEach(problem -> {
                var record = ErrorRecordDto.builder()
                        .checkTask(EntityRef.builder().id(checkTaskId).build())
                        .problem(EntityRef.builder().id(problem.getId()).build())
                        .numberFactorIndex(0)
                        .scaleFactorIndex(0)
                        .level(ErrorLevel.values()[problem.getId() % ErrorLevel.values().length])
                        .problemPart("问题部位_" + problem.getId())
                        .hierarchyName(getProblemHierarchyName(problem, hierarchy))
                        .hierarchyNames(getProblemHierarchyNames(problem, hierarchy))
                        .suggestion("建议_" + problem.getId())
                        .remark("备注_" + problem.getId())
                        .images(List.of(EntityRef.of(1), EntityRef.of(4), EntityRef.of(5)))
                        .build();
                errorRecords.add(record);

                var warningRecord = WarningRecordDto.builder()
                        .checkTask(EntityRef.builder().id(checkTaskId).build())
                        .problem(EntityRef.builder().id(problem.getId()).build())
                        .hierarchyName(getProblemHierarchyName(problem, hierarchy))
                        .hierarchyNames(getProblemHierarchyNames(problem, hierarchy))
                        .content("正确做法_" + problem.getId())
                        .remark("备注_" + problem.getId())
                        .images(List.of(EntityRef.of(1)))
                        .build();
                warningRecords.add(warningRecord);
            });

            log.info("Importing system {} error records, {} warning to task {}",
                    errorRecords.size(), warningRecords.size(), checkTaskId);
            errorRecordService.addDtos(userService.getSystemUser(), errorRecords);
            warningRecordService.addDtos(userService.getSystemUser(), warningRecords);
        }
    }

    public void importDeviceRecords() {
        for(var checkTask : checkTaskService.findAll()) {
            final int checkTaskId = checkTask.getId();
            var errorRecords = new ArrayList<ErrorRecordDto>();
            var warningRecords = new ArrayList<WarningRecordDto>();

            ImportUtil.doImport(i -> {
                var device = deviceService.get(i);;

                var deviceModel = checkTask.getPlan().getDeviceModel();
                var problems = checkTaskService.getDeviceProblems(checkTaskId, device.getType().hierarchyPath());

                var hierarchy = deviceModel.getHierarchy();
                // 给每个设备添加问题
                randomSelect(problems, checkTaskId).forEach(problem -> {
                    var record = ErrorRecordDto.builder()
                            .checkTask(EntityRef.builder().id(checkTaskId).build())
                            .device(EntityRef.builder().id(i).build())
                            .problem(EntityRef.builder().id(problem.getId()).build())
                            .numberFactorIndex(0)
                            .scaleFactorIndex(0)
                            .level(ErrorLevel.values()[problem.getId() % ErrorLevel.values().length])
                            .problemPart("问题部位_" + problem.getId())
                            .hierarchyName(getProblemHierarchyName(problem, hierarchy))
                            .hierarchyNames(getProblemHierarchyNames(problem, hierarchy))
                            .suggestion("建议_" + problem.getId())
                            .remark("备注_" + problem.getId())
                            .images(List.of(EntityRef.of(1), EntityRef.of(4)))
                            .build();
                    errorRecords.add(record);

                    // 目前设备不需要正确做法
//                    var warningRecord = WarningRecordDto.builder()
//                            .device(EntityRef.builder().id(i).build())
//                            .checkTask(EntityRef.builder().id(checkTaskId).build())
//                            .problem(EntityRef.builder().id(problem.getId()).build())
//                            .hierarchyName(getProblemHierarchyName(problem, hierarchy))
//                            .device(EntityRef.builder().id(1).build())
//                            .content("正确做法_" + problem.getId())
//                            .remark("备注_" + problem.getId())
//                            .build();
//                    warningRecords.add(warningRecord);
                });
            });

            log.info("Importing device {} error records, {} warning to checkTask  {}",
                    errorRecords.size(), warningRecords.size(), checkTask.getId());
            errorRecordService.addDtos(userService.getSystemUser(), errorRecords);
            warningRecordService.addDtos(userService.getSystemUser(), warningRecords);
        }
    }

    public void importByPassInfo(){
        log.info("Importing by pass info...");
        for(var checkTask : checkTaskService.findAll()) {
            var byPassInfo = ByPassInfoDto.builder()
                            .advice("建议_" + checkTask.getId())
                            .conclusion("结论_" + checkTask.getId())
                            .systemHierarchyPaths(List.of("/0/0/0/0/0/0", "/0/0/0/0/1/0"))
                            .build();
            checkTaskService.setByPassInfo(checkTask.getId(), byPassInfo);
        }
    }

    public void closeCheckTasks() {
        log.info("Closing check tasks...");
        for(var checkTask : checkTaskService.findAll()) {
            checkTaskService.close(checkTask.getId());
        }
    }

    public void importReviewTasks()
    {
        var systemUser = userService.getSystemUser();
        var checkTasks = checkTaskService.findAll();
        for(var checkTask : checkTasks) {
            var reviewTaskDto = ReviewTaskDto.builder()
                    .assemblePlace("集合点_" + checkTask.getId())
                    .reviewProblems(List.of(List.of(ErrorLevel.P1, ErrorLevel.P2, ErrorLevel.P3, ErrorLevel.P4).toArray(new ErrorLevel[0])))
                    .date(LocalDate.now())
                    .groupLeaders(Set.of(EntityRef.builder().id(1).build()))
                    .groupMembers(Set.of(EntityRef.builder().id(2).build()))
                    .drivers(List.of(EntityRef.builder().id(3).build()))
                    .cars(List.of(EntityRef.builder().id(1).build()))
                    .build();
            checkTaskService.addReviewTask(systemUser, checkTask.getId(), reviewTaskDto);
        }
    }

    public void importReviewRecords()
    {
        var systemUser = userService.getSystemUser();
        for(var checkTask : checkTaskService.findAll()) {
            log.info("Importing review records to task {}", checkTask.getId());
            errorRecordService.listSystemErrors(checkTask.getId()).forEach(errorRecord -> {
                var reviewRecord = ErrorReviewDto.builder()
                        .content("复核内容_" + errorRecord.getId())
                        .state(ErrorReviewState.REFORMED)
                        .images(List.of(EntityRef.of(1)))
                        .build();
                errorRecordService.addReview(systemUser, errorRecord.getId(), reviewRecord);
            });
        }
    }

    public void closeReviewTasks() {
        for(var checkTask : checkTaskService.findAll()) {
            checkTaskService.closeReviewTask(checkTask.getId());
        }
    }
}