package com.webportals.commons.uri;

import com.webportals.commons.uri.UrlWriterFactory.UrlResult;
import org.apache.commons.lang.StringUtils;

import javax.servlet.*;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

public class UrlWriterFilter implements Filter {

    @Override
    public void doFilter(ServletRequest request, ServletResponse response,
                         FilterChain chain) throws IOException, ServletException {
        HttpServletRequest httpServletRequest = (HttpServletRequest) request;
        HttpServletResponse httpServletResponse = (HttpServletResponse) response;
        String requestURL = httpServletRequest.getRequestURL().toString();
        UrlResult urlResult = UrlWriterFactory.getUrlResult(requestURL);
        if (null != urlResult && StringUtils.isNotBlank(urlResult.getUrl())) {
            String realPath = urlResult.getUrl();
            if (urlResult.isFlag()) {
                httpServletResponse.sendRedirect(realPath);
                return;
            } else {
                if (!realPath.startsWith("/")) {
                    realPath = "/" + realPath;
                }
                httpServletRequest.getRequestDispatcher(realPath).forward(request, response);
                return;
            }
        } else {
            chain.doFilter(request, response);
        }
    }

    @Override
    public void destroy() {
    }

    @Override
    public void init(FilterConfig arg0) throws ServletException {
    }


}