package com.ard.sample.wayline.service.impl;

import com.ard.route.service.IArdRouteInfoService;
import com.ard.sample.common.model.Pagination;
import com.ard.sample.common.model.PaginationData;
import com.ard.sample.manage.model.dto.DeviceDTO;
import com.ard.sample.manage.service.IDeviceService;
import com.ard.sample.media.service.IFileService;
import com.ard.sample.wayline.mapper.IWaylineJobHistoryMapper;
import com.ard.sample.wayline.model.dto.WaylineJobDTO;
import com.ard.sample.wayline.model.entity.WaylineJobEntity;
import com.ard.sample.wayline.model.entity.WaylineJobHistoryEntity;
import com.ard.sample.wayline.model.enums.WaylineJobStatusEnum;
import com.ard.sample.wayline.service.IWaylineJobHistoryService;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;

import java.time.Instant;
import java.time.LocalDateTime;
import java.time.ZoneId;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.stream.Collectors;

@Service
@Transactional
@Slf4j
@SuppressWarnings("all")
public class WaylineJobHistoryServiceImpl implements IWaylineJobHistoryService {
    @Autowired
    private IWaylineJobHistoryMapper mapper;

    @Autowired
    private IFileService fileService;

    @Autowired
    private IDeviceService deviceService;

    @Override
    public Optional<WaylineJobDTO> createWaylineJob(WaylineJobHistoryEntity entity) {
        int id = mapper.insert(entity);
        if (id <= 0) {
            return Optional.empty();
        }
        return Optional.ofNullable(this.entity2Dto(entity));
    }

    @Override
    public Boolean updateJob(WaylineJobDTO dto) {
        int id = mapper.update(dto2Entity(dto),
                new LambdaQueryWrapper<WaylineJobHistoryEntity>()
                        .eq(WaylineJobHistoryEntity::getJobId, dto.getJobId()
                ));
        return id>0;
    }

    @Override
    public PaginationData<WaylineJobDTO> getJobsList(long page, long pageSize, Integer status,
                                                     Long startTime, Long endTime, Integer taskType,
                                                     String name, Boolean isComplete) {
        Page<WaylineJobHistoryEntity> pageData = mapper.selectPage(
                new Page<WaylineJobHistoryEntity>(page, pageSize),
                new LambdaQueryWrapper<WaylineJobHistoryEntity>()
                        .isNull(WaylineJobHistoryEntity::getParentId)
                        // 实际开始时间筛选，允许为空，为空则不进行筛选
                        .ge(startTime != null, WaylineJobHistoryEntity::getBeginTime, startTime)
                        .le(endTime != null, WaylineJobHistoryEntity::getBeginTime, endTime)
                        // 任务类型筛选，允许为空，为空则不进行筛选
                        .eq(taskType != null, WaylineJobHistoryEntity::getTaskType, taskType)
                        // 任务名称筛选，允许为空，为空则不进行筛选;....模糊查询
                        .like(name != null, WaylineJobHistoryEntity::getName, "%" + name + "%")
                        // 根据是否完成筛选，允许为空，为空则不进行筛选
                        .eq(isComplete != null && isComplete, WaylineJobHistoryEntity::getStatus, 3)
                        .ne(isComplete != null && !isComplete, WaylineJobHistoryEntity::getStatus, 3)
                        // 任务状态筛选，允许为空，为空则不进行筛选
                        .eq(status != null, WaylineJobHistoryEntity::getStatus, status)
                        .orderByDesc(WaylineJobHistoryEntity::getBeginTime));
        List<WaylineJobDTO> records = pageData.getRecords()
                .stream()
                .map(this::entity2Dto)
                .peek(job -> {
                    List<WaylineJobDTO> childOpt = this.getJobByJobParentId(job.getJobId());
                    job.setChildren(childOpt);
                })
                .collect(Collectors.toList());

        return new PaginationData<WaylineJobDTO>(records, new Pagination(pageData));
    }

    public List<WaylineJobDTO> getJobByJobParentId(String parentId) {
        List<WaylineJobHistoryEntity> jobEntities = mapper.selectList(
                new LambdaQueryWrapper<WaylineJobHistoryEntity>()
                        .eq(WaylineJobHistoryEntity::getParentId, parentId));
        if (CollectionUtils.isEmpty(jobEntities)) {
            return Collections.emptyList(); // 返回不可变空列表，更安全
        }
        return jobEntities.stream()
                .map(this::entity2Dto) // 调用原有的单个转换方法
                .collect(Collectors.toList());
    }

    @Override
    public Optional<WaylineJobDTO> getJobByJobId(String jobId) {
        WaylineJobHistoryEntity jobEntities = mapper.selectOne(
                new LambdaQueryWrapper<WaylineJobHistoryEntity>()
                        .eq(WaylineJobHistoryEntity::getJobId, jobId));
        return Optional.ofNullable(this.entity2Dto(jobEntities));
    }

    /**
     * 将WaylineJobDTO对象转换为WaylineJobHistoryEntity对象
     *
     * @param dto WaylineJobDTO对象
     * @return WaylineJobHistoryEntity对象
     */
    @Override
    public WaylineJobHistoryEntity dto2Entity(WaylineJobDTO dto) {
        WaylineJobHistoryEntity.WaylineJobHistoryEntityBuilder builder = WaylineJobHistoryEntity.builder();

        if (dto == null) {
            return builder.build();
        }
        if (Objects.nonNull(dto.getBeginTime())) {
            builder.beginTime(dto.getBeginTime().atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
        }
        if (Objects.nonNull(dto.getEndTime())) {
            builder.endTime(dto.getEndTime().atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
        }
        if (Objects.nonNull(dto.getExecuteTime())) {
            builder.executeTime(dto.getExecuteTime().atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
        }
        if (Objects.nonNull(dto.getCompletedTime())) {
            builder.completedTime(dto.getCompletedTime().atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
        }
        return builder.status(dto.getStatus())
                .mediaCount(dto.getMediaCount())
                .name(dto.getJobName())
                .errorCode(dto.getCode())
                .jobId(dto.getJobId())
                .fileId(dto.getFileId())
                .filename(dto.getFileName())
                .dockSn(dto.getDockSn())
                .taskType(dto.getTaskType())
                .waylineType(dto.getWaylineType())
                .username(dto.getUsername())
                .rthAltitude(dto.getRthAltitude())
                .outOfControlAction(dto.getOutOfControlAction())
                .parentId(dto.getParentId())
                .loopSum(dto.getLoopSum())
                .build();
    }

    /**
     * 将实体对象转换为数据传输对象
     *
     * @param entity WaylineJobEntity 实体对象
     * @return WaylineJobDTO 数据传输对象
     */
    private WaylineJobDTO entity2Dto(WaylineJobHistoryEntity entity) {
        if (entity == null) {
            return null;
        }
        // 修改航线文件名称，从ard_route_info航线表中取;
        String fileId = entity.getFileId();
        WaylineJobDTO.WaylineJobDTOBuilder builder = WaylineJobDTO.builder()
                .jobId(entity.getJobId())
                .jobName(entity.getName())
                .fileId(entity.getFileId())
                .fileName(entity.getFilename())
                .dockSn(entity.getDockSn())
                .dockName(deviceService.getDeviceBySn(entity.getDockSn())
                        .orElse(DeviceDTO.builder().build()).getNickname())
                .username(entity.getUsername())
                .status(entity.getStatus())
                .code(entity.getErrorCode())
                .beginTime(LocalDateTime.ofInstant(Instant.ofEpochMilli(entity.getBeginTime()), ZoneId.systemDefault()))
                .endTime(Objects.nonNull(entity.getEndTime()) ?
                        LocalDateTime.ofInstant(Instant.ofEpochMilli(entity.getEndTime()), ZoneId.systemDefault()) : null)
                .executeTime(Objects.nonNull(entity.getExecuteTime()) ?
                        LocalDateTime.ofInstant(Instant.ofEpochMilli(entity.getExecuteTime()), ZoneId.systemDefault()) : null)
                .completedTime(WaylineJobStatusEnum.find(entity.getStatus()).getEnd() ?
                        LocalDateTime.ofInstant(Instant.ofEpochMilli(entity.getUpdateTime()), ZoneId.systemDefault()) : null)
                .taskType(entity.getTaskType())
                .waylineType(entity.getWaylineType())
                .rthAltitude(entity.getRthAltitude())
                .rthMode(entity.getRthMode())
                .outOfControlAction(entity.getOutOfControlAction())
                .mediaCount(entity.getMediaCount())
                .loopSum(entity.getLoopSum());


//        if (Objects.nonNull(entity.getEndTime())) {
//            builder.endTime(LocalDateTime.ofInstant(Instant.ofEpochMilli(entity.getEndTime()), ZoneId.systemDefault()));
//        }
//        if (WaylineJobStatusEnum.IN_PROGRESS.getVal() == entity.getStatus() && RedisOpsUtils.getExpire(entity.getJobId()) > 0) {
//            EventsReceiver<WaylineTaskProgressReceiver> taskProgress = (EventsReceiver<WaylineTaskProgressReceiver>) RedisOpsUtils.get(RedisConst.WAYLINE_JOB_RUNNING_PREFIX + entity.getDockSn());
//            if (Objects.nonNull(taskProgress.getOutput()) && Objects.nonNull(taskProgress.getOutput().getProgress())) {
//                builder.progress(taskProgress.getOutput().getProgress().getPercent());
//            }
//        }
//
//        if (entity.getMediaCount() == 0) {
//            return builder.build();
//        }
//
//        // sync the number of media files
//        String key = RedisConst.MEDIA_HIGHEST_PRIORITY_PREFIX + entity.getDockSn();
//        String countKey = RedisConst.MEDIA_FILE_PREFIX + entity.getDockSn();
//        Object mediaFileCount = RedisOpsUtils.hashGet(countKey, entity.getJobId());
//        if (Objects.nonNull(mediaFileCount)) {
//            builder.uploadedCount(((MediaFileCountDTO) mediaFileCount).getUploadedCount())
//                    .uploading(RedisOpsUtils.checkExist(key) && entity.getJobId().equals(((MediaFileCountDTO) RedisOpsUtils.get(key)).getJobId()));
//            return builder.build();
//        }

//        int uploadedSize = fileService.getFilesByWorkspaceAndJobId(entity.getJobId()).size();
//        // All media for this job have been uploaded.
//        if (uploadedSize >= entity.getMediaCount()) {
//            return builder.uploadedCount(uploadedSize).build();
//        }
//        RedisOpsUtils.hashSet(countKey, entity.getJobId(),
//                MediaFileCountDTO.builder()
//                        .jobId(entity.getJobId())
//                        .mediaCount(entity.getMediaCount())
//                        .uploadedCount(uploadedSize).build());
        return builder.build();
    }
}