/**
    * Copyright 2004-2013 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.apache.torque.mojo;
  
  /*
   * Licensed to the Apache Software Foundation (ASF) under one
   * or more contributor license agreements.  See the NOTICE file
   * distributed with this work for additional information
   * regarding copyright ownership.  The ASF licenses this file
   * to you under the Apache 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.apache.org/licenses/LICENSE-2.0
   *
   * 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.
   */
  
  import java.io.File;
  import java.util.Collections;
  import java.util.Comparator;
  import java.util.List;
  import java.util.Vector;
  
  import org.apache.commons.collections.comparators.ReverseComparator;
  import org.apache.maven.plugin.MojoExecutionException;
  import org.apache.torque.util.SimpleScanner;
  import org.apache.torque.util.TransactionComparator;
  import org.kuali.db.Transaction;
  
  /**
   * Executes SQL for creating tables, primary keys, constraints, indexes, relationships, and views inside a database.
   * Also populates the tables with a default set of data.
   * 
   * @goal import
   */
  public class ImportMojo extends AbstractSQLExecutorMojo {
  	private static final String FS = System.getProperty("file.separator");
  
  	public enum Order {
  		ASCENDING, DESCENDING, NONE;
  	}
  
  	/**
  	 * This is the directory to scan for SQL files to import. ${targetDatabase} gets appended to this value
  	 * 
  	 * @parameter expression="${importDir}" default-value="${project.build.directory}/classes/sql"
  	 * @required
  	 */
  	private File importDir;
  
  	/**
  	 * @parameter expression="${importDirIncludes}" default-value="*.sql"
  	 */
  	private String importDirIncludes;
  
  	/**
  	 * @parameter expression="${importDirExcludes}" default-value=""
  	 */
  	private String importDirExcludes = "";
  
  	/**
  	 * Set the order in which the SQL files will be executed. Possible values are <code>ASCENDING</code> and
  	 * <code>DESCENDING</code> and <code>NONE</code>. Default value is <code>ASCENDING</code>
  	 * 
  	 * @parameter expression="${order}"
  	 */
  	private String order = Order.ASCENDING.toString();
  
  	protected void updateImportDir() {
  		String path = importDir.getAbsolutePath();
  		if (!path.endsWith(FS)) {
  			path += FS;
  		}
  		path += getTargetDatabase();
  		importDir = new File(path);
  	}
  
  	protected Vector<Transaction> getTransactions(List<File> files) {
  		Vector<Transaction> transactions = new Vector<Transaction>();
  		for (File file : files) {
 			Transaction t = new Transaction();
 			t.setResourceLocation(file.getAbsolutePath());
 			transactions.add(t);
 		}
 		return transactions;
 	}
 
 	@Override
 	protected void configureTransactions() throws MojoExecutionException {
 		updateImportDir();
 		List<File> files = new SimpleScanner(importDir, importDirIncludes, importDirExcludes).getFiles();
 		transactions = getTransactions(files);
 		sortTransactions(transactions);
 	}
 
 	/**
 	 * Sort the transaction list.
 	 */
 	@SuppressWarnings("unchecked")
 	protected void sortTransactions(Vector<Transaction> transactions) {
 		Comparator<Transaction> comparator = new TransactionComparator<Transaction>(getProject().getArtifactId());
 		if (Order.ASCENDING.toString().equals(order)) {
 			Collections.sort(transactions, comparator);
 		} else if (Order.DESCENDING.toString().equals(order)) {
 			Collections.sort(transactions, new ReverseComparator(comparator));
 		}
 	}
 
 	public File getImportDir() {
 		return importDir;
 	}
 
 	public void setImportDir(File importDirectory) {
 		this.importDir = importDirectory;
 	}
 
 	public String getImportDirIncludes() {
 		return importDirIncludes;
 	}
 
 	public void setImportDirIncludes(String importDirectoryIncludes) {
 		this.importDirIncludes = importDirectoryIncludes;
 	}
 
 	public String getImportDirExcludes() {
 		return importDirExcludes;
 	}
 
 	public void setImportDirExcludes(String importDirectoryExcludes) {
 		this.importDirExcludes = importDirectoryExcludes;
 	}
 
 	public String getOrder() {
 		return order;
 	}
 
 	public void setOrder(String order) {
 		this.order = order;
 	}
 }