package com.ruoyi.safety.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ruoyi.safety.mapper.SafetyAssessmentTaskEmployeeMapper;
import com.ruoyi.safety.pojo.SafetyAssessmentTask;
import com.ruoyi.safety.pojo.SafetyAssessmentTaskEmployee;
import com.ruoyi.safety.service.SafetyAssessmentTaskEmployeeService;
import com.ruoyi.safety.service.SafetyAssessmentTaskService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.time.LocalDateTime;

@Service
public class SafetyAssessmentTaskEmployeeServiceImpl
        extends SafetyBaseServiceImpl<SafetyAssessmentTaskEmployeeMapper, SafetyAssessmentTaskEmployee>
        implements SafetyAssessmentTaskEmployeeService {

    @Autowired
    private SafetyAssessmentTaskService taskService;

    @Override
    public IPage<SafetyAssessmentTaskEmployee> queryPage(Page<SafetyAssessmentTaskEmployee> page,
                                                         SafetyAssessmentTaskEmployee query) {
        QueryWrapper<SafetyAssessmentTaskEmployee> wrapper =
                new QueryWrapper<SafetyAssessmentTaskEmployee>().orderByDesc("create_time");
        if (query != null) {
            if (query.getTaskId() != null) {
                wrapper.eq("task_id", query.getTaskId());
            }
            if (query.getEmployeeId() != null) {
                wrapper.eq("employee_id", query.getEmployeeId());
            }
            if (hasText(query.getEmployeeName())) {
                wrapper.like("employee_name", query.getEmployeeName());
            }
            if (query.getStatus() != null) {
                wrapper.eq("status", query.getStatus());
            }
        }
        return page(page, wrapper);
    }

    @Override
    public boolean start(SafetyAssessmentTaskEmployee request) {
        if (request == null || request.getTaskId() == null) {
            return false;
        }
        fillCurrentEmployeeIfMissing(request);
        if (request.getEmployeeId() == null) {
            return false;
        }
        SafetyAssessmentTaskEmployee existing = getOne(new QueryWrapper<SafetyAssessmentTaskEmployee>()
                .eq("task_id", request.getTaskId())
                .eq("employee_id", request.getEmployeeId())
                .last("limit 1"));
        if (existing != null) {
            return true;
        }
        if (request.getStatus() == null) {
            request.setStatus(0);
        }
        updateTaskStatus(request.getTaskId(), 1);
        return saveSafety(request);
    }

    @Override
    public boolean submit(SafetyAssessmentTaskEmployee request) {
        if (request == null) {
            return false;
        }
        fillCurrentEmployeeIfMissing(request);
        SafetyAssessmentTaskEmployee target = findSubmitTarget(request);
        if (target == null) {
            if (request.getTaskId() == null || request.getEmployeeId() == null) {
                return false;
            }
            request.setStatus(1);
            if (request.getCompleteTime() == null) {
                request.setCompleteTime(LocalDateTime.now());
            }
            boolean saved = saveSafety(request);
            finishTaskWhenAllSubmitted(request.getTaskId());
            return saved;
        }
        target.setStatus(1);
        target.setScore(request.getScore());
        target.setRemark(request.getRemark());
        target.setCompleteTime(request.getCompleteTime() == null ? LocalDateTime.now() : request.getCompleteTime());
        boolean updated = updateSafety(target);
        finishTaskWhenAllSubmitted(target.getTaskId());
        return updated;
    }

    private SafetyAssessmentTaskEmployee findSubmitTarget(SafetyAssessmentTaskEmployee request) {
        if (request.getId() != null) {
            SafetyAssessmentTaskEmployee byId = getById(request.getId());
            if (byId != null) {
                return byId;
            }
        }
        if (request.getTaskId() != null && request.getEmployeeId() != null) {
            return getOne(new QueryWrapper<SafetyAssessmentTaskEmployee>()
                    .eq("task_id", request.getTaskId())
                    .eq("employee_id", request.getEmployeeId())
                    .last("limit 1"));
        }
        return null;
    }

    private void fillCurrentEmployeeIfMissing(SafetyAssessmentTaskEmployee request) {
        if (request.getEmployeeId() == null) {
            request.setEmployeeId(currentUserId());
        }
        if (!hasText(request.getEmployeeName())) {
            request.setEmployeeName(currentUsername());
        }
    }

    private void updateTaskStatus(Long taskId, Integer status) {
        SafetyAssessmentTask task = taskService.getById(taskId);
        if (task != null && (task.getStatus() == null || task.getStatus() < status)) {
            task.setStatus(status);
            taskService.updateSafety(task);
        }
    }

    private void finishTaskWhenAllSubmitted(Long taskId) {
        if (taskId == null) {
            return;
        }
        long pending = count(new QueryWrapper<SafetyAssessmentTaskEmployee>()
                .eq("task_id", taskId)
                .eq("status", 0));
        if (pending == 0) {
            updateTaskStatus(taskId, 2);
        }
    }
}