package com.people.model;

import android.text.TextUtils;

import com.wd.base.log.Logger;
import com.wd.foundation.bean.launch.LaunchPageBean;
import com.wd.foundation.bean.response.AreaTreeselectBean;
import com.wd.foundation.bean.response.GdDistrictBean;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;


/**
 * @author libo
 * @version [V1.0.0, 2022/07/25]
 * @since V1.0.0
 */
public class ProvinceDataManager {
    private static final String TAG = "ProvinceDataManager";

    /**
     * 直辖市:[{"provinceCode":"500000","provinceName":"重庆市"},{"provinceCode":"310000","provinceName":"上海市"},{"provinceCode":"110000","provinceName":"北京市"},{"provinceCode":"120000","provinceName":"天津市"}]
     */
    public static List<String> directCityCodes = new ArrayList<>();

    public static List<String> directCityNames = new ArrayList<>();

    private final Object lock = new Object();

    private List<AreaTreeselectBean> provinceList;


    static {
        try {
            directCityCodes = Arrays.asList("500000","310000","110000","120000");
        } catch (Exception e) {
            Logger.t(TAG).e("directCity init fail");
        }

        try {
            directCityNames = Arrays.asList("重庆市","上海市","北京市","天津市");
        } catch (Exception e) {
            Logger.t(TAG).e("directCity init fail");
        }
    }

    private static class SingletonHolder {

        private static final ProvinceDataManager INSTANCE = new ProvinceDataManager();
    }

    private ProvinceDataManager() {
    }

    public static ProvinceDataManager getInstance() {
        return SingletonHolder.INSTANCE;
    }

    /**
     * 获取缓存的省份数据
     *
     * @return 省份数据
     */
    public List<AreaTreeselectBean> getProvinceList(){
        return provinceList;
    }


    public void init(LaunchPageBean.AreasBean bean) {
        if (provinceList != null && !provinceList.isEmpty()) {
            Logger.t(TAG).e("province info already init");
            return;
        }
        ProvinceDataFetcher provinceDataFetcher = new ProvinceDataFetcher();
        provinceDataFetcher.setAreaRevealListener(new ProvinceDataFetcher.IGetAreaRevealListener() {
            @Override
            public void onSuccess(List<AreaTreeselectBean> datas) {
                synchronized (lock){
                    provinceList = datas;
                }
            }

            @Override
            public void onFailed(String error) {

            }
        });
        //不调用接口了，使用合并统一接口
//        provinceDataFetcher.getAreaReveal();
        provinceDataFetcher.saveAreaData(bean);

    }


    /**
     * 检测是否是直辖市
     * @param proCode
     * @return
     */
    public boolean isTwoAreaInfor(String proCode){
        return directCityCodes.contains(proCode);
    }

    /**
     * 查找城市信息
     *
     * @param provinceCode 省份信息
     * @param cityCode 城市信息
     * @param districtCode 区信息
     * @return 城市信息
     */
    public String find(String provinceCode, String cityCode, String districtCode) {
        if (TextUtils.isEmpty(provinceCode)) {
            return "";
        }
        if (provinceList == null || provinceList.isEmpty()) {
            return "";
        }
        for (AreaTreeselectBean areaRevealBean : provinceList) {
            if (areaRevealBean == null || TextUtils.isEmpty(areaRevealBean.getCode())) {
                continue;
            }

            if (provinceCode.equals(areaRevealBean.getCode())) {
                List<AreaTreeselectBean.City> cityBeanList = areaRevealBean.getChildren();
                if (cityBeanList == null || cityBeanList.isEmpty()) {
                    return areaRevealBean.getLabel();
                } else {
                    String cityName = findCityName(cityCode, districtCode, cityBeanList);
                    // 如果是直辖市，只返回"市·区县"
                    if (directCityCodes.contains(areaRevealBean.getCode())) {
                        return cityName;
                    } else {
                        //其他情况返回"省·市·区县"
                        return TextUtils.isEmpty(cityName) ? areaRevealBean.getLabel()
                            : areaRevealBean.getLabel() + cityName;
                    }
                }
            }
        }
        return null;
    }

    public String findProvinceName(String provinceCode) {
        if (TextUtils.isEmpty(provinceCode)) {
            return "";
        }
        if (provinceList == null || provinceList.isEmpty()) {
            return "";
        }
        for (AreaTreeselectBean areaRevealBean : provinceList) {
            if (areaRevealBean == null || TextUtils.isEmpty(areaRevealBean.getCode())) {
                continue;
            }

            if (provinceCode.equals(areaRevealBean.getCode())) {
                return areaRevealBean.getLabel();
            }
        }
        return null;
    }

    private String findCityName(String cityCode, String countyCode, List<AreaTreeselectBean.City> cityBeanList) {
        if (TextUtils.isEmpty(cityCode)) {
            return null;
        }
        for (AreaTreeselectBean.City city : cityBeanList) {
            if (city == null || TextUtils.isEmpty(city.getCode())) {
                continue;
            }

            if (cityCode.equals(city.getCode())) {
                List<AreaTreeselectBean.Country> counties = city.getChildren();
                if (counties == null || counties.isEmpty()) {
                    return city.getLabel();
                } else {
                    String countryName = findCountryName(countyCode, counties);
                    return TextUtils.isEmpty(countryName) ? city.getLabel() : city.getLabel() + countryName;
                }
            }
        }
        return null;
    }

    private String findCountryName(String countryCode, List<AreaTreeselectBean.Country> countryBeanList) {
        if (TextUtils.isEmpty(countryCode)) {
            return null;
        }
        for (AreaTreeselectBean.Country country : countryBeanList) {
            if (country == null || TextUtils.isEmpty(country.getCode())) {
                continue;
            }
            if (countryCode.equals(country.getCode())) {
                return country.getLabel();
            }
        }
        return null;
    }

    /**
     * 查找城市信息"省·区县"(只返回省和区县)
     *
     * @param provinceCode 省份信息
     * @param cityCode 城市信息
     * @param countyCode 区信息
     * @return 城市信息
     */
    public String findProvinceAndCounty(String provinceCode, String cityCode, String countyCode) {
        if (TextUtils.isEmpty(provinceCode)) {
            return "";
        }
        if (provinceList == null || provinceList.isEmpty()) {
            return "";
        }
        for (AreaTreeselectBean areaRevealBean : provinceList) {
            if (areaRevealBean == null || TextUtils.isEmpty(areaRevealBean.getCode())) {
                continue;
            }

            if (provinceCode.equals(areaRevealBean.getCode())) {
                List<AreaTreeselectBean.City> cityBeanList = areaRevealBean.getChildren();
                if (cityBeanList == null || cityBeanList.isEmpty()) {
                    return areaRevealBean.getLabel();
                } else {
                    String countyName = findProvinceAndCountyName(cityCode, countyCode, areaRevealBean.getChildren());
                    return TextUtils.isEmpty(countyName) ? areaRevealBean.getLabel()
                        : areaRevealBean.getLabel() + countyName;
                }
            }
        }
        return null;
    }

    private String findProvinceAndCountyName(String cityCode, String countyCode, List<AreaTreeselectBean.City> cityList) {
        if (TextUtils.isEmpty(cityCode)) {
            return null;
        }
        for (AreaTreeselectBean.City city : cityList) {
            if (city == null || TextUtils.isEmpty(city.getCode())) {
                continue;
            }

            if (cityCode.equals(city.getCode())) {
                List<AreaTreeselectBean.Country> counties = city.getChildren();
                if (counties == null || counties.isEmpty()) {
                    return city.getLabel();
                } else {
                    String countryName = findCountryName(countyCode, counties);
                    return TextUtils.isEmpty(countryName) ? city.getLabel() : countryName;
                }
            }
        }
        return null;
    }

    /**
     * 关键字搜索：1.优先显示市区信息;2.再显示省市信息
     *
     * @param keywords 关键字
     * @return 搜索结果
     */
    public List<GdDistrictBean> searchByKeyWord(String keywords) {
        List<GdDistrictBean> searchResultList = new ArrayList<>();
        if (TextUtils.isEmpty(keywords)) {
            return searchResultList;
        }
        if (provinceList == null || provinceList.isEmpty()) {
            return searchResultList;
        }

        // 1.优先匹配区
        for (AreaTreeselectBean areaRevealBean : provinceList) {
            if (areaRevealBean == null || TextUtils.isEmpty(areaRevealBean.getCode())
                || areaRevealBean.getChildren() == null) {
                continue;
            }
            for (AreaTreeselectBean.City city : areaRevealBean.getChildren()) {
                List<GdDistrictBean> result = searchCountryByKeyWord(keywords, city,areaRevealBean);
                if (result != null && !result.isEmpty()) {
                    searchResultList.addAll(result);
                }
            }
        }
        if (searchResultList != null && !searchResultList.isEmpty()) {
            return searchResultList;
        }

        // 2.再匹配市
        for (AreaTreeselectBean areaRevealBean : provinceList) {
            if (areaRevealBean == null || TextUtils.isEmpty(areaRevealBean.getCode())) {
                continue;
            }
            searchResultList = searchCityByKeyWord(keywords, areaRevealBean.getChildren(),areaRevealBean);
            if (searchResultList != null && !searchResultList.isEmpty()) {
                return searchResultList;
            }
        }
        // 3.再匹配省
        return searchProvinceByKeyWord(keywords);
    }

    private List<GdDistrictBean> searchProvinceByKeyWord(String keywords) {
        List<GdDistrictBean> searchResultList = new ArrayList<>();
        if (TextUtils.isEmpty(keywords)) {
            return searchResultList;
        }
        if (provinceList == null || provinceList.isEmpty()) {
            return searchResultList;
        }
        for (AreaTreeselectBean areaRevealBean : provinceList) {
            if (areaRevealBean == null || TextUtils.isEmpty(areaRevealBean.getCode())) {
                continue;
            }
            if (areaRevealBean.getLabel().contains(keywords) && areaRevealBean.getChildren() != null
                && !areaRevealBean.getChildren().isEmpty()) {
                    List<GdDistrictBean> gdDistrictBeans = new ArrayList<>();
                    for(AreaTreeselectBean.City city:areaRevealBean.getChildren()){
                        GdDistrictBean gdDistrictBean = new GdDistrictBean();
                        gdDistrictBean.setParentName(areaRevealBean.getLabel());
                        gdDistrictBean.setAdcode(areaRevealBean.getCode());
                        gdDistrictBean.setName(city.getLabel());
                        gdDistrictBean.setCitycode(city.getCode());
                        gdDistrictBeans.add(gdDistrictBean);
                    }
                    return gdDistrictBeans;
//                return areaRevealBean.getChildren().stream().map(city -> {
//
//                    GdDistrictBean gdDistrictBean = new GdDistrictBean();
//                    gdDistrictBean.setParentName(areaRevealBean.getLabel());
//                    gdDistrictBean.setAdcode(areaRevealBean.getCode());
//                    gdDistrictBean.setName(city.getLabel());
//                    gdDistrictBean.setCitycode(city.getCode());
//
//                    return gdDistrictBean;
//
//                }).collect(Collectors.toList());
            }
        }
        if (searchResultList == null || searchResultList.isEmpty()) {
            return new ArrayList<>();
        }
        return searchResultList;
    }

    private List<GdDistrictBean> searchCityByKeyWord(String keywords, List<AreaTreeselectBean.City> cities, AreaTreeselectBean areaRevealBean) {
        List<GdDistrictBean> searchResultList = new ArrayList<>();
        if (TextUtils.isEmpty(keywords)) {
            return searchResultList;
        }
        if (cities == null || cities.isEmpty()) {
            return searchResultList;
        }
        for (AreaTreeselectBean.City city : cities) {
            if (city == null || TextUtils.isEmpty(city.getLabel())) {
                continue;
            }

            if ((city.getLabel().contains(keywords) && city.getChildren() != null && !city.getChildren().isEmpty())) {
                List<GdDistrictBean> gdDistrictBeans = new ArrayList<>();
                for(AreaTreeselectBean.Country country: city.getChildren()){
                    GdDistrictBean gdDistrictBean = new GdDistrictBean();
                    gdDistrictBean.setParentName(city.getLabel());
                    gdDistrictBean.setAdcode(city.getCode());
                    gdDistrictBean.setName(country.getLabel());
                    gdDistrictBean.setCitycode(country.getCode());
                    gdDistrictBean.setProvince(areaRevealBean.getLabel());
                    gdDistrictBeans.add(gdDistrictBean);
                }
                return gdDistrictBeans;
//                return city.getChildren().stream().map(country -> {
//                    GdDistrictBean gdDistrictBean = new GdDistrictBean();
//                    gdDistrictBean.setParentName(city.getLabel());
//                    gdDistrictBean.setAdcode(city.getCode());
//                    gdDistrictBean.setName(country.getLabel());
//                    gdDistrictBean.setCitycode(country.getCode());
//                    gdDistrictBean.setProvince(areaRevealBean.getLabel());
//                    return gdDistrictBean;
//                }).collect(Collectors.toList());
            }
        }
        return null;
    }

    private List<GdDistrictBean> searchCountryByKeyWord(String keywords, AreaTreeselectBean.City city,AreaTreeselectBean areaRevealBean) {
        List<GdDistrictBean> searchResultList = new ArrayList<>();
        if (TextUtils.isEmpty(keywords)) {
            return searchResultList;
        }
        if (city == null) {
            return searchResultList;
        }
        List<AreaTreeselectBean.Country> countries = city.getChildren();
        for (AreaTreeselectBean.Country country : countries) {
            if (country == null || TextUtils.isEmpty(country.getLabel())) {
                continue;
            }

            if (country.getLabel().contains(keywords)) {
                GdDistrictBean gdDistrictBean = new GdDistrictBean();
                gdDistrictBean.setParentName(city.getLabel());
                gdDistrictBean.setAdcode(city.getCode());
                gdDistrictBean.setName(country.getLabel());
                gdDistrictBean.setCitycode(country.getCode());
                gdDistrictBean.setProvince(areaRevealBean.getLabel());
                searchResultList.add(gdDistrictBean);
            }
        }
        return searchResultList;
    }

}