package com.ruoyi.system.service.impl;


import cn.hutool.core.bean.BeanUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.common.redis.service.RedisService;
import com.ruoyi.system.mapper.TbRegionMapper;
import com.ruoyi.system.model.TbRegion;
import com.ruoyi.system.service.TbRegionService;
import com.ruoyi.system.vo.RegionVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

/**
 * <p>
 * 省市区三级联动 服务实现类
 * </p>
 *
 * @author administrator
 * @since 2025-05-26
 */
@Service
public class TbRegionServiceImpl extends ServiceImpl<TbRegionMapper, TbRegion> implements TbRegionService {

    @Autowired
    private RedisService redisService;

    @Override
    public List<RegionVo> listCityVo() {
        // 缓存
        List<Object> regionList = redisService.getCacheList("region_list");
        if (regionList!=null && !regionList.isEmpty()) {
            List<RegionVo> regionVoList = BeanUtil.copyToList(redisService.getCacheList("region_list"), RegionVo.class);
            return regionVoList;
        } else {
            // 省级
            List<TbRegion> provinceList = this.baseMapper.selectList(
                    new LambdaQueryWrapper<TbRegion>().eq(TbRegion::getParentId, 0));
            List<TbRegion> childrenList = this.baseMapper.selectList(
                    new LambdaQueryWrapper<TbRegion>().ne(TbRegion::getParentId, 0));
            List<RegionVo> provinceVoList = BeanUtil.copyToList(provinceList, RegionVo.class);
            provinceVoList = provinceVoList.stream().peek(province -> {
                province.setLevel(1);
                List<RegionVo> cityVOList = getChildrenRegionVoList(province,
                        childrenList);
                if (cityVOList!=null && !cityVOList.isEmpty()) {
                    cityVOList = cityVOList.stream().peek(city -> {
                        city.setLevel(2);
                        List<RegionVo> countyVoList = getChildrenRegionVoList(city,
                                childrenList);
                        if (countyVoList!=null && !countyVoList.isEmpty()) {
                            countyVoList = countyVoList.stream().peek(county -> county.setLevel(3))
                                    .collect(Collectors.toList());
                            city.setChildren(countyVoList);
                        }
                    }).collect(Collectors.toList());
                    province.setChildren(cityVOList);
                }
            }).collect(Collectors.toList());
            // 缓存
            redisService.setCacheList("region_list", provinceVoList);
            return provinceVoList;
        }
    }

    private static List<RegionVo> getChildrenRegionVoList(RegionVo province,
                                                          List<TbRegion> childrenList) {
        List<RegionVo> cityVOList = childrenList.stream()
                .filter(city -> city.getParentId().equals(province.getId()))
                .map(item -> BeanUtil.copyProperties(item, RegionVo.class))
                .filter(Objects::nonNull)
                .collect(Collectors.toList());
        // 将符合条件的元素从childrenList中移除
        childrenList.removeIf(item -> item.getParentId().equals(province.getId()));
        return cityVOList;
    }
}