/*
 *      Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are met:
 *
 *  Redistributions of source code must retain the above copyright notice,
 *  this list of conditions and the following disclaimer.
 *  Redistributions in binary form must reproduce the above copyright
 *  notice, this list of conditions and the following disclaimer in the
 *  documentation and/or other materials provided with the distribution.
 *  Neither the name of the dreamlu.net developer nor the names of its
 *  contributors may be used to endorse or promote products derived from
 *  this software without specific prior written permission.
 *  Author: Chill 庄骞 (smallchill@163.com)
 */
package org.sxkj.system.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.apache.commons.lang3.StringUtils;
import org.apache.logging.log4j.util.Strings;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.tool.utils.*;
import org.springframework.stereotype.Service;
import org.sxkj.common.func.Streams;
import org.sxkj.system.entity.Region;
import org.sxkj.system.excel.RegionExcel;
import org.sxkj.system.excel.RegionImportExcel;
import org.sxkj.system.mapper.RegionMapper;
import org.sxkj.system.param.RegionExportParam;
import org.sxkj.system.param.RegionPageParam;
import org.sxkj.system.service.IRegionService;
import org.sxkj.system.vo.RegionVO;
import org.sxkj.system.vo.TreeVo;

import java.io.Serializable;
import java.util.*;

import static org.sxkj.system.cache.RegionCache.*;


/**
 * 行政区划表 服务实现类
 *
 * @author Chill
 */
@Service
public class RegionServiceImpl extends ServiceImpl<RegionMapper, Region> implements IRegionService {

	@Override
	public Region getById(Serializable id) {
		return Optional.ofNullable(id)
			.map(val -> baseMapper.selectOne(new LambdaQueryWrapper<Region>()
				// 1. 处理 id = ? 部分，强制转为 bigint
				.apply("id = {0}::bigint", val)
				.or()
				// 2. 处理 code = ? 部分，强制转为 varchar
				.apply("code = {0}::varchar", val)))
			.orElse(null);
	}

	@Override
	public boolean submit(Region region) {
		// 设置市级编号格式
		String regionCode = region.getCode();
		String regionParentCode = region.getParentCode();
		if (regionCode.startsWith(MAIN_CODE)) {
			region.setCode(StringUtil.removePrefix(regionCode, MAIN_CODE));
		}
		if (regionParentCode.startsWith(MAIN_CODE)) {
			region.setParentCode(StringUtil.removePrefix(regionParentCode, MAIN_CODE));
		}
		region.setCityName("吉安市");
		region.setCityCode("360800000000");
		region.setProvinceCode("360000000000");
		region.setProvinceName("江西省");
		if (region.getRegionLevel().equals(DISTRICT_LEVEL)) {
			region.setDistrictCode(region.getCode());
			region.setDistrictName(region.getName());
		}
		if (region.getRegionLevel().equals(TOWN_LEVEL)) {
			String parentCode = region.getParentCode();
			Region byCode = getByCode(parentCode);
			region.setDistrictCode(byCode.getCode());
			region.setDistrictName(byCode.getName());
			region.setTownCode(region.getCode());
			region.setTownName(region.getName());
		}
		if (region.getRegionLevel().equals(VILLAGE_LEVEL)) {
			String parentCode = region.getParentCode();
			Region byCode = getByCode(parentCode);
			region.setTownCode(byCode.getCode());
			region.setTownName(byCode.getName());
			Region byCode2 = getByCode(byCode.getParentCode());
			region.setDistrictCode(byCode2.getCode());
			region.setDistrictName(byCode2.getName());
			region.setVillageCode(region.getCode());
			region.setVillageName(region.getName());
		}
		// 查询是否已存在
		Long cnt = baseMapper.selectCount(Wrappers.<Region>query().lambda().eq(Region::getCode, region.getCode()));
		if (cnt > 0L) {
			return this.updateById(region);
		}
		// 设置祖区划编号
		Region parent = getByCode(region.getParentCode());
		if (Func.isNotEmpty(parent) || Func.isNotEmpty(parent.getCode())) {
			String ancestors = parent.getAncestors() + StringPool.COMMA + parent.getCode();
			region.setAncestors(ancestors);
		}
		// 设置省、市、区、镇、村
		Integer level = region.getRegionLevel();
		String code = region.getCode();
		String name = region.getName();
		if (level == PROVINCE_LEVEL) {
			region.setProvinceCode(code);
			region.setProvinceName(name);
		} else if (level == CITY_LEVEL) {
			region.setCityCode(code);
			region.setCityName(name);
		} else if (level == DISTRICT_LEVEL) {
			region.setDistrictCode(code);
			region.setDistrictName(name);
		} else if (level == TOWN_LEVEL) {
			region.setTownCode(code);
			region.setTownName(name);
		} else if (level == VILLAGE_LEVEL) {
			region.setVillageCode(code);
			region.setVillageName(name);
		}
		return this.save(region);
	}

	@Override
	public boolean removeRegion(String id) {
		Long cnt = baseMapper.selectCount(Wrappers.<Region>query().lambda().eq(Region::getParentCode, id));
		if (cnt > 0L) {
			throw new ServiceException("请先删除子节点!");
		}
		Long idLong = Long.valueOf(id);
		return removeById(idLong);
	}

	@Override
	public List<RegionVO> lazyList(String parentCode, Map<String, Object> param) {
		return baseMapper.lazyList(parentCode, param);
	}

	@Override
	public List<RegionVO> lazyTree(String parentCode, Map<String, Object> param) {
		return baseMapper.lazyTree(parentCode, param);
	}

	// WHEN
	// 	region_level = '5' THEN town_name
	// WHEN
	// 	region_level = '4' THEN district_name
	// WHEN
	// 	region_level = '3' THEN city_name
	// WHEN
	// 	region_level = '2' THEN province_name
	@Override
	public void importRegion(List<RegionImportExcel> data, Boolean isCovered) {
		List<Region> list = new ArrayList<>();
		data.forEach(regionExcel -> {
			Region region = BeanUtil.copy(regionExcel, Region.class);
			region.setRegionLevel(Integer.valueOf(regionExcel.getRegionLevel()));
			region.setCityName("吉安市");
			region.setCityCode("360800000000");
			region.setProvinceCode("360000000000");
			region.setProvinceName("江西省");
			if (region.getRegionLevel().equals(DISTRICT_LEVEL)) {
				region.setDistrictCode(region.getCode());
				region.setDistrictName(region.getName());
			}
			if (region.getRegionLevel().equals(TOWN_LEVEL)) {
				String parentCode = region.getParentCode();
				Region byCode = getByCode(parentCode);
				region.setDistrictCode(byCode.getCode());
				region.setDistrictName(byCode.getName());
				region.setTownCode(region.getCode());
				region.setTownName(region.getName());
			}
			if (region.getRegionLevel().equals(VILLAGE_LEVEL)) {
				String parentCode = region.getParentCode();
				Region byCode = getByCode(parentCode);
				region.setTownCode(byCode.getCode());
				region.setTownName(byCode.getName());
				Region byCode2 = getByCode(byCode.getParentCode());
				region.setDistrictCode(byCode2.getCode());
				region.setDistrictName(byCode2.getName());
				region.setVillageCode(region.getCode());
				region.setVillageName(region.getName());
			}
			list.add(region);
		});
		if (isCovered) {
			this.saveOrUpdateBatch(list);
		} else {
			this.saveBatch(list);
		}
	}

	@Override
	public List<RegionExcel> exportRegion(RegionExportParam region) {
		return baseMapper.exportRegion(region);
	}

	/**
	 * 区域数据处理
	 */
	@Override
	public void regionDataHandle() {
		Integer size = 1000;
		// 查询总数
		Integer count = baseMapper.getCodeLess12DataCount();
		if (count > 0) {
			int num = count / size;
			for (int i = 0; i <= num + 1; i++) {
				// 查询所有区划数据
				List<Region> regionList = baseMapper.getCodeLess12DataList(size);
				// 遍历处理
				for (Region region : regionList) {
					// 设置code
					region.setCode(StringUtils.rightPad(region.getCode(), 12, "0"));
					// 设置parent_code
					region.setParentCode(StringUtils.rightPad(region.getParentCode(), 12, "0"));
					// 设置Ancestors
					setAncestors(region);
					// 设置ProvinceCode
					region.setProvinceCode(StringUtils.rightPad(region.getProvinceCode(), 12, "0"));
					// 设置CityCode
					region.setCityCode(StringUtils.rightPad(region.getCityCode(), 12, "0"));
					// 设置DistrictCode
					region.setDistrictCode(StringUtils.rightPad(region.getDistrictCode(), 12, "0"));
					// 设置TownCode
					region.setTownCode(StringUtils.rightPad(region.getTownCode(), 12, "0"));
					// 更新
//					updateById(region);
				}
				// 批量更新
				this.updateBatchById(regionList);
			}
		}
	}

	/**
	 * Ancestors 字段处理
	 *
	 * @param region
	 */
	private void setAncestors(Region region) {
		String[] split = region.getAncestors().split(",");
		List<String> list = new ArrayList<>();
		for (String s : split) {
			list.add(StringUtils.rightPad(s, 12, "0"));
		}
		region.setAncestors(Strings.join(list, ','));
	}

	@Override
	public TreeVo buildRegionTree(String parentCode) {
		if (StringUtil.isBlank(parentCode)) {
			return null;
		}
		LambdaQueryWrapper<Region> query = Wrappers.<Region>lambdaQuery()
			.le(Region::getRegionLevel, 3)
			.likeRight(Region::getParentCode, parentCode.replaceAll("0+$", ""))
			.or(regionLambdaQueryWrapper ->
				regionLambdaQueryWrapper.eq(Region::getCode, parentCode)
			);
		List<Region> list = list(query);
		List<TreeVo> regionVOS = Streams.toList(list, region -> {
			TreeVo vo = new TreeVo();
			vo.setId(vo.getId());
			vo.setParentId(region.getParentCode());
			vo.setId(region.getCode());
			vo.setName(region.getName());
			vo.setRegionLevel(region.getRegionLevel());
			return vo;
		});
		Map<String, TreeVo> map = Streams.toMap(regionVOS, TreeVo::getId, S -> S);

		List<TreeVo> regionVOS1 = buildRegionTree(regionVOS, parentCode);
		TreeVo treeVo = map.get(parentCode);
		if (Objects.nonNull(treeVo)) {
			treeVo.setChildrens(regionVOS1);
		}

		return treeVo;
	}

	@Override
	public IPage<RegionVO> selectRegionPage(IPage<RegionVO> page, RegionPageParam regionPageParam) {
		return baseMapper.selectRegionPage(page, regionPageParam);
	}

	/**
	 * Query regions by parentCode and build a recursive tree structure
	 *
	 * @param allRegions List of all regions from database
	 * @param parentCode The parent code to start building the tree from
	 * @return List of RegionVO trees starting from the specified parentCode
	 */
	public List<TreeVo> buildRegionTree(List<TreeVo> allRegions, String parentCode) {
		if (CollectionUtil.isEmpty(allRegions)) {
			return Collections.EMPTY_LIST;
		}

		Map<String, List<TreeVo>> regionMap = new HashMap<>();
		for (TreeVo region : allRegions) {
			String pCode = region.getParentId();
			regionMap.computeIfAbsent(pCode, k -> new ArrayList<>()).add(region);
		}
		return buildTree(regionMap, parentCode);
	}

	/**
	 * Recursive method to build the region tree
	 *
	 * @param regionMap   Map of regions grouped by parentCode
	 * @param currentCode Current region code to process
	 * @return List of RegionVOs with children populated
	 */
	private List<TreeVo> buildTree(Map<String, List<TreeVo>> regionMap, String currentCode) {
		List<TreeVo> nodes = regionMap.get(currentCode);
		if (nodes == null) {
			return new ArrayList<>();
		}
		for (TreeVo node : nodes) {
			List<TreeVo> children = buildTree(regionMap, node.getId());
			if (!children.isEmpty()) {
				node.setChildrens(children);
				node.setHasChildren(true);
			}
		}
		return nodes;
	}

}