/*******************************************************************************
 *******************************************************************************/
package at.asitplus.eidas.specific.connector;

import java.util.Arrays;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.ServletRegistration;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.beans.factory.xml.XmlBeanDefinitionReader;
import org.springframework.context.ApplicationContext;
import org.springframework.context.support.ClassPathXmlApplicationContext;
import org.springframework.context.support.GenericApplicationContext;
import org.springframework.core.io.ClassPathResource;
import org.springframework.web.WebApplicationInitializer;
import org.springframework.web.context.ContextLoaderListener;
import org.springframework.web.context.request.RequestContextListener;
import org.springframework.web.context.support.GenericWebApplicationContext;
import org.springframework.web.context.support.ServletContextResource;
import org.springframework.web.servlet.DispatcherServlet;

import at.gv.egiz.components.spring.api.SpringLoader;
import at.gv.egiz.eaaf.core.api.IStatusMessenger;
import at.gv.egiz.eaaf.core.impl.logging.LogMessageProviderFactory;
import at.gv.egiz.eaaf.core.impl.utils.Random;
import at.gv.egiz.eaaf.modules.pvp2.impl.opensaml.initialize.EAAFDefaultSAML2Bootstrap;

/**
 * Web application initializer
 * 
 * @author Thomas Lenz
 */
public class SpringInitializer implements WebApplicationInitializer {

	private static final Logger log = LoggerFactory.getLogger(SpringInitializer.class);
	
    private String[] rootServletContexts = null;
	private String[] servletContexts = null;
	private String[] activeProfiles = null;

	public SpringInitializer() {
		this.rootServletContexts = null;
		this.servletContexts = new String[] {
				"/applicationContext.xml",
				
		};
		this.activeProfiles = null;
	}
    
    
	/* (non-Javadoc)
	 * @see org.springframework.web.WebApplicationInitializer#onStartup(javax.servlet.ServletContext)
	 */
	@Override
	public void onStartup(ServletContext servletContext) throws ServletException {
		try {						
			log.info("=============== Loading Config Root Context! ===============");
			ApplicationContext cfgRootContext = 
			    	new ClassPathXmlApplicationContext(new String[] {
			    			"/applicationContext.xml"
			    			});


			log.info("=============== Loading Root Context! ===============");
			GenericWebApplicationContext rootContext = new GenericWebApplicationContext();
			rootContext.setServletContext(servletContext);
			rootContext.setParent(cfgRootContext);
												
//			log.info("=============== Setting active profiles! ===============");
//			if (this.activeProfiles != null) {
//				for (String profile : this.activeProfiles) {
//					rootContext.getEnvironment().addActiveProfile(profile);
//				}
//			}
			
			log.info("Spring-context was initialized with active profiles: " + 
					Arrays.asList(rootContext.getEnvironment().getActiveProfiles()));
			
			log.info("=============== Loading Local Contexts! ===============");
			XmlBeanDefinitionReader xmlReader = new XmlBeanDefinitionReader(
					rootContext);
			if (rootServletContexts != null) {
				for (String rootServletContext : rootServletContexts) {
					log.debug("Loading: "+ rootServletContext);
					xmlReader.loadBeanDefinitions(new ServletContextResource(
							servletContext, rootServletContext));
				}
			}
			// Manage the lifecycle of the root application context
			servletContext.addListener(new ContextLoaderListener(rootContext));

			// log.debug("Beans after logAMQP in {}", rootContext);
			// dumpBeanDefinitions(rootContext);
			
			log.info("=============== Loading SPI Context! ===============");
			if (rootContext instanceof BeanDefinitionRegistry) {
				log.debug("Loading modules and components");
				SpringLoader.loadSpringServices(rootContext);
				
			} else
				log.warn("Failed to load external Spring since no BeanDefinitionRegistry");
			
			log.trace("Beans after SPI in "+ rootContext);
			dumpBeanDefinitions(rootContext);

			log.debug("Loading servlet config in "+ rootContext);
			if (servletContexts != null) {
				for (String servletContextString : servletContexts)
					xmlReader.loadBeanDefinitions(new ClassPathResource(servletContextString, SpringInitializer.class));
				
			}
						
			log.debug("Refreshing context "+ rootContext);
			rootContext.refresh();

			log.info("=============== Register Dispatcher Servlet! ===============");

			log.trace("Final Beans in "+ rootContext);
			dumpBeanDefinitions(rootContext);
											
			log.info("Registering dispatcher configuration");
			ServletRegistration.Dynamic dispatcher = servletContext.addServlet("dispatcher", new DispatcherServlet(rootContext));
			if (dispatcher != null) {  
				dispatcher.setLoadOnStartup(1);
				dispatcher.addMapping("/");
				dispatcher.setAsyncSupported(true);
				
			} else
				log.error("Failed to register dispatcher server in servlet context!");

			
			log.info("=============== Register RequestContextListener! ===============");
			servletContext.addListener(new RequestContextListener());
									
			//initialize status messenger
			LogMessageProviderFactory.setStatusMessager(rootContext.getBean(IStatusMessenger.class));
			
			log.info("Bootstrap openSAML .... ");
			EAAFDefaultSAML2Bootstrap.bootstrap();

			log.info("Seed random number generator ... ");
			Random.seedRandom();
			
			log.info("Initialization of MS-specific eIDAS-connector finished.");
				
						
		} catch (Throwable e) {
			log.error("MS-specific eIDAS-connector initialization FAILED!", e);
			
		}
				
	}
	
	private void dumpBeanDefinitions(GenericApplicationContext context) {
		log.trace("Registered Bean in context " + context.toString());
		
		String[] registeredBeans = context.getBeanDefinitionNames();
		for (String registeredBean : registeredBeans) {
			BeanDefinition beanDefinition = context
					.getBeanDefinition(registeredBean);
			log.trace(registeredBean + " -> " + 	beanDefinition.getBeanClassName());
			
		}
		
		log.trace("Registered Bean in context --"+ context);
	}
}