/*
    * Copyright (c) 2004 International Decision Systems, Inc.  All Rights Reserved.
    *
    * By using this Software, You acknowledge that the Software is a valuable asset
    * and trade secret of either International Decision Systems, Inc. ("IDSI") or a
    * third party supplier of IDSI and constitutes confidential and proprietary
    * information.
    *
    * NEITHER IDSI NOR ANY AGENT OR PERSON ACTING FOR OR WITH IDSI HAS MADE OR DOES
   * MAKE ANY STATEMENTS, AFFIRMATIONS, REPRESENTATIONS OR WARRANTIES WHATSOEVER
   * TO YOU, WHETHER EXPRESS OR IMPLIED, AS TO THE SOFTWARE, THE QUALITY OR
   * CONDITION OF THE SOFTWARE, OR THE OPERATING CHARACTERISTICS OR RELIABILITY OF
   * THE SOFTWARE, OR ITS SUITABILITY FOR ANY GENERAL OR PARTICULAR PURPOSE, OR AS
   * TO ANY OTHER MATTER WHATSOEVER; ANY AND ALL OTHER WARRANTIES INCLUDING
   * WITHOUT LIMITATION ANY WARRANTIES IMPLIED BY LAW, SUCH AS THE IMPLIED
   * WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND TITLE,
   * USE AND NON-INFRINGEMENT; ARE HEREBY EXPRESSLY DISCLAIMED AND EXCLUDED.
  */
  package net.sourceforge.addam.ddlgen;
  
  import org.apache.tools.ant.BuildException;
  
  import java.io.*;
  import java.util.ArrayList;
  import java.util.List;
  import java.util.ResourceBundle;
  import java.util.regex.Pattern;
  
  /**
   * Used to define what the output files written by a generator, allows include/exclude for specific
   * objects, and can generate a master script based on the objects generated.
   * Date: Nov 11, 2004
   */
  public class GeneratorSpec implements Filter {
  
      public void setType(String type) {
          this.type = type;
      }
  
      public void setVendor(String vendor) {
          this.vendor = vendor;
      }
  
      public void setMasterScript(String file) {
          this.masterScript = file;
      }
  
      public void setPrefix(String prefix) {
          if (prefix == null) prefix = "";
          this.prefix = prefix;
      }
  
      public void setExtension(String extension) {
          if (extension == null) extension = "";
          this.extension = extension;
      }
  
      public void setIncludes(String includes) {
          String[] includeArray = includes.split(",");
          for (String anIncludeArray : includeArray) {
              Table table = new Table();
              table.setName(anIncludeArray);
              addConfiguredInclude(table);
          }
      }
  
      public void setIncludesPattern(String pattern) {
          includesPattern = Pattern.compile(pattern);
      }
  
      public void setIncludesFile(File includesFile) {
          try {
              BufferedReader reader = new BufferedReader(new FileReader(includesFile));
              for (String line = reader.readLine(); line != null; line = reader.readLine()) {
                  Table table = new Table();
                  table.setName(line);
                  addConfiguredInclude(table);
              }
          } catch (FileNotFoundException e) {
              throw new BuildException("file not found:" + includesFile.getAbsolutePath());
          } catch (IOException e) {
              throw new BuildException("error reading file:" + includesFile.getAbsolutePath());
          }
      }
  
      public void setExcludes(String excludes) {
          String[] excludeArray = excludes.split(",");
          for (String anExcludeArray : excludeArray) {
              Table table = new Table();
              table.setName(anExcludeArray);
              addConfiguredExclude(table);
          }
      }
  
      public void setExcludesPattern(String pattern) {
          excludesPattern = Pattern.compile(pattern);
      }
  
      public void setExcludesFile(File excludesFile) {
         try {
             BufferedReader reader = new BufferedReader(new FileReader(excludesFile));
             for (String line = reader.readLine(); line != null;) {
                 Table table = new Table();
                 table.setName(line);
                 addConfiguredInclude(table);
             }
         } catch (FileNotFoundException e) {
             throw new BuildException("file not found:" + excludesFile.getAbsolutePath());
         } catch (IOException e) {
             throw new BuildException("error reading file:" + excludesFile.getAbsolutePath());
         }
     }
 
     public void setDir(String dir) {
         this.dir = dir;
     }
 
     public void addConfiguredInclude(Table table) {
         includedTables.add(table.getName().trim());
     }
 
     public void addConfiguredExclude(Table table) {
         excludedTables.add(table.getName().trim());
     }
 
     public boolean includes(String objectName) {
         boolean includes = true;
         boolean excludes = false;
         String normalizedTableName = objectName.trim();
         if (!includedTables.isEmpty()) {
             includes = includedTables.contains(normalizedTableName);
         }
         if (!includes && includesPattern != null) {
             includes = includesPattern.matcher(objectName).matches();
         }
         if (!excludedTables.isEmpty()) {
             excludes = excludedTables.contains(normalizedTableName);
         }
         if (!excludes && excludesPattern != null) {
             excludes = excludesPattern.matcher(objectName).matches();
         }
         return includes && !excludes;
     }
 
     public String getVendor() {
         return vendor;
     }
 
     public Generator getGenerator() throws Exception {
         String classname;
         try {
             ResourceBundle bundle = ResourceBundle.getBundle("net.sourceforge.addam.ddlgen.RegisteredGenerators");
             classname = bundle.getString(type.toLowerCase());
         } catch (Exception e) {
             throw new BuildException("invalid type: " + type);
         }
         return (Generator)Class.forName(classname).newInstance();
     }
 
     public FileWriter getMasterScriptWriter() throws IOException {
         File scriptDir = new File(dir);
         if (masterScriptWriter == null && masterScript != null) {
             if (!scriptDir.exists()) {
                 scriptDir.mkdirs();
             }
             masterScriptWriter = new FileWriter(new File(scriptDir,masterScript));
         }
         return masterScriptWriter;
     }
 
     public String getFileName(String tableName) {
         // prepend a . on the suffix if the extension doesn't start with one
         String suffix = extension;
         if (suffix.length() > 0 && !suffix.startsWith(".")) {
             suffix = "." + suffix;
         }
         if (tableName == null) tableName = "";
         return prefix + tableName + suffix;
     }
 
     public File getFile(String tableName) {
         return new File(dir, getFileName(tableName));
     }
 
     public FileWriter getFileWriter(String object) throws IOException {
         // prepend a . on the suffix if the extension doesn't start with one
         String suffix = extension;
         if (suffix.length() > 0 && !suffix.startsWith(".")) {
             suffix = "." + suffix;
         }
         File file = new File(dir, prefix + object + suffix);
         File dir = file.getParentFile();
         if (!dir.exists()) dir.mkdirs();
         return new FileWriter(file);
     }
 
     private String dir;
     private String type;
     private String prefix = "";
     private String extension = "";
     private String vendor;
     private String masterScript;
     private FileWriter masterScriptWriter = null;
     private final List<String> includedTables = new ArrayList<String>();
     private final List<String> excludedTables = new ArrayList<String>();
     private Pattern includesPattern;
     private Pattern excludesPattern;
 }