/*

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

 }