package org.ngbw.utils;

import java.util.List;

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

import org.ngbw.sdk.Workbench;
import org.ngbw.sdk.database.ConnectionManager;
import org.ngbw.sdk.database.DriverConnectionSource;
import org.ngbw.sdk.database.Task;
import org.ngbw.sdk.database.User;
import org.ngbw.sdk.database.UserDataItem;


/**
 * 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.
 *
 * Also deletes UserDataItem records over the same age.
 *
 * Caller should check and report exit status. Exit status of -1 means there was
 * an error. Once an error occurs no further records are deleted.
 */
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.closeConnSourceQuietly();
            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 delete task " + tid + ", jh=" + jh + ", uid=" + uid + ", date created = " + cdate);
                }
            }
            
            List<UserDataItem> dataCandidates = UserDataItem.findDataItemsByAge(age);
            
            for (UserDataItem udi : dataCandidates) {
                long uid = udi.getUserId();
                long dataId = udi.getUserDataId();
                long sid = udi.getSourceDocumentId();
                String cdate = udi.getCreationDate().toString();
                log.debug("Found user data item older than " + age + " days: " + dataId);
                
                if (m_restOnly) {
                    User user = new User(uid);
                    if (!user.getRole().isRestEndUser()) {
                        log.debug("Skipping " + user.getRole() + " user data record " + uid);
                        continue;
                    }
                }
                
                if (m_debug == false) {
                    udi.delete();
                    log.debug("Deleted user data record " + dataId + ", sourceDocument=" + sid +
                            ", date created = " + cdate);
                }
                else {
                    log.debug("Would delete user data record " + dataId + ", sourceDocument=" + sid + 
                            ", date created = " + cdate);
                }
            }
            System.exit(0);
        }
        catch ( Exception e ) {
            log.error(e.getMessage(), e);
            System.exit(-1);
        }
        finally {
            ConnectionManager.closeConnSourceQuietly();
        }
    }

}