All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • jdplus.sql.desktop.plugin.jdbc.DbExplorerUtil Maven / Gradle / Ivy

    The newest version!
    /*
 * Copyright 2013 National Bank of Belgium
 * 
 * Licensed under the EUPL, Version 1.1 or - as soon they will be approved 
 * by the European Commission - subsequent versions of the EUPL (the "Licence");
 * You may not use this work except in compliance with the Licence.
 * You may obtain a copy of the Licence at:
 * 
 * http://ec.europa.eu/idabc/eupl
 * 
 * Unless required by applicable law or agreed to in writing, software 
 * distributed under the Licence is distributed on an "AS IS" basis,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the Licence for the specific language governing permissions and 
 * limitations under the Licence.
 */
package jdplus.sql.desktop.plugin.jdbc;

import java.util.Optional;
import java.util.Properties;
import lombok.NonNull;
import org.netbeans.api.db.explorer.ConnectionManager;
import org.netbeans.api.db.explorer.DatabaseConnection;
import org.netbeans.api.db.explorer.DatabaseException;
import org.netbeans.api.db.explorer.JDBCDriver;
import org.netbeans.api.db.explorer.JDBCDriverManager;
import org.openide.DialogDescriptor;
import org.openide.DialogDisplayer;
import org.openide.nodes.Node;
import org.openide.util.Exceptions;
import org.openide.util.Lookup;

/**
 * http://code.metager.de/source/xref/netbeans/db/src/org/netbeans/modules/db/resources/mf-layer.xml
 *
 * @author Philippe Charles
 */
public final class DbExplorerUtil {

    private DbExplorerUtil() {
        // static class
    }

    static boolean isConnected(@NonNull DatabaseConnection conn) {
        return conn.getJDBCConnection() != null;
    }

    static boolean isTableOrView(@NonNull Node node) {
        return lookupContains(node.getLookup(),
                getClassName("TableNode"),
                getClassName("ViewNode"));
    }

    private static String getClassName(String name) {
        return "org.netbeans.modules.db.explorer.node." + name;
    }

    // some part of the db api is private; we need to cheat a bit
    private static boolean lookupContains(Lookup lookup, String... classNames) {
        for (Lookup.Item o : lookup.lookupResult(Object.class).allItems()) {
            for (String className : classNames) {
                if (o.getType().getName().equals(className)) {
                    return true;
                }
            }
        }
        return false;
    }

    @NonNull
    static Optional findConnection(@NonNull Node node) {
        DatabaseConnection result = null;
        Node current = node;
        while (current != null && (result = current.getLookup().lookup(DatabaseConnection.class)) == null) {
            current = current.getParentNode();
        }
        return Optional.ofNullable(result);
    }

    @NonNull
    static Optional getDriverByClass(@NonNull String driverClass) {
        for (JDBCDriver o : JDBCDriverManager.getDefault().getDrivers(driverClass)) {
            return Optional.of(o);
        }
        return Optional.empty();
    }

    @NonNull
    static Optional getConnectionByDisplayName(@NonNull String displayName) {
        for (DatabaseConnection o : ConnectionManager.getDefault().getConnections()) {
            if (o.getDisplayName().equals(displayName)) {
                return Optional.of(o);
            }
        }
        return Optional.empty();
    }

    public static void importConnection(@NonNull DriverBasedConfig config) {
        Optional driver = getDriverByClass(config.getDriverClass());
        if (!driver.isPresent()) {
            String msg = "Cannot find driver '" + config.getDriverClass() + "'";
            DialogDisplayer.getDefault().notify(new DialogDescriptor.Message(msg, DialogDescriptor.ERROR_MESSAGE));
            return;
        }
        Optional conn = getConnectionByDisplayName(config.getDisplayName());
        if (conn.isPresent()) {
            String msg = "A connection with the same name already exist: '" + config.getDisplayName() + "'";
            DialogDisplayer.getDefault().notify(new DialogDescriptor.Message(msg, DialogDescriptor.ERROR_MESSAGE));
            return;
        }
        Properties properties = new Properties();
        config.getParams().forEach(properties::put);
        DatabaseConnection newConn = DatabaseConnection.create(driver.orElseThrow(), config.getDatabaseUrl(), "", config.getSchema(), "", false, config.getDisplayName(), properties);
        try {
            ConnectionManager.getDefault().addConnection(newConn);
        } catch (DatabaseException ex) {
            Exceptions.printStackTrace(ex);
        }
    }

    @NonNull
    public static DriverBasedConfig exportConnection(@NonNull DatabaseConnection conn) {
        DriverBasedConfig.Builder b = DriverBasedConfig.builder(conn.getDriverClass(), conn.getDatabaseURL(), conn.getSchema(), conn.getDisplayName());
        Properties properties = conn.getConnectionProperties();
        for (String o : properties.stringPropertyNames()) {
            b.param(o, properties.getProperty(o));
        }
        return b.build();
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy