package org.argeo.slc.core.test.tree;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import org.argeo.slc.core.structure.SimpleSElement;
import org.argeo.slc.core.structure.StructureElement;
import org.argeo.slc.core.structure.StructureRegistry;
import org.argeo.slc.core.structure.tree.TreeSPath;
import org.argeo.slc.core.structure.tree.TreeSRegistry;
import org.argeo.slc.dao.structure.tree.TreeSPathDao;
import org.argeo.slc.dao.structure.tree.TreeSRegistryDao;
import org.argeo.slc.dao.test.TestResultDao;
import org.argeo.slc.dao.test.tree.TreeTestResultDao;

/**
 * Listener persisting tree-based results.
 * 
 * @see TreeTestResult
 */
public class TreeTestResultPersister extends AsynchronousTreeTestResultListener {
	private static Log log = LogFactory.getLog(TreeTestResultPersister.class);

	private TreeTestResultDao testResultDao;
	private TreeSPathDao treeSPathDao;
	private TreeSRegistryDao treeSRegistryDao;

	@Override
	protected void resultPartAdded(PartStruct partStruct) {
		try {
			TreeTestResult persistedResult = testResultDao
					.getTestResult(partStruct.resultId);
			
			if(persistedResult==null){
				testResultDao.create(partStruct.result);
			}
			else{
				testResultDao.update(partStruct.result);
			}
/*
			TreeSPath path = treeSPathDao.getOrCreate(partStruct.path);

			StructureRegistry<TreeSPath> localRegistry = partStruct.result
					.getRegistry();
			TreeSRegistry registry = getOrCreateTreeSRegistry(path);
			treeSRegistryDao.syncPath(registry, localRegistry, path);

			if (persistedResult == null) {
				persistedResult = new TreeTestResult();
				persistedResult.setNumericResultId(partStruct.resultId);
				PartSubList subList = new PartSubList();
				subList.getParts().add(partStruct.part);
				persistedResult.getResultParts().put(path, subList);

				testResultDao.create(persistedResult);
			} else {
				PartSubList subList = persistedResult.getResultParts()
						.get(path);
				if (subList == null) {
					subList = new PartSubList();
					persistedResult.getResultParts().put(path, subList);
				}
				persistedResult.getResultParts().get(path).getParts().add(
						partStruct.part);

				if (log.isTraceEnabled()) {
					log.trace("ResultId:" + persistedResult.getTestResultId());
					log.trace("ResultParts size:"
							+ persistedResult.getResultParts().size());
					log.trace("Sublist size:" + subList.getParts().size());
					log.trace("Part: " + partStruct.part);
				}
				testResultDao.update(persistedResult);
			}*/
		} catch (Exception e) {
			log.error("Could not persist part for result #"
					+ partStruct.resultId, e);
		}
	}

	@Override
	protected void postClose(TreeTestResult testResult) {
		TreeTestResult persistedResult = (TreeTestResult) testResultDao
				.getTestResult(testResult.getTestResultId());

		if (persistedResult != null) {
			persistedResult.setCloseDate(testResult.getCloseDate());
			testResultDao.update(persistedResult);
		}
		if (log.isDebugEnabled())
			log.debug("Closed result persister for result "
					+ testResult.getNumericResultId());
	}

	private TreeSRegistry getOrCreateTreeSRegistry(TreeSPath path) {
		TreeSRegistry registry = treeSRegistryDao.getActiveTreeSRegistry();
		if (registry == null) {
			registry = new TreeSRegistry();
			registry.setStatus(TreeSRegistry.STATUS_ACTIVE);
			treeSRegistryDao.create(registry);
			return treeSRegistryDao.getActiveTreeSRegistry();
		} else {
			return registry;
		}
	}

	/** Sets the DAO to use in order to persist the results. */
	public void setTestResultDao(TreeTestResultDao testResultDao) {
		this.testResultDao = testResultDao;
	}

	/** Sets the tree structure path DAO. */
	public void setTreeSPathDao(TreeSPathDao treeSPathDao) {
		this.treeSPathDao = treeSPathDao;
	}

	/** Sets the tree structure registry DAO. */
	public void setTreeSRegistryDao(TreeSRegistryDao treeSRegistryDao) {
		this.treeSRegistryDao = treeSRegistryDao;
	}

}