package cn.besteam.common.util;

import cn.besteam.common.constant.ErrorCode;
import cn.besteam.common.exception.BusinessException;
import cn.besteam.modules.system.entity.SysDepartment;
import cn.besteam.modules.system.entity.SysUser;
import cn.besteam.modules.system.entity.SysUserDepartment;
import cn.besteam.modules.system.service.ISysDepartmentService;
import cn.besteam.modules.system.service.ISysUserDepartmentService;
import cn.besteam.modules.shiro.authc.util.EnvUtils;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;

/**
 * 用户上下文工具类
 * 用于获取当前用户信息和权限检查
 */
@Component
@Slf4j
public class UserContextUtil {

    @Autowired
    private ISysDepartmentService sysDepartmentService;

    @Autowired
    private ISysUserDepartmentService sysUserDepartmentService;

    /**
     * 机构类型：区域
     */
    private static final Integer ORG_TYPE_REGION = 1;

    /**
     * 机构类型：单位（医院）
     */
    private static final Integer ORG_TYPE_HOSPITAL = 2;

    /**
     * 机构类型：部门
     */
    private static final Integer ORG_TYPE_DEPARTMENT = 3;

    /**
     * 机构类型：科室
     */
    private static final Integer ORG_TYPE_OFFICE = 4;

    /**
     * 获取当前用户信息
     * @return 当前用户信息
     */
    public SysUser getCurrentUser() {
        SysUser currentUser = EnvUtils.getUser();
        if (currentUser == null) {
            throw new BusinessException(ErrorCode.ERR_500, "用户未登录");
        }
        return currentUser;
    }

    /**
     * 获取当前用户的部门信息
     * @return 当前用户部门信息
     */
    public SysDepartment getCurrentUserDepartment() {
        SysUser currentUser = getCurrentUser();
        
        // 查询用户部门关联
        SysUserDepartment userDepartment = sysUserDepartmentService.getOne(
            new com.baomidou.mybatisplus.core.conditions.query.QueryWrapper<SysUserDepartment>()
                .eq("user_id", currentUser.getId())
        );
        
        if (userDepartment == null || userDepartment.getDepartmentId() == null) {
            throw new BusinessException(ErrorCode.ERR_500, "用户部门信息不存在");
        }
        
        return sysDepartmentService.getById(userDepartment.getDepartmentId());
    }

    /**
     * 检查用户是否有操作权限
     * 区域用户不能进行操作
     */
    public void checkUserPermission() {
        SysDepartment department = getCurrentUserDepartment();
        if (department.getOrgType().equals(ORG_TYPE_REGION)) {
            throw new BusinessException(ErrorCode.ERR_500, "区域用户不能进行此操作");
        }
    }

    /**
     * 获取当前用户所属医院的ID
     * @return 医院ID
     */
    public Integer getCurrentUserHospitalId() {
        SysDepartment department = getCurrentUserDepartment();
        return calculateHospitalId(department);
    }

    /**
     * 根据部门信息计算医院ID
     * @param department 部门信息
     * @return 医院ID
     */
    private Integer calculateHospitalId(SysDepartment department) {
        if (department == null) {
            throw new BusinessException(ErrorCode.ERR_500, "部门信息不存在");
        }

        Integer orgType = department.getOrgType();
        
        if (orgType.equals(ORG_TYPE_HOSPITAL)) {
            // 当前部门就是医院
            return department.getId();
        } else if (orgType.equals(ORG_TYPE_DEPARTMENT)) {
            // 部门用户，医院是父级部门
            return department.getParentId();
        } else if (orgType.equals(ORG_TYPE_OFFICE)) {
            // 科室用户，医院是父级部门的父级部门
            SysDepartment parentDepartment = sysDepartmentService.getById(department.getParentId());
            if (parentDepartment == null) {
                throw new BusinessException(ErrorCode.ERR_500, "父级部门信息不存在");
            }
            return parentDepartment.getParentId();
        } else {
            throw new BusinessException(ErrorCode.ERR_500, "不支持的用户类型");
        }
    }

    /**
     * 验证医院ID是否有效
     * @param hospitalId 医院ID
     */
    public void validateHospitalId(Integer hospitalId) {
        Integer currentUserHospitalId = getCurrentUserHospitalId();
        if (!hospitalId.equals(currentUserHospitalId)) {
            throw new BusinessException(ErrorCode.ERR_500, "无权操作此医院的数据");
        }
    }
}