package org.jeecg.common.api.vo;

import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.experimental.Accessors;
import org.apache.commons.lang.StringUtils;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;

@SuppressWarnings("rawtypes")
@Data
@NoArgsConstructor
@Accessors(chain = true)
public class CommonGenericTree<T> {

	private String key;// 要显示的节点的ID

	private String title;// 要显示的子节点的名称

	private String icon; // 节点的图标

	private String parentId;// 父节点的ID

	private boolean isLeaf;// 是否为叶子节点

	private boolean selectable = true;// 是否可选

	private boolean disabled = false;// 是否可选

	private List<CommonGenericTree> children;// 子节点的List
	
	private String rField1;// 绑定扩展属性1

	private String rField2;// 绑定扩展属性2

	private String rField3;// 绑定扩展属性3

	private ScopedSlots scopedSlots = new ScopedSlots();

	private Integer type;

	private T entity;

	private String value;

	/**
	 * @Description: 添加子节点
	 */
	@SuppressWarnings("unchecked")
	public CommonGenericTree<T> addChildren(CommonGenericTree node) {
		if (this.children == null) {
			this.children = new ArrayList<CommonGenericTree>();
		}
		this.children.add(node);
		return node;
	}

	public boolean hasChildren() {
		boolean bu = true;
		if (getChildren() == null || getChildren().isEmpty()) {
			bu = false;
			return bu;
		}
		return bu;
	}

	/**
	 * @Description: 获取树的所有叶子节点
	 */
	@SuppressWarnings("unchecked")
	public static Map<String, CommonGenericTree> getLeafNode(Map<String, CommonGenericTree> map,
			CommonGenericTree tree) {
		if (tree == null)
			return map;
		if (tree.hasChildren()) {
			List<CommonGenericTree> childrenNodes = tree.getChildren();
			for (CommonGenericTree child : childrenNodes) {
				map = getLeafNode(map, child);
			}
			map.put(tree.getKey(), tree);
		} else {
			map.put(tree.getKey(), tree);
		}
		return map;
	}

	public static Map<String, CommonGenericTree> getAllParentNode(Map<String, CommonGenericTree> map,
			Map<String, CommonGenericTree> allNode, CommonGenericTree childNode) {
		if (childNode == null || allNode == null || allNode.isEmpty())
			return map;
		String parentId = childNode.getParentId();
		if (StringUtils.isNotBlank(parentId)) {
			CommonGenericTree parent;
			if (allNode.containsKey(parentId)) {
				parent = allNode.get(childNode.getParentId());
				map.put(childNode.getKey(), childNode);
				map = getAllParentNode(map, allNode, parent);
			} else {
				map.put(childNode.getKey(), childNode);
			}

		} else {
			map.put(childNode.getKey(), childNode);
		}
		return map;
	}

	@SuppressWarnings("unchecked")
	public static CommonGenericTree copyToNewTree(CommonGenericTree oldNode, CommonGenericTree newNode,
			Map<String, CommonGenericTree> map) {
		if (oldNode == null || map == null || map.isEmpty())
			return newNode;
		if (oldNode.hasChildren()) {
			List<CommonGenericTree> childrenNodes = oldNode.getChildren();
			CommonGenericTree newChild;
			for (CommonGenericTree child : childrenNodes) {
				if (map.containsKey(child.getKey())) {
					newChild = new CommonGenericTree();
					newChild.copyNoneChild(child);
					newNode.addChildren(newChild);
					copyToNewTree(child, newChild, map);
				}
			}
		}
		return newNode;

	}

	public static List<CommonGenericTree> convertMapToList(Map<String, CommonGenericTree> map) {
		if (map == null || map.isEmpty())
			return Collections.emptyList();
		List<CommonGenericTree> tree = new ArrayList<>();
		CommonGenericTree node;
		CommonGenericTree parent;
		for (Map.Entry<String, CommonGenericTree> en : map.entrySet()) {
			node = en.getValue();
			if (node != null) {
				if (node.getParentId() == null) {
					tree.add(en.getValue());
				} else {
					if (map.containsKey(node.getParentId())) {
						parent = map.get(node.getParentId());
						parent.addChildren(node);
					}
				}
			}
		}
		return tree;
	}

	@SuppressWarnings("unchecked")
	public CommonGenericTree copyNoneChild(CommonGenericTree oldNode) {
		this.setKey(oldNode.getKey()).setTitle(oldNode.getTitle()).setType(oldNode.getType()).setIcon(oldNode.getIcon())
				.setEntity((T) oldNode.getEntity()).setParentId(oldNode.getParentId()).setRField1(oldNode.getRField1())
				.setRField2(oldNode.getRField2());
		return this;
	}

}