package com.ruoyi.system.service.impl;

import cn.hutool.core.bean.BeanUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.core.domain.entity.TDept;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.system.dto.asset.AddInventoryTaskDTO;
import com.ruoyi.system.mapper.AssetInventoryTaskMapper;
import com.ruoyi.system.mapper.TDeptMapper;
import com.ruoyi.system.model.AssetInventoryTask;
import com.ruoyi.system.model.AssetInventoryTaskItem;
import com.ruoyi.system.query.AssertInventoryQuery;
import com.ruoyi.system.service.AssetInventoryTaskItemService;
import com.ruoyi.system.service.AssetInventoryTaskService;
import com.ruoyi.system.vo.asset.AssetInventoryTaskVO;
import lombok.RequiredArgsConstructor;
import org.apache.commons.lang3.StringUtils;
import org.springframework.context.annotation.Lazy;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.time.LocalDateTime;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;

/**
 * <p>
 * 盘点任务表 服务实现类
 * </p>
 *
 * @author WuGuanFengYue
 * @since 2025-09-15
 */
@Service
@RequiredArgsConstructor(onConstructor_ = {@Lazy})
public class AssetInventoryTaskServiceImpl extends ServiceImpl<AssetInventoryTaskMapper, AssetInventoryTask> implements AssetInventoryTaskService {

    private final TDeptMapper tDeptMapper;

    private final AssetInventoryTaskItemService assetInventoryTaskItemService;

    @Override
    public IPage<AssetInventoryTaskVO> getInventoryTaskPage(AssertInventoryQuery query) {
        // 1. 构建Lambda查询条件
        LambdaQueryWrapper<AssetInventoryTask> wrapper = new LambdaQueryWrapper<>();
        wrapper.like(StringUtils.isNotBlank(query.getTaskName()),
                     AssetInventoryTask::getTaskName, query.getTaskName())
               .eq(AssetInventoryTask::getDisabled, false)
               .orderByDesc(AssetInventoryTask::getCreateTime);

        // 2. 构建分页对象
        Page<AssetInventoryTask> page = new Page<>(query.getPageNum(), query.getPageSize());

        // 3. 执行分页查询
        IPage<AssetInventoryTask> taskPage = this.page(page, wrapper);

        // 4. 收集部门ID并去重
        List<Integer> deptIds = taskPage.getRecords().stream()
                .map(AssetInventoryTask::getDeptId)
                .filter(Objects::nonNull)
                .distinct()
                .collect(Collectors.toList());

        // 5. 批量查询部门信息
        List<TDept> depts = deptIds.isEmpty() ?
                Collections.emptyList() :
                tDeptMapper.selectBatchIds(deptIds);

        // 6. 构建部门ID->名称的映射Map
        Map<Integer, String> deptNameMap = depts.stream()
                .collect(Collectors.toMap(TDept::getId, TDept::getDeptName));

        // 7. 使用BeanUtil.copyToList转换VO并填充部门名称
        List<AssetInventoryTaskVO> voList = BeanUtil.copyToList(taskPage.getRecords(), AssetInventoryTaskVO.class);
        voList.forEach(vo -> {
            if (vo.getDeptId() != null) {
                vo.setDeptName(deptNameMap.get(vo.getDeptId()));
            }
        });

        // 8. 构建返回的分页结果
        IPage<AssetInventoryTaskVO> resultPage = new Page<>(taskPage.getCurrent(), taskPage.getSize(), taskPage.getTotal());
        resultPage.setRecords(voList);

        return resultPage;
    }

    @Override
    @Transactional(rollbackFor = Exception.class)
    public boolean createInventoryTask(AddInventoryTaskDTO dto) {
        // 1. DTO转Entity
        AssetInventoryTask task = new AssetInventoryTask();
        BeanUtil.copyProperties(dto, task);

        // 2. 设置默认值
        SysUser currentUser = SecurityUtils.getLoginUser().getUser();
        task.setUserId(Math.toIntExact(currentUser.getUserId()));
        task.setStatus(0); // 未开始状态
        task.setCreateTime(LocalDateTime.now());
        task.setCreateBy(currentUser.getUserName());
        task.setDisabled(false);

        // 3. 保存主任务
        boolean saved = this.save(task);
        if (!saved) {
            throw new RuntimeException("保存盘点任务失败");
        }

        // 4. 批量创建资产明细
        List<AssetInventoryTaskItem> items = buildTaskItems(task.getId(), dto.getAssetMainIds());
        return assetInventoryTaskItemService.saveBatch(items);
    }

    /**
     * 构建盘点任务资产明细列表
     *
     * @param taskId 盘点任务ID
     * @param assetMainIds 资产ID列表
     * @return 资产明细列表
     */
    private List<AssetInventoryTaskItem> buildTaskItems(Integer taskId, List<Integer> assetMainIds) {
        SysUser currentUser = SecurityUtils.getLoginUser().getUser();
        Integer currentUserId = Math.toIntExact(currentUser.getUserId());

        return assetMainIds.stream()
                .map(assetId -> {
                    AssetInventoryTaskItem item = new AssetInventoryTaskItem();
                    item.setInventoryTaskId(taskId);
                    item.setAssetMainId(assetId);
                    item.setUserId(currentUserId);
                    item.setResultStatus(0); // 未盘点状态
                    return item;
                })
                .collect(Collectors.toList());
    }
}