/* ===================================================================
    * Copyright 2002-06 SRI International.
    * Released under the MOZILLA PUBLIC LICENSE Version 1.1
    * which can be obtained at http://www.mozilla.org/MPL/MPL-1.1.html
    * This software is distributed on an "AS IS"
    * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied.
    * See the License for the specific language governing rights and
    * limitations under the License.
    * =================================================================== */
  package com.sri.emo.commandline;
  
  import java.io.BufferedInputStream;
  import java.io.File;
  import java.io.FileInputStream;
  import java.io.InputStream;
  import java.io.InputStreamReader;
  import java.io.LineNumberReader;
  import java.util.Vector;
  
  import org.apache.log4j.Logger;
  
  import com.jcorporate.expresso.core.ExpressoSchema;
  import com.jcorporate.expresso.core.db.DBConnection;
  import com.jcorporate.expresso.core.db.DBConnectionPool;
  import com.jcorporate.expresso.core.db.DBException;
  import com.jcorporate.expresso.core.dbobj.Schema;
  import com.jcorporate.expresso.core.dbobj.SchemaFactory;
  import com.jcorporate.expresso.kernel.ConsoleInstallLog;
  import com.jcorporate.expresso.services.test.SchemaCreator;
  import com.jcorporate.expresso.services.test.SchemaDeleter;
  import com.jcorporate.expresso.services.test.TestSystemInitializer;
  import com.sri.emo.EmoSchema;
  
  /***
   * Allows import of a file of sql statements to be dropped into a database.
   * @author Michael Rimov
   * @todo If this turns out to be useful, add --no-drop arg to skip the
   * drop/create steps.
   */
  public class Import extends AbstractEmoUtility {
  	
  	/***
  	 * Logger.
  	 */
  	private static final Logger log = Logger.getLogger(Import.class);
  
  	/***
  	 * 
  	 * @param args
  	 * @throws Exception 
  	 */
  	public static void main(String[] args) throws Exception {
  		// TODO Auto-generated method stub
      	if (!AbstractEmoUtility.checkArgs(args)) {
      		usage();
      		System.exit(-1);
      	}
      	
      	//Default import context is 'import'
      	setupExpresso("import");
      	
      	boolean skipPrompt = false;
      	if (args.length > 1 && args[0].equalsIgnoreCase("--force")) {
      		skipPrompt = true;    		    		
      	} 
      	
     	
      	
      	if (!skipPrompt) {
      		System.out.println("*********************************************");
      		System.out.println("Warning, the import will destroy all data");
      		System.out.println("On the database context labelled: " + TestSystemInitializer.getTestContext());    		
      		System.out.println("*********************************************");
      		//The fork that is running in ANT doesn't seem to
      		//print this line if print() is used even if
      		//System.out.flush() is called afterwards :(
      		System.out.println("Are you sure you want to continue? Y/[N]");
      		char oneChar = (char)System.in.read();
      		if (!(oneChar == 'Y' || oneChar == 'y')) {
      			System.out.println("Import aborted");
      			System.exit(0);
      		}    		
      	}
      	
      	createFreshSchemas();
      	
      	String inputFile = getInputFile(args);
      	
      	System.out.println("Importing file " + inputFile);
      	File f= new File(inputFile);
      	if (!f.exists()) {
      		throw new IllegalArgumentException("Could not find file " + inputFile);
      	}
      	
      	InputStream is = new BufferedInputStream(new FileInputStream(f));
      	
      	LineNumberReader reader = new LineNumberReader(new InputStreamReader(is));
      	
      	DBConnection connection = DBConnectionPool.getInstance(TestSystemInitializer.getTestContext()).getConnection("Database import");
     	int totalRecords = 0;
     	int totalErrors = 0;
     	int currentLine = 0;
     	try {
     		String oneStatement = reader.readLine();
     		while (oneStatement != null) {
         		currentLine = reader.getLineNumber();
     			try {
     				if (log.isDebugEnabled()) {
     					log.debug("Executing insert " + oneStatement);
     				}
     				connection.execute(oneStatement);
     				totalRecords++;
     				if (log.isDebugEnabled()) {
     					log.debug("Total Records Imported = " + totalRecords);
     				}
     			} catch (DBException ex) {
     				log.error("Error executing statement from file " + inputFile + " line " + currentLine, ex);
     				totalErrors++;
     			}
 
     			oneStatement = reader.readLine();    			
     		}
     	} finally {
     		connection.release();
     		DBConnectionPool.getInstance(TestSystemInitializer.getTestContext()).disconnectAll();
     		is.close();
     	}
 		
     	if (totalErrors == 0) {
     		System.out.println("Successfully imported " + totalRecords + " into database.");
     	} else {
     		System.out.println("Import Errors.");
     		System.out.println("Import failed with " + totalErrors + " errors");
     		System.out.println(totalRecords + " were successfully imported");    		
     	}
     	
         System.exit(0);    	
 	}
 
 	private static void createFreshSchemas() throws Exception {
 		SchemaDeleter deleter = new SchemaDeleter();
 		deleter.setDataContext(TestSystemInitializer.getTestContext());
 		
 		SchemaFactory schemaFactory = SchemaFactory.getInstance();
 		Schema emo = schemaFactory.getSchema(EmoSchema.class.getName());
 		//TODO: which is necessary -- set db context here or in deleter or both?
 		//Better safe than sorry.
 		Schema expresso = schemaFactory.getSchema(ExpressoSchema.class.getName());
 		emo.setDataContext(TestSystemInitializer.getTestContext());
 		expresso.setDataContext(TestSystemInitializer.getTestContext());
 		
 		
 		log.info("Dropping any existing exprseso or emo schema on the existing database");
 		Vector toDelete = new Vector();
 		toDelete.add(emo);
 		toDelete.add(expresso);
 		SchemaDeleter.setSchemas(toDelete);
 		SchemaDeleter.deleteSchemas(TestSystemInitializer.getTestContext());
 		
 		log.info("Schemas dropped.  Creating fresh ones");
 		
 		
 		SchemaCreator.ensureSchemaExists(expresso, new ConsoleInstallLog());
 		SchemaCreator.ensureSchemaExists(emo, new ConsoleInstallLog());
 		log.info("Fresh schemas created.");
 	}
 
 	private static String getInputFile(String[] args) {
 		for (int i =0; i < args.length; i++) {
 			String commandParam = args[i];
 			if (! (commandParam.startsWith("-") || commandParam.startsWith("--")) ) {
 				return commandParam;
 			}
 		}
 		
 		usage();
 		throw new IllegalArgumentException("Could nto find file name in command line");
 	}
 
 	private static void usage() {
 		System.out.println("Usage:  ");
 		System.out.println("\t [--force] <import filename>");		
 	}
 
 }