package org.apache.torque.mojo;

 import java.io.File;

 import java.util.List;

 import java.util.Properties;

 import java.util.Set;

 import java.util.TreeSet;

 import org.apache.maven.plugin.MojoExecutionException;

 import org.apache.maven.plugin.MojoFailureException;

 import org.apache.tools.ant.DirectoryScanner;

 import org.apache.torque.engine.database.model.Database;

 import org.apache.torque.engine.database.model.Table;

 import org.kuali.core.db.torque.SetUtils;

 import org.kuali.core.db.torque.Utils;

 /**

  * This mojo identifies data files that are present on the file system but are not present in schema.xml. This can

  * happen if a table is removed from the schema.

  * 

  * It sets a project property called "impex.data.invalid". This property is a comma delimited list of filenames that

  * have no match in the db schema.

  * 

  * @goal id-invalid-data-files

  */

 public class IdentifyInvalidDataFiles extends BaseMojo {

         private static final String FS = System.getProperty("file.separator");

         /**

          * @parameter expression="${extension}" default-value=".xml"

          * @required

          */

         private String extension;

         /**

          * @parameter expression="${dataDir}" default-value="${project.basedir}/src/main/impex"

          * @required

          */

         private File dataDir;

         /**

          * @parameter expression="${dataDirIncludes}" default-value="*.xml"

          */

         private String dataDirIncludes;

         /**

          * @parameter expression="${dataDirExcludes}" default-value="schema.xml"

          */

         private String dataDirExcludes;

         /**

          * @parameter expression="${schemaXMLFile}" default-value="src/main/impex/schema.xml"

          */

         private String schemaXMLFile;

         /**

          * @parameter expression="${targetDatabase}" default-value="oracle"

          */

         private String targetDatabase;

         @Override

         protected void executeMojo() throws MojoExecutionException, MojoFailureException {

                 Utils utils = new Utils();

                 try {

                         getLog().info("Examining " + dataDir.getAbsolutePath());

                         Database db = utils.getDatabase(schemaXMLFile, targetDatabase);

                         DirectoryScanner ds = getDirectoryScanner();

                         Set<File> existing = getExistingFiles(ds);

                         Set<File> allowed = getDatabaseFiles(db);

                         Set<File> invalid = SetUtils.difference(existing, allowed);

                         getLog().info(existing.size() + " data files currently exist");

                         getLog().info(invalid.size() + " of those are invalid");

                         StringBuilder sb = new StringBuilder();

                         int count = 0;

                         for (File file : invalid) {

                                 if (count != 0) {

                                         sb.append(",");

                                 }

                                 sb.append("**/src/main/impex/" + file.getName());

                                 getLog().info("Marked for removal: " + file.getName());

                                 count++;

                         }

                         Properties properties = getProject().getProperties();

                         properties.setProperty("impex.data.invalid", sb.toString());

                 } catch (Exception e) {

                         throw new MojoExecutionException("Error executing mojo", e);

                 }

         }

         protected Set<File> getDatabaseFiles(Database db) {

                 List<?> tables = db.getTables();

                 Set<File> files = new TreeSet<File>();

                 for (Object object : tables) {

                         Table table = (Table) object;

                         String tableName = table.getName();

                         String filename = dataDir.getAbsolutePath() + FS + tableName + extension;

                         File file = new File(filename);

                         files.add(file);

                 }

                 return files;

         }

         protected Set<File> getExistingFiles(DirectoryScanner ds) {

                 ds.scan();

                 String[] relativeFilenames = ds.getIncludedFiles();

                 Set<File> files = new TreeSet<File>();

                 for (int i = 0; i < relativeFilenames.length; i++) {

                         String filename = ds.getBasedir().getAbsolutePath() + FS + relativeFilenames[i];

                         File file = new File(filename);

                         files.add(file);

                 }

                 return files;

         }

         protected DirectoryScanner getDirectoryScanner() {

                 DirectoryScanner ds = new DirectoryScanner();

                 ds.setBasedir(dataDir);

                 ds.setIncludes(new String[] { dataDirIncludes });

                 ds.setExcludes(new String[] { dataDirExcludes });

                 return ds;

         }

         protected File getFile(Table table) {

                 String tableName = table.getName();

                 String filename = dataDir.getAbsolutePath() + FS + tableName + extension;

                 File file = new File(filename);

                 return file;

         }

         public File getDataDir() {

                 return dataDir;

         }

         public void setDataDir(File dataDir) {

                 this.dataDir = dataDir;

         }

         public String getDataDirIncludes() {

                 return dataDirIncludes;

         }

         public void setDataDirIncludes(String dataDirIncludes) {

                 this.dataDirIncludes = dataDirIncludes;

         }

         public String getDataDirExcludes() {

                 return dataDirExcludes;

         }

         public void setDataDirExcludes(String dataDirExcludes) {

                 this.dataDirExcludes = dataDirExcludes;

         }

         public String getSchemaXMLFile() {

                 return schemaXMLFile;

         }

         public void setSchemaXMLFile(String schemaXMLFile) {

                 this.schemaXMLFile = schemaXMLFile;

         }

         public String getExtension() {

                 return extension;

         }

         public void setExtension(String extension) {

                 this.extension = extension;

         }

         public String getTargetDatabase() {

                 return targetDatabase;

         }

         public void setTargetDatabase(String targetDatabase) {

                 this.targetDatabase = targetDatabase;

         }

 }