/*

  * Copyright 2007 The Kuali Foundation

  *

  * Licensed under the Educational Community License, Version 2.0 (the "License");

  * you may not use this file except in compliance with the License.

  * You may obtain a copy of the License at

  *

  * http://www.opensource.org/licenses/ecl2.php

  *

  * Unless required by applicable law or agreed to in writing, software

  * distributed under the License is distributed on an "AS IS" BASIS,

  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  * See the License for the specific language governing permissions and

  * limitations under the License.

  */

 package org.kuali.rice.test;

 import java.io.File;

 import java.sql.DriverManager;

 import org.apache.log4j.Logger;

 import org.kuali.rice.core.api.config.ConfigurationException;

 import org.kuali.rice.core.api.config.property.Config;

 import org.kuali.rice.core.api.config.property.ConfigContext;

 import org.kuali.rice.core.api.lifecycle.Lifecycle;

 public class DerbyDBCreationLifecycle implements Lifecycle {

         private static final Logger LOG = Logger.getLogger(DerbyDBCreationLifecycle.class);

         private String sqlFile;

         public DerbyDBCreationLifecycle(String sqlFile) {

                 this.setSqlFile(sqlFile);

         }

         public boolean isStarted() {

                 return false;

         }

         public void start() throws Exception {

                 if (! isDoingDerby()) {

                         LOG.info("Not using the Derby database for testing or no ddl file found");

                         return;

                 }

                 //just checking that the driver's on the classpath and the url is valid

         Class.forName("org.apache.derby.jdbc.EmbeddedDriver");

         DriverManager.getConnection(ConfigContext.getCurrentContextConfig().getProperty("datasource.url")).close();

                 String dbLocation = ConfigContext.getCurrentContextConfig().getProperty("db.location");

                 File db = new File(dbLocation);

                 if (! db.exists()) {

                         throw new ConfigurationException("Can't find db file " + dbLocation);

                 }

                 if (isDerbyDBReadyForTests()) {

                         LOG.info("Derby ready for testing");

                         return;

                 }

                 LOG.info("Setting up Derby for testing");

                 LOG.info("Derby connection string: " + ConfigContext.getCurrentContextConfig().getProperty("datasource.url"));

                 SQLDataLoader dataLoader = new SQLDataLoader(this.getSqlFile(), ";");

                 dataLoader.runSql();

         }

         public void stop() throws Exception {

         }

         private boolean isDerbyDBReadyForTests() {

                 return new ClearDatabaseLifecycle().isTestTableInSchema(TestHarnessServiceLocator.getDataSource());

         }

         protected boolean isDoingDerby() {

                 if (this.getSqlFile() == null) {

                         return false;

                 }

                 String dbDriverName = ConfigContext.getCurrentContextConfig().getProperty(Config.DATASOURCE_DRIVER_NAME);

                 if (dbDriverName == null) {

                         throw new ConfigurationException("No property '" + Config.DATASOURCE_DRIVER_NAME + "' found");

                 }

                 return dbDriverName.toLowerCase().contains("derby");

         }

         public String getSqlFile() {

                 return sqlFile;

         }

         public void setSqlFile(String sqlFile) {

                 this.sqlFile = sqlFile;

         }

 }