/*

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

 import org.apache.velocity.VelocityContext;

 import java.sql.*;

 import java.util.ArrayList;

 import java.util.List;

 import java.io.BufferedWriter;

 /**

  * User: mkrishna

  * Date: Nov 18, 2004

  */

 public class InsertGenerator extends VelocityGenerator {

     private List rows = new ArrayList();

     public void generate(String catalog, String schema, String object,

                          String vendor, Connection connection, DatabaseMetaData metadata, BufferedWriter writer)

             throws Exception {

         rows.clear();

         if (metadata.storesUpperCaseIdentifiers()) {

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

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

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

         } else if (metadata.storesLowerCaseIdentifiers()) {

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

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

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

         }

         super.generate(catalog,schema,object,vendor,connection,metadata,writer);

     }

     protected VelocityContext getContext(String catalog, String schema, String object,

                                          Connection connection, DatabaseMetaData metadata)

             throws Exception {

         VelocityContext context = new VelocityContext();

         ResultSet resultSet = null;

         Statement statement = null;

         try {

             statement = connection.createStatement();

             statement.execute("select * from " + object + " order by 1");

             resultSet = statement.getResultSet();

             while (resultSet.next()) {

                 rows.add(getRow(resultSet));

             }

             context.put("tableName", object);

             context.put("rows", rows);

         } catch (Exception e) {

             e.printStackTrace();

         } finally {

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

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

         }

         return context;

     }

     private List<String> getRow(ResultSet resultSet) throws SQLException {

         List<String> table = new ArrayList<String>();

         ResultSetMetaData metaData = resultSet.getMetaData();

         int columnCount = metaData.getColumnCount();

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

             table.add(getFormattedString(resultSet.getString(i), metaData.getColumnType(i)));

         }

         return table;

     }

     protected String getFormattedString(String unformattedString, int sqlType) {

         if (Types.DATE == sqlType) {

             return "to_timestamp ('" + unformattedString + "', 'yyyy-mm-dd HH24:MI:SS.FF')";

         } else if (unformattedString == null) {

             return "null";

         }

         return "'" + unformattedString.replaceAll("'", "''") + "'";

     }

     protected List getRows() {

         return rows;

     }

 }