package com.dsh.upms.controller;


import cn.mb.cloud.common.api.dto.MenuRouterTree;
import cn.mb.cloud.common.api.vo.MenuVO;
import cn.mb.cloud.common.api.vo.TreeUtil;
import cn.mb.cloud.common.core.constant.CommonConstants;
import cn.mb.cloud.common.core.constant.enums.ErrorCodeConstants;
import cn.mb.cloud.common.core.exception.BusinessException;
import cn.mb.cloud.common.core.util.ResponseData;
import cn.mb.cloud.common.log.annotation.SysLog;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.dsh.upms.entity.SysMenu;
import com.dsh.upms.entity.SysRoleMenu;
import com.dsh.upms.model.vo.LoginUserVo;
import com.dsh.upms.service.ISysMenuService;
import com.dsh.utils.login.LoginHelper;
import com.dsh.utils.login.Tree;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * <p>
 * 权限菜单 前端控制器
 * </p>
 *
 * @author null123
 * @since 2021-04-29
 */
@Api("权限菜单")
@RestController
@RequestMapping("/sysMenu")
public class SysMenuController {


    @Autowired
    private ISysMenuService sysMenuService;

    /**
     * 分页查询
     *
     * @param page    分页对象
     * @param sysMenu 菜单权限表
     * @return
     */
    @GetMapping("/page")
    @ApiOperation(value = "分页查询", tags = {"权限菜单"}, notes = "", response = SysMenu.class)
    @ApiImplicitParams({
            @ApiImplicitParam(value = "每页条数", name = "size", required = true, dataType = "Integer"),
            @ApiImplicitParam(value = "当前页数", name = "current", required = true, dataType = "Integer"),
            @ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
    })
    public ResponseData getSysMenuPage(Page page, SysMenu sysMenu) {
        Page pages = sysMenuService.page(page, Wrappers.query(sysMenu));
        return ResponseData.success(pages);
    }


    /**
     * 通过id查询菜单权限
     *
     * @param id id
     * @return SysMenu
     */
    @GetMapping("/getById")
    @ApiOperation(value = "通过id查询菜单权限", tags = {"权限菜单"}, notes = "", response = SysMenu.class)
    @ApiImplicitParams({
            @ApiImplicitParam(value = "菜单ID", name = "id", required = true, dataType = "Long"),
            @ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
    })
    public ResponseData getById(Long id) {
        SysMenu sysMenu = sysMenuService.getById(id);
        return ResponseData.success(sysMenu);
    }

    /**
     * 新增菜单权限表
     *
     * @param sysMenu 菜单权限表
     * @return SysMenu
     */
    @SysLog("新增菜单权限表")
    @PostMapping("/create")
    public ResponseData save(@RequestBody SysMenu sysMenu) {
        boolean result = sysMenuService.save(sysMenu);
        return ResponseData.success(result);
    }

    /**
     * 修改菜单权限表
     *
     * @param sysMenu 菜单权限表
     * @return R
     */
    @SysLog("修改菜单权限表")
    @PostMapping("updateById")
    @ApiOperation(value = "修改菜单权限表", tags = {"权限菜单"}, notes = "")
    @ApiImplicitParams({
            @ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
    })
    public ResponseData updateById(@RequestBody SysMenu sysMenu) {
        boolean result = sysMenuService.updateById(sysMenu);
        return ResponseData.success(result);
    }

    /**
     * 通过id删除菜单权限表
     *
     * @param id id
     * @return R
     */
    @SysLog("删除菜单权限表")
    @DeleteMapping("/removeById")
    public ResponseData removeById(@PathVariable Long id) {
        boolean result = sysMenuService.removeById(id);
        return ResponseData.success(result);
    }

    /**
     * 启用禁用
     *
     * @param id
     * @return
     */
    @GetMapping("enable")
    @ApiOperation(value = "启用禁用", tags = {"权限菜单"}, notes = "")
    @ApiImplicitParams({
            @ApiImplicitParam(value = "菜单ID", name = "id", required = true, dataType = "Long"),
            @ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
    })
    public ResponseData<Boolean> enable(@RequestParam("id") Long id) {
        SysMenu menu = sysMenuService.getById(id);
        if (null != menu) {
            menu.setEnable(!menu.getEnable());
        } else {
            return ResponseData.fail("权限不存在");
        }
        return ResponseData.success(sysMenuService.updateById(menu));
    }

    /**
     * 返回所有树形菜单集合
     *
     * @return 树形菜单
     */
    @GetMapping("/tree")
    @ApiOperation(value = "返回所有树形菜单集合", tags = {"权限菜单"}, notes = "")
    @ApiImplicitParams({
            @ApiImplicitParam(value = "菜单名称", name = "title", required = false, dataType = "String"),
            @ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
    })
    public ResponseData getTree(@RequestParam(value = "name", required = false) String title) {
        try {
            List<SysMenu> menus = sysMenuService.getBaseMapper()
                    .selectList(Wrappers.<SysMenu>query().lambda()
                            .eq(SysMenu::getDelFlag, 0).like(StringUtils.isNotBlank(title), SysMenu::getName, title)
                            .orderByAsc(SysMenu::getSort));
            List<Tree> treeList = new ArrayList<Tree>();
            Tree rootTree = new Tree("系统菜单");
            for (SysMenu menu : menus) {
                Long parent = menu.getParentId();
                if (null == parent) {
                    Tree tree = new Tree(menu.getId().toString(), menu.getName(), Tree.TREE_NODE_STATE_DEFAULT);
                    tree.setPermission(menu.getPermission());
                    tree.setType(menu.getType());
                    tree.setName(menu.getName());
                    tree.setLable(menu.getName());
                    tree.setPath(menu.getPath());
                    tree.setIcon(menu.getIcon());
                    tree.setEnable(menu.getEnable());
                    tree.setSort(menu.getSort());
                    SysMenu.initTree(menus, tree);
                    treeList.add(tree);
                }
            }
            rootTree.setChildren(treeList);
            return new ResponseData<>(treeList);
        } catch (Exception e) {
            e.printStackTrace();
            return ResponseData.fail("操作失败");
        }
    }

    /**
     * 返回当前登录用户的树形菜单集合
     *
     * @return 当前用户的树形菜单
     */
    @GetMapping("/user/tree")
    @ApiOperation(value = "返回当前登录用户的树形菜单集合", tags = {"权限菜单"}, notes = "", response = MenuRouterTree.class)
    @ApiImplicitParams({
            @ApiImplicitParam(name = "Authorization", value = "Bearer +token", required = true, dataType = "String", paramType = "header", defaultValue = "Bearer eyJhbGciOiJIUzUxMiJ9.....")
    })
    public ResponseData getUserMenu() throws BusinessException {
        // 获取当前登录用户信息
        LoginUserVo loginUserVo = LoginHelper.getUser();
        if (Objects.isNull(loginUserVo)) {
            throw new BusinessException("请登录", ErrorCodeConstants.FAIL.getValue());
        }
        // 获取符合条件的菜单
        Set<MenuVO> all = sysMenuService.findMenuByUserId(loginUserVo.getId());
        List<MenuRouterTree> menuTreeList = all.stream()
                .filter(menuVo -> CommonConstants.MENU.equals(menuVo.getType()))
                .map(MenuRouterTree::new)
                .sorted(Comparator.comparingInt(MenuRouterTree::getSort))
                .collect(Collectors.toList());
        return ResponseData.success(TreeUtil.bulidRouter(menuTreeList));
    }

}