package org.dromara.eims.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import org.dromara.common.core.utils.MapstructUtils;
import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.mybatis.core.page.TableDataInfo;
import org.dromara.common.mybatis.core.page.PageQuery;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import lombok.RequiredArgsConstructor;
import org.dromara.eims.domain.EimsEqu;
import org.dromara.eims.domain.EimsEquType;
import org.dromara.eims.domain.vo.EimsEquTypeVo;
import org.dromara.eims.mapper.EimsEquTypeMapper;
import org.springframework.stereotype.Service;
import org.dromara.eims.domain.bo.EimsInventoryDetailBo;
import org.dromara.eims.domain.vo.EimsInventoryDetailVo;
import org.dromara.eims.domain.EimsInventoryDetail;
import org.dromara.eims.mapper.EimsInventoryDetailMapper;
import org.dromara.eims.service.IEimsInventoryDetailService;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Collection;

/**
 * 盘点明细Service业务层处理
 *
 * @author zhuguifei
 * @date 2025-02-06
 */
@RequiredArgsConstructor
@Service
public class EimsInventoryDetailServiceImpl implements IEimsInventoryDetailService {

    private final EimsInventoryDetailMapper baseMapper;
    private final EimsEquTypeMapper equTypeMapper;

    /**
     * 查询盘点明细
     *
     * @param id 主键
     * @return 盘点明细
     */
    @Override
    public EimsInventoryDetailVo queryById(Long id) {
        return baseMapper.selectVoById(id);
    }


    /**
     * 分页查询盘点明细列表-多表查询
     *
     * @param bo        查询条件
     * @param pageQuery 分页参数
     * @return 盘点明细分页列表
     */
    @Override
    public TableDataInfo<EimsInventoryDetailVo> queryPageListCustom(EimsInventoryDetailBo bo, PageQuery pageQuery) {
        //TODO 特殊处理-多表同字段导致问题
        String orderByColumn = pageQuery.getOrderByColumn();
        if (StringUtils.isNotEmpty(orderByColumn) && orderByColumn.contains("status")) {
            pageQuery.setOrderByColumn(orderByColumn.replace("status", "a.status"));
        }
        Page<EimsInventoryDetailVo> page = baseMapper.selectInventoryDetailList(pageQuery.build(), buildQueryWrapper(bo));
        return TableDataInfo.build(page);
    }

    /**
     * 查询符合条件的盘点明细列表
     *
     * @param bo 查询条件
     * @return 盘点明细列表
     */
    @Override
    public List<EimsInventoryDetailVo> queryList(EimsInventoryDetailBo bo) {
        QueryWrapper<EimsInventoryDetail> qw = buildQueryWrapper(bo);
        return baseMapper.selectVoList(qw);
    }

    private QueryWrapper<EimsInventoryDetail> buildQueryWrapper(EimsInventoryDetailBo bo) {
        Map<String, Object> params = bo.getParams();
        QueryWrapper<EimsInventoryDetail> qw = Wrappers.query();
        qw.eq(bo.getInventoryId() != null, "a.inventory_id", bo.getInventoryId());
        qw.eq(bo.getEquId() != null, "a.equ_id", bo.getEquId());
        qw.eq(StringUtils.isNotBlank(bo.getStatus()), "a.status", bo.getStatus());

        qw.like(StringUtils.isNotBlank(bo.getEquName()), "c.equ_name", bo.getEquName());
        qw.like(StringUtils.isNotBlank(bo.getEquCode()), "c.equ_code", bo.getEquCode());
        qw.like(StringUtils.isNotBlank(bo.getEquAssetNo()), "c.asset_no", bo.getEquAssetNo());
        qw.like(StringUtils.isNotBlank(bo.getLocation()), "c.location", bo.getLocation());
        qw.like(StringUtils.isNotBlank(bo.getEquStatus()), "c.status", bo.getEquStatus());
        /**
         * equTypeId = 0 时查询所有设备（默认根目录id为0，详见SysEquTypeServiceImpl中selectEquTypeTreeList）
         * equTypeId   为其他值时只查当前设备类型id和后代设备类型id
         */
        if (bo.getEquTypeId() != null && bo.getEquTypeId() > 0) {
            List<Long> allDescendantIds = getAllDescendantIds(bo.getEquTypeId());
            qw.in("c.equ_type_id", allDescendantIds);
        }

        return qw;
    }

    /**
     * 根据id，获取所有后代id
     * @param rootId
     * @return
     */
    public List<Long> getAllDescendantIds(Long rootId) {
        List<Long> result = new ArrayList<>();
        result.add(rootId);
        collectDescendants(rootId, result);
        return result;
    }

    private void collectDescendants(Long currentId, List<Long> collector) {
        QueryWrapper<EimsEquType> equTypeWrapper = new QueryWrapper<>();
        equTypeWrapper.lambda().eq(EimsEquType::getParentId, currentId);

        List<EimsEquTypeVo> children = equTypeMapper.selectVoList(equTypeWrapper);
        if (children != null && !children.isEmpty()) {
            for (EimsEquTypeVo child : children) {
                Long childId = child.getEquTypeId();
                collector.add(childId);
                collectDescendants(childId, collector);
            }
        }
    }

    /**
     * 新增盘点明细
     *
     * @param bo 盘点明细
     * @return 是否新增成功
     */
    @Override
    public Boolean insertByBo(EimsInventoryDetailBo bo) {
        EimsInventoryDetail add = MapstructUtils.convert(bo, EimsInventoryDetail.class);
        validEntityBeforeSave(add);
        boolean flag = baseMapper.insert(add) > 0;
        if (flag) {
            bo.setId(add.getId());
        }
        return flag;
    }

    /**
     * 修改盘点明细
     *
     * @param bo 盘点明细
     * @return 是否修改成功
     */
    @Override
    public Boolean updateByBo(EimsInventoryDetailBo bo) {
        EimsInventoryDetail update = MapstructUtils.convert(bo, EimsInventoryDetail.class);
        validEntityBeforeSave(update);
        return baseMapper.updateById(update) > 0;
    }

    /**
     * 保存前的数据校验
     */
    private void validEntityBeforeSave(EimsInventoryDetail entity) {
        //TODO 做一些数据校验,如唯一约束
    }

    /**
     * 校验并批量删除盘点明细信息
     *
     * @param ids     待删除的主键集合
     * @param isValid 是否进行有效性校验
     * @return 是否删除成功
     */
    @Override
    public Boolean deleteWithValidByIds(Collection<Long> ids, Boolean isValid) {
        if (isValid) {
            //TODO 做一些业务上的校验,判断是否需要校验
        }
        return baseMapper.deleteByIds(ids) > 0;
    }
}