package com.xzx.gc.role.util;



import com.xzx.gc.entity.CoreMenu;
import com.xzx.gc.role.rbac.tree.MenuItem;

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

public class MenuBuildUtil {
    private MenuBuildUtil() {

    }

    public static MenuItem buildMenuTree(List<CoreMenu> list) {
        CoreMenu root = new CoreMenu();
        root.setId(0L);
        root.setType("");
        root.setName("主菜单");
        MenuItem rootMenu = new MenuItem(root);
        buildTreeNode(rootMenu, list);
        return rootMenu;
    }

    private static void buildTreeNode(MenuItem parent, List<CoreMenu> list) {

        if (parent.getData().getType().equals(CoreMenu.TYPE_MENUITEM)) {
            return;
        }


        Long id = parent.getId();
        List<CoreMenu> dels = new ArrayList<>();
        for (CoreMenu sysMenu : list) {
            if (id.equals(sysMenu.getParentMenuId() )) {
                MenuItem item = new MenuItem(sysMenu);
                item.setParent(parent);
                dels.add(sysMenu);
            }
        }
        list.removeAll(dels);

        if (list.isEmpty()) {
            return;
        }
        
        sortMenu(parent.getChildren());
        
        for (MenuItem child : parent.getChildren()) {
            buildTreeNode(child, list);
        }

    }

	private static void sortMenu(List<MenuItem> children) {
		Collections.sort(children, new Comparator<MenuItem>() {

			@Override
			public int compare(MenuItem o1, MenuItem o2) {
				return o1.getSeq().compareTo(o2.getSeq());
			}
			
		});
		
	}

}