package com.hxzk.gps.service.impl.Role;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.github.pagehelper.Page;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.hxzk.gps.controller.Role.Results.AddRoleDto;
import com.hxzk.gps.controller.Role.Results.RoleCompanyList;
import com.hxzk.gps.controller.Role.Results.RoleResult;
import com.hxzk.gps.controller.Role.Results.RoleTreeTableDto;
import com.hxzk.gps.entity.Role.Rolemenu;
import com.hxzk.gps.entity.System.TbSystemOperationLog;
import com.hxzk.gps.mapper.Role.RoleMenuMapper;
import com.hxzk.gps.service.Role.RoleMenuService;
import com.hxzk.gps.service.System.TbSystemOperationLogService;
import com.hxzk.gps.util.GetUser.UserInfoUtil;
import com.hxzk.gps.util.Result.ReturnMessage;
import com.hxzk.gps.util.Result.TableDto;
import com.hxzk.gps.entity.Role.Role;
import com.hxzk.gps.mapper.Role.RoleMapper;
import com.hxzk.gps.service.Company.TbCompanyService;
import com.hxzk.gps.service.Role.IRoleService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.hxzk.gps.util.MessageUtils.MessageUtils;
import com.hxzk.gps.controller.User.Results.UserTreeTableDto;
import com.hxzk.gps.controller.System.SystemLogUtil;
import com.hxzk.gps.util.Time.TimeUtil;
import com.sun.org.apache.bcel.internal.generic.PUSH;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

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

/**
 * <p>
 *  服务实现类
 * </p>
 *
 * @author YuZhiTong
 * @since 2025-05-08
 */
@Service
public class RoleServiceImpl extends ServiceImpl<RoleMapper, Role> implements IRoleService {
    /*
    * Service注入
    * */
    @Autowired
    private TbCompanyService companyService;
    @Autowired
    private RoleMenuService roleMenuService;
    @Autowired
    public TbSystemOperationLogService systemOperationLogService;


    /*
    * 角色列表信息
    * */
    @Override
    public RoleResult FindRoleInfo(RoleTreeTableDto roleTreeTableDto) {
        // 创建查询包装器
        QueryWrapper<Role> queryWrapper = new QueryWrapper<>();

        // 处理部门 ID 为空的情况
        if (StringUtils.isEmpty(roleTreeTableDto.getDepartmentId())) {
            List<Long> list = companyService.getChildCompanyIds(UserInfoUtil.getUserCompanyId());
            queryWrapper.in("companyid", list);
        } else {
            queryWrapper.eq("companyid", roleTreeTableDto.getDepartmentId());
        }

        // 处理角色名称查询条件
        Optional.ofNullable(roleTreeTableDto.getName()).ifPresent(name ->
                queryWrapper.like("name", name));

        // 使用PageHelper进行分页查询
        Page<Role> page = PageHelper.startPage(roleTreeTableDto.getPageNum(), roleTreeTableDto.getPageSize())
                .doSelectPage(() -> baseMapper.selectList(queryWrapper));

        // 封装角色列表
        TableDto tableDto = new TableDto(page.getResult(), page.getPageNum(), page.getPageSize(), page.getTotal());

        // 获取成功消息
        String message = MessageUtils.getMessage("Success");

        // 返回结果 - 根据RoleResult的构造函数调整参数
        return new RoleResult(200, message, tableDto);
    }


    /*
    * 获取当前登录用户公司所属角色信息列表
    * @return 登录用户公司角色信息列表   lable -- value
    * */
    @Override
    public List<RoleCompanyList> FindCompanyRole(String companyid) {
        // 使用泛型指定 QueryWrapper 的类型
        if (companyid.isEmpty()){
           companyid = UserInfoUtil.getUserCompanyId();
        }
        QueryWrapper<Role> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("companyid", companyid);
        // 查询角色列表
        List<Role> roleList = baseMapper.selectList(queryWrapper);
        // 使用 Stream API 简化循环操作
        return roleList.stream()
                .map(role -> {
                    RoleCompanyList roleLists = new RoleCompanyList();
                    roleLists.setLabel(role.getName());
                    roleLists.setValue(String.valueOf(role.getId()));
                    return roleLists;
                })
                .collect(Collectors.toList());
    }

    /*
    * 获取角色名称
    * @return 角色名称
    * @param id 角色ID
    * */
    @Override
    public String FindRoleName(String id) {
        QueryWrapper<Role> queryWrapper = new QueryWrapper();
        queryWrapper.eq("id", id).or().eq("name", id);
        return baseMapper.selectOne(queryWrapper).getName();

    }

    @Override
    public ReturnMessage add(AddRoleDto addRoleDto) {
         /*
        声明操作状态
        * */
        boolean isSuccess = false;
        QueryWrapper queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("name", addRoleDto.getName());
        queryWrapper.eq("companyid", addRoleDto.getCompanyid());
        //查重
        Integer count = Math.toIntExact(baseMapper.selectCount(queryWrapper));
        if (count > 0) {
            //存在
            return ReturnMessage.ReturnMessageError("RoleRepeat");
        } else {
            Role role = new Role();
            role.setName(addRoleDto.getName());
            if (addRoleDto.getCompanyid().equals("")){
                addRoleDto.setCompanyid(UserInfoUtil.getUserCompanyId());
            }
            role.setCompanyname(companyService.getChildCompanyName(addRoleDto.getCompanyid()));
            role.setUsername(UserInfoUtil.getUserName());
            role.setCompanyid(addRoleDto.getCompanyid());
            role.setAddtime(TimeUtil.GetTime());
            if (addRoleDto.isAddrole()){
                role.setAddrole(true);
            }
            if (addRoleDto.isEditrole()){
                role.setEditrole(true);
            }
            if (addRoleDto.isDeleterole()){
                role.setDeleterole(true);
            }
            baseMapper.insert(role);
            isSuccess = true;
            //添加角色成功
            QueryWrapper queryWrapperselect = new QueryWrapper<>();
            queryWrapperselect.eq("name", role.getName());
            queryWrapperselect.eq("companyname", role.getCompanyname());
            //查询角色ID
            Role role1 = baseMapper.selectOne(queryWrapperselect);
            Integer roleId = role1.getId();
            Map<Integer,Integer> map = new HashMap<>();
            for (int k = 0; k < addRoleDto.getRolist().size();k++){
                map.put(addRoleDto.getRolist().get(k).getId(),addRoleDto.getRolist().get(k).getId());
            }
            for (int i = 0; i < addRoleDto.getRolist().size();i++){
                if (addRoleDto.getRolist().get(i).getParentId() != null){
                    if (!map.containsKey(addRoleDto.getRolist().get(i).getParentId())) {
                        //不包含
                        Rolemenu rolemenu = new Rolemenu();
                        rolemenu.setRoleId(roleId);
                        rolemenu.setMenuId(addRoleDto.getRolist().get(i).getParentId());
                        rolemenu.setIsall(0);
                        roleMenuService.save(rolemenu);
                        map.put(addRoleDto.getRolist().get(i).getParentId(),addRoleDto.getRolist().get(i).getParentId());
                    }
                }
                Rolemenu rolemenu = new Rolemenu();
                rolemenu.setRoleId(roleId);
                rolemenu.setMenuId(addRoleDto.getRolist().get(i).getId());
                roleMenuService.save(rolemenu);
            }
            return SystemLogUtil.handleAddResult(isSuccess,"RoleAdd",role.getName());

        }
    }

    @Override
    public ReturnMessage update(AddRoleDto role) {
         /*
        声明操作状态
        * */
        boolean isSuccess = false;
        Role role1 = new Role();
        role1.setId(role.getId());
        role1.setName(role.getName());
        role1.setAddtime(TimeUtil.GetTime());
        role1.setAddrole(role.isAddrole());
        role1.setEditrole(role.isEditrole());
        role1.setDeleterole(role.isDeleterole());
        role1.setCompanyname(role.getCompanyname());
        role1.setUsername(UserInfoUtil.getUserName());
        role1.setCompanyid(role.getCompanyid());
        QueryWrapper queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("id", role.getId());
        try {
            baseMapper.update(role1, queryWrapper);
            roleMenuService.deleteRoleMenu(role.getId());
            Integer roleId = role1.getId();
            Map<Integer,Integer> map = new HashMap<>();
            for (int k = 0; k < role.getRolist().size();k++){
                map.put(role.getRolist().get(k).getId(),role.getRolist().get(k).getId());
            }
            for (int i = 0; i < role.getRolist().size();i++){
                if (role.getRolist().get(i).getParentId() != null){
                    if (!map.containsKey(role.getRolist().get(i).getParentId())) {
                        //不包含
                        Rolemenu rolemenu = new Rolemenu();
                        rolemenu.setRoleId(roleId);
                        rolemenu.setMenuId(role.getRolist().get(i).getParentId());
                        rolemenu.setIsall(0);
                        roleMenuService.save(rolemenu);
                        map.put(role.getRolist().get(i).getParentId(),role.getRolist().get(i).getParentId());
                    }
                }
                Rolemenu rolemenu = new Rolemenu();
                rolemenu.setRoleId(roleId);
                rolemenu.setMenuId(role.getRolist().get(i).getId());
                roleMenuService.save(rolemenu);
            }
            isSuccess = true;
        }catch (Exception e){
        }
        return SystemLogUtil.handleUpdateResult(isSuccess,"RoleUpdate",role.getName());

    }

    @Override
    public ReturnMessage delete(AddRoleDto role) {
         /*
        声明操作状态
        * */
        boolean isSuccess = false;
        QueryWrapper queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("id", role.getId());
        roleMenuService.deleteRoleMenu(role.getId());
        try {
            baseMapper.delete(queryWrapper);
            isSuccess = true;
        }catch (Exception e){

        }
        return SystemLogUtil.handleDeleteResult(isSuccess,"RoleDelete",role.getName());

    }

}