package org.argeo.slc.core.execution;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.argeo.slc.SlcException;
import org.argeo.slc.execution.ExecutionFlowDescriptor;
import org.argeo.slc.execution.ExecutionModule;
import org.argeo.slc.execution.ExecutionModuleDescriptor;
import org.argeo.slc.execution.ExecutionModulesManager;
import org.argeo.slc.process.RealizedFlow;
import org.argeo.slc.process.SlcExecution;
import org.argeo.slc.process.SlcExecutionNotifier;
import org.springframework.util.Assert;

public class DefaultModulesManager implements ExecutionModulesManager {
	private final static Log log = LogFactory
			.getLog(DefaultModulesManager.class);

	private List<ExecutionModule> executionModules = new ArrayList<ExecutionModule>();
	private List<SlcExecutionNotifier> slcExecutionNotifiers = new ArrayList<SlcExecutionNotifier>();

	protected ExecutionModule getExecutionModule(String moduleName,
			String version) {
		for (ExecutionModule moduleT : executionModules) {
			if (moduleT.getName().equals(moduleName)) {
				if (moduleT.getVersion().equals(version)) {
					return moduleT;
				}
			}
		}
		return null;
	}

	public ExecutionModuleDescriptor getExecutionModuleDescriptor(
			String moduleName, String version) {
		ExecutionModule module = getExecutionModule(moduleName, version);

		Assert.notNull(module);

		return module.getDescriptor();
	}

	public List<ExecutionModule> listExecutionModules() {
		return executionModules;
	}

	public void setExecutionModules(List<ExecutionModule> executionModules) {
		this.executionModules = executionModules;
	}

	public void process(SlcExecution slcExecution) {
		new ProcessThread(slcExecution).start();
	}

	protected void dispatchUpdateStatus(SlcExecution slcExecution,
			String oldStatus, String newStatus) {
		for (Iterator<SlcExecutionNotifier> it = slcExecutionNotifiers
				.iterator(); it.hasNext();) {
			it.next().updateStatus(slcExecution, oldStatus, newStatus);
		}
	}

	public void setSlcExecutionNotifiers(
			List<SlcExecutionNotifier> slcExecutionNotifiers) {
		this.slcExecutionNotifiers = slcExecutionNotifiers;
	}

	private class ProcessThread extends Thread {
		private final SlcExecution slcExecution;
		private final List<RealizedFlow> flowsToProcess = new ArrayList<RealizedFlow>();

		public ProcessThread(SlcExecution slcExecution) {
			this.slcExecution = slcExecution;
		}

		public void run() {
			log.info("\n##\n## Process SLC Execution " + slcExecution
					+ "\n##\n");

			// FIXME: hack to let the SlcExecution be registered on server
			try {
				Thread.sleep(500);
			} catch (InterruptedException e1) {
				// silent
			}

			slcExecution.setStatus(SlcExecution.STATUS_RUNNING);
			dispatchUpdateStatus(slcExecution, SlcExecution.STATUS_SCHEDULED,
					SlcExecution.STATUS_RUNNING);

			flowsToProcess.addAll(slcExecution.getRealizedFlows());

			while (flowsToProcess.size() > 0) {
				RealizedFlow flow = flowsToProcess.remove(0);
				ExecutionModule module = getExecutionModule(flow
						.getModuleName(), flow.getModuleVersion());
				if (module != null) {
					ExecutionThread thread = new ExecutionThread(this, flow
							.getFlowDescriptor(), module);
					thread.start();
				} else {
					throw new SlcException("ExecutionModule "
							+ flow.getModuleName() + ", version "
							+ flow.getModuleVersion() + " not found.");
				}

				synchronized (this) {
					try {
						wait();
					} catch (InterruptedException e) {
						// silent
					}
				}
			}

			slcExecution.setStatus(SlcExecution.STATUS_RUNNING);
			dispatchUpdateStatus(slcExecution, SlcExecution.STATUS_RUNNING,
					SlcExecution.STATUS_FINISHED);
			/*
			 * for (RealizedFlow flow : slcExecution.getRealizedFlows()) {
			 * ExecutionModule module = getExecutionModule(flow.getModuleName(),
			 * flow.getModuleVersion()); if (module != null) { ExecutionThread
			 * thread = new ExecutionThread(flow .getFlowDescriptor(), module);
			 * thread.start(); } else { throw new
			 * SlcException("ExecutionModule " + flow.getModuleName() +
			 * ", version " + flow.getModuleVersion() + " not found."); } }
			 */
		}

		public synchronized void flowCompleted() {
			notifyAll();
		}
	}

	private class ExecutionThread extends Thread {
		private final ExecutionFlowDescriptor executionFlowDescriptor;
		private final ExecutionModule executionModule;
		private final ProcessThread processThread;

		public ExecutionThread(ProcessThread processThread,
				ExecutionFlowDescriptor executionFlowDescriptor,
				ExecutionModule executionModule) {
			super("SLC Execution #" /* + executionContext.getUuid() */);
			this.executionFlowDescriptor = executionFlowDescriptor;
			this.executionModule = executionModule;
			this.processThread = processThread;
		}

		public void run() {
			try {
				executionModule.execute(executionFlowDescriptor);
			} catch (Exception e) {
				// TODO: re-throw exception ?
				log.error("Execution "/* + executionContext.getUuid() */
						+ " failed.", e);
			} finally {
				processThread.flowCompleted();
			}
		}
	}
}