package com.by4cloud.platformx.business.service.impl;

import cn.hutool.core.lang.tree.Tree;
import cn.hutool.core.lang.tree.TreeNode;
import cn.hutool.core.lang.tree.TreeUtil;
import cn.hutool.core.util.ObjUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.by4cloud.platformx.admin.api.entity.SysDept;
import com.by4cloud.platformx.business.dto.TaxCodeQueryDTO;
import com.by4cloud.platformx.business.entity.TaxCode;
import com.by4cloud.platformx.business.mapper.TaxCodeMapper;
import com.by4cloud.platformx.business.service.TaxCodeService;
import com.github.yulichang.wrapper.MPJLambdaWrapper;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;

import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * 税务编码
 *
 * @author syt
 * @date 2026-05-13 16:35:23
 */
@Service
public class TaxCodeServiceImpl extends ServiceImpl<TaxCodeMapper, TaxCode> implements TaxCodeService {

	@Override
	public Page pageNew(Page page, TaxCodeQueryDTO queryDTO) {
		MPJLambdaWrapper<TaxCode> wrapper = new MPJLambdaWrapper<TaxCode>()
				.selectAll(TaxCode.class)
				.selectAs("(select count(1) from tax_code t1 where t1.parent_bm = t.bm)",TaxCode::getChildNum)
				.like(StrUtil.isNotBlank(queryDTO.getMc()),TaxCode::getMc,queryDTO.getMc());
		if (ObjUtil.isNotNull(queryDTO.getChildNum())){
			if (queryDTO.getChildNum()==0){
				wrapper.eq("(select count(1) from tax_code t1 where t1.parent_bm = t.bm)",0);
			}else {
				wrapper.gt("(select count(1) from tax_code t1 where t1.parent_bm = t.bm)",0);
			}
		}
//		List<TaxCode> taxCodeList = baseMapper.selectPage(wrapper);
//		List<TaxCode> treeList=buildTreeByStream(taxCodeList, "1");
		return baseMapper.selectPage(page,wrapper);
	}

	/**
	 * 使用 Stream 递归构建树
	 * @param allNodes 所有节点
	 * @param parentId 当前层级的父ID
	 * @return 树形列表
	 */
	public static List<TaxCode> buildTreeByStream(List<TaxCode> allNodes, String parentId) {
		return allNodes.stream()
				.filter(node -> {
					// 过滤出父编码等于当前 parentId 的节点
					if (parentId == null) {
						return node.getParentBm() == null || node.getParentBm().isEmpty();
					}
					return parentId.equals(node.getParentBm());
				})
				.peek(node -> {
					// 递归设置子节点
					node.setChildren(buildTreeByStream(allNodes, node.getBm()));
				})
				.collect(Collectors.toList());
	}

	public <T> Page<T> convert(List<T> dataList, Page<T> pageParam) {
		if (dataList == null) {
			dataList = java.util.Collections.emptyList();
		}

		long total = dataList.size();
		long current = pageParam.getCurrent();
		long size = pageParam.getSize();

		// 计算总页数
		long pages = (total + size - 1) / size;

		// 计算截取范围
		long startIndex = (current - 1) * size;
		long endIndex = Math.min(startIndex + size, total);

		List<T> records;
		if (startIndex >= total) {
			records = java.util.Collections.emptyList();
		} else {
			records = dataList.subList((int) startIndex, (int) endIndex);
		}

		// 设置结果
		pageParam.setRecords(records);
		pageParam.setTotal(total);
		pageParam.setPages(pages);

		return pageParam;
	}
}