com.firefly.mvc.web.AnnotationWebContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of firefly-jdk7 Show documentation
There is a newer version: 4.0.3.2
package com.firefly.mvc.web;

import java.lang.reflect.Method;
import java.util.Collections;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;

import javax.servlet.ServletContext;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.firefly.annotation.RequestMapping;
import com.firefly.core.XmlApplicationContext;
import com.firefly.core.support.BeanDefinition;
import com.firefly.core.support.annotation.ConfigReader;
import com.firefly.core.support.xml.XmlBeanReader;
import com.firefly.mvc.web.servlet.SystemHtmlPage;
import com.firefly.mvc.web.support.ControllerBeanDefinition;
import com.firefly.mvc.web.support.ControllerMetaInfo;
import com.firefly.mvc.web.support.InterceptorBeanDefinition;
import com.firefly.mvc.web.support.InterceptorMetaInfo;
import com.firefly.mvc.web.support.MethodParam;
import com.firefly.mvc.web.support.WebBeanReader;
import com.firefly.mvc.web.view.JsonView;
import com.firefly.mvc.web.view.JspView;
import com.firefly.mvc.web.view.TemplateView;
import com.firefly.mvc.web.view.TextView;

/**
 * Web context
 * 
 * @author AlvinQiu
 * 
 */
public class AnnotationWebContext extends XmlApplicationContext implements WebContext {
	
	protected final Resource resource;
	protected final List interceptorList = new LinkedList();

	public AnnotationWebContext(String file) {
		super(file);
		resource = new Resource(getEncoding());
		initContext();
	}
	
	public AnnotationWebContext(String file, ServletContext servletContext) {
		this(file);
		
		if (servletContext != null)
			TemplateView.init(servletContext.getRealPath(getViewPath()), getEncoding());
	}

	private void initContext() {
		for (BeanDefinition beanDef : beanDefinitions) {
			if (beanDef instanceof ControllerBeanDefinition) {
				ControllerBeanDefinition beanDefinition = (ControllerBeanDefinition) beanDef;
				List list = beanDefinition.getReqMethods();
				if (list != null) {
					for (Method m : list) {
						m.setAccessible(true);
						final String uri = m.getAnnotation(RequestMapping.class).value();
						ControllerMetaInfo c = new ControllerMetaInfo(beanDefinition.getInjectedInstance(), m);
						resource.add(uri, c);
					}
				}
			} else if (beanDef instanceof InterceptorBeanDefinition) {
				InterceptorBeanDefinition beanDefinition = (InterceptorBeanDefinition) beanDef;
				if(beanDefinition.getDisposeMethod() != null) {
					beanDefinition.getDisposeMethod().setAccessible(true);
					InterceptorMetaInfo interceptor = new InterceptorMetaInfo(beanDefinition.getInjectedInstance(), 
							beanDefinition.getDisposeMethod(),
							beanDefinition.getUriPattern(), 
							beanDefinition.getOrder());
					interceptorList.add(interceptor);
				}
			}
		}
		if(interceptorList.size() > 0)
			Collections.sort(interceptorList);
		
		TextView.setEncoding(getEncoding());
		JsonView.setEncoding(getEncoding());
		JspView.setViewPath(getViewPath());
	}

	@Override
	protected List getBeanDefinitions(String file) {
		List list1 = new WebBeanReader(file)
				.loadBeanDefinitions();
		List list2 = new XmlBeanReader(file)
				.loadBeanDefinitions();
		if (list1 != null && list2 != null) {
			list1.addAll(list2);
			return list1;
		} else if (list1 != null)
			return list1;
		else if (list2 != null)
			return list2;
		return null;
	}

	@Override
	public String getEncoding() {
		return ConfigReader.getInstance().getConfig().getEncoding();
	}

	@Override
	public String getViewPath() {
		return ConfigReader.getInstance().getConfig().getViewPath();
	}
	
	@Override
	public HandlerChain match(String uri, String servletURI) {
		final HandlerChainImpl chain = new HandlerChainImpl();
		addInterceptor(uri, servletURI, chain);
		addLastHandler(uri, servletURI, chain);
		chain.init();
		return chain;
	}
	
	protected void addLastHandler(String uri, String servletURI, final HandlerChainImpl chain) {
		if(servletURI == null)
			return;
		
		WebHandler last = resource.match(servletURI);
		if(last != null)
			chain.add(last);
	}
	
	protected void addInterceptor(String uri, String servletURI, final HandlerChainImpl chain) {
		if(servletURI == null)
			return;
		
		for(final InterceptorMetaInfo interceptor : interceptorList) {
			if(interceptor.getPattern().match(servletURI) != null) {
				chain.add(new WebHandler(){

					@Override
					public View invoke(HttpServletRequest request, HttpServletResponse response) {
						return interceptor.invoke(getParams(request, response));
					}
					
					private Object[] getParams(HttpServletRequest request, HttpServletResponse response) {
						byte[] methodParam = interceptor.getMethodParam();
						Object[] p = new Object[methodParam.length];

						for (int i = 0; i < p.length; i++) {
							switch (methodParam[i]) {
							case MethodParam.REQUEST:
								p[i] = request;
								break;
							case MethodParam.RESPONSE:
								p[i] = response;
								break;
							case MethodParam.HANDLER_CHAIN:
								p[i] = chain;
								break;
							}
						}
						return p;
					}
				});
			}
		}
	}
	
	protected class HandlerChainImpl implements HandlerChain {
		private List list = new LinkedList();
		private Iterator iterator;
		
		public void add(WebHandler webHandler) {
			list.add(webHandler);
		}
		
		private void init() {
			if(list.size() == 0) { // If web handler is not found, response 404
				list.add(new WebHandler(){

					@Override
					public View invoke(HttpServletRequest request, HttpServletResponse response) {
						String msg = request.getRequestURI() + " not found";
						SystemHtmlPage.responseSystemPage(request, response, getEncoding(), HttpServletResponse.SC_NOT_FOUND, msg);
						return null;
					}
				});
			}
			
			if(iterator == null)
				iterator = list.iterator();
		}

		@Override
		public View doNext(HttpServletRequest request, HttpServletResponse response, HandlerChain chain) {
			if(iterator.hasNext())
				return iterator.next().invoke(request, response);
			else
				return null;
		}
	}

}