//

 // Copyright (c) 2004, International Decision Systems

 // all rights reserved

 /*

  * Copyright (c) 2004 International Decision Systems, Inc.  All Rights Reserved.

  *

  * By using this Software, You acknowledge that the Software is a valuable asset

  * and trade secret of either International Decision Systems, Inc. ("IDSI") or a

  * third party supplier of IDSI and constitutes confidential and proprietary

  * information.

  *

  * NEITHER IDSI NOR ANY AGENT OR PERSON ACTING FOR OR WITH IDSI HAS MADE OR DOES

  * MAKE ANY STATEMENTS, AFFIRMATIONS, REPRESENTATIONS OR WARRANTIES WHATSOEVER

  * TO YOU, WHETHER EXPRESS OR IMPLIED, AS TO THE SOFTWARE, THE QUALITY OR

  * CONDITION OF THE SOFTWARE, OR THE OPERATING CHARACTERISTICS OR RELIABILITY OF

  * THE SOFTWARE, OR ITS SUITABILITY FOR ANY GENERAL OR PARTICULAR PURPOSE, OR AS

  * TO ANY OTHER MATTER WHATSOEVER; ANY AND ALL OTHER WARRANTIES INCLUDING

  * WITHOUT LIMITATION ANY WARRANTIES IMPLIED BY LAW, SUCH AS THE IMPLIED

  * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND TITLE,

  * USE AND NON-INFRINGEMENT; ARE HEREBY EXPRESSLY DISCLAIMED AND EXCLUDED.

 */

 package net.sourceforge.addam.ddlrun.runners;

 import java.io.IOException;

 import java.sql.SQLException;

 import java.util.*;

 import net.sourceforge.addam.ddlrun.filters.Filter;

 import net.sourceforge.addam.ddlrun.utils.ResourceReader;

 import net.sourceforge.addam.ddlrun.utils.ResourceReaderFactory;

 import net.sourceforge.addam.ddlrun.utils.RunLogger;

 /**

  * Used to run a group (folder) of scripts.

  *

  * @author TIM3

  * @since Mar 5, 2005

  */

 public class ScriptGroupRunner implements Runner {

     private static final String MASTER_DRIVER = "master.drv";

     private static final String SEPARATOR_CHAR = "/";

     /**

      * Constructs a ScriptGroupRunner

      *

      * @param readerFactory   used to read the resources

      * @param scriptFilter    used to determine which scripts need to run

      * @param scriptRunnerMap registry of runners by type

      */

     public ScriptGroupRunner(ResourceReaderFactory readerFactory,

                              Filter scriptFilter, Map scriptRunnerMap) {

         this.readerFactory = readerFactory;

         this.scriptFilter = scriptFilter;

         this.scriptRunnerMap = scriptRunnerMap;

         this.runLoggers = new ArrayList();

     }

     public void addRunLogger(RunLogger logger) {

         runLoggers.add(logger);

     }

     private void logRunComplete(String group, long time) throws Exception {

         for (Iterator it = runLoggers.iterator(); it.hasNext();) {

             RunLogger logger = (RunLogger) it.next();

             logger.logRunComplete(group, time);

         }

     }

     private void logRunScript(String script, long time) throws Exception {

         for (Iterator it = runLoggers.iterator(); it.hasNext();) {

             RunLogger logger = (RunLogger) it.next();

             logger.logRunScript(script, time);

         }

     }

     private void logRunFailure(String group, String script, Exception e) throws Exception {

         for (Iterator it = runLoggers.iterator(); it.hasNext();) {

             RunLogger logger = (RunLogger) it.next();

             logger.logRunFailure(group, script, e);

         }

     }

     /**

      * @param group the group that contains the list of scripts to run

      * @throws IOException

      */

     public void run(String group) throws Exception {

         ResourceReader reader = readerFactory.getReader(group + SEPARATOR_CHAR + MASTER_DRIVER);

         List allScripts = new ArrayList();

         for (String line = reader.readLine(); line != null; line = reader.readLine()) {

             allScripts.add(line);

         }

         Date groupStart = new Date();

         for (Iterator it = scriptFilter.getIterator(allScripts); it.hasNext();) {

             String script = it.next().toString();

             int i = script.lastIndexOf(".");

             if (i >= 0) {

                 String type = getExtension(script);

                 Runner scriptRunner = (Runner) scriptRunnerMap.get(type);

                 if (scriptRunner == null) {

                     Exception e = new RuntimeException("unknown type " + type + " for script " + script);

                     logRunFailure(group, script, e);

                     throw e;

                 }

                 try {

                     // make sure to include the group so the script can be located

                     Date scriptStart = new Date();

                     scriptRunner.run(group + SEPARATOR_CHAR + script);

                     Date scriptStop = new Date();

                     logRunScript(script, scriptStop.getTime() - scriptStart.getTime());

                 } catch (SQLException e) {

                     logRunFailure(group, script, e);

                     e.printStackTrace();

                     throw e;

                 } catch (Exception e) {

                     logRunFailure(group, script, e);

                     throw e;

                 }

             } else {

                 Exception e = new RuntimeException("no type found for script " + script);

                 logRunFailure(group, script, e);

                 throw e;

             }

         }

         Date groupStop = new Date();

         logRunComplete(group, groupStop.getTime() - groupStart.getTime());

     }

     // nonprivate to facilitate testing

     static String getExtension(String resource) {

         String[] parts = resource.split("\\.");

         return (parts.length > 1) ? parts[parts.length - 1] : null;

     }

     private final ResourceReaderFactory readerFactory;

     private final Filter scriptFilter;

     private final Map scriptRunnerMap;

     private final ArrayList runLoggers;

 }