/*
 * The contents of this file are subject to the terms of the Common Development
 * and Distribution License (the License). You may not use this file except in
 * compliance with the License.
 *
 * You can obtain a copy of the License at http://www.netbeans.org/cddl.html
 * or http://www.netbeans.org/cddl.txt.
 *
 * When distributing Covered Code, include this CDDL Header Notice in each file
 * and include the License file at http://www.netbeans.org/cddl.txt.
 * If applicable, add the following below the CDDL Header, with the fields
 * enclosed by brackets [] replaced by your own identifying information:
 * "Portions Copyrighted [year] [name of copyright owner]"
 *
 * The Original Software is NetBeans. The Initial Developer of the Original
 * Software is Sun Microsystems, Inc. Portions Copyright 1997-2007 Sun
 * Microsystems, Inc. All Rights Reserved.
 */
package org.netbeans.modules.j2ee.common.ui;

import org.netbeans.modules.j2ee.common.source.*;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;
import javax.swing.SwingUtilities;
import org.netbeans.api.db.explorer.ConnectionManager;
import org.netbeans.api.db.explorer.DatabaseConnection;
import org.netbeans.api.db.explorer.JDBCDriver;
import org.netbeans.api.db.explorer.JDBCDriverManager;
import org.netbeans.api.project.Project;
import org.netbeans.modules.j2ee.deployment.common.api.ConfigurationException;
import org.netbeans.modules.j2ee.deployment.devmodules.spi.J2eeModuleProvider;
import org.netbeans.modules.j2ee.deployment.common.api.Datasource;

/**
 *
 * @author John Baker
 */
public class DatabaseConnectionResolver {
    private Set<Datasource> brokenDatasources = null;
    private final static JDBCDriver[] NULL_JDBC_DRIVER_ARRAY = new JDBCDriver[0];
    private boolean connStatus = false;
    
    // show Add Connection Dialog to allow users to add a connection
    public boolean addDatabaseConnection(Datasource brokenDatasource, String datasourceName) {
        
        // if data source is not null then pass data source parameters to the Add Connection dialog
        if (brokenDatasource != null) {
            // next check to see if the driver has been added
            String driverName = brokenDatasource.getDriverClassName();
            final JDBCDriver matchingDriver = findMatchingDriver(driverName);
            final Datasource edtDS = brokenDatasource;
            
            // OK if matchingDriver is null, user can add driver in the Add Connection dialog
            SwingUtilities.invokeLater(new Runnable() {
                public void run() {
                    setConnStatus(ConnectionManager.getDefault().showAddConnectionDialogFromEventThread(matchingDriver, edtDS.getUrl(), edtDS.getUsername(), edtDS.getPassword()));
                }
            });
        } else {
            // user to add a driver from the Add Connection dialog if not already registered
            SwingUtilities.invokeLater(new Runnable() {
                public void run() {
                    setConnStatus(ConnectionManager.getDefault().showAddConnectionDialogFromEventThread(null));
                }
            });
        }
        
        // setConnStatus sets instance variable, connStatus
        return connStatus;
    }
    
    // Set the connection status result of adding a connection
    private synchronized void setConnStatus(DatabaseConnection dbConn) {
        if (dbConn == null) {
            connStatus = false;
        } else {
            connStatus = true;
        }
    }
    
    // get project data sources
    public Set<Datasource> getBrokenDataSources(Project project) {
        J2eeModuleProvider jmp =
                (J2eeModuleProvider)project.getLookup().lookup(J2eeModuleProvider.class);
        
        Set<Datasource> dss = null;
        try {
            dss = jmp.getModuleDatasources();
        } catch (ConfigurationException e) {
            dss = new HashSet<Datasource>();
            // TODO: give some feedback to the user
        }
        
        brokenDatasources = new HashSet<Datasource>();
        
        Iterator it = dss.iterator();
        while (it.hasNext()) {
            Datasource ds = (Datasource)it.next();
            if(!isFound(ds)){
                brokenDatasources.add(ds);
            }
        }
        return brokenDatasources;
    }
    
     // get a specific broken project data source
    public Datasource getBrokenDataSource(Project project, String item) {
        Set <Datasource> dss = getBrokenDataSources(project);
        Iterator it = dss.iterator();
        Datasource brokenDs = null;
                
        while (it.hasNext()) {
            brokenDs = (Datasource)it.next();
            String name = brokenDs.getJndiName();
            name = name.substring(name.indexOf("/")+1);
            if (name.equals(item))
                break;
        }
        
        return brokenDs;
    }
    
    private boolean isFound(Datasource ds) {
        boolean found = false;
        String url = ds.getUrl();
        String username = ds.getUsername();
        DatabaseConnection[] dbConns = ConnectionManager.getDefault().getConnections();
        for(int i=0; i<dbConns.length; i++ ){
            DatabaseConnection dbCon = dbConns[i];
            String url1 = dbCon.getDatabaseURL();
            String username1 = dbCon.getUser();
            if (matchURL(url, url1, true) && matchString(username, username1, false)) {
                found = true;
            }
        }
        return found;
    }
    
    /**
     * The strings match if both or null
     */
    private boolean matchString(String str1, String str2, boolean ignoreCase) {
        if ((str1 != null) && (str2 != null)) {
            if (ignoreCase) {
                return str1.trim().equalsIgnoreCase(str2.trim());
            } else {
                return str1.trim().equals(str2.trim());
            }
        } else if ((str1 == null) && (str2 == null)) {
            return true;
        } else {
            return false;
        }
    }
    
    /**
     * Separate match method for URLs due to url problems when jdbc url doesn't contain a port #
     *         jbaker hack around for 6481339/6494241
     */
    private boolean matchURL(String jdbcResourceUrl, String dsInfoUrl, boolean ignoreCase) {
        if (ignoreCase){
            jdbcResourceUrl = jdbcResourceUrl.toLowerCase();
            dsInfoUrl = dsInfoUrl.toLowerCase();
        }
        if (jdbcResourceUrl.equals(dsInfoUrl)){
            return true;
        }
        
        if (jdbcResourceUrl.contains("derby")) {
            String newJdbcResourceUrl = jdbcResourceUrl.substring(0, jdbcResourceUrl.lastIndexOf(":")) + jdbcResourceUrl.substring(jdbcResourceUrl.lastIndexOf("/"));
            if (newJdbcResourceUrl.equals(dsInfoUrl)){
                return true;
            }
        }
        
        int nextIndex = 0;
        if ((jdbcResourceUrl != null) && (dsInfoUrl != null)){
            char[] jdbcResourceUrlChars = jdbcResourceUrl.toCharArray();
            char[] dsInfoUrlChars = dsInfoUrl.toCharArray();
            for(int i = 0; i < jdbcResourceUrlChars.length - 1; i++){
                if ((jdbcResourceUrlChars[i] != dsInfoUrlChars[i]) && jdbcResourceUrlChars[i] == ':'){
                    nextIndex = 1;
                }else if(jdbcResourceUrlChars[i + nextIndex] != dsInfoUrlChars[i]){
                    return false;
                }
            }
        }
        return true;
    }
    
    // Find a matching driver registered with the IDE
    // public: to-be-used outside this class
    public JDBCDriver findMatchingDriver(String driverClass) {
        int i = 0;
        
        JDBCDriver[] newDrivers;
        List theDriver = new ArrayList();
        newDrivers = JDBCDriverManager.getDefault().getDrivers();
        boolean found = false;
        
        if (newDrivers.length == 0)
            newDrivers = (JDBCDriver[]) theDriver.toArray(NULL_JDBC_DRIVER_ARRAY);
        
        if (newDrivers != null) {
            for (i=0; i<newDrivers.length; i++) {
                if (newDrivers[i].getClassName().equals(driverClass)) {
                    found = true;
                    break;
                }
            }
        }
        
        if (!found)  {
            i = 0;
            return null;
        } else
            return newDrivers[i];
    }
}