wms / wms_shandonghaiwang · Files

package com.huaheng.framework.config;

import java.util.LinkedHashMap;
import java.util.Map;
import javax.servlet.Filter;

import org.apache.shiro.cache.ehcache.EhCacheManager;
import org.apache.shiro.codec.Base64;
import org.apache.shiro.mgt.SecurityManager;
import org.apache.shiro.spring.security.interceptor.AuthorizationAttributeSourceAdvisor;
import org.apache.shiro.spring.web.ShiroFilterFactoryBean;
import org.apache.shiro.web.mgt.CookieRememberMeManager;
import org.apache.shiro.web.mgt.DefaultWebSecurityManager;
import org.apache.shiro.web.servlet.SimpleCookie;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import com.huaheng.common.utils.StringUtils;
import com.huaheng.framework.shiro.realm.UserRealm;
import com.huaheng.framework.shiro.session.OnlineSessionDAO;
import com.huaheng.framework.shiro.session.OnlineSessionFactory;
import com.huaheng.framework.shiro.web.filter.LogoutFilter;
import com.huaheng.framework.shiro.web.filter.online.OnlineSessionFilter;
import com.huaheng.framework.shiro.web.filter.sync.SyncOnlineSessionFilter;
import com.huaheng.framework.shiro.web.session.OnlineWebSessionManager;
import com.huaheng.framework.shiro.web.session.SpringSessionValidationScheduler;
import at.pollux.thymeleaf.shiro.dialect.ShiroDialect;

/**
 * 权限配置加载

 *

 * @author huaheng
 */
@Configuration

public class ShiroConfig {

    public static final String PREMISSION_STRING = "perms[\"{0}\"]";

    @Value("${shiro.session.redisEnabled}")
    private boolean redisEnabled;


    // Session超时时间，单位为毫秒

    @Value("${shiro.session.expireTime}")
    private int expireTime;

    // 相隔多久检查一次session的有效性，单位毫秒

    @Value("${shiro.session.validationInterval}")
    private int validationInterval;

    // 验证码开关
    @Value("${shiro.user.captchaEnabled}")
    private boolean captchaEnabled;

    // 验证码类型
    @Value("${shiro.user.captchaType}")
    private String captchaType;

    // 设置Cookie的域名
    @Value("${shiro.cookie.domain}")
    private String domain;

    // 设置cookie的有效访问路径
    @Value("${shiro.cookie.path}")
    private String path;

    // 设置HttpOnly属性
    @Value("${shiro.cookie.httpOnly}")
    private boolean httpOnly;

    // 设置Cookie的过期时间，秒为单位
    @Value("${shiro.cookie.maxAge}")
    private int maxAge;

    // 登录地址
    @Value("${shiro.user.loginUrl}")
    private String loginUrl;

    // 登录地址
    @Value("/admin/login")
    private String loginUrls;

    // 权限认证失败地址
    @Value("${shiro.user.unauthorizedUrl}")
    private String unauthorizedUrl;

    @Autowired
    private CacheManagerConfig cacheManagerConfig;

    /**
     * 缓存管理器 使用Ehcache实现
     */

    //@Bean
    //public EhCacheManager getEhCacheManager() {
    //    net.sf.ehcache.CacheManager cacheManager = net.sf.ehcache.CacheManager.getCacheManager("huaheng");
    //    EhCacheManager em = new EhCacheManager();
    //    if (StringUtils.isNull(cacheManager)) {
    //        em.setCacheManagerConfigFile("classpath:ehcache/ehcache-shiro.xml");
    //        return em;
    //    } else {
    //        em.setCacheManager(cacheManager);
    //        return em;
    //    }
    //}

    /**
     * 自定义Realm
     */
    @Bean

    public UserRealm userRealm() {

        UserRealm userRealm = new UserRealm();

        userRealm.setCacheManager(
                redisEnabled ? cacheManagerConfig.getRedisCacheManager() : cacheManagerConfig.getEhCacheManager());

        return userRealm;
    }

    /**
     * 自定义sessionDAO会话
     */
    @Bean

    public OnlineSessionDAO sessionDAO() {

        OnlineSessionDAO sessionDAO = new OnlineSessionDAO();
        return sessionDAO;
    }

    /**
     * 自定义sessionFactory会话
     */
    @Bean

    public OnlineSessionFactory sessionFactory() {

        OnlineSessionFactory sessionFactory = new OnlineSessionFactory();
        return sessionFactory;
    }

    /**
     * 自定义sessionFactory调度器
     */
    @Bean

    public SpringSessionValidationScheduler sessionValidationScheduler() {

        SpringSessionValidationScheduler sessionValidationScheduler = new SpringSessionValidationScheduler();
        // 相隔多久检查一次session的有效性，单位毫秒，默认就是10分钟

        sessionValidationScheduler.setSessionValidationInterval(validationInterval);

        // 设置会话验证调度器进行会话验证时的会话管理器
        sessionValidationScheduler.setSessionManager(sessionValidationManager());
        return sessionValidationScheduler;
    }

    /**
     * 会话管理器
     */
    @Bean

    public OnlineWebSessionManager sessionValidationManager() {

        OnlineWebSessionManager manager = new OnlineWebSessionManager();
        // 加入缓存管理器

        manager.setCacheManager(redisEnabled ? cacheManagerConfig.getRedisCacheManager() : cacheManagerConfig.getEhCacheManager());

        // 删除过期的session
        manager.setDeleteInvalidSessions(true);
        // 设置全局session超时时间

        manager.setGlobalSessionTimeout(expireTime);

        // 去掉 JSESSIONID
        manager.setSessionIdUrlRewritingEnabled(false);
        // 是否定时检查session
        manager.setSessionValidationSchedulerEnabled(true);
        // 自定义SessionDao
        manager.setSessionDAO(sessionDAO());
        // 自定义sessionFactory
        manager.setSessionFactory(sessionFactory());
        return manager;
    }

    /**
     * 会话管理器
     */
    @Bean

    public OnlineWebSessionManager sessionManager() {

        OnlineWebSessionManager manager = new OnlineWebSessionManager();
        // 加入缓存管理器

        manager.setCacheManager(redisEnabled ? cacheManagerConfig.getRedisCacheManager() : cacheManagerConfig.getEhCacheManager());

        // 删除过期的session
        manager.setDeleteInvalidSessions(true);
        // 设置全局session超时时间

        manager.setGlobalSessionTimeout(expireTime);

        // 去掉 JSESSIONID
        manager.setSessionIdUrlRewritingEnabled(false);
        // 定义要使用的无效的Session定时调度器
        manager.setSessionValidationScheduler(sessionValidationScheduler());
        // 是否定时检查session
        manager.setSessionValidationSchedulerEnabled(true);
        // 自定义SessionDao
        manager.setSessionDAO(sessionDAO());
        // 自定义sessionFactory
        manager.setSessionFactory(sessionFactory());
        return manager;
    }

    /**
     * 安全管理器
     */
    @Bean

    public SecurityManager securityManager(UserRealm userRealm) {

        DefaultWebSecurityManager securityManager = new DefaultWebSecurityManager();
        // 设置realm.
        securityManager.setRealm(userRealm);
        // 记住我
        securityManager.setRememberMeManager(rememberMeManager());
        // 注入缓存管理器;

        securityManager.setCacheManager(
                redisEnabled ? cacheManagerConfig.getRedisCacheManager() : cacheManagerConfig.getEhCacheManager());

        // session管理器
        securityManager.setSessionManager(sessionManager());
        return securityManager;
    }

    /**
     * 退出过滤器
     */

    public LogoutFilter logoutFilter() {

        LogoutFilter logoutFilter = new LogoutFilter();
        logoutFilter.setLoginUrl(loginUrl);
        return logoutFilter;
    }

    public LogoutFilter logoutFilters() {

        LogoutFilter logoutFilter = new LogoutFilter();
        logoutFilter.setLoginUrl(loginUrls);
        return logoutFilter;
    }

    /**
     * Shiro过滤器配置
     */
    @Bean

    public ShiroFilterFactoryBean shiroFilterFactoryBean(SecurityManager securityManager) {

        ShiroFilterFactoryBean shiroFilterFactoryBean = new ShiroFilterFactoryBean();
        // Shiro的核心安全接口,这个属性是必须的
        shiroFilterFactoryBean.setSecurityManager(securityManager);
        // 身份认证失败，则跳转到登录页面的配置
        shiroFilterFactoryBean.setLoginUrl(loginUrl);
        // 权限认证失败，则跳转到指定页面
        shiroFilterFactoryBean.setUnauthorizedUrl(unauthorizedUrl);
        // Shiro连接约束配置，即过滤链的定义
        LinkedHashMap<String, String> filterChainDefinitionMap = new LinkedHashMap<>();
        // 对静态资源设置匿名访问
        filterChainDefinitionMap.put("/favicon.ico**", "anon");
        filterChainDefinitionMap.put("/huaheng.png**", "anon");

        filterChainDefinitionMap.put("/logo.png", "anon");

        filterChainDefinitionMap.put("/css/**", "anon");
        filterChainDefinitionMap.put("/docs/**", "anon");
        filterChainDefinitionMap.put("/fonts/**", "anon");
        filterChainDefinitionMap.put("/img/**", "anon");

        filterChainDefinitionMap.put("/apk/**", "anon");

        filterChainDefinitionMap.put("/reservation/**", "anon");

        filterChainDefinitionMap.put("/ajax/**", "anon");
        filterChainDefinitionMap.put("/js/**", "anon");
        filterChainDefinitionMap.put("/huaheng/**", "anon");
        filterChainDefinitionMap.put("/druid/**", "anon");
        filterChainDefinitionMap.put("/captcha/captchaImage**", "anon");

        filterChainDefinitionMap.put("/reservation/add", "anon");

        // 退出 logout地址，shiro去清除session
        filterChainDefinitionMap.put("/logout", "logout");
        filterChainDefinitionMap.put("/admin/logout", "adminlogout");
        // 不需要拦截的访问
//        filterChainDefinitionMap.put("/admin/home", "anon,captchaValidate");

        filterChainDefinitionMap.put("/mobile/download/*", "anon");

        //filterChainDefinitionMap.put("/config/zone/*", "anon");
        //filterChainDefinitionMap.put("/receipt/receiptHeader/listPda", "anon");
        //filterChainDefinitionMap.put("/receipt/receiving/*", "anon");
        //filterChainDefinitionMap.put("/task/taskHeader/*", "anon");
        //filterChainDefinitionMap.put("/mobile/inventory/completeTaskListByWMS", "anon");

        filterChainDefinitionMap.put("/admin/login", "anon");
        filterChainDefinitionMap.put("/login", "anon");

        filterChainDefinitionMap.put("/file/**", "anon");

        filterChainDefinitionMap.put("/api/login", "anon");
        filterChainDefinitionMap.put("/mobile/login", "anon");

        filterChainDefinitionMap.put("/api/getTokenForMobile", "anon");

        filterChainDefinitionMap.put("/getWarehouseByUserCode", "anon");

        //积木报表排除
        filterChainDefinitionMap.put("/jmreport/**", "anon");
        filterChainDefinitionMap.put("/**/*.js.map", "anon");
        filterChainDefinitionMap.put("/**/*.css.map", "anon");

//        filterChainDefinitionMap.put("/websocket", "anon");
//        filterChainDefinitionMap.put("/websocket/*", "anon");
//        filterChainDefinitionMap.put("/user/lincoln/message", "anon");

        filterChainDefinitionMap.put("/API/WMS/v2/login", "anon");

        filterChainDefinitionMap.put("/api/**", "anon");

        filterChainDefinitionMap.put("/endpoint/**", "anon");

        filterChainDefinitionMap.put("/manager/**", "anon");

        filterChainDefinitionMap.put("/API/WMS/v2/**", "anon");
        filterChainDefinitionMap.put("/mobile/receipt/batch/**", "anon");

        //todo pda出入库查询，测试后注释

        //filterChainDefinitionMap.put("/receipt/receiptHeader/listPda", "anon");
        //filterChainDefinitionMap.put("/receipt/receiptDetail/list", "anon");
        //filterChainDefinitionMap.put("/receipt/receiptDetail/pdaList", "anon");
        //filterChainDefinitionMap.put("/task/taskHeader/getTaskByTaskId", "anon");
        //filterChainDefinitionMap.put("/task/taskHeader/getTaskByShipmentCode", "anon");
        //filterChainDefinitionMap.put("/mobile/inventory/completeTaskListByWMS", "anon");
        //filterChainDefinitionMap.put("/receipt/receiving/saveBatch", "anon");
        //filterChainDefinitionMap.put("/config/zone/getAllFlatLocation", "anon");

        //filterChainDefinitionMap.put("/mobile/getModules2", "anon");

        // 系统权限列表
//        filterChainDefinitionMap.putAll(SpringUtils.getBean(IMenuService.class).selectPermsAll());

        //websocket

        Map<String, Filter> filters = new LinkedHashMap<>();
        filters.put("onlineSession", onlineSessionFilter());
        filters.put("syncOnlineSession", syncOnlineSessionFilter());
        // 注销成功，则跳转到指定页面
        filters.put("logout", logoutFilter());
        filters.put("adminlogout", logoutFilters());
        shiroFilterFactoryBean.setFilters(filters);

        // 所有请求需要认证
        filterChainDefinitionMap.put("/**", "user,onlineSession,syncOnlineSession");
        shiroFilterFactoryBean.setFilterChainDefinitionMap(filterChainDefinitionMap);

        return shiroFilterFactoryBean;
    }

    /**
     * 自定义在线用户处理过滤器
     */
    @Bean

    public OnlineSessionFilter onlineSessionFilter() {

        OnlineSessionFilter onlineSessionFilter = new OnlineSessionFilter();
        onlineSessionFilter.setLoginUrl(loginUrl);
        return onlineSessionFilter;
    }

    /**
     * 自定义在线用户同步过滤器
     */
    @Bean

    public SyncOnlineSessionFilter syncOnlineSessionFilter() {

        SyncOnlineSessionFilter syncOnlineSessionFilter = new SyncOnlineSessionFilter();
        return syncOnlineSessionFilter;
    }

    /**
     * cookie 属性设置
     */

    public SimpleCookie rememberMeCookie() {

        SimpleCookie cookie = new SimpleCookie("rememberMe");
        cookie.setDomain(domain);
        cookie.setPath(path);
        cookie.setHttpOnly(httpOnly);

        cookie.setMaxAge(maxAge * 24 * 60 * 60);

        return cookie;
    }

    /**
     * 记住我
     */

    public CookieRememberMeManager rememberMeManager() {

        CookieRememberMeManager cookieRememberMeManager = new CookieRememberMeManager();
        cookieRememberMeManager.setCookie(rememberMeCookie());
        cookieRememberMeManager.setCipherKey(Base64.decode("fCq+/xW488hMTCD+cmJ3aQ=="));
        return cookieRememberMeManager;
    }

    /**
     * thymeleaf模板引擎和shiro框架的整合
     */
    @Bean

    public ShiroDialect shiroDialect() {

        return new ShiroDialect();
    }

    /**
     * 开启Shiro注解通知器
     */
    @Bean
    public AuthorizationAttributeSourceAdvisor authorizationAttributeSourceAdvisor(

            @Qualifier("securityManager") SecurityManager securityManager) {

        AuthorizationAttributeSourceAdvisor authorizationAttributeSourceAdvisor = new AuthorizationAttributeSourceAdvisor();
        authorizationAttributeSourceAdvisor.setSecurityManager(securityManager);
        return authorizationAttributeSourceAdvisor;
    }
}