/**
    * Copyright 2011-2013 The Kuali Foundation Licensed under the Educational
    * Community 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.osedu.org/licenses/ECL-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 org.kuali.mobility.computerlabs.dao;
  
  import java.io.IOException;
  import java.net.MalformedURLException;
  import java.net.URL;
  import java.util.ArrayList;
  import java.util.LinkedHashMap;
  import java.util.List;
  import java.util.Map;
  import org.apache.log4j.Logger;
  import org.kuali.mobility.computerlabs.entity.Lab;
  import org.kuali.mobility.computerlabs.entity.LabGroup;
  import org.kuali.mobility.computerlabs.entity.Location;
  import org.kuali.mobility.util.mapper.DataMapper;
  import org.springframework.context.ApplicationContext;
  import org.springframework.context.ApplicationContextAware;
  
  /**
   *
   * @author Joe Swanson <joseswan@umich.edu>
   */
  public class ComputerLabsInitBean implements ApplicationContextAware {
  
  	private static Logger LOG = Logger.getLogger(ComputerLabsInitBean.class);
  	private ApplicationContext applicationContext;
  	private ComputerLabsDao dao;
  	private int waitInterval = 300;
  	private static Thread backgroundThread = null;
  	private Map<String, List<String>> labUrls;
  	private DataMapper dataMapper;
  	private String dataMappingUrl;
  	private boolean shouldRun = true;
  
  	public void init() {
  		backgroundThread = new Thread(new ComputerLabsInitBean.BackgroundThread());
  		backgroundThread.setDaemon(true);
  		backgroundThread.start();
  	}
  
  	public void cleanup() {
  		LOG.info("Cleaning up computer lab data.");
  	}
  
  	/**
  	 * @return the dao
  	 */
  	public ComputerLabsDao getDao() {
  		return dao;
  	}
  
  	/**
  	 * @param dao the dao to set
  	 */
  	public void setDao(ComputerLabsDao dao) {
  		this.dao = dao;
  	}
  
  	/**
  	 * @return the applicationContext
  	 */
  	public ApplicationContext getApplicationContext() {
  		return applicationContext;
  	}
  
  	/**
  	 * @param applicationContext the applicationContext to set
  	 */
  	public void setApplicationContext(ApplicationContext applicationContext) {
  		this.applicationContext = applicationContext;
  	}
  
  	/**
  	 * @return the waitInterval
  	 */
  	public int getWaitInterval() {
  		return waitInterval;
  	}
  
  	/**
  	 * @param waitInterval the waitInterval to set
  	 */
  	public void setWaitInterval(int waitInterval) {
  		this.waitInterval = waitInterval;
  	}
  
 	/**
 	 * @return the labUrls
 	 */
 	public Map<String, List<String>> getLabUrls() {
 		return labUrls;
 	}
 
 	/**
 	 * @param labUrls the labUrls to set
 	 */
 	public void setLabUrls(Map<String, List<String>> labUrls) {
 		this.labUrls = labUrls;
 	}
 
 	/**
 	 * @return the dataMapper
 	 */
 	public DataMapper getDataMapper() {
 		return dataMapper;
 	}
 
 	/**
 	 * @param dataMapper the dataMapper to set
 	 */
 	public void setDataMapper(DataMapper dataMapper) {
 		this.dataMapper = dataMapper;
 	}
 
 	/**
 	 * @return the dataMappingUrl
 	 */
 	public String getDataMappingUrl() {
 		return dataMappingUrl;
 	}
 
 	/**
 	 * @param dataMappingUrl the dataMappingUrl to set
 	 */
 	public void setDataMappingUrl(String dataMappingUrl) {
 		this.dataMappingUrl = dataMappingUrl;
 	}
 
 	public void loadData() {
 		List<LabGroup> groups = new ArrayList<LabGroup>();
 		for (String key : getLabUrls().keySet()) {
 			LabGroup group = (LabGroup) getApplicationContext().getBean("computerLabGroupBean");
 			group.setName(key);
 			LOG.debug("Processing lab group " + key);
 			for (String groupUrl : getLabUrls().get(key)) {
 				try {
 					LOG.debug("SourceUrl: " + groupUrl);
 					List<Lab> groupLabs = new ArrayList<Lab>();
 					try {
 						URL url = new URL(groupUrl);
 						if (getDataMappingUrl() != null && !"".equals(getDataMappingUrl().trim())) {
 							groupLabs = getDataMapper().mapData(groupLabs, url, new URL(getDataMappingUrl()));
 						} else {
 							groupLabs = getDataMapper().mapData(groupLabs, url, "labMapping.xml");
 						}
 					} catch (MalformedURLException mue) {
 						LOG.error(mue.getLocalizedMessage(), mue);
 						LOG.debug( "Bad URL, attempting to load data as file in classpath: "+groupUrl);
 						groupLabs = getDataMapper().mapData(groupLabs, groupUrl, "labMapping.xml");
 					}
 					LOG.debug("Loaded " + groupLabs.size() + " labs for group " + group.getName());
 					Map<String, Location> locations = new LinkedHashMap<String, Location>();
 					for (Lab lab : groupLabs) {
 						Location location = null;
 						if (locations.containsKey(lab.getBuildingCode())) {
 							location = locations.get(lab.getBuildingCode());
 						} else {
 							location = (Location) getApplicationContext().getBean("computerLabLocationBean");
 							location.setName(lab.getBuildingCode());
 							locations.put(lab.getBuildingCode(), location);
 						}
 						location.addLab(lab);
 					}
 					LOG.debug("Sorted into " + locations.size() + " locations.");
 					group.setLocations(new ArrayList(locations.values()));
 				} catch (MalformedURLException mue) {
 					LOG.error(mue.getLocalizedMessage(), mue);
 				} catch (ClassNotFoundException cnfe) {
 					LOG.error(cnfe.getLocalizedMessage(), cnfe);
 				} catch (IOException io) {
 					LOG.error(io.getLocalizedMessage(), io);
 				} catch (Exception e) {
 					LOG.error(e.getLocalizedMessage(), e);
 				}
 			}
 			groups.add(group);
 		}
 		LOG.debug("Preparing to set " + groups.size() + " lab groups in dao.");
 		getDao().setLabGroups(groups);
 	}
 
 	/**
 	 * @return the shouldRun
 	 */
 	public boolean shouldRun() {
 		return shouldRun;
 	}
 
 	/**
 	 * @param shouldRun the shouldRun to set
 	 */
 	public void setShouldRun(boolean shouldRun) {
 		this.shouldRun = shouldRun;
 	}
 
 	private class BackgroundThread implements Runnable {
 
 		@Override
 		public void run() {
 			while (shouldRun()) {
 				try {
 					/*
 					 * Sleeping first for base KME since the test data
 					 * is self referencing and the URLs won't be valid
 					 * when the bean is initialized by Spring but will
 					 * be once the web app is fully up and running.
 					 */
 					Thread.sleep(1000 * getWaitInterval());
 					loadData();
 					// Loop over campuses
 				} catch (InterruptedException ie) {
 					LOG.error(ie.getLocalizedMessage(), ie);
 				} catch (NullPointerException npe) {
 					// No idea why we get these yet.
 					LOG.error(npe.getLocalizedMessage(), npe);
 				} catch (Exception e) {
 					LOG.error(e.getLocalizedMessage(), e);
 				}
 			}
 		}
 	}
 
 	/**
 	 * @return the backgroundThread
 	 */
 	public static Thread getBackgroundThread() {
 		return backgroundThread;
 	}
 
 	/**
 	 * @param aBackgroundThread the backgroundThread to set
 	 */
 	public static void setBackgroundThread(Thread aBackgroundThread) {
 		backgroundThread = aBackgroundThread;
 	}
 }