package fr.ifremer.quadrige2.core.test;
   
   /*-
    * #%L
    * Quadrige2 Core :: Quadrige2 Core Shared
    * $Id:$
    * $HeadURL:$
    * %%
    * Copyright (C) 2017 Ifremer
   * %%
   * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU Affero General Public License as published by
   * the Free Software Foundation, either version 3 of the License, or
   * (at your option) any later version.
   * 
   * This program is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU General Public License for more details.
   * 
   * You should have received a copy of the GNU Affero General Public License
   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
   * #L%
   */
  
  
  import com.google.common.base.Charsets;
  import com.google.common.base.Preconditions;
  import com.google.common.collect.Lists;
  import com.google.common.collect.Sets;
  import com.google.common.io.Files;
  import fr.ifremer.quadrige2.core.config.Quadrige2Configuration;
  import fr.ifremer.quadrige2.core.config.Quadrige2ConfigurationOption;
  import fr.ifremer.quadrige2.core.dao.technical.Daos;
  import fr.ifremer.quadrige2.core.dao.technical.DatabaseSchemaDao;
  import fr.ifremer.quadrige2.core.dao.technical.hibernate.DatabaseSchemaDaoImpl;
  import fr.ifremer.quadrige2.core.service.ServiceLocator;
  import org.apache.commons.collections4.CollectionUtils;
  import org.apache.commons.io.FileUtils;
  import org.apache.commons.lang3.StringUtils;
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.junit.Assume;
  import org.junit.rules.TestRule;
  import org.junit.runner.Description;
  import org.junit.runners.model.Statement;
  import org.nuiton.i18n.I18n;
  import org.nuiton.i18n.init.DefaultI18nInitializer;
  import org.nuiton.i18n.init.UserI18nInitializer;
  
  import java.io.*;
  import java.sql.Connection;
  import java.sql.SQLException;
  import java.util.List;
  import java.util.Locale;
  import java.util.Properties;
  import java.util.Set;
  
  /**
   * To be able to manage database connection for unit test.
   *
   * @author blavenie <benoit.lavenier@e-is.pro>
   * @since 3.3.3
   */
  public abstract class DatabaseResource implements TestRule {
  
      /** Logger. */
      protected static final Log log = LogFactory.getLog(DatabaseResource.class);
  
      /** Constant <code>BUILD_ENVIRONMENT_DEFAULT="hsqldb"</code> */
      public static final String BUILD_ENVIRONMENT_DEFAULT = "hsqldb";
      /** Constant <code>HSQLDB_SRC_DATABASE_DIRECTORY= ie : "../quadrige2-core-client/src/test/db"</code> */
      public static final String HSQLDB_SRC_DATABASE_DIRECTORY_PATTERN = "../%s/src/test/db";
      public static final String HSQLDB_SRC_DATABASE_NAME = "quadrige2";
      public static final String HSQLDB_SRC_DATABASE_SCRIPT_FILE = HSQLDB_SRC_DATABASE_NAME + ".script";
      public static final String HSQLDB_SRC_DATABASE_PROPERTIES_FILE = HSQLDB_SRC_DATABASE_NAME + ".properties";
  
      /** Constant <code>BUILD_TIMESTAMP=System.nanoTime()</code> */
      public static long BUILD_TIMESTAMP = System.nanoTime();
  
      private File resourceDirectory;
  
      private String dbDirectory;
  
      protected final String beanFactoryReferenceLocation;
  
      protected final String beanRefFactoryReferenceId;
  
      private final boolean writeDb;
  
      private String configName;
  
      private boolean witherror = false;
  
      protected Class<?> testClass;
  
      /**
       * <p>Constructor for DatabaseResource.</p>
       *
      * @param configName a {@link String} object.
      * @param beanFactoryReferenceLocation a {@link String} object.
      * @param beanRefFactoryReferenceId a {@link String} object.
      * @param writeDb a boolean.
      */
     protected DatabaseResource(String configName, String beanFactoryReferenceLocation,
             String beanRefFactoryReferenceId,
             boolean writeDb) {
         this.configName = configName;
         this.beanFactoryReferenceLocation = beanFactoryReferenceLocation;
         this.beanRefFactoryReferenceId = beanRefFactoryReferenceId;
         this.writeDb = writeDb;
     }
 
     /**
      * Return configuration files prefix (i.e. 'quadrige2-test')
      * Could be override by external project
      *
      * @return the prefix to use to retrieve configuration files
      */
     protected abstract String getConfigFilesPrefix();
 
     protected abstract String getModuleDirectory();
 
     protected String getHsqldbSrcDatabaseDirectory() {
         return String.format(HSQLDB_SRC_DATABASE_DIRECTORY_PATTERN, getModuleDirectory());
     }
 
     protected String getHsqldbSrcCreateScript() {
         return String.format("%s/%s", getHsqldbSrcDatabaseDirectory(), HSQLDB_SRC_DATABASE_SCRIPT_FILE);
     }
 
     /**
      * <p>Getter for the field <code>resourceDirectory</code>.</p>
      *
      * @param name a {@link String} object.
      * @return a {@link File} object.
      */
     public File getResourceDirectory(String name) {
         return new File(resourceDirectory, name);
     }
 
     /**
      * <p>Getter for the field <code>resourceDirectory</code>.</p>
      *
      * @return a {@link File} object.
      */
     public File getResourceDirectory() {
         return resourceDirectory;
     }
 
     /**
      * <p>isWriteDb.</p>
      *
      * @return a boolean.
      */
     protected boolean isWriteDb() {
         return writeDb;
     }
 
     /** {@inheritDoc} */
     @Override
     public Statement apply(final Statement base, final Description description) {
 
         return new Statement() {
             @Override
             public void evaluate() throws Throwable {
                 before(description);
                 try {
                     base.evaluate();
                 } catch (Throwable e) {
                     witherror = true;
                     log.error(e);
                 } finally {
                     after(description);
                 }
             }
         };
     }
 
     /**
      * <p>before.</p>
      *
      * @param description a {@link org.junit.runner.Description} object.
      * @throws Throwable if any.
      */
     protected void before(Description description) throws Throwable {
         testClass = description.getTestClass();
 
         boolean defaultDbName = StringUtils.isEmpty(configName);
 
         dbDirectory = null;
 //        if (defaultDbName) {
 //            configName = "db";
 //        }
 
         if (log.isInfoEnabled()) {
             log.info("Prepare test " + testClass);
         }
 
         resourceDirectory = getTestSpecificDirectory(testClass, "");
         addToDestroy(resourceDirectory);
 
         // Load building env
         String buildEnvironment = getBuildEnvironment();
 
         // check that config file is in classpath (avoid to find out why it does not works...)
         String configFilename = getConfigFilesPrefix();
         if (enableDb()) {
             configFilename += "-" + (writeDb ? "write" : "read");
         }
         if (!defaultDbName) {
             configFilename += "-" + configName;
         }
         String configFilenameNoEnv = configFilename + ".properties";
         if (StringUtils.isNotBlank(buildEnvironment)) {
             configFilename += "-" + buildEnvironment;
         }
         configFilename += ".properties";
 
         InputStream resourceAsStream = getClass().getResourceAsStream("/" + configFilename);
         if (resourceAsStream == null && StringUtils.isNotBlank(buildEnvironment)) {
             resourceAsStream = getClass().getResourceAsStream("/" + configFilenameNoEnv);
             Preconditions.checkNotNull(resourceAsStream, "Could not find " + configFilename + " or " + configFilenameNoEnv + " in test class-path");
             configFilename = configFilenameNoEnv;
         }
         else {
             Preconditions.checkNotNull(resourceAsStream, "Could not find " + configFilename + " in test class-path");
         }
 
         // Prepare DB
         if (enableDb() && BUILD_ENVIRONMENT_DEFAULT.equalsIgnoreCase(buildEnvironment)) {
 
             dbDirectory = getHsqldbSrcDatabaseDirectory();
             if (!defaultDbName) {
                 dbDirectory += configName;
             }
             Tests.checkDbExists(testClass, dbDirectory);
 
             if (writeDb) {
                 Properties p = new Properties();
                 p.load(resourceAsStream);
                 String jdbcUrl = p.getProperty(Quadrige2ConfigurationOption.JDBC_URL.getKey());
                 boolean serverMode = jdbcUrl != null && jdbcUrl.startsWith("jdbc:hsqldb:hsql://");
 
                 // If hsqld run on server mode
                 if (serverMode) {
                     // Do not copy DB files, but display a warn
                     log.warn(String.format("Database running in server mode ! Please remove the property '%s' in file %s, to use a file database.",
                             Quadrige2ConfigurationOption.JDBC_URL.getKey(), configFilename));
                 }
                 else {
                     // Copy DB files into test directory
                     copyDb(new File(dbDirectory), "db", !writeDb, null);
 
                     // Update db directory with the new path
                     dbDirectory = new File(resourceDirectory, "db").getAbsolutePath();
                     dbDirectory = dbDirectory.replaceAll("[\\\\]", "/");
                 }
             } else {
                 // Load db config properties
                 File dbConfig = new File(dbDirectory, getTestDbName() + ".properties");
                 Properties p = new Properties();
                 BufferedReader reader = Files.newReader(dbConfig, Charsets.UTF_8);
                 p.load(reader);
                 reader.close();
 
                 if (log.isDebugEnabled()) {
                     log.debug("Db config: " + dbConfig + "\n" + p);
                 }
 
                 // make sure db is on readonly mode
                 String readonly = p.getProperty("readonly");
                 Preconditions.checkNotNull(readonly, "Could not find readonly property on db confg: " + dbConfig);
                 Preconditions.checkState("true".equals(readonly), "readonly property must be at true value in read mode test in  db confg: "
                         + dbConfig);
             }
         }
 
         // Initialize configuration
         initConfiguration(configFilename);
 
         // Init i18n
         initI18n();
 
         // Initialize spring context
         if (beanFactoryReferenceLocation != null) {
             ServiceLocator.instance().init(
                     beanFactoryReferenceLocation,
                     beanRefFactoryReferenceId);
         }
     }
 
     protected final Set<File> toDestroy = Sets.newHashSet();
 
     /**
      * <p>addToDestroy.</p>
      *
      * @param dir a {@link File} object.
      */
     public void addToDestroy(File dir) {
         toDestroy.add(dir);
     }
 
     /**
      * <p>setProperty.</p>
      *
      * @param file a {@link File} object.
      * @param key a {@link String} object.
      * @param value a {@link String} object.
      * @throws IOException if any.
      */
     public void setProperty(File file, String key, String value) throws IOException {
         // Load old properties values
         Properties props = new Properties();
         BufferedReader reader = Files.newReader(file, Charsets.UTF_8);
         props.load(reader);
         reader.close();
 
         // Store new properties values
         props.setProperty(key, value);
         BufferedWriter writer = Files.newWriter(file, Charsets.UTF_8);
         props.store(writer, "");
         writer.close();
     }
 
     /**
      * <p>copyDb.</p>
      *
      * @param sourceDirectory a {@link File} object.
      * @param targetDbDirectoryName a {@link String} object.
      * @param readonly a boolean.
      * @param p a {@link Properties} object.
      * @throws IOException if any.
      */
     public void copyDb(File sourceDirectory, String targetDbDirectoryName, boolean readonly, Properties p) throws IOException {
         File targetDirectory = getResourceDirectory(targetDbDirectoryName);
         copyDb(sourceDirectory, targetDirectory, readonly, p, true);
     }
 
     /**
      * <p>copyDb.</p>
      *
      * @param sourceDirectory a {@link File} object.
      * @param targetDirectory a {@link File} object.
      * @param readonly a boolean.
      * @param p a {@link Properties} object.
      * @param destroyAfterTest a boolean.
      * @throws IOException if any.
      */
     public void copyDb(File sourceDirectory, File targetDirectory, boolean readonly, Properties p, boolean destroyAfterTest) throws IOException {
         if (!sourceDirectory.exists()) {
 
             if (log.isWarnEnabled()) {
                 log.warn("Could not find db at " + sourceDirectory + ", test [" +
                         testClass + "] is skipped.");
             }
             Assume.assumeTrue(false);
         }
 
         if (p != null) {
             String jdbcUrl = Daos.getJdbcUrl(targetDirectory, getTestDbName());
             Daos.fillConnectionProperties(p, jdbcUrl, "SA", "");
         }
 
         // Add to destroy files list
         if (destroyAfterTest) {
             addToDestroy(targetDirectory);
         }
 
         log.debug(String.format("Copy directory %s at %s", sourceDirectory.getPath(), targetDirectory.getPath()));
         FileUtils.copyDirectory(sourceDirectory, targetDirectory);
 
         // Set readonly property
         log.debug(String.format("Set database properties with readonly=%s", readonly));
         File dbConfig = new File(targetDirectory, getTestDbName() + ".properties");
         setProperty(dbConfig, "readonly", String.valueOf(readonly));
     }
 
     /**
      * <p>after.</p>
      *
      * @param description a {@link org.junit.runner.Description} object.
      * @throws IOException if any.
      */
     protected void after(Description description) throws IOException {
         if (log.isInfoEnabled()) {
             log.info("After test " + testClass);
         }
 
         ServiceLocator serviceLocator = ServiceLocator.instance();
 
         // If service and database has been started
         if (enableDb() && serviceLocator.isOpen()) {
             Properties connectionProperties = Quadrige2Configuration.getInstance().getConnectionProperties();
 
             // Shutdown if HSQLDB database is a file database (not server mode)
             if (Daos.isFileDatabase(Daos.getUrl(connectionProperties))) {
                 try {
                     Daos.shutdownDatabase(connectionProperties);
                 } catch (Exception e) {
                     if (log.isErrorEnabled()) {
                         log.error("Could not close database.", e);
                     }
                     witherror = true;
                 }
             }
 
             // Shutdown spring context
             serviceLocator.shutdown();
         }
 
         if (!witherror) {
             destroyDirectories(toDestroy, true);
         }
 
         if (beanFactoryReferenceLocation != null) {
 
             // push back default configuration
             ServiceLocator.instance().init(null, null);
         }
     }
 
     /**
      * <p>getTestSpecificDirectory.</p>
      *
      * @param testClass a {@link Class} object.
      * @param name a {@link String} object.
      * @return a {@link File} object.
      * @throws IOException if any.
      */
     public static File getTestSpecificDirectory(Class<?> testClass,
                                                 String name) throws IOException {
         // Trying to look for the temporary folder to store data for the test
         String tempDirPath = System.getProperty("java.io.tmpdir");
         if (tempDirPath == null) {
             // can this really occur ?
             tempDirPath = "";
             if (log.isWarnEnabled()) {
                 log.warn("'\"java.io.tmpdir\" not defined");
             }
         }
         File tempDirFile = new File(tempDirPath);
 
         // create the directory to store database data
         String dataBasePath = testClass.getName()
                 + File.separator // a directory with the test class name
                 + name // a sub-directory with the method name
                 + '_'
                 + BUILD_TIMESTAMP; // and a timestamp
         File databaseFile = new File(tempDirFile, dataBasePath);
         FileUtils.forceMkdir(databaseFile);
 
         return databaseFile;
     }
 
     /**
      * <p>createEmptyDb.</p>
      *
      * @param dbDirectory a {@link String} object.
      * @param dbName a {@link String} object.
      * @return a {@link Connection} object.
      * @throws IOException if any.
      * @throws SQLException if any.
      */
     public Connection createEmptyDb(String dbDirectory,
             String dbName) throws IOException, SQLException {
         File externalDbFile = getResourceDirectory(dbDirectory);
         File scriptFile = new File(getHsqldbSrcCreateScript());
         return createEmptyDb(externalDbFile, dbName, null, scriptFile);
     }
 
     /**
      * <p>createEmptyDb.</p>
      *
      * @param dbDirectory a {@link String} object.
      * @param dbName a {@link String} object.
      * @param p a {@link Properties} object.
      * @return a {@link Connection} object.
      * @throws IOException if any.
      * @throws SQLException if any.
      */
     public Connection createEmptyDb(String dbDirectory,
             String dbName, Properties p) throws IOException, SQLException {
         File externalDbFile = getResourceDirectory(dbDirectory);
         File scriptFile = new File(getHsqldbSrcCreateScript());
         return createEmptyDb(externalDbFile, dbName, p, scriptFile);
     }
 
     /**
      * <p>createEmptyDb.</p>
      *
      * @param directory a {@link File} object.
      * @param dbName a {@link String} object.
      * @param p a {@link Properties} object.
      * @param scriptFile a {@link File} object.
      * @return a {@link Connection} object.
      * @throws SQLException if any.
      * @throws IOException if any.
      */
     protected Connection createEmptyDb(
             File directory,
             String dbName,
             Properties p,
             File scriptFile) throws SQLException, IOException {
 
         Quadrige2Configuration config = Quadrige2Configuration.getInstance();
 
         if (log.isInfoEnabled()) {
             log.info("Create new db at " + directory);
         }
         addToDestroy(directory);
         String jdbcUrl = Daos.getJdbcUrl(directory, dbName);
         String user = "SA";
         String password = "";
 
         p = (p == null) ? config.getConnectionProperties() : p;
         Daos.fillConnectionProperties(p, jdbcUrl, user, password);
 
         Preconditions.checkState(scriptFile.exists(), "Could not find db script at " + scriptFile);
 
         DatabaseSchemaDao schemaDao = new DatabaseSchemaDaoImpl(config);
         schemaDao.generateNewDb(directory, true, scriptFile, p, true/*isTemporaryDb*/);
         Connection connection = Daos.createConnection(jdbcUrl, user, password);
 
         if (log.isInfoEnabled()) {
             log.info("Created connection at " + connection.getMetaData().getURL());
         }
         return connection;
     }
 
     /**
      * <p>getBuildEnvironment.</p>
      *
      * @return a {@link String} object.
      */
     public String getBuildEnvironment() {
         return getBuildEnvironment(null);
     }
 
     /**
      * -- protected methods--
      *
      * @param defaultEnvironment a {@link String} object.
      * @return a {@link String} object.
      */
     protected String getBuildEnvironment(String defaultEnvironment) {
         String buildEnv = System.getProperty("env");
 
         // Check validity
         if (buildEnv == null && StringUtils.isNotBlank(defaultEnvironment)) {
             buildEnv = defaultEnvironment;
             log.warn("Could not find build environment. Please add -Denv=<hsqldb|oracle|pgsql>. Test [" +
                     testClass + "] will use default environment : " + defaultEnvironment);
         } else if (!"hsqldb".equals(buildEnv)
                 && !"oracle".equals(buildEnv)
                 && !"pgsql".equals(buildEnv)) {
 
             if (log.isWarnEnabled()) {
                 log.warn("Could not find build environment. Please add -Denv=<hsqldb|oracle|pgsql>. Test [" +
                         testClass + "] will be skipped.");
             }
             Assume.assumeTrue(false);
         }
         return buildEnv;
     }
 
     /**
      * <p>getConfigArgs.</p>
      *
      * @return an array of {@link String} objects.
      */
     protected String[] getConfigArgs() {
         List<String> configArgs = Lists.newArrayList();
         configArgs.addAll(Lists.newArrayList(
                 "--option", Quadrige2ConfigurationOption.BASEDIR.getKey(), resourceDirectory.getAbsolutePath()));
         if (dbDirectory != null) {
             configArgs.addAll(Lists.newArrayList("--option", Quadrige2ConfigurationOption.DB_DIRECTORY.getKey(), dbDirectory));
         }
         return configArgs.toArray(new String[configArgs.size()]);
     }
 
     /**
      * Convenience methods that could be override to initialize other configuration
      *
      * @param configFilename a {@link String} object.
      */
     protected void initConfiguration(String configFilename) {
         String[] configArgs = getConfigArgs();
         Quadrige2Configuration config = new Quadrige2Configuration(configFilename, configArgs);
         Quadrige2Configuration.setInstance(config);
     }
 
     /**
      * <p>initI18n.</p>
      *
      * @throws IOException if any.
      */
     protected void initI18n() throws IOException {
         Quadrige2Configuration config = Quadrige2Configuration.getInstance();
 
         // --------------------------------------------------------------------//
         // init i18n
         // --------------------------------------------------------------------//
         File i18nDirectory = new File(config.getDataDirectory(), "i18n");
         if (i18nDirectory.exists()) {
             // clean i18n cache
             FileUtils.cleanDirectory(i18nDirectory);
         }
 
         FileUtils.forceMkdir(i18nDirectory);
 
         if (log.isDebugEnabled()) {
             log.debug("I18N directory: " + i18nDirectory);
         }
 
         Locale i18nLocale = config.getI18nLocale();
 
         if (log.isDebugEnabled()) {
             log.debug(String.format("Starts i18n with locale [%s] at [%s]",
                     i18nLocale, i18nDirectory));
         }
         I18n.init(new UserI18nInitializer(
                 i18nDirectory, new DefaultI18nInitializer(getI18nBundleName())),
                 i18nLocale);
     }
 
     /**
      * <p>getI18nBundleName.</p>
      *
      * @return a {@link String} object.
      */
     protected String getI18nBundleName() {
         return "quadrige2-core-shared-i18n";
     }
 
     /**
      * <p>getTestDbName.</p>
      *
      * @return a {@link String} object.
      */
     protected String getTestDbName() {
         return HSQLDB_SRC_DATABASE_NAME;
     }
 
     /* -- Internal methods -- */
 
     private boolean enableDb() {
         return beanFactoryReferenceLocation == null || !beanFactoryReferenceLocation.contains("WithNoDb");
     }
 
     private void destroyDirectories(Set<File> toDestroy, boolean retry) {
         if (CollectionUtils.isEmpty(toDestroy)) {
             return;
         }
 
         Set<File> directoriesToRetry = Sets.newHashSet(toDestroy);
         for (File file : toDestroy) {
             if (file.exists()) {
                 if (log.isInfoEnabled()) {
                     log.info("Destroy directory: " + file);
                 }
                 try {
                     FileUtils.deleteDirectory(file);
 
                 } catch (IOException e) {
                     if (retry) {
                         if (log.isErrorEnabled()) {
                             log.error("Could not delete directory: " + file + ". Will retry later.");
                         }
                         directoriesToRetry.add(file);
                     }
                     else {
                         if (log.isErrorEnabled()) {
                             log.error("Could not delete directory: " + file + ". Please delete it manually.");
                         }
                     }
                 }
             }
         }
 
         if (retry && CollectionUtils.isEmpty(directoriesToRetry)) {
             destroyDirectories(directoriesToRetry, false);
         }
     }
 }