//

 // Copyright (c) 2004, International Decision Systems

 // all rights reserved

 /*

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

 import net.sourceforge.addam.impexp.csv.CSVPrinter;

 import net.sourceforge.addam.util.DatabaseMetaDataUtil;

 import net.sourceforge.addam.util.TableUtil;

 import org.apache.tools.ant.BuildException;

 import org.apache.tools.ant.taskdefs.JDBCTask;

 import java.io.File;

 import java.io.FileWriter;

 import java.io.IOException;

 import java.io.Writer;

 import java.sql.*;

 import java.util.ArrayList;

 import java.util.Iterator;

 import java.util.List;

 public class ExportDataTask extends JDBCTask {

     private final List<Tables> includes = new ArrayList<Tables>();

     private final List<Tables> excludes = new ArrayList<Tables>();

     protected File dir;

     private String schema;

     private String catalog;

     /**

      * Selects all data from the named tables and extracts to the named files.

      *

      * @throws org.apache.tools.ant.BuildException

      *          if anything goes wrong

      */

     public void execute() throws BuildException {

         // run for each table

         for (String tableName : getTableNames()) {

             Connection connection = null;

             try {

                 connection = this.getConnection();

                 this.exportTable(tableName, connection);

             } finally {

                 if (connection != null)

                     try {

                         connection.close();

                     } catch (SQLException e) {

                         // ignore

                     }

             }

         }

     }

     /**

      * Selects all data from the named table and extracts to the named file.

      *

      * @throws org.apache.tools.ant.BuildException

      *          if anything goes wrong

      */

     public void exportTable(String tableName, Connection connection) throws BuildException {

         File toFile = new File(dir, TableUtil.getFileName(tableName));

         log("extracting " + tableName + " to file " + toFile);

         int count = 0;

         Writer writer = null;

         try {

             writer = new FileWriter(toFile);

             CSVPrinter printer = new CSVPrinter(writer);

             Statement stmt = connection.createStatement();

             ResultSet rs = stmt.executeQuery("SELECT * FROM " + tableName + " ORDER BY 1");

             ResultSetMetaData rsmd = rs.getMetaData();

             int numCols = rsmd.getColumnCount();

             String line[] = new String[numCols];

             // output column names

             for (int i = 1; i <= numCols; i++) {

                 line[i - 1] = rsmd.getColumnName(i);

             }

             printer.writeRecord(line);

             // output data

             while (rs.next()) {

                 for (int i = 1; i <= numCols; i++) {

                     String className = rsmd.getColumnClassName(i);

                     DataCodec codec = DataCodec.getCodec(className);

                     Object value;

                     if (className.equals("java.sql.Timestamp")) {

                         value = rs.getTimestamp(i);

                     } else if (className.equals("java.lang.String")) {

                         value = rs.getString(i);

                     } else {

                         value = rs.getObject(i);

                     }

                     String o = codec.encode(value);

                     line[i - 1] = (o == null) ? "" : o;

                 }

                 printer.writeRecord(line);

                 count++;

             }

         } catch (BuildException e) {

             throw e; // don't re-wrap BuildExceptions

         } catch (Exception e) {

             e.printStackTrace();

             throw new BuildException(e);

         } finally {

             try {

                 if (writer != null) {

                     writer.close();

                 }

             } catch (IOException e) {

                 log(e.toString());

             }

         }

         log(" ... " + count + " records");

     }

     /**

      * this is a shortcut for having an <includeTables> subelement by simply providing a filename as an attribute to this task

      *

      * @param tableFile a file containing an ordered list of tables

      */

     public void setIncludeFile(File tableFile) {

         Tables tables = new Tables();

         tables.setFile(tableFile);

         includes.add(tables);

     }

     /**

      * this is a shortcut for having a <excludeTables> subelement by simply providing a filename as an attribute to this task

      *

      * @param tableFile a file containing an ordered list of tables

      */

     public void setExcludeFile(File tableFile) {

         Tables tables = new Tables();

         tables.setFile(tableFile);

         excludes.add(tables);

     }

     /**

      * @param pattern a regexp pattern to compare table names to

      */

     public void setIncludesPattern(String pattern) {

         Tables tables = new Tables();

         tables.setPattern(pattern);

         includes.add(tables);

     }

     /**

      * @param pattern a regexp pattern to compare table names to

      */

     public void setExcludesPattern(String pattern) {

         Tables tables = new Tables();

         tables.setPattern(pattern);

         excludes.add(tables);

     }

     /**

      * @param names a comma separated list of table names

      */

     public void setIncludes(String names) {

         Tables tables = new Tables();

         tables.setNames(names);

         includes.add(tables);

     }

     /**

      * @param names a comma separated list of table names

      */

     public void setExcludes(String names) {

         Tables tables = new Tables();

         tables.setNames(names);

         excludes.add(tables);

     }

     /**

      * @param tables a set of tables to include or exclude

      */

     public void addIncludeTables(Tables tables) {

         includes.add(tables);

     }

     /**

      * @param tables a set of tables to include or exclude

      */

     public void addExcludeTables(Tables tables) {

         excludes.add(tables);

     }

     /**

      * @param tableDir the directory to write the files into

      */

     public void setDir(File tableDir) {

         dir = tableDir;

     }

     public void setSchema(String schema) {

         this.schema = schema;

     }

     public void setCatalog(String catalog) {

         this.catalog = catalog;

     }

     /**

      * returns all Tables included by not excluded

      *

      * @return List of Strings each naming a table

      */

     protected List<String> getTableNames() {

         List<String> tableNamesFromDB;

         Connection connection = null;

         try {

             connection = this.getConnection();

             tableNamesFromDB = DatabaseMetaDataUtil.getTables(catalog,schema,connection);

         } catch (SQLException e) {

             throw new BuildException(e);

         } finally {

             if (connection != null)

                 try {

                     connection.close();

                 } catch (SQLException e) {

                     // ignore

                 }

         }

         return Tables.narrow(tableNamesFromDB,includes,excludes);

     }

 }