wms / wms_changshushengyi · Files

package com.huaheng.common.utils.security;

import java.util.List;

import javax.annotation.Resource;

import org.apache.shiro.SecurityUtils;

import org.apache.shiro.authc.UsernamePasswordToken;

import org.apache.shiro.mgt.RealmSecurityManager;
import org.apache.shiro.session.Session;
import org.apache.shiro.subject.PrincipalCollection;
import org.apache.shiro.subject.SimplePrincipalCollection;
import org.apache.shiro.subject.Subject;

import com.huaheng.common.utils.StringUtils;
import com.huaheng.common.utils.ZoneAreaUtils;
import com.huaheng.common.utils.bean.BeanUtils;

import com.huaheng.framework.shiro.realm.UserRealm;

import com.huaheng.pc.config.warehouse.service.WarehouseService;
import com.huaheng.pc.system.dept.domain.Dept;

import com.huaheng.pc.system.user.domain.User;

/**
 * shiro 工具类
 * @author huaheng
 */

public class ShiroUtils {

    @Resource
    private WarehouseService warehouseService;

    public static Subject getSubjct() {

        return SecurityUtils.getSubject();
    }

    public static Session getSession() {

        return SecurityUtils.getSubject().getSession();
    }

    public static void logout() {

        getSubjct().logout();
    }

    public static User getUser() {

        try {
            Object obj = getSubjct().getPrincipal();

            if (StringUtils.isNotNull(obj)) {

                User user = new User();
                BeanUtils.copyBeanProp(user, obj);
                return user;

            } else {

                return null;
            }

        } catch (Exception e) {

            return null;
        }
    }

    public static void setUser(User user) {

        Subject subject = getSubjct();
        PrincipalCollection principalCollection = subject.getPrincipals();
        String realmName = principalCollection.getRealmNames().iterator().next();
        PrincipalCollection newPrincipalCollection = new SimplePrincipalCollection(user, realmName);
        // 重新加载Principal
        subject.runAs(newPrincipalCollection);
    }

    public static void clearCachedAuthorizationInfo() {
        RealmSecurityManager rsm = (RealmSecurityManager)SecurityUtils.getSecurityManager();
        UserRealm realm = (UserRealm)rsm.getRealms().iterator().next();

        realm.clearCachedAuthorizationInfo();
    }

    public static Integer getUserId() {

        return getUser().getId();
    }

    public static String getLoginName() {
        if (getUser() == null) {

            return null;
        }

        return getUser().getLoginName();
    }

    public static String getWarehouseCode() {
        return getUser().getWarehouseCode();

    }

    public static void setUserByWcs(String loginname, String warehouseCode, String area) {
        User user = new User();

        user.setWarehouseCode(warehouseCode);
        user.setLoginName(loginname);

        user.setArea(area);

        ShiroUtils.setUser(user);
    }

    public static void setUserByWcsZone(String loginname, String warehouseCode, String zoneCode) {
        User user = new User();

        user.setWarehouseCode(warehouseCode);
        user.setLoginName(loginname);
        user.setZoneCode(zoneCode);

        user.setId(0);
        Dept dept = new Dept();
        dept.setDeptName("11");
        user.setDept(dept);
//        user.setr
        ShiroUtils.setUser1(user);
    }

    public static void setUser1(User user) {
        Subject subject = SecurityUtils.getSubject();
        UsernamePasswordToken token = new UsernamePasswordToken("test", "123456", false, "1111");
        subject.login(token);
        PrincipalCollection principalCollection = subject.getPrincipals();
        String realmName = principalCollection.getRealmNames().iterator().next();
        PrincipalCollection newPrincipalCollection = new SimplePrincipalCollection(user, realmName);
        // 重新加载Principal
        subject.runAs(newPrincipalCollection);

    }

    public static String getZoneCode() {
        return getUser().getZoneCode();
    }

    public static String getZoneName() {
        return getUser().getZoneName();
    }

    public static void setZoneCode(String zoneCode) {
        getUser().setZoneCode(zoneCode);
    }

    public static String getArea() {
        String zoneCode = getUser().getZoneCode();
        if (StringUtils.isNotEmpty(zoneCode)) {

            return ZoneAreaUtils.getAreaByZoneCode(zoneCode).toString();
        }
        return getUser().getArea();
    }

    public static List<Integer> getCompanyIdList() {

        return getUser().getCompanyIdList();
    }

    public static List<String> getCompanyCodeList() {

        return getUser().getCompanyCodeList();
    }

    public static String getIp() {

        return getSubjct().getSession().getHost();
    }

    public static String getSessionId() {

        return String.valueOf(getSubjct().getSession().getId());
    }

    public static String getCompanyCode() {

        return getUser().getCompanyCodeList().get(0);
    }

}