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;
 	}
 }