net.ymate.framework.webmvc.intercept.UserSessionCheckInterceptor.java Source code

Introduction

Here is the source code for net.ymate.framework.webmvc.intercept.UserSessionCheckInterceptor.java

Source

/*
 * Copyright 2007-2016 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package net.ymate.framework.webmvc.intercept;

import net.ymate.framework.core.Optional;
import net.ymate.framework.core.util.WebUtils;
import net.ymate.framework.webmvc.ErrorCode;
import net.ymate.framework.webmvc.WebResult;
import net.ymate.framework.webmvc.support.UserSessionBean;
import net.ymate.platform.core.beans.intercept.IInterceptor;
import net.ymate.platform.core.beans.intercept.InterceptContext;
import net.ymate.platform.core.lang.BlurObject;
import net.ymate.platform.core.util.ExpressionUtils;
import net.ymate.platform.webmvc.context.WebContext;
import net.ymate.platform.webmvc.view.View;
import org.apache.commons.lang.StringUtils;

import javax.servlet.http.HttpServletRequest;

/**
 * ???
 *
 * @author  (suninformation@163.com) on 16/3/12 ?9:54
 * @version 1.0
 */
public class UserSessionCheckInterceptor implements IInterceptor {

    public Object intercept(InterceptContext context) throws Exception {
        // ??
        switch (context.getDirection()) {
        case BEFORE:
            UserSessionBean _sessionBean = UserSessionBean.current();
            if (_sessionBean == null) {
                if (UserSessionBean.getSessionHandler() != null) {
                    _sessionBean = UserSessionBean.getSessionHandler().handle(context);
                }
            } else if (!_sessionBean.isVerified()) {
                _sessionBean.destroy();
                _sessionBean = null;
            }
            if (_sessionBean == null) {
                // URL?
                HttpServletRequest _request = WebContext.getRequest();
                StringBuffer _returnUrlBuffer = _request.getRequestURL();
                String _queryStr = _request.getQueryString();
                if (StringUtils.isNotBlank(_queryStr)) {
                    _returnUrlBuffer.append("?").append(_queryStr);
                }
                //
                String _redirectUrl = WebUtils.buildRedirectURL(context,
                        StringUtils.defaultIfBlank(WebUtils.buildRedirectCustomURL(context, null),
                                "login?redirect_url=${redirect_url}"),
                        true);
                _redirectUrl = ExpressionUtils.bind(_redirectUrl)
                        .set(Optional.REDIRECT_URL, WebUtils.encodeURL(_returnUrlBuffer.toString())).getResult();
                //
                String _message = WebUtils.i18nStr(context.getOwner(), Optional.SYSTEM_SESSION_TIMEOUT_KEY,
                        "???");
                //
                if (WebUtils.isAjax(WebContext.getRequest(), true, true)) {
                    WebResult _result = WebResult.CODE(ErrorCode.USER_SESSION_INVALID_OR_TIMEOUT).msg(_message)
                            .attr(Optional.REDIRECT_URL, _redirectUrl);
                    return WebResult.formatView(_result);
                }
                if (context.getContextParams().containsKey(Optional.OBSERVE_SILENCE)) {
                    return View.redirectView(_redirectUrl);
                }
                return WebUtils.buildErrorView(WebContext.getContext().getOwner(),
                        ErrorCode.USER_SESSION_INVALID_OR_TIMEOUT, _message, _redirectUrl,
                        BlurObject.bind(context.getOwner().getConfig().getParam(Optional.REDIRECT_TIME_INTERVAL))
                                .toIntValue());
            } else {
                // ??
                _sessionBean.touch();
            }
            break;
        }
        return null;
    }
}