package com.gkhy.labRiskManage.domain.riskReport.utils;

import com.gkhy.labRiskManage.commons.enums.UserRoleEnum;
import com.gkhy.labRiskManage.domain.account.model.dto.SysUserRoleBindDomainDTO;
import com.gkhy.labRiskManage.domain.account.model.dto.UserInfoDomainDTO;
import org.springframework.stereotype.Component;

@Component
public class GetRoleTagUtils {

    public static int GetRoleTagUtils (UserInfoDomainDTO userInfo){

        int roleTag = 0;
        for (SysUserRoleBindDomainDTO role : userInfo.getRoles()) {
            if (role.getRoleId() == UserRoleEnum.USER_ROLE_4.getCode().byteValue()){
                roleTag = 1;
            }
        }
        for (SysUserRoleBindDomainDTO role : userInfo.getRoles()) {
            if (role.getRoleId() == UserRoleEnum.USER_ROLE_6.getCode().byteValue()){
                roleTag = 2;
            }
            if (role.getRoleId() == UserRoleEnum.USER_ROLE_7.getCode().byteValue()){
                roleTag = 2;
            }
        }

        return  roleTag;
    }
}