package com.tievd.cube.modules.system.controller;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.fasterxml.jackson.databind.JsonNode;
import com.tievd.cube.commons.constant.CommonConst;
import com.tievd.cube.commons.base.CubeController;
import com.tievd.cube.commons.mybatisplus.QueryGenerator;
import com.tievd.cube.commons.utils.SystemContextUtil;
import com.tievd.cube.commons.annotations.AutoLog;
import com.tievd.cube.commons.base.Result;
import com.tievd.cube.commons.easyexcel.model.ImportExcel;
import com.tievd.cube.commons.utils.web.HttpServletUtil;
import com.tievd.cube.modules.system.entity.SysPermission;
import com.tievd.cube.modules.system.entity.SysPermissionDataRule;
import com.tievd.cube.modules.system.entity.SysRole;
import com.tievd.cube.modules.system.entity.SysRolePermission;
import com.tievd.cube.modules.system.model.TreeModel;
import com.tievd.cube.modules.system.service.ISysPermissionDataRuleService;
import com.tievd.cube.modules.system.service.ISysPermissionService;
import com.tievd.cube.modules.system.service.ISysRolePermissionService;
import com.tievd.cube.modules.system.service.ISysRoleService;
import io.swagger.v3.oas.annotations.tags.Tag;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.*;

/**
 * 角色
 *
 * @author xinwuy
 * @version V2.3.0
 * @since 2021-10-08
 */
@Slf4j
@Tag(name = "角色相关接口")
@RestController
@RequestMapping("/sys/role")
public class SysRoleController extends CubeController<SysRole, ISysRoleService> {

    @Autowired
    private ISysPermissionDataRuleService sysPermissionDataRuleService;
    @Autowired
    private ISysRolePermissionService sysRolePermissionService;
    @Autowired
    private ISysPermissionService sysPermissionService;

    /**
     * 分页列表查询
     *
     * @param role     查询参数
     * @param pageNo   页码
     * @param pageSize 每页大小
     */
    @GetMapping("/list")
    public Result<IPage<SysRole>> queryPageList(SysRole role, @RequestParam(defaultValue = "1") Integer pageNo, @RequestParam(defaultValue = "10") Integer pageSize, HttpServletRequest req) {
        QueryWrapper<SysRole> queryWrapper = QueryGenerator.initQueryWrapper(role, req.getParameterMap());
        Page<SysRole> page = new Page<>(pageNo, pageSize);
        IPage<SysRole> pageList = service.page(page, queryWrapper);
        return Result.ok(pageList);
    }

    /**
     * 新增
     *
     * @param role 角色信息
     */
    @AutoLog("角色管理-添加角色")
    @PostMapping("/add")
    public Result<SysRole> add(@RequestBody SysRole role) {
        service.save(role);
        return Result.ok();
    }

    /**
     * 更新
     *
     * @param role 角色信息
     */
    @AutoLog("角色管理-编辑角色")
    @PutMapping("/edit")
    public Result<?> edit(@RequestBody SysRole role) {
        SysRole sysrole = service.getById(role.getId());
        if (sysrole == null) {
            return Result.error("未找到对应实体！");
        }
        service.updateById(role);
        return Result.ok();
    }

    /**
     * 通过id删除角色
     *
     * @param id 角色id
     */
    @AutoLog("角色管理-通过id删除角色")
    @DeleteMapping("/delete")
    public Result<?> delete(@RequestParam String id) {
        service.deleteRole(id);
        return Result.ok();
    }

    /**
     * 批量删除角色
     *
     * @param ids 角色id（多个逗号分隔）
     */
    @AutoLog("角色管理-批量删除角色")
    @DeleteMapping("/deleteBatch")
    public Result<?> deleteBatch(@RequestParam String ids) {
        if (StrUtil.isEmpty(ids)) {
            return Result.error("未选中角色！");
        }
        service.deleteBatchRole(ids.split(","));
        return Result.ok();
    }

    /**
     * 查询所有角色
     */
    @GetMapping("/queryAll")
    public Result<List<SysRole>> queryAll() {
        List<SysRole> list = service.list();
        if (list == null || list.size() <= 0) {
            return Result.error("未找到角色信息！");
        }
        return Result.ok(list);
    }

    /**
     * 校验角色编码唯一
     *
     * @param id       角色id
     * @param roleCode 角色编码
     */
    @GetMapping("/checkRoleCode")
    public Result<?> checkUsername(String id, String roleCode) {
        SysRole role = null;
        if (StrUtil.isNotEmpty(id)) {
            role = service.getById(id);
        }
        SysRole newRole = service.getOne(new QueryWrapper<SysRole>().lambda().eq(SysRole::getRoleCode, roleCode));
        if (newRole != null) {
            //如果根据传入的roleCode查询到信息了，那么就需要做校验了。
            if (role == null || !id.equals(newRole.getId())) {
                //role为空=>新增模式=>只要roleCode存在则返回false
                //否则=>编辑模式=>判断两者ID是否一致
                return Result.error("角色编码已存在！");
            }
        }
        return Result.ok();
    }

    /**
     * 导出角色数据为Excel
     *
     * @param sysRole 查询条件
     */
    @GetMapping("/exportXls")
    public void exportXls(SysRole sysRole, HttpServletRequest request, HttpServletResponse response) throws IOException {
        QueryWrapper<SysRole> queryWrapper = QueryGenerator.initQueryWrapper(sysRole, request.getParameterMap());
        List<SysRole> pageList = service.list(queryWrapper);
        HttpServletUtil.addDownloadHeader(response, "角色数据-" + DateUtil.format(new Date(), "yyyyMMdd") + easyExcel.getExtension());
        easyExcel.export(pageList, response.getOutputStream(), SystemContextUtil.dictTranslator());
    }

    /**
     * 通过Excel导入角色数据
     */
    @AutoLog("角色管理-通过Excel导入角色数据")
    @PostMapping("/importExcel")
    public Result<?> importExcel(HttpServletRequest request) throws Exception {
        MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
        Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
        for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
            MultipartFile file = entity.getValue();
            ImportExcel excel = new ImportExcel();
            excel.setInputStream(file.getInputStream());
            List<SysRole> sysRoles = easyExcel.read(SysRole.class, excel, SystemContextUtil.dictTranslator());
            service.saveBatch(sysRoles);
        }
        return Result.ok();
    }

    /**
     * 查询数据规则数据
     *
     * @param permissionId 菜单id
     * @param roleId       角色id
     */
    @GetMapping("/dataRule/{permissionId}/{roleId}")
    public Result<?> loadDataRule(@PathVariable String permissionId, @PathVariable String roleId) {
        List<SysPermissionDataRule> list = sysPermissionDataRuleService.getPermRuleListByPermId(permissionId);
        if (list == null || list.size() == 0) {
            return Result.error("未找到权限配置信息");
        }
        Map<String, Object> returnData = new HashMap<>();
        returnData.put("datarule", list);
        LambdaQueryWrapper<SysRolePermission> query = new LambdaQueryWrapper<SysRolePermission>().eq(SysRolePermission::getPermissionId, permissionId).isNotNull(SysRolePermission::getDataRuleIds).eq(SysRolePermission::getRoleId, roleId);
        SysRolePermission sysRolePermission = sysRolePermissionService.getOne(query);
        if (sysRolePermission != null) {
            String drChecked = sysRolePermission.getDataRuleIds();
            if (StrUtil.isNotEmpty(drChecked)) {
                returnData.put("drChecked", drChecked.endsWith(",") ? drChecked.substring(0, drChecked.length() - 1) : drChecked);
            }
        }
        return Result.ok(returnData);
    }

    /**
     * 保存数据规则至角色菜单关联表
     *
     * @param jsonObject 参数
     */
    @PostMapping("/dataRule")
    public Result<?> saveDataRule(@RequestBody JsonNode jsonObject) {
        String permissionId = jsonObject.get("permissionId").asText();
        String roleId = jsonObject.get("roleId").asText();
        String dataRuleIds = jsonObject.get("dataRuleIds").asText();
        LambdaQueryWrapper<SysRolePermission> query = new LambdaQueryWrapper<SysRolePermission>().eq(SysRolePermission::getPermissionId, permissionId).eq(SysRolePermission::getRoleId, roleId);
        SysRolePermission sysRolePermission = sysRolePermissionService.getOne(query);
        if (sysRolePermission == null) {
            return Result.error("请先保存角色菜单权限！");
        }
        sysRolePermission.setDataRuleIds(dataRuleIds);
        this.sysRolePermissionService.updateById(sysRolePermission);
        return Result.ok();
    }

    /**
     * 用户角色授权功能
     *
     * @apiNote 查询菜单权限树
     */
    @GetMapping("/queryTreeList")
    public Result<?> queryTreeList() {
        //全部权限ids
        List<String> ids = new ArrayList<>();
        LambdaQueryWrapper<SysPermission> query = new LambdaQueryWrapper<>();
        query.eq(SysPermission::getDelFlag, CommonConst.NOT_DELETED);
        query.orderByAsc(SysPermission::getSortNo);
        List<SysPermission> list = sysPermissionService.list(query);
        for (SysPermission sysPer : list) {
            ids.add(sysPer.getId());
        }
        List<TreeModel> treeList = new ArrayList<>();
        getTreeModelList(treeList, list, null);
        Map<String, Object> returnData = new HashMap<>();
        //全部树节点数据
        returnData.put("treeList", treeList);
        //全部树ids
        returnData.put("ids", ids);
        return Result.ok(returnData);
    }

    private void getTreeModelList(List<TreeModel> treeList, List<SysPermission> metaList, TreeModel temp) {
        for (SysPermission permission : metaList) {
            String tempPid = permission.getParentId();
            TreeModel tree = new TreeModel(permission.getId(), tempPid, permission.getName(), permission.getRuleFlag(), permission.isLeaf());
            if (temp == null && StrUtil.isEmpty(tempPid)) {
                treeList.add(tree);
                if (!tree.isLeaf()) {
                    getTreeModelList(treeList, metaList, tree);
                }
            } else if (temp != null && tempPid != null && tempPid.equals(temp.getKey())) {
                temp.getChildren().add(tree);
                if (!tree.isLeaf()) {
                    getTreeModelList(treeList, metaList, tree);
                }
            }
        }
    }
}