// BSD License (http://www.galagosearch.org/license)

 package org.galagosearch.tupleflow.typebuilder;

 import java.io.File;

 import java.io.IOException;

 import org.antlr.runtime.RecognitionException;

 import org.apache.maven.plugin.AbstractMojo;

 import org.apache.maven.plugin.MojoExecutionException;

 /**

  * This is the Maven plugin that generates Type objects from galagotype specification

  * files.  See the build.xml file and the galagotype files in the galagosearch-core

  * project to see some examples of this.

  * 

  * @phase generate-sources

  * @goal typebuilder

  * @author trevor

  */

 public class TypeBuilderMojo extends AbstractMojo {  

     /**

      * @parameter expression="${basedir}/src/main/galagotype" default-value="y"

      */

     private String sourceDirectory;

     /**

      * @parameter expression="${basedir}" default-value="y"

      */

     private String baseDirectory;

     public String getOutputFilename(String packageName, String typeName) {

         String[] packagePathComponents = packageName.split("\\.");

         String packageStringPath = baseDirectory;

         if (baseDirectory == null)

             System.err.println("baseDir == null");

         packageStringPath += File.separator + "src" +

                              File.separator + "main" +

                              File.separator + "java";

         for (String component : packagePathComponents) {

             packageStringPath += File.separator + component;

         }

         // Create the filename:

         return packageStringPath + File.separator + typeName + ".java";

     }

     public void execute() throws MojoExecutionException {

         if (sourceDirectory == null) return;

         File[] files = new File(sourceDirectory).listFiles();

         if (files == null) return;

         for (File f : files) {

             if (f.isFile() && f.getName().endsWith("galagotype")) {

                 TypeSpecification spec = null;

                 java.io.FileWriter writer = null;

                 try {

                     spec = ParserDriver.getTypeSpecification(f.getAbsolutePath());

                 } catch (IOException ex) {

                     throw new MojoExecutionException("Couldn't open file: " + f.getAbsolutePath(), ex);

                 } catch (RecognitionException ex) {

                     throw new MojoExecutionException("Parsing failed: " + f.getAbsolutePath(), ex);

                 }

                 String outputFilename =

                         getOutputFilename(spec.getPackageName(), spec.getTypeName());

                 File outputFile = new File(outputFilename);

                 if (!outputFile.exists() || f.lastModified() > outputFile.lastModified()) {

                     System.err.println("Generating " + spec.getTypeName());

                     new File(outputFile.getParent()).mkdirs();

                     TemplateTypeBuilder builder = new TemplateTypeBuilder(spec);

                     try {

                         writer = new java.io.FileWriter(outputFilename);

                     } catch(IOException ex) {

                         throw new MojoExecutionException("Trouble creating " + outputFilename, ex);

                     }

                     String comment =

                             "// This file was automatically generated with the command: \n" +

                             "//     java org.galagosearch.tupleflow.typebuilder.TypeBuilderMojo ...\n";

                     try {

                         writer.write(comment);

                         writer.write(builder.toString());

                         writer.close();

                     } catch(IOException e) {

                         throw new MojoExecutionException("Trouble writing " + outputFilename);

                     }

                 }

             }

         }

     }

 }