package org.dromara.erp.service.impl;

import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import lombok.RequiredArgsConstructor;
import org.dromara.erp.domain.ErpModelKind;
import org.dromara.erp.domain.query.ErpModelKindQuery;
import org.dromara.erp.domain.vo.ErpModelKindVo;
import org.dromara.erp.mapper.ErpModelKindMapper;
import org.dromara.erp.service.IErpModelKindService;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * ERP 型号类别 Service 业务层处理
 *
 * @Author: Trae
 */
@RequiredArgsConstructor
@Service
public class ErpModelKindServiceImpl implements IErpModelKindService {

    private final ErpModelKindMapper baseMapper;

    @Override
    public List<ErpModelKindVo> queryList(ErpModelKindQuery query) {
        if (query != null && StrUtil.isNotBlank(query.getParentRowId())) {
            LambdaQueryWrapper<ErpModelKind> lqw = buildQueryWrapper(query);
            return baseMapper.selectVoList(lqw);
        }
        LambdaQueryWrapper<ErpModelKind> lqw = buildFirstLevelWrapper(query);
        List<ErpModelKindVo> all = baseMapper.selectVoList(lqw);
        Set<String> rowIds = all.stream()
            .map(ErpModelKindVo::getRowId)
            .filter(ObjectUtil::isNotEmpty)
            .collect(Collectors.toSet());
        return all.stream()
            .filter(item -> {
                String pid = item.getParentRowId();
                return ObjectUtil.isEmpty(pid) || !rowIds.contains(pid);
            })
            .collect(Collectors.toList());
    }

    private LambdaQueryWrapper<ErpModelKind> buildFirstLevelWrapper(ErpModelKindQuery query) {
        LambdaQueryWrapper<ErpModelKind> lqw = Wrappers.lambdaQuery();
        if (query != null) {
            lqw.like(StrUtil.isNotBlank(query.getNum()), ErpModelKind::getNum, query.getNum());
            lqw.like(StrUtil.isNotBlank(query.getName()), ErpModelKind::getName, query.getName());
            lqw.eq(StrUtil.isNotBlank(query.getModelKind()), ErpModelKind::getModelKind, query.getModelKind());
        }
        lqw.eq(ErpModelKind::getStopUse, 0);
        lqw.orderByAsc(ErpModelKind::getNum);
        return lqw;
    }

    private LambdaQueryWrapper<ErpModelKind> buildQueryWrapper(ErpModelKindQuery query) {
        LambdaQueryWrapper<ErpModelKind> lqw = Wrappers.lambdaQuery();
        lqw.like(StrUtil.isNotBlank(query.getNum()), ErpModelKind::getNum, query.getNum());
        lqw.like(StrUtil.isNotBlank(query.getName()), ErpModelKind::getName, query.getName());
        lqw.eq(StrUtil.isNotBlank(query.getModelKind()), ErpModelKind::getModelKind, query.getModelKind());
        lqw.eq(ErpModelKind::getParentRowId, query.getParentRowId());
        lqw.eq(ErpModelKind::getStopUse, 0);
        lqw.orderByAsc(ErpModelKind::getNum);
        return lqw;
    }

}
