/*

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

 import java.io.InputStreamReader;

 import java.sql.Connection;

 import java.sql.SQLException;

 import java.sql.Statement;

 import org.apache.commons.lang.StringUtils;

 import org.apache.log4j.Logger;

 import org.springframework.core.io.DefaultResourceLoader;

 import org.springframework.jdbc.core.ConnectionCallback;

 import org.springframework.jdbc.core.JdbcTemplate;

 import org.springframework.transaction.TransactionStatus;

 import org.springframework.transaction.support.TransactionCallback;

 import org.springframework.transaction.support.TransactionTemplate;

 public class SQLDataLoader {

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

     public static final String SQL_LINE_COMMENT_PREFIX = "--";

     private String fileLoc;

     private String seperatorChar;

     private String statement;

     public SQLDataLoader(String statement) {

         this.fileLoc = null;

         this.seperatorChar = null;

         this.statement = statement;

     }

     public SQLDataLoader(String fileLoc, String seperatorChar) {

         this.fileLoc = fileLoc;

         this.seperatorChar = seperatorChar;

         this.statement = null;

     }

     public void runSql() throws Exception {

         String[] sqlStatements = null;

         if (statement == null) {

             String sqlStatementsContent = getContentsAsString(fileLoc);

             // separator char must be the last non-whitespace char on the line

             // to avoid splitting in the middle of data that might contain the separator char

             sqlStatements = sqlStatementsContent.split("(?m)" + getSeperatorChar() + "\\s*$");

         } else {

             sqlStatements = new String[]{statement};

         }

         final String[] finalSqlStatements = sqlStatements;

         new TransactionTemplate(TestHarnessServiceLocator.getJtaTransactionManager()).execute(new TransactionCallback() {

             public Object doInTransaction(TransactionStatus status) {

                 return new JdbcTemplate(TestHarnessServiceLocator.getDataSource()).execute(new ConnectionCallback() {

                     public Object doInConnection(Connection connection) throws SQLException {

                         Statement statement = connection.createStatement();

                         LOG.info("################################");

                         LOG.info(fileLoc != null ? "#" + fileLoc : "#");

                         LOG.info("#");

                         for (String sqlStatement : finalSqlStatements) {

                             if (StringUtils.isNotBlank(sqlStatement)) {

                                 LOG.info("# Executing sql statement ->" + sqlStatement + "<-");

                                 statement.execute(sqlStatement);

                             }

                         }

                         LOG.info("#");

                         LOG.info("#");

                         LOG.info("################################");

                         statement.close();

                         return null;

                     }

                 });

             }

         });

     }

     private String getContentsAsString(String fileLoc) throws Exception {

         DefaultResourceLoader resourceLoader = new DefaultResourceLoader();

         String data = "";

         BufferedReader reader = null;

         try {

             reader = new BufferedReader(new InputStreamReader(resourceLoader.getResource(fileLoc).getInputStream()));

             String line = "";

             while ((line = reader.readLine()) != null) {

                 // discard comments...commented single line statements

                 // will result in errors when executed because there are no

                 // results

                 if (!line.trim().startsWith(SQL_LINE_COMMENT_PREFIX)) {

                     data += line + "\r\n ";

                 }

             }

         } finally {

             if (reader != null) {

                 try {

                     reader.close();

                 } catch (Exception e) {

                     LOG.error(e);

                 }

             }

         }

         return data;

     }

     public String getSeperatorChar() {

         if (this.seperatorChar == null) {

             return ";";

         }

         return seperatorChar;

     }

 }