package com.ruoyi.basic.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.ruoyi.basic.entity.City;
import com.ruoyi.basic.entity.District;
import com.ruoyi.basic.entity.Province;
import com.ruoyi.basic.entity.dto.TreeSelectOption;
import com.ruoyi.basic.mapper.CityMapper;
import com.ruoyi.basic.mapper.DistrictMapper;
import com.ruoyi.basic.mapper.ProvinceMapper;
import com.ruoyi.basic.service.ProvinceService;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * <p>
 * 省表 服务实现类
 * </p>
 *
 * @author ruoyi
 * @since 2025-05-31
 */
@Service
@RequiredArgsConstructor
public class ProvinceServiceImpl extends ServiceImpl<ProvinceMapper, Province> implements ProvinceService {

    @Autowired
    private ProvinceMapper provinceMapper;

    @Autowired
    private CityMapper cityMapper;

    @Autowired
    private DistrictMapper districtMapper;

    @Override
    public List<TreeSelectOption> getAreaOptions() {
        List<TreeSelectOption> res = new ArrayList<>();

        // 批量查询所有省份
        List<Province> provinces = provinceMapper.selectList(null);
        if (provinces.isEmpty()) {
            return res;
        }

        // 获取所有省份ID用于批量查询城市
        List<Long> provinceIds = provinces.stream()
                .map(Province::getId)
                .collect(Collectors.toList());

        // 批量查询所有城市
        List<City> allCities = cityMapper.selectList(
                new LambdaQueryWrapper<City>().in(City::getProvinceId, provinceIds));

        // 将城市按省份ID分组
        Map<Long, List<City>> cityMap = allCities.stream()
                .collect(Collectors.groupingBy(City::getProvinceId));

        // 获取所有城市ID用于批量查询区域
        List<Long> cityIds = allCities.stream()
                .map(City::getId)
                .collect(Collectors.toList());

        // 批量查询所有区域
        List<District> allDistricts = districtMapper.selectList(
                new LambdaQueryWrapper<District>().in(District::getCityId, cityIds));

        // 将区域按城市ID分组
        Map<Long, List<District>> districtMap = allDistricts.stream()
                .collect(Collectors.groupingBy(District::getCityId));

        // 构建树形结构
        for (Province province : provinces) {
            List<TreeSelectOption> cities = new ArrayList<>();

            List<City> cityList = cityMap.get(province.getId());
            if (cityList != null && !cityList.isEmpty()) {
                for (City city : cityList) {
                    List<TreeSelectOption> districts = new ArrayList<>();

                    List<District> districtList = districtMap.get(city.getId());
                    if (districtList != null && !districtList.isEmpty()) {
                        for (District district : districtList) {
                            districts.add(new TreeSelectOption(district.getId().toString(),
                                    district.getName(), district.getId(), null));
                        }
                    }

                    TreeSelectOption cityTreeSelectOption = new TreeSelectOption();
                    cityTreeSelectOption.setValue(city.getId().toString());
                    cityTreeSelectOption.setLabel(city.getName());
                    cityTreeSelectOption.setId(city.getId());
                    cityTreeSelectOption.setChildren(districts);

                    cities.add(cityTreeSelectOption);
                }
            }

            TreeSelectOption treeSelectOption = new TreeSelectOption();
            treeSelectOption.setValue(province.getId().toString());
            treeSelectOption.setLabel(province.getName());
            treeSelectOption.setId(province.getId());
            treeSelectOption.setChildren(cities);
            res.add(treeSelectOption);
        }

        return res;
    }


    @Override
    public String getAreaDisplay(Long provinceId, Long cityId, Long districtId) {
        Province province = provinceMapper.selectById(provinceId);
        if (province == null) {
            throw new RuntimeException("省不存在");
        }
        City city = cityMapper.selectById(cityId);
        if (city == null) {
            throw new RuntimeException("市不存在");
        }
        District district = districtMapper.selectById(districtId);
        if (district == null) {
            throw new RuntimeException("区不存在");
        }

        // 如果省和市同名
        if (province.getName().equals(city.getName())) {
            return province.getName() + " " + district.getName();
        }
        return province.getName() + city.getName() + district.getName();
    }
}