package com.mzl.flower.service.system;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.mzl.flower.config.exception.ValidationException;
import com.mzl.flower.config.security.SecurityUtils;
import com.mzl.flower.constant.Constants;
import com.mzl.flower.dto.request.system.CreateRoleDTO;
import com.mzl.flower.dto.request.system.SearchRoleDTO;
import com.mzl.flower.dto.request.system.UpdateRoleDTO;
import com.mzl.flower.dto.response.system.RoleDTO;
import com.mzl.flower.dto.response.system.RoleListDTO;
import com.mzl.flower.entity.system.Role;
import com.mzl.flower.entity.system.RoleMenu;
import com.mzl.flower.entity.system.UserRole;
import com.mzl.flower.mapper.system.RoleMapper;
import com.mzl.flower.mapper.system.RoleMenuMapper;
import com.mzl.flower.mapper.system.UserRoleMapper;
import com.mzl.flower.service.BaseService;
import com.mzl.flower.utils.UUIDGenerator;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.List;

@Service
@Transactional
public class RoleService extends BaseService {
    @Autowired
    private RoleMapper roleMapper;

    @Autowired
    private RoleMenuMapper roleMenuMapper;

    @Autowired
    private UserRoleMapper userRoleMapper;

    public String addRole(CreateRoleDTO dto) {
        Role role1 = roleMapper.selectOne(new QueryWrapper<Role>().eq("ROLE_NAME", dto.getRoleName()));
        if(role1!=null){
            throw new ValidationException("角色名字重复");
        }

        Role role = new Role();
        role.setId(UUIDGenerator.getUUID());
        role.setRoleName(dto.getRoleName());
        role.setRoleDesc(dto.getRoleDesc());
        //设置非系统默认
        role.setIsSys(Constants.N);
        role.create(SecurityUtils.getUserId());
        roleMapper.insert(role);

        addRoleMenus(role.getId(), dto.getMenuIds());
        return role.getId();
    }

    public Page<RoleListDTO> searchRole(SearchRoleDTO dto, Page page) {
        //查询角色列表
        List<RoleListDTO> result = roleMapper.searchRole(page, dto);
        //给分页添加结果集
        page.setRecords(result);
        return page;
    }

    public RoleDTO getRoleDetail(final String id) {
        //获取角色详情
        Role role = roleMapper.selectById(id);
        //获取角色下的菜单
        QueryWrapper queryWrapper = new QueryWrapper<RoleMenu>().eq("ROLE_ID", id);
        List<RoleMenu> roleMenus = roleMenuMapper.selectList(queryWrapper);

        List<String> menuIds = new ArrayList<>();
        for (RoleMenu roleMenu : roleMenus) {
            menuIds.add(roleMenu.getMenuId());
        }

        //返回结果集
        RoleDTO result = new RoleDTO();
        BeanUtils.copyProperties(role, result);
        result.setMenuIds(menuIds);
        return result;
    }

    private void addRoleMenus(String id, List<String> menuIds){
        //删除角色已存在的菜单
        QueryWrapper deleteQuery = new QueryWrapper<RoleMenu>().eq("ROLE_ID", id);
        roleMenuMapper.delete(deleteQuery);
        //给角色添加可以访问的菜单
        for (String menuId : menuIds) {
            RoleMenu roleMenu = new RoleMenu();
            roleMenu.setId(UUIDGenerator.getUUID());
            roleMenu.setRoleId(id);
            roleMenu.setMenuId(menuId);
            roleMenuMapper.insert(roleMenu);
        }
    }

    public void updateRole(UpdateRoleDTO dto) {
        //获取角色详情
        Role role = roleMapper.selectById(dto.getId());

        Role role1 = roleMapper.selectOne(new QueryWrapper<Role>().eq("ROLE_NAME", dto.getRoleName()));
        if(role1!=null&&!role1.getId().equals(role.getId())){
            throw new ValidationException("角色名字重复");
        }

        role.setRoleName(dto.getRoleName());
        role.setRoleDesc(dto.getRoleDesc());
        role.update(SecurityUtils.getUserId());
        roleMapper.updateById(role);

        addRoleMenus(role.getId(), dto.getMenuIds());
    }

    /**
     * 删除角色
     *
     * @param id 角色主键
     */
    public void deleteRole(String id) {
        //删除角色
        roleMapper.deleteById(id);

        //删除角色关联的菜单
        QueryWrapper deleteQuery = new QueryWrapper<RoleMenu>().eq("ROLE_ID", id);
        roleMenuMapper.delete(deleteQuery);

        //删除角色关联的用户
        deleteQuery = new QueryWrapper<UserRole>().eq("ROLE_ID", id);
        userRoleMapper.delete(deleteQuery);
    }

    public List<Role> getUserRoleList(String userId) {
        return roleMapper.getUserRoleList(userId);
    }
}