package cn.iocoder.yudao.module.system.controller.admin.user;

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjUtil;
import cn.iocoder.yudao.framework.common.enums.UserTypeEnum;
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
import cn.iocoder.yudao.framework.datapermission.core.annotation.DataPermission;
import cn.iocoder.yudao.module.system.controller.admin.user.vo.profile.UserProfileRespVO;
import cn.iocoder.yudao.module.system.controller.admin.user.vo.profile.UserProfileUpdatePasswordReqVO;
import cn.iocoder.yudao.module.system.controller.admin.user.vo.profile.UserProfileUpdateReqVO;
import cn.iocoder.yudao.module.system.convert.user.UserConvert;
import cn.iocoder.yudao.module.system.dal.dataobject.dept.DeptDO;
import cn.iocoder.yudao.module.system.dal.dataobject.dept.PostDO;
import cn.iocoder.yudao.module.system.dal.dataobject.permission.RoleDO;
import cn.iocoder.yudao.module.system.dal.dataobject.social.SocialUserDO;
import cn.iocoder.yudao.module.system.dal.dataobject.user.AdminUserDO;
import cn.iocoder.yudao.module.system.service.dept.DeptService;
import cn.iocoder.yudao.module.system.service.dept.PostService;
import cn.iocoder.yudao.module.system.service.permission.PermissionService;
import cn.iocoder.yudao.module.system.service.permission.RoleService;
import cn.iocoder.yudao.module.system.service.social.SocialUserService;
import cn.iocoder.yudao.module.system.service.user.AdminUserService;
import io.swagger.v3.oas.annotations.tags.Tag;
import io.swagger.v3.oas.annotations.Operation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import javax.validation.Valid;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
import static cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils.getLoginUserId;
import static cn.iocoder.yudao.module.infra.enums.ErrorCodeConstants.FILE_IS_EMPTY;

@Tag(name = "管理后台 - 用户个人中心")
@RestController
@RequestMapping("/system/user/profile")
@Validated
@Slf4j
public class UserProfileController {

    @Resource
    private AdminUserService userService;
    @Resource
    private DeptService deptService;
    @Resource
    private PostService postService;
    @Resource
    private PermissionService permissionService;
    @Resource
    private RoleService roleService;
    @Resource
    private SocialUserService socialService;

    @GetMapping("/get")
    @Operation(summary = "获得登录用户信息")
    @DataPermission(enable = false) // 关闭数据权限，避免只查看自己时，查询不到部门。
    public CommonResult<UserProfileRespVO> getUserProfile() {
        // 获得用户基本信息
        AdminUserDO user = userService.getUser(getLoginUserId());
        // 获得用户角色
        List<RoleDO> userRoles = roleService.getRoleListFromCache(permissionService.getUserRoleIdListByUserId(user.getId()));
        Set<String> roleCodes = new HashSet<>();
        if (CollUtil.isNotEmpty(userRoles)) {
            userRoles.forEach(userRole -> {
                roleCodes.add(userRole.getCode());
            });
        }
        // 获得部门信息
        DeptDO dept = user.getDeptId() != null ? deptService.getDept(user.getDeptId()) : null;
        // 瑞都日报填报 菜单标识 移动端使用
        boolean rdReportFlag = false;
        // 瑞都日报审批 菜单标识 移动端使用
        boolean rdReportApprovalFlag = false;
        // 瑞恒日报 菜单标识 移动端使用 157
        boolean rhReportFlag = false;
        // 瑞恒日报 菜单标识 移动端使用
        boolean rhReportApprovalFlag = false;
        // 瑞鹰钻井日报 菜单标识 移动端使用 158
        boolean ryReportFlag = false;
        // 瑞鹰钻井日报 菜单标识 移动端使用
        boolean ryReportApprovalFlag = false;
        // 瑞鹰修井日报 菜单标识 移动端使用
        boolean ryXjReportFlag = false;
        // 瑞鹰修井日报 菜单标识 移动端使用
        boolean ryXjReportApprovalFlag = false;
        // 瑞都日报 修改非生产时效标识 移动端使用
        boolean rdNonProductFlag = false;
        // 瑞都日报 统计报表 移动端使用
        boolean rdReportStatisticsFlag = false;
        // 获取当前登录人的公司级部门 如果 是 四川瑞都163 则设置 标识 rdReportFlag = true
        // 查询瑞都下所有部门
        Set<Long> childDeptIds = deptService.getChildDeptIdListFromCache(163l);
        childDeptIds.add(163l);
        childDeptIds.add(156l);

        // 添加瑞恒日报 填报 审批 标识
        Set<Long> rhChildDeptIds = deptService.getChildDeptIdListFromCache(157l);
        rhChildDeptIds.add(157l);

        // 添加瑞恒日报 填报 审批 标识
        Set<Long> ryChildDeptIds = deptService.getChildDeptIdListFromCache(158l);
        ryChildDeptIds.add(158l);

        if (ObjUtil.isNotEmpty(dept)) {
            if (childDeptIds.contains(dept.getId()) && roleCodes.contains("日报填报RD")) {
                rdReportFlag = true;
            }
            if (childDeptIds.contains(dept.getId()) && roleCodes.contains("项目部日报审批RD")) {
                rdReportApprovalFlag = true;
            }
            // 瑞都日报 非生产时效 填报
            if (childDeptIds.contains(dept.getId()) && roleCodes.contains("非生产时效RD")) {
                rdNonProductFlag = true;
            }
            // 瑞都日报 统计报表
            if (childDeptIds.contains(dept.getId()) && roleCodes.contains("项目部日报管理RD")) {
                rdReportStatisticsFlag = true;
            }
            // 瑞恒日报标识
            if (rhChildDeptIds.contains(dept.getId()) && roleCodes.contains("日报填报RH")) {
                rhReportFlag = true;
            }
            if (rhChildDeptIds.contains(dept.getId()) && roleCodes.contains("项目部日报审批RH")) {
                rhReportApprovalFlag = true;
            }
            // 瑞鹰钻井日报标识
            if (ryChildDeptIds.contains(dept.getId()) && roleCodes.contains("日报填报RYZJ")) {
                ryReportFlag = true;
            }
            if (ryChildDeptIds.contains(dept.getId()) && roleCodes.contains("项目部日报审批RY")) {
                ryReportApprovalFlag = true;
            }
            // 瑞鹰修井日报标识
            if (ryChildDeptIds.contains(dept.getId()) && roleCodes.contains("日报填报RYXJ")) {
                ryXjReportFlag = true;
            }
            if (ryChildDeptIds.contains(dept.getId()) && roleCodes.contains("项目部日报审批RYXJ")) {
                ryXjReportApprovalFlag = true;
            }
        }

        // 获得岗位信息
        List<PostDO> posts = CollUtil.isNotEmpty(user.getPostIds()) ? postService.getPostList(user.getPostIds()) : null;
        // 获得社交用户信息
        List<SocialUserDO> socialUsers = socialService.getSocialUserList(user.getId(), UserTypeEnum.ADMIN.getValue());
        return success(UserConvert.INSTANCE.convert(user, userRoles, dept, posts, socialUsers,
                rdReportFlag, rdNonProductFlag, rhReportFlag, rhReportApprovalFlag, ryReportFlag, rdReportApprovalFlag, rdReportStatisticsFlag,
                ryReportApprovalFlag, ryXjReportFlag, ryXjReportApprovalFlag));
    }

    @PutMapping("/update")
    @Operation(summary = "修改用户个人信息")
    public CommonResult<Boolean> updateUserProfile(@Valid @RequestBody UserProfileUpdateReqVO reqVO) {
        userService.updateUserProfile(getLoginUserId(), reqVO);
        return success(true);
    }

    @PutMapping("/update-password")
    @Operation(summary = "修改用户个人密码")
    public CommonResult<Boolean> updateUserProfilePassword(@Valid @RequestBody UserProfileUpdatePasswordReqVO reqVO) {
        userService.updateUserPassword(getLoginUserId(), reqVO);
        return success(true);
    }

    @RequestMapping(value = "/update-avatar",
            method = {RequestMethod.POST, RequestMethod.PUT}) // 解决 uni-app 不支持 Put 上传文件的问题
    @Operation(summary = "上传用户个人头像")
    public CommonResult<String> updateUserAvatar(@RequestParam("avatarFile") MultipartFile file) throws Exception {
        if (file.isEmpty()) {
            throw exception(FILE_IS_EMPTY);
        }
        String avatar = userService.updateUserAvatar(getLoginUserId(), file.getInputStream());
        return success(avatar);
    }

}
