89 lines
3.9 KiB
Java
89 lines
3.9 KiB
Java
|
package com.zhgd.xmgl.security;
|
|||
|
|
|
|||
|
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
|||
|
|
import com.zhgd.xmgl.modules.basicdata.entity.Company;
|
|||
|
|
import com.zhgd.xmgl.modules.basicdata.enums.SystemUserAccountTypeEnum;
|
|||
|
|
import com.zhgd.xmgl.modules.basicdata.service.ICompanyService;
|
|||
|
|
import com.zhgd.xmgl.modules.project.service.IProjectService;
|
|||
|
|
import com.zhgd.xmgl.security.entity.UserInfo;
|
|||
|
|
import com.zhgd.xmgl.security.util.SecurityUtils;
|
|||
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|||
|
|
import org.springframework.context.annotation.Lazy;
|
|||
|
|
import org.springframework.stereotype.Component;
|
|||
|
|
|
|||
|
|
@Component("perm")
|
|||
|
|
public class PermissionEvaluator {
|
|||
|
|
@Lazy
|
|||
|
|
@Autowired
|
|||
|
|
private ICompanyService companyService;
|
|||
|
|
@Lazy
|
|||
|
|
@Autowired
|
|||
|
|
private IProjectService projectService;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 有企业sn的权限
|
|||
|
|
*
|
|||
|
|
* @param sn 企业sn
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
public boolean hasCompanySnAccess(String sn) {
|
|||
|
|
UserInfo user = SecurityUtils.getUser();
|
|||
|
|
if (user.getAccountType().equals(SystemUserAccountTypeEnum.ENTERPRISE_ADMINISTRATOR_ACCOUNT.getValue())) {
|
|||
|
|
return companyService.hasCompanySnAccessBy1(user.getUserId(), sn);
|
|||
|
|
} else if (user.getAccountType().equals(SystemUserAccountTypeEnum.ENTERPRISE_DISTRICT_ACCOUNT.getValue())) {
|
|||
|
|
return companyService.hasCompanySnAccessBy2(user.getUserId(), sn);
|
|||
|
|
} else if (user.getAccountType().equals(SystemUserAccountTypeEnum.ENTERPRISE_CITY_ACCOUNT.getValue())) {
|
|||
|
|
return companyService.hasCompanySnAccessBy3(user.getUserId(), sn);
|
|||
|
|
} else if (user.getAccountType().equals(SystemUserAccountTypeEnum.ENTERPRISE_SUB_ACCOUNT.getValue())) {
|
|||
|
|
return companyService.hasCompanySnAccessBy4(user.getUserId(), sn);
|
|||
|
|
}
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 有企业id的权限
|
|||
|
|
*
|
|||
|
|
* @param companyId 企业id
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
public boolean hasCompanyIdAccess(String companyId) {
|
|||
|
|
Company company = companyService.getById(companyId);
|
|||
|
|
if (company == null) {
|
|||
|
|
return false;
|
|||
|
|
}
|
|||
|
|
return hasCompanySnAccess(company.getCompanySn());
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 有sn（企业或项目）的权限
|
|||
|
|
*
|
|||
|
|
* @param sn 企业或项目sn
|
|||
|
|
* @return
|
|||
|
|
*/
|
|||
|
|
public boolean hasSnAccess(String sn) {
|
|||
|
|
UserInfo user = SecurityUtils.getUser();
|
|||
|
|
int c = companyService.count(new LambdaQueryWrapper<Company>()
|
|||
|
|
.eq(Company::getCompanySn, sn));
|
|||
|
|
if (c > 0) {
|
|||
|
|
return hasCompanySnAccess(sn);
|
|||
|
|
} else {
|
|||
|
|
if (user.getAccountType().equals(SystemUserAccountTypeEnum.ENTERPRISE_ADMINISTRATOR_ACCOUNT.getValue())) {
|
|||
|
|
return companyService.hasProjectSnAccessBy1(user.getUserId(), sn);
|
|||
|
|
} else if (user.getAccountType().equals(SystemUserAccountTypeEnum.ENTERPRISE_DISTRICT_ACCOUNT.getValue())) {
|
|||
|
|
return companyService.hasProjectSnAccessBy2(user.getUserId(), sn);
|
|||
|
|
} else if (user.getAccountType().equals(SystemUserAccountTypeEnum.ENTERPRISE_CITY_ACCOUNT.getValue())) {
|
|||
|
|
return companyService.hasProjectSnAccessBy3(user.getUserId(), sn);
|
|||
|
|
} else if (user.getAccountType().equals(SystemUserAccountTypeEnum.ENTERPRISE_SUB_ACCOUNT.getValue())) {
|
|||
|
|
return companyService.hasProjectSnAccessBy4(user.getUserId(), sn);
|
|||
|
|
} else if (user.getAccountType().equals(SystemUserAccountTypeEnum.PROJECT_ACCOUNT.getValue())) {
|
|||
|
|
return projectService.hasProjectSnAccess(user.getUserId(), sn);
|
|||
|
|
} else if (user.getAccountType().equals(SystemUserAccountTypeEnum.PROJECT_SUB_ACCOUNT.getValue())) {
|
|||
|
|
return projectService.hasProjectSnAccess(user.getUserId(), sn);
|
|||
|
|
} else if (user.getAccountType().equals(SystemUserAccountTypeEnum.NEW_USER.getValue())) {
|
|||
|
|
return projectService.hasProjectSnAccessByNewUser(user.getUserId(), sn);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return true;
|
|||
|
|
}
|
|||
|
|
}
|