package org.apache.torque.mojo;
   
   import java.io.File;
   import java.io.IOException;
   
   import org.apache.commons.io.FileUtils;
   import org.apache.maven.plugin.MojoExecutionException;
   import org.kuali.core.db.torque.KualiTorqueDataDumpTask;
   
  /**
   * Reads the content of tables from a database and exports the data in a database agnostic format to XML files.
   * 
   * @goal exportdata
   * @phase generate-sources
   */
  public class ExportDataMojo extends ExportMojo {
  
      /**
       * The format to use for dates/timestamps
       * 
       * @parameter expression="${dateFormat}" default-value="yyyyMMddHHmmss z"
       * @required
       */
      private String dateFormat;
  
      /**
       * The directory where data XML files will be written
       * 
       * @parameter expression="${dataXMLDir}" default-value="${basedir}/src/main/impex"
       * @required
       */
      private File dataXMLDir;
  
      /**
       * Configure the Ant task
       */
      @Override
      protected void configureTask() throws MojoExecutionException {
          KualiTorqueDataDumpTask task = new KualiTorqueDataDumpTask();
          task.setBuildDirectory(new File(getProject().getBuild().getDirectory()));
          setAntTask(task);
          super.configureTask();
          makeOutputDir();
      }
  
      protected void makeOutputDir() throws MojoExecutionException {
          if (getDataXMLDir().exists()) {
              return;
          }
          try {
              FileUtils.forceMkdir(getDataXMLDir());
          } catch (IOException e) {
              throw new MojoExecutionException("Error creating output directory", e);
          }
      }
  
      public File getDataXMLDir() {
          return dataXMLDir;
      }
  
      public void setDataXMLDir(File outputDir) {
          this.dataXMLDir = outputDir;
      }
  
      public String getDateFormat() {
          return dateFormat;
      }
  
      public void setDateFormat(String dateFormat) {
          this.dateFormat = dateFormat;
      }
  }