/*
 * Copyright (c) 2018-2999 广州市蓝海创新科技有限公司 All rights reserved.
 *
 * https://www.mall4j.com/
 *
 * 未经允许，不可做商业用途！
 *
 * 版权所有，侵权必究！
 */

package com.yami.trading.sys.service.impl;


import cn.hutool.core.collection.CollectionUtil;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.yami.trading.sys.constant.Constant;
import com.yami.trading.sys.dao.SysMenuMapper;
import com.yami.trading.sys.dao.SysRoleMenuMapper;
import com.yami.trading.sys.model.SysMenu;
import com.yami.trading.sys.service.SysMenuService;
import lombok.AllArgsConstructor;
import org.apache.ibatis.annotations.Param;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.stream.Collectors;

/**
 * @author lgh
 */
@Service("sysMenuService")
@AllArgsConstructor
public class SysMenuServiceImpl extends ServiceImpl<SysMenuMapper, SysMenu> implements SysMenuService {

	private final SysRoleMenuMapper sysRoleMenuMapper;

	private final SysMenuMapper sysMenuMapper;

	@Override
	public List<SysMenu> listMenuByUserId(Long userId,String appType) {
		// 用户的所有菜单信息
		List<SysMenu> sysMenus ;
		//系统管理员，拥有最高权限
		if(userId == Constant.SUPER_ADMIN_ID||userId == Constant.SUPER_ROOT_ID){
			sysMenus = sysMenuMapper.listMenu(appType);
		}else {
			sysMenus = sysMenuMapper.listMenuByUserId(userId,appType);

			//记录全勾选
			Map<Long,SysMenu> mapSysMenu = new HashMap<>();
            for (SysMenu sysMenu : sysMenus) {
                mapSysMenu.put(sysMenu.getMenuId(), sysMenu);
            }

			List<Long> bigIntArrays = new Vector<>();

			//权限
			List<SysMenu> sysMenus3 = sysMenuMapper.queryAllPerms2(userId);
			for (SysMenu sysMenu : sysMenus3) {
				if (!mapSysMenu.containsKey(sysMenu.getParentId())
						&& !bigIntArrays.contains(sysMenu.getParentId())
						&& sysMenu.getParentId() != 0) {
					bigIntArrays.add(sysMenu.getParentId());
				}
			}

			//补充
			if(!bigIntArrays.isEmpty()){
				List<SysMenu> sysMenus2 = sysMenuMapper.listMenuByUserId2(bigIntArrays,appType);
				for (SysMenu sysMenu : sysMenus2) {
					mapSysMenu.put(sysMenu.getMenuId(), sysMenu);
					//权限按钮的父节点
					if (!mapSysMenu.containsKey(sysMenu.getParentId())
							&& !bigIntArrays.contains(sysMenu.getParentId())
							&& sysMenu.getParentId() != 0) {
						bigIntArrays.add(sysMenu.getParentId());
					}
				}
				sysMenus.addAll(sysMenus2);
			}
			bigIntArrays.clear();

			//子菜单
            for (SysMenu sysMenu : sysMenus) {
                if (!mapSysMenu.containsKey(sysMenu.getParentId())
                        && !bigIntArrays.contains(sysMenu.getParentId())
                        && sysMenu.getParentId() != 0) {
                    bigIntArrays.add(sysMenu.getParentId());
                }
            }

			//补充
			if(!bigIntArrays.isEmpty()){
				List<SysMenu> sysMenus2 = sysMenuMapper.listMenuByUserId2(bigIntArrays,appType);
//				for (SysMenu sysMenu : sysMenus2) {
//					mapSysMenu.put(sysMenu.getMenuId(), sysMenu);
//					//权限按钮的父节点
//					if (!mapSysMenu.containsKey(sysMenu.getParentId())
//							&& !bigIntArrays.contains(sysMenu.getParentId())
//							&& sysMenu.getParentId() != 0) {
//						bigIntArrays.add(sysMenu.getParentId());
//					}
//				}
				sysMenus.addAll(sysMenus2);
			}





		}

		Map<Long, List<SysMenu>> sysMenuLevelMap = sysMenus.stream()
				.sorted(Comparator.comparing(SysMenu::getOrderNum))
				.collect(Collectors.groupingBy(SysMenu::getParentId));

		//System.out.println("sys = " +sysMenuLevelMap.values());

		// 一级菜单
		List<SysMenu> rootMenu = sysMenuLevelMap.get(0L);
		if (CollectionUtil.isEmpty(rootMenu)) {
			return Collections.emptyList();
		}

		// 二级菜单
		for (SysMenu sysMenu : rootMenu) {
			sysMenu.setList(sysMenuLevelMap.get(sysMenu.getMenuId()));
		}

		// 三级菜单
		for (SysMenu sysMenu : rootMenu) {
			List list = sysMenu.getList();
			if(list != null){
				for (Object object : list) {
					SysMenu sysMenu2 = (SysMenu)object;
					sysMenu2.setList(sysMenuLevelMap.get(sysMenu2.getMenuId()));
				}
			}
		}

		return rootMenu;
	}

	@Override
	public void deleteMenuAndRoleMenu(Long menuId){
		//删除菜单
		this.removeById(menuId);
		//删除菜单与角色关联
		sysRoleMenuMapper.deleteByMenuId(menuId);
	}


	@Override
	public List<Long> listMenuIdByRoleId(Long roleId,String appType) {
		return sysMenuMapper.listMenuIdByRoleId(roleId,appType);
	}


	@Override
	public List<SysMenu> listSimpleMenuNoButton(String appType) {
		return sysMenuMapper.listSimpleMenuNoButton(appType);
	}

	@Override
	public List<SysMenu> listRootMenu() {
		return sysMenuMapper.listRootMenu();
	}

	@Override
	public List<SysMenu> listChildrenMenuByParentId(Long parentId) {
		return sysMenuMapper.listChildrenMenuByParentId(parentId);
	}

	@Override
	public List<SysMenu> listMenuAndBtn( String appType) {
		return sysMenuMapper.listMenuAndBtn(appType);
	}

}