/*
    * Copyright 2004-2006 the Seasar Foundation and the Others.
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *     http://www.apache.org/licenses/LICENSE-2.0
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, 
   * either express or implied. See the License for the specific language
   * governing permissions and limitations under the License.
   */
  package org.seasar.tuigwaa.database;
  
  import java.io.UnsupportedEncodingException;
  import java.sql.Connection;
  import java.sql.SQLException;
  import java.text.ParseException;
  import java.text.SimpleDateFormat;
  
  import javax.sql.DataSource;
  
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.hibernate.util.StringHelper;
  import org.seasar.framework.container.factory.SingletonS2ContainerFactory;
  import org.seasar.tuigwaa.util.TgwResource;
  
  import com.isenshi.util.extlib.HSQLDBUtil;
  
  public class DatabaseUtils {
  
  	private static final String BACKUP_SQL_PREFIX = "SELECT * FROM ";
  
  	private static final String CLEAN_SQL_PREFIX = "DELETE FROM ";
  
  	private static final String RESTORE_SQL_PREFIX = "INSERT INTO ";
  
  	private static Log log_ = LogFactory.getLog(DatabaseUtils.class);
  	
  	private static final SimpleDateFormat SQL_TIMESTAMP_FORMAT = 
  		new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS");
  
  	private static final SimpleDateFormat SQL_DATA_FORMAT = 
  		new SimpleDateFormat("yyyy-MM-dd");	
  
  	public static String buildBackupQuery(String schema, String table) {
  		StringBuffer buf = new StringBuffer(BACKUP_SQL_PREFIX);
  		String fullTableName = (schema != null && schema.length() > 0) ? schema
  				+ "." + table
  				: table;
  		buf.append(fullTableName);
  		return buf.toString();
  	}
  
  	public static String buildCleanQuery(String schema, String table) {
  		StringBuffer buf = new StringBuffer(CLEAN_SQL_PREFIX);
  		String fullTableName = (schema != null && schema.length() > 0) ? schema
  				+ "." + table
  				: table;
  		buf.append(fullTableName);
  		return buf.toString();
  	}
  
  	public static String buildRestoreQuery(String schema, String table,
  			String[] columns) {
  		StringBuffer buf = new StringBuffer(RESTORE_SQL_PREFIX);
  		String fullTableName = (schema != null && schema.length() > 0) ? schema
  				+ "." + table
  				: table;
  		buf.append(fullTableName);
  
  		int size = 0;
  		if (columns != null && (size = columns.length) > 0) {
  			buf.append(" (");
  			buf.append(StringHelper.join(",", columns));
  			buf.append(" ) VALUES (");
  			for (int i = 0; i < size; i++) {
  				buf.append("?");
  				if (i != size - 1 && size != 1)
  					buf.append(",");
  			}
  			buf.append(")");
  		}
  		return buf.toString();
  	}
  
  	
  	public static void closeQueitly(Connection con){
  		if(con != null){
  			try {
  				con.close();
  			} catch (SQLException e) {
  				//do nothing
  			}
  		}
 	}
 	
 	public static String getJDBCURL() {
 		if (isEmbededHsql()) {
 			try {
 				return HSQLDBUtil.getJDBCURL();
 			} catch (UnsupportedEncodingException e) {
 				e.printStackTrace();
 				throw new RuntimeException(e);
 			}
 		} else {
 			return TgwResource.getProperty("database.url");
 		}
 	}
 
 	public static void shutdown() {
 		if (isEmbededHsql()) {
 			DataSource dataSource = (DataSource) SingletonS2ContainerFactory
 					.getContainer().getComponent("basedb.dataSource");
 			try {
 				HSQLDBUtil.shutdown(dataSource);
 			} catch (SQLException e) {
 				e.printStackTrace();
 			}
 		}
 	}
 
 	public static boolean isEscapeClauseNeeded(){ // temporary
 		
 		String driver = getBasedbDriver();
 		boolean flag = false;
 		if("org.hsqldb.jdbcDriver".equals(driver) ||
 			"com.ibm.db2.jcc.DB2Driver".equals(driver)){
 			flag = true;
 		}		
 		return flag;		
 	}
 	
 	private static boolean isEmbededHsql() {
 		return "org.hsqldb.jdbcDriver".equals(getBasedbDriver());
 	}	
 	
 	private static String getBasedbDriver(){
 		return TgwResource.getProperty("database.driver");
 	}
 	
 	public static Object[] bindArgs(String[] data, String[] type) {
 
 		if (data.length != type.length)
 			return null;
 
 		int size = type.length;
 		Object[] objs = new Object[size];
 
 		for (int i = 0; i < size; i++) {
 			String className = type[i];
 			Object o = null;
 
 			if (data[i] != null && !data[i].equals("")) {
 
 				if (className.equals("java.lang.Long")) {
 					o = new Long(data[i]);
 				} else if (className.equals("java.lang.Integer")) {
 					o = new Integer(data[i]);
 				} else if (className.equals("java.lang.Double")) {
 					o = new Double(data[i]);
 				} else if (className.equals("java.lang.String")) {
 					o = data[i];
 				} else if (className.equals("java.lang.Boolean")) {
 					o = new Boolean(data[i]);
 				} else if (className.equals("java.sql.Date")) {
 					try {
 						o = SQL_DATA_FORMAT.parse(data[i]);
 					} catch (ParseException pe) {
 						log_.error("failed to parse. " + pe.getMessage());
 					}
 				} else if (className.equals("java.sql.Timestamp")) {
 					try {
 						o = SQL_TIMESTAMP_FORMAT.parse(data[i]);
 					} catch (ParseException pe) {
 						log_.error("failed to parse. " + pe.getMessage());
 					}
 				} else {
 					log_.error("unknown types " + className);
 					// throws some exception...
 				}
 			}			
 			objs[i] = o;
 		}
 		return objs;
 	}	
 	
 }