/*

  * 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 org.apache.tools.ant.taskdefs.JDBCTask;

 import java.io.*;

 import java.sql.Connection;

 import java.sql.DatabaseMetaData;

 import java.sql.ResultSet;

 import java.sql.SQLException;

 import java.util.ArrayList;

 import java.util.List;

 /**

  * User: mkrishna

  * Date: Nov 10, 2004

  */

 public class GenerateTask extends JDBCTask {

     private String schema = null;

     private String catalog = null;

     private final List<GeneratorSpec> generatorSpecs = new ArrayList<GeneratorSpec>();

     public void setSchema(String schema) {

         this.schema = schema;

     }

     public void setCatalog(String catalog) {

         this.catalog = catalog;

     }

     public void addGenerator(GeneratorSpec spec) {

         generatorSpecs.add(spec);

     }

     public void execute() throws BuildException {

         Exception exception = null;

         ResultSet tablesRS = null;

         Connection connection = null;

         try {

             connection = getConnection();

             DatabaseMetaData dbmd = connection.getMetaData();

             if (dbmd.storesUpperCaseIdentifiers()) {

                 catalog = (catalog == null) ? null : catalog.toUpperCase();

                 schema = (schema == null) ? null : schema.toUpperCase();

             } else if (dbmd.storesLowerCaseIdentifiers()) {

                 catalog = (catalog == null) ? null : catalog.toLowerCase();

                 schema = (schema == null) ? null : schema.toLowerCase();

             }

             for (GeneratorSpec spec : generatorSpecs) {

                 Generator generator = spec.getGenerator();

                 List<String> objects = generator.getObjects(catalog,schema,spec,connection,dbmd);

                 for (String dbObject: objects) {

                     if (spec.includes(dbObject)) {

                         String vendor = spec.getVendor();

                         log("generating " + dbObject + " with " + generator.getClass().getName());

                         BufferedWriter writer = null;

                         Writer masterScriptWriter = spec.getMasterScriptWriter();

                         BufferedWriter bufferedMSW = null;

                         if (masterScriptWriter != null) {

                             bufferedMSW = new BufferedWriter(masterScriptWriter);

                         }

                         try {

                             writer = new BufferedWriter(spec.getFileWriter(dbObject));

                             generator.generate(catalog, schema, dbObject, vendor, connection, dbmd, writer);

                             if (bufferedMSW != null) {

                                 bufferedMSW.write(spec.getFileName(dbObject));

                                 bufferedMSW.newLine();

                                 bufferedMSW.flush();

                             }

                         } finally {

                             if (writer != null) writer.close();

                         }

                     }

                 }

             }

         } catch (Exception e) {

             exception = e;

         } finally {

             try {

                 for (GeneratorSpec spec : generatorSpecs) {

                     Writer writer = spec.getMasterScriptWriter();

                     if (writer != null) writer.close();

                 }

                 if (tablesRS != null) tablesRS.close();

                 if (connection != null) connection.close();

             } catch (SQLException e) {

                 exception = e;

             } catch (IOException e) {

                 exception = e;

             }

         }

         if (exception != null) throw new BuildException(exception);

     }

 }