/*

  * 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;

 import java.io.*;

 import java.lang.reflect.Method;

 import java.net.URL;

 import java.net.URLClassLoader;

 import java.sql.Connection;

 import java.sql.Driver;

 import java.sql.SQLException;

 import java.util.*;

 import net.sourceforge.addam.ddlrun.custom.IDSCompletedScriptRetriever;

 import net.sourceforge.addam.ddlrun.custom.IDSRunLogger;

 import net.sourceforge.addam.ddlrun.custom.IDSStartFolderRetriever;

 import net.sourceforge.addam.ddlrun.grammars.ScriptGrammar;

 import net.sourceforge.addam.ddlrun.grammars.ScriptGrammarManager;

 import net.sourceforge.addam.ddlrun.runners.DeploymentRunner;

 import net.sourceforge.addam.ddlrun.runners.DeploymentRunnerFactory;

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

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

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

 /**

  * this class is intended to be bundled as the main for a jar file that includes

  * the scripts necessary to run the upgrade

  */

 public class Main {

     private static final String separator = "/";

     // getting the VERSION only works when the classes in addam.jar have been copied to

     // another jar file with a valid manifest

     private static final Package PKG = Main.class.getPackage();

     private static final String VERSION = PKG.getSpecificationTitle() + " " + PKG.getSpecificationVersion() + " (" + PKG.getImplementationVersion() + ")";

     private static final String URLPREFIX = "jar:file:/";

     private static final String JAR_FILE_NAME = getJarFileName();

     private static final String JAR_FILE_DIR = getJarFileDir();

     private Main() {

     }

     public static void main(String args[]) {

         String configFileName = JAR_FILE_NAME.replace(".jar",".conf");

         if (args.length < 1 || args.length > 2) {

             usage(configFileName);

             return;

         }

         if (args.length == 2) {

             configFileName = args[1];

         }

         Map<String,String> conf;

         try {

             conf = loadConfig(configFileName);

         } catch (IOException e) {

             System.err.println("unable to load file " + configFileName);

             e.printStackTrace();

             return;

         }

         String url = getConf(conf,"url");

         String username = getConf(conf,"username");

         String password = getConf(conf,"password");

         String driverjar = getConf(conf,"driverjar");

         String driverName = getConf(conf,"driver");

         ScriptGrammar grammar;

         Driver driver;

         try {

             ClassPathHacker.addFile(driverjar);

             Class driverClass = Class.forName(driverName);

             driver = (Driver) driverClass.newInstance();

         } catch (ClassNotFoundException e) {

             throw new RuntimeException("unknown driver class " + driverName + "\n" +

                     "make sure to place the driver .jar file in the same directory as the script jar file");

         } catch (IllegalAccessException e) {

             throw new RuntimeException("can't instantiate driver " + driverName, e);

         } catch (InstantiationException e) {

             throw new RuntimeException("can't instantiate driver " + driverName, e);

         } catch (IOException e) {

             throw new RuntimeException("can't instantiate driver " + driverName, e);

         }

         String type;

         if ("-install".equals(args[0])) {

             type = "install";

         } else if ("-upgrade".equals(args[0])) {

             type = "upgrade";

         } else if ("-version".equals(args[0])) {

             System.out.println(VERSION);

             return;

         } else {

             usage(configFileName);

             return;

         }

         // use internal scripts, e.g. upgrade/oracle/upgrade.drv

         String script = type + ".drv";

         DeploymentRunner runner;

         Connection connection = null;

         try {

             System.out.println("connecting to " + url);

             Properties props = new Properties();

             props.put("user", username);

             props.put("password", password);

             connection = driver.connect(url, props);

             grammar = new ScriptGrammarManager().getGrammar(connection);

             if (grammar == null) {

                 throw new RuntimeException("unsupported database vendor");

             }

             String resourcePath = "database" + separator + type + separator + grammar.name();

             ResourceReaderFactory factory = new ClassLoaderReaderFactory(resourcePath);

             if ("install".equals(type)) {

                 runner = DeploymentRunnerFactory.getInstance().getInstallRunner(factory,

                         connection,

                         grammar);

             } else if ("upgrade".equals(type)) {

                 runner = DeploymentRunnerFactory.getInstance().getUpgradeRunner(factory,

                         connection,

                         grammar,

                         new IDSStartFolderRetriever(connection),

                         new IDSCompletedScriptRetriever(connection));

                 runner.addRunLogger(new IDSRunLogger(connection, VERSION));

             } else {

                 throw new RuntimeException("action must be install or upgrade");

             }

             runner.addRunLogger(new RunLogger() {

                 int count = 0;

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

                     System.out.println("executed " + script + " (" + time + "ms)");

                 }

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

                     System.out.println("completed execution of available scripts in " + group + " (" + time + "ms)");

                 }

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

                     System.out.println(group + separator + script + " failed!");

                 }

             });

             runner.run(script);

         } catch (Exception e) {

             System.err.println(e.getMessage());

             e.printStackTrace(System.err);

             System.exit(1);

         } finally {

             try {

                 connection.close();

             } catch (SQLException e) {

                 e.printStackTrace();

             }

         }

     }

     private static String getConf(Map<String,String> map, String key) {

         String value = map.get(key);

         if (value == null || value.length()==0) {

             System.err.println("missing config entry for " + key);

             System.exit(-1);

         }

         return value;

     }

     private static Map<String,String> loadConfig(String configFileName) throws IOException {

         Properties props = new Properties();

         File file = new File(configFileName);

         if (!file.exists()) {

              file = new File(JAR_FILE_DIR,configFileName);

         }

         FileInputStream fis = new FileInputStream(file);

         props.load(fis);

         Map<String,String> config = new HashMap<String,String>();

         for (Map.Entry entry : props.entrySet()) {

             config.put(entry.getKey().toString(),entry.getValue().toString());

         }

         String password = props.getProperty("password");

         if (password != null && password.length()>0) {

             props.setProperty("password","");

             System.out.println("password stripped from " + configFileName + " for security purposes");

             FileOutputStream fos = new FileOutputStream(file);

             props.store(fos,"updated " + new Date() + " by " + JAR_FILE_NAME);

         }

         return config;

     }

     private static void usage(String configFileName) {

         System.err.printf("usage " + JAR_FILE_NAME + " -version | -install | -upgrade [config file]\n");

         System.err.println();

         System.err.println("if config file is not specified, \"" + configFileName + "\" will be used");

         System.err.println("config file must contain elements from the following example:");

         System.err.println(" url = jdbc:oracle:thin:@host:1521:sid");

         System.err.println(" driver = oracle.jdbc.driver.OracleDriver");

         System.err.println(" driverjar = ojdbc14.jar");

         System.err.println(" username = " + System.getProperty("user.name"));

         System.err.println(" password = <password>");

     }

     private static String getJarFileName() {

         String jarURL = ClassLoader.getSystemResource("net/sourceforge/addam/Main.class").toString();

         int i = jarURL.indexOf('!');

         if (i > 0) jarURL = jarURL.substring(0, i);

         i = jarURL.lastIndexOf('/');

         if (i > 0 && jarURL.length() > 1) jarURL = jarURL.substring(i + 1);

         return jarURL;

     }

     private static String getJarFileDir() {

         String jarURL = ClassLoader.getSystemResource("net/sourceforge/addam/Main.class").toString();

         int i = jarURL.indexOf('!');

         if (i > 0) jarURL = jarURL.substring(0, i);

         if (jarURL.startsWith(URLPREFIX) && jarURL.length() > URLPREFIX.length()) {

             jarURL = jarURL.substring(URLPREFIX.length());

         }

         i = jarURL.lastIndexOf('/');

         if (i > 0) jarURL = jarURL.substring(0,i);

         return jarURL;

     }

     public static class ClassPathHacker {

         public static void addFile(String fileName) throws IOException {

             File file = new File(fileName);

             if (!file.exists() && !file.isAbsolute()) {

                 file = new File(JAR_FILE_DIR,fileName);

             }

             if (!file.exists()) {

                 throw new FileNotFoundException(fileName);

             }

             URLClassLoader classloader = (URLClassLoader)ClassLoader.getSystemClassLoader();

             try {

                 Method method = URLClassLoader.class.getDeclaredMethod("addURL", URL.class);

                 method.setAccessible(true);

                 method.invoke(classloader, file.toURL());

             } catch (Throwable t) {

                 t.printStackTrace();

                 throw new IOException("Error, could not add URL to system classloader");

             }

         }

     }

 }