/*
 *                 Sun Public License Notice
 *
 * The contents of this file are subject to the Sun Public License
 * Version 1.0 (the "License"). You may not use this file except in
 * compliance with the License. A copy of the License is available at
 * http://www.sun.com/
 *
 * The Original Code is NetBeans. The Initial Developer of the Original
 * Code is Sun Microsystems, Inc. Portions Copyright 1997-2004 Sun
 * Microsystems, Inc. All Rights Reserved.
 */

package org.netbeans.lib.ddl.impl;

import java.beans.*;
import java.sql.*;
import java.util.*;

import org.openide.*;

import org.netbeans.lib.ddl.*;
import org.netbeans.lib.ddl.adaptors.*;
import org.netbeans.lib.ddl.impl.*;

public class DriverSpecification {
    
    /** Used DBConnection */
    private HashMap desc;
    
    private String catalog, schema;
    
    private DatabaseMetaData dmd;
    
    private ResultSet rs;
    
    private String quoteString;
    
    /** Owned factory */
    SpecificationFactory factory;
    
    /** Constructor */
    public DriverSpecification(HashMap description) {
        desc = description;
        quoteString = null;
    }
    
    public DriverSpecificationFactory getDriverSpecificationFactory() {
        return factory;
    }
    
    public void setDriverSpecificationFactory(DriverSpecificationFactory fac) {
        factory = (SpecificationFactory) fac;
    }
    
    public void setCatalog(String catalog) {
        if (catalog == null || dmd == null) {
            this.catalog = catalog;
            return;
        } else
            catalog.trim();
        
        ResultSet rs;
        LinkedList list = new LinkedList();
        
        try {
            rs = dmd.getCatalogs();
            while (rs.next())
                list.add(rs.getString(1).trim());
            rs.close();
        } catch (SQLException exc) {
            ErrorManager.getDefault().notify(ErrorManager.INFORMATIONAL, exc);
            
//            this.catalog = catalog;
            this.catalog = null;  //hack for IBM ODBC driver
            rs = null;
            return;
        }
        
        if (list.contains(catalog))
            this.catalog = catalog;
        else
            this.catalog = null; //hack for Sybase ODBC driver
    }
    
    public void setSchema(String schema) {
        this.schema = schema;
    }
    
    public String getSchema() {
        return schema;
    }
    
    public void setMetaData(DatabaseMetaData dmd) {
        this.dmd = dmd;
    }
    
    public void getTables(String tableNamePattern, String[] types) {
        try {
            tableNamePattern = quoteString(tableNamePattern);
            rs = dmd.getTables(catalog, schema, tableNamePattern, types);
        } catch (SQLException exc) {
            ErrorManager.getDefault().notify(ErrorManager.INFORMATIONAL, exc);
            rs = null;
        }
    }
    
    public void getProcedures(String procedureNamePattern) {
        try {
            procedureNamePattern = quoteString(procedureNamePattern);
            rs = dmd.getProcedures(catalog, schema, procedureNamePattern);
        } catch (SQLException exc) {
            ErrorManager.getDefault().notify(ErrorManager.INFORMATIONAL, exc);
            rs = null;
        }
    }
    
    public void getPrimaryKeys(String table) {
        try {
            table = quoteString(table);
            rs = dmd.getPrimaryKeys(catalog, schema, table);
        } catch (SQLException exc) {
            ErrorManager.getDefault().notify(ErrorManager.INFORMATIONAL, exc);
            rs = null;
        }
    }
    
    public void getIndexInfo(String table, boolean unique, boolean approximate) {
        try {
            table = quoteString(table);
            rs = dmd.getIndexInfo(catalog, schema, table, unique, approximate);
        } catch (SQLException exc) {
            ErrorManager.getDefault().notify(ErrorManager.INFORMATIONAL, exc);
            rs = null;
        }
    }
    
    public void getColumns(String tableNamePattern, String columnNamePattern) {
        try {
            tableNamePattern = quoteString(tableNamePattern);
            columnNamePattern = quoteString(columnNamePattern);
            rs = dmd.getColumns(catalog, schema, tableNamePattern, columnNamePattern);
        } catch (SQLException exc) {
            ErrorManager.getDefault().notify(ErrorManager.INFORMATIONAL, exc);
            rs = null;
        }
    }
    
    public void getProcedureColumns(String procedureNamePattern, String columnNamePattern) {
        try {
            procedureNamePattern = quoteString(procedureNamePattern);
            columnNamePattern = quoteString(columnNamePattern);
            rs = dmd.getProcedureColumns(catalog, schema, procedureNamePattern, columnNamePattern);
        } catch (SQLException exc) {
            ErrorManager.getDefault().notify(ErrorManager.INFORMATIONAL, exc);
            rs = null;
        }
    }
    
    public void getExportedKeys(String table) {
        try {
            table = quoteString(table);
            rs = dmd.getExportedKeys(catalog, schema, table);
        } catch (SQLException exc) {
            ErrorManager.getDefault().notify(ErrorManager.INFORMATIONAL, exc);
            rs = null;
        }
    }
    
    public void getImportedKeys(String table) {
        try {
            table = quoteString(table);
            rs = dmd.getImportedKeys(catalog, schema, table);
        } catch (SQLException exc) {
            ErrorManager.getDefault().notify(ErrorManager.INFORMATIONAL, exc);
            rs = null;
        }
    }
    
    public ResultSet getResultSet() {
        return rs;
    }
    
    public HashMap getRow() {
        HashMap rset = new HashMap();
        Object value;
        
        try {
            int count = rs.getMetaData().getColumnCount();
            
            for (int i = 1; i <= count; i++) {
                value = null;
                try {
                    value = rs.getString(i);
//                    value = rs.getObject(i); //cannot use getObject() because of problems with MSSQL ODBC driver
                }  catch (SQLException exc) {
                    ErrorManager.getDefault().notify(ErrorManager.INFORMATIONAL, exc);
                    rset = null;
                    break;
                }
                rset.put(new Integer(i), value);
            }
        } catch (SQLException exc) {
            ErrorManager.getDefault().notify(ErrorManager.INFORMATIONAL, exc);
            rset = null;
        }
        
        return rset;
    }
    
    //another patches
    
    public boolean areViewsSupported() {
        
        boolean bViewsSupported = false;
		
		LinkedList list = new LinkedList();
         
        list.add("PointBase"); // NOI18N
        list.add("MySQL"); // NOI18N
        list.add("HypersonicSQL"); // NOI18N
        // list.add("InstantDB"); // NOI18N - isn't necessary in the list - getTables() returns empty result set for views
         
		try {
            
			String productName = dmd.getDatabaseProductName().trim();
			
			if (list.contains(productName)) {
				
				if (productName.equalsIgnoreCase("PointBase")) {

					bViewsSupported = ((dmd.getDriverMajorVersion() == 4 && dmd.getDriverMinorVersion() >= 1) || dmd.getDriverMajorVersion() > 4);
				
				} else if(productName.equalsIgnoreCase("MySQL")) 	{

					bViewsSupported = (dmd.getDriverMajorVersion() >= 5);

				}

			} else {

				bViewsSupported = true;
			
			}

		} catch(SQLException exc) {
			
			//PENDING
			bViewsSupported = true;

		} finally {

			return bViewsSupported;

		}
        
        /*boolean bViewsSupported = false;
        
        try {
            
            ResultSet tableTypes = dmd.getTableTypes();
            
            tableTypes.beforeFirst(); // isn'r really neccessary, only a habit...   :)
            
            while(tableTypes.next() && !bViewsSupported) {
                
                String strTableType = tableTypes.getString(1);
                
                bViewsSupported = strTableType.equalsIgnoreCase("VIEW");
            
            }
            
        } catch(SQLException e) {
            
            // error-handling code?
            bViewsSupported = false;
            
        } finally {
            
            return bViewsSupported;
            
        }*/
        
    }
    
    private String getQuoteString() {
        if (quoteString == null) {
            try {
                quoteString = dmd.getIdentifierQuoteString();
                if (quoteString == null || quoteString.equals(" ")) //NOI18N
                    quoteString = ""; //NOI18N
                else
                    quoteString.trim();
            } catch (SQLException exc) {
                quoteString = ""; //NOI18N
            }
        }
        
        return quoteString;
    }
    
    private String quoteString(String str) {
        try {
            if (dmd.getDatabaseProductName().trim().equals("PointBase")) { //NOI18N
                //hack for PointBase - DatabaseMetaData methods require quoted arguments for case sensitive identifiers
                String quoteStr = getQuoteString();
                if (str != null && !str.equals("%") && !quoteStr.equals("")) //NOI18N
                    str = quoteStr + str + quoteStr;
            }
        } catch (SQLException exc) {
            //PENDING
        }
        
        return str;
    }
    
    public String getDBName() {
        try {
            return dmd.getDatabaseProductName().trim();
        } catch (SQLException exc) {
            //PENDING
            return null;
        }
    }
}