package com.yanzhen.controller;

import com.yanzhen.entity.Menu;
import com.yanzhen.entity.User;
import com.yanzhen.service.MenuService菜单;
import com.yanzhen.utils.Result;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletRequest;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
@RestController // 声明这是一个控制器，并且返回的数据直接写入 HTTP 响应体中，而不是解析为跳转路径。
@RequestMapping("/menu") // 设置请求的根路径为 "/menu"
public class MenuController {

    @Autowired // 自动注入 MenuService 对象
    private MenuService菜单 menuService;

    @GetMapping("/query") // 处理对 "/menu/query" 的 GET 请求
    public Result query(HttpServletRequest request){
        List<Menu> menus = new ArrayList<>(); // 创建一个空的菜单列表
        if(request.getAttribute("user") != null){ // 如果请求中有 "user" 属性
            User user = (User) request.getAttribute("user"); // 获取用户信息
            menus = menuService.query(user.getId()); // 根据用户ID查询菜单
        }else if(request.getAttribute("student") != null){ // 如果请求中有 "student" 属性
            menus = menuService.queryByType(); // 查询所有学生类型的菜单
        }

        List<Menu> menuList1 = new ArrayList<>(); // 创建一级菜单列表
        //找出一级菜单
        for (Menu menu : menus) { // 遍历所有菜单项
            if(menu.getParentId() == 0){ // 如果父ID为0，则为一级菜单
                menuList1.add(menu); // 添加到一级菜单列表中
            }
        }

        //嵌套循环找出关联设置child属性
        for (Menu parent : menuList1) { // 遍历一级菜单
            List<Menu> child = new ArrayList<>(); // 创建子菜单列表
            for (Menu entity : menus) { // 遍历所有菜单项
                if(parent.getId() == entity.getParentId()){ // 如果当前菜单是某个一级菜单的子菜单
                    child.add(entity); // 添加到子菜单列表中
                }
            }
            parent.setChild(child); // 设置一级菜单的子菜单列表
        }

        return Result.ok(menuList1); // 返回包含一级菜单及其子菜单的结果
    }

    @GetMapping("/tree") // 处理对 "/menu/tree" 的 GET 请求
    public Result tree(Integer checked,HttpServletRequest request){
        //check查询的时候是否选中状态
        List<Integer> checkedMenuId = new ArrayList<>(); // 创建已选中菜单ID列表
        if(checked != null){ // 如果请求参数中包含 "checked"
            User user = (User) request.getAttribute("user"); // 获取用户信息
            //查询出来自己已经设置过得menuId
            checkedMenuId = menuService.queryCheckMenuId(user.getId()); // 查询用户已设置的菜单ID
        }
        List<Menu> list = menuService.list(); // 获取所有菜单列表
        List<Map<String,Object>> menus = new ArrayList<>(); // 创建用于存储树形结构的菜单列表
        for (Menu menu : list) { // 遍历所有菜单项
            if(menu.getParentId()==0){ // 如果父ID为0，则为一级菜单
                Map<String,Object> map = new HashMap<>(); // 创建存储一级菜单信息的映射
                map.put("id",menu.getId()); // 添加菜单ID
                map.put("name",menu.getTitle()); // 添加菜单名称
                map.put("isParent",true); // 标记为父节点
                map.put("open",true); // 默认展开
                if(checked != null){ // 如果存在已选中的菜单ID
                    map.put("checked",checkedMenuId.contains(menu.getId())); // 设置是否选中
                }
                List<Map<String,Object>> child = new ArrayList<>(); // 创建子菜单列表
                for (Menu menu1 : list) { // 遍历所有菜单项
                    if(menu1.getParentId()!=0 && menu.getId() == menu1.getParentId()){ // 如果当前菜单是某个一级菜单的子菜单
                        Map<String,Object> map2 = new HashMap<>(); // 创建存储子菜单信息的映射
                        map2.put("id",menu1.getId()); // 添加子菜单ID
                        map2.put("name",menu1.getTitle()); // 添加子菜单名称
                        map2.put("isParent",false); // 标记为子节点
                        if(checked != null){ // 如果存在已选中的菜单ID
                            map2.put("checked",checkedMenuId.contains(menu1.getId())); // 设置是否选中
                        }
                        child.add(map2); // 添加到子菜单列表中
                    }
                }
                map.put("children",child); // 将子菜单列表添加到一级菜单的映射中
                menus.add(map); // 将一级菜单的映射添加到结果列表中
            }
        }
        return Result.ok(menus); // 返回包含树形结构菜单的结果
    }

}