package cn.iocoder.yudao.module.wms.service.md.item;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.wms.controller.admin.md.item.vo.category.WmsItemCategoryListReqVO;
import cn.iocoder.yudao.module.wms.controller.admin.md.item.vo.category.WmsItemCategorySaveReqVO;
import cn.iocoder.yudao.module.wms.dal.dataobject.md.item.WmsItemCategoryDO;
import cn.iocoder.yudao.module.wms.dal.mysql.md.item.WmsItemCategoryMapper;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;
import org.springframework.validation.annotation.Validated;

import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet;
import static cn.iocoder.yudao.module.wms.enums.ErrorCodeConstants.*;

/**
 * WMS 商品分类 Service 实现类
 *
 * @author 芋道源码
 */
@Service
@Validated
public class WmsItemCategoryServiceImpl implements WmsItemCategoryService {

    @Resource
    private WmsItemCategoryMapper categoryMapper;

    @Resource
    private WmsItemService itemService;

    @Override
    public Long createItemCategory(WmsItemCategorySaveReqVO createReqVO) {
        // 校验数据
        validateCategorySaveData(createReqVO);

        // 插入
        WmsItemCategoryDO category = BeanUtils.toBean(createReqVO, WmsItemCategoryDO.class);
        categoryMapper.insert(category);
        return category.getId();
    }

    @Override
    public void updateItemCategory(WmsItemCategorySaveReqVO updateReqVO) {
        // 校验存在
        validateItemCategoryExists(updateReqVO.getId());
        // 校验数据
        validateCategorySaveData(updateReqVO);

        // 更新
        WmsItemCategoryDO updateObj = BeanUtils.toBean(updateReqVO, WmsItemCategoryDO.class);
        categoryMapper.updateById(updateObj);
    }

    private void validateCategorySaveData(WmsItemCategorySaveReqVO reqVO) {
        validateCategoryCodeUnique(reqVO.getId(), reqVO.getCode());
        validateCategoryNameUnique(reqVO.getId(), reqVO.getParentId(), reqVO.getName());
        validateParentCategory(reqVO.getId(), reqVO.getParentId());
    }

    private void validateCategoryCodeUnique(Long id, String code) {
        WmsItemCategoryDO category = categoryMapper.selectByCode(code);
        if (category == null) {
            return;
        }
        // 如果 id 为空，说明新增；和数据库已有 code 重复
        if (id == null) {
            throw exception(ITEM_CATEGORY_CODE_DUPLICATE);
        }
        if (ObjectUtil.notEqual(category.getId(), id)) {
            throw exception(ITEM_CATEGORY_CODE_DUPLICATE);
        }
    }

    private void validateCategoryNameUnique(Long id, Long parentId, String name) {
        WmsItemCategoryDO category = categoryMapper.selectByParentIdAndName(parentId, name);
        if (category == null) {
            return;
        }
        // 如果 id 为空，说明不用比较是否为相同 id 的商品分类
        if (id == null) {
            throw exception(ITEM_CATEGORY_NAME_DUPLICATE);
        }
        if (ObjectUtil.notEqual(category.getId(), id)) {
            throw exception(ITEM_CATEGORY_NAME_DUPLICATE);
        }
    }

    private void validateParentCategory(Long id, Long parentId) {
        if (parentId == null || WmsItemCategoryDO.PARENT_ID_ROOT.equals(parentId)) {
            return;
        }
        // 1. 不能设置自己为父分类
        if (ObjectUtil.equal(parentId, id)) {
            throw exception(ITEM_CATEGORY_PARENT_ERROR);
        }
        // 2. 父分类不存在
        WmsItemCategoryDO parentCategory = categoryMapper.selectById(parentId);
        if (parentCategory == null) {
            throw exception(ITEM_CATEGORY_PARENT_NOT_EXISTS);
        }
        // 3. 递归校验父分类，如果父分类是自己的子分类，则报错，避免形成环路
        if (id == null) { // id 为空，说明新增，不需要考虑环路
            return;
        }
        for (int i = 0; i < Short.MAX_VALUE; i++) {
            // 3.1 校验环路
            parentId = parentCategory.getParentId();
            if (ObjectUtil.equal(id, parentId)) {
                throw exception(ITEM_CATEGORY_PARENT_IS_CHILD);
            }
            // 3.2 继续递归上级父分类
            if (parentId == null || WmsItemCategoryDO.PARENT_ID_ROOT.equals(parentId)) {
                break;
            }
            parentCategory = categoryMapper.selectById(parentId);
            if (parentCategory == null) {
                break;
            }
        }
    }

    @Override
    public void deleteItemCategory(Long id) {
        // 校验存在
        validateItemCategoryExists(id);
        // 有子分类不能删
        if (categoryMapper.selectCountByParentId(id) > 0) {
            throw exception(ITEM_CATEGORY_HAS_CHILDREN);
        }
        // 校验分类下不存在商品
        if (itemService.getItemCountByCategoryId(id) > 0) {
            throw exception(ITEM_CATEGORY_HAS_ITEM);
        }

        // 删除
        categoryMapper.deleteById(id);
    }

    @Override
    public WmsItemCategoryDO validateItemCategoryExists(Long id) {
        WmsItemCategoryDO category = categoryMapper.selectById(id);
        if (category == null) {
            throw exception(ITEM_CATEGORY_NOT_EXISTS);
        }
        return category;
    }

    @Override
    public WmsItemCategoryDO getItemCategory(Long id) {
        return categoryMapper.selectById(id);
    }

    @Override
    public List<WmsItemCategoryDO> getItemCategoryList(WmsItemCategoryListReqVO listReqVO) {
        return categoryMapper.selectList(listReqVO);
    }

    @Override
    public List<WmsItemCategoryDO> getItemCategoryList(Collection<Long> ids) {
        if (CollUtil.isEmpty(ids)) {
            return Collections.emptyList();
        }
        return categoryMapper.selectByIds(ids);
    }

    @Override
    public Set<Long> getSelfAndChildItemCategoryIdList(Long id) {
        if (id == null) {
            return null;
        }
        Set<Long> ids = new HashSet<>();
        ids.add(id);
        Collection<Long> parentIds = Collections.singleton(id);
        for (int i = 0; i < Short.MAX_VALUE; i++) {
            List<WmsItemCategoryDO> children = categoryMapper.selectListByParentIds(parentIds);
            if (CollUtil.isEmpty(children)) {
                break;
            }
            ids.addAll(convertSet(children, WmsItemCategoryDO::getId));
            parentIds = convertSet(children, WmsItemCategoryDO::getId);
        }
        return ids;
    }

}