package org.ngbw.utils;
import java.util.List;
import java.util.concurrent.*;

import org.ngbw.sdk.Workbench;
import org.ngbw.sdk.tool.BaseProcessWorker;
import org.ngbw.sdk.tool.TaskMonitor;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.ngbw.sdk.core.shared.UserRole;
import org.ngbw.sdk.database.ConnectionManager;
import org.ngbw.sdk.database.ConnectionSource;
import org.ngbw.sdk.database.DriverConnectionSource;
import org.ngbw.sdk.database.Task;
import org.ngbw.sdk.database.User;

/*
	For the rest service, we don't store tasks indefinitely.  This deletes all
	rest tasks over a certain age.  Will most likely run this as a cron job.
	See sdk/scripts/DeleteTasksD.
*/
public class DeleteTasks
{
	private static final Log log = LogFactory.getLog(DeleteTasks.class.getName());
	private static String m_maxDays;
	private static Boolean m_debug;
	private static Boolean m_restOnly = true;

	public static void main(String[] args)
	{
		try
		{
			Workbench wb = Workbench.getInstance();
			log.debug("DELETE TASKS: workbench initialized.");

			//These 2 lines close the connection pool created in Workbench.getInstance() and open an 
			// unpooled connection source. 
			ConnectionManager.getConnectionSource().close();
			ConnectionManager.setConnectionSource(new DriverConnectionSource());

			m_debug = Boolean.valueOf(System.getProperty("debug"));
			m_restOnly = Boolean.valueOf(System.getProperty("restOnly"));
			m_maxDays = System.getProperty("age");
			if (m_maxDays == null)
			{
				throw new Exception("Missing system property age");
			}
			int age;
			try
			{
				age = Integer.valueOf(m_maxDays);
			}
			catch(NumberFormatException nfe)
			{
				throw new Exception("age must be an integer");
			}
			log.debug("age=" + age + ", debug=" + m_debug + ", restOnly=" + m_restOnly);
			List<Task> candidates = Task.findTasksByAge(age);
			for (Task task : candidates)
			{
				log.debug("Found task older than " + age + " days: " + task.getTaskId());
				long uid = task.getUserId();
				long tid = task.getTaskId();
				String jh = task.getJobHandle();
				String cdate = task.getCreationDate().toString();
				if (m_restOnly)
				{
					User user = new User(uid);
					if (!user.getRole().isRestEndUser())
					{
						log.debug("Skipping " + user.getRole() + " user task " + tid); 
						continue;
					}
				}
				if (m_debug == false)
				{
					task.delete();
					log.debug("Deleted task " + tid + ", jh=" + jh + ", uid=" + uid + ", date created = " + cdate);
				} else
				{
					log.debug("Would deleted task " + tid + ", jh=" + jh + ", uid=" + uid + ", date created = " + cdate);
				}
			}
		}
		catch (Exception e)
		{
			log.error("", e);
			return;
		}
		finally
		{
			ConnectionSource source = ConnectionManager.getConnectionSource();
			if (source != null)
			{
				try
				{
					source.close();
				}
				catch(Exception e)
				{
					log.error("", e);
				}
			}
		}
	}

}