MVCBeanPostProcess.java

package org.theonefx.wcframework.mvc;

import org.theonefx.wcframework.core.ClassWrapper;
import org.theonefx.wcframework.ioc.MergedBeanDefinitionPostProcessor;
import org.theonefx.wcframework.ioc.RootBeanDefinition;
import org.theonefx.wcframework.ioc.RuntimeBeanReference;

public class MVCBeanPostProcess implements MergedBeanDefinitionPostProcessor {

	@Override
	public void postProcessMergedBeanDefinition(RootBeanDefinition beanDefinition, ClassWrapper<?> classWrapper, String id) {
		if (classWrapper.canCastToDirectly(ServletContextAware.class)) {
			beanDefinition.getPropertyValues().addPropertyValue("servletContext", new RuntimeBeanReference(WebApplicationContext.SERVLET_BEAN_ID));
		}
	}

}