org.datacleaner.widgets.properties.SchemaNamePropertyWidget Maven / Gradle / Ivy

Go to download
/**
 * DataCleaner (community edition)
 * Copyright (C) 2014 Free Software Foundation, Inc.
 *
 * This copyrighted material is made available to anyone wishing to use, modify,
 * copy, or redistribute it subject to the terms and conditions of the GNU
 * Lesser General Public License, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this distribution; if not, write to:
 * Free Software Foundation, Inc.
 * 51 Franklin Street, Fifth Floor
 * Boston, MA  02110-1301  USA
 */
package org.datacleaner.widgets.properties;

import javax.swing.DefaultComboBoxModel;

import org.apache.metamodel.schema.MutableSchema;
import org.apache.metamodel.schema.Schema;
import org.apache.metamodel.util.CollectionUtils;
import org.apache.metamodel.util.MutableRef;
import org.datacleaner.connection.Datastore;
import org.datacleaner.connection.DatastoreConnection;
import org.datacleaner.descriptors.ConfiguredPropertyDescriptor;
import org.datacleaner.job.builder.ComponentBuilder;
import org.datacleaner.widgets.DCComboBox;
import org.datacleaner.widgets.DCComboBox.Listener;
import org.datacleaner.widgets.SchemaStructureComboBoxListRenderer;

/**
 * Alternative String property widget, specifically built for components that
 * need a {@link Schema} name drop down.
 *
 * @author Kasper Sørensen
 */
public class SchemaNamePropertyWidget extends AbstractPropertyWidget {

    private final DCComboBox _comboBox;
    private final MutableRef _datastoreRef;

    public SchemaNamePropertyWidget(final ComponentBuilder componentBuilder,
            final ConfiguredPropertyDescriptor propertyDescriptor) {
        super(componentBuilder, propertyDescriptor);
        _comboBox = new DCComboBox<>();
        _comboBox.setRenderer(new SchemaStructureComboBoxListRenderer(false));
        _comboBox.setEditable(false);
        addComboListener(item -> fireValueChanged());
        add(_comboBox);
        _datastoreRef = new MutableRef<>();

        setValue(getCurrentValue());
    }

    public void addComboListener(final Listener listener) {
        _comboBox.addListener(listener);
    }

    public void setDatastore(final Datastore datastore) {
        final String previousValue = getValue();
        _datastoreRef.set(datastore);
        if (datastore == null) {
            _comboBox.setModel(new DefaultComboBoxModel<>(new Schema[1]));
        } else {
            try (DatastoreConnection con = datastore.openConnection()) {
                Schema[] schemas = con.getSchemaNavigator().getSchemas();
                schemas = CollectionUtils.array(new Schema[1], schemas);
                _comboBox.setModel(new DefaultComboBoxModel<>(schemas));
                Schema newValue = null;
                if (previousValue != null) {
                    newValue = con.getSchemaNavigator().getSchemaByName(previousValue);
                }
                if (newValue == null) {
                    newValue = con.getSchemaNavigator().getDefaultSchema();
                }
                _comboBox.setSelectedItem(newValue);
            }
        }
    }

    @Override
    public String getValue() {
        final Schema schema = getSchema();
        if (schema == null) {
            return null;
        }
        return schema.getName();
    }

    @Override
    protected void setValue(String value) {
        final Datastore datastore = _datastoreRef.get();
        if (value == null && datastore != null) {
            try (DatastoreConnection con = datastore.openConnection()) {
                value = con.getSchemaNavigator().getDefaultSchema().getName();
            }
        }

        if (getValue() == value) {
            return;
        }

        final Schema schema;
        if (value == null) {
            schema = null;
        } else if (datastore == null) {
            schema = new MutableSchema(value);
        } else {
            try (DatastoreConnection con = datastore.openConnection()) {
                schema = con.getSchemaNavigator().getSchemaByName(value);
            }
        }

        _comboBox.setEditable(true);
        _comboBox.setSelectedItem(schema);
        _comboBox.setEditable(false);
    }

    public Schema getSchema() {
        return _comboBox.getSelectedItem();
    }

    public void connectToTableNamePropertyWidget(final SingleTableNamePropertyWidget tableNamePropertyWidget) {
        addComboListener(item -> {
            // update the table name when schema is selected
            tableNamePropertyWidget.setSchema(_datastoreRef.get(), item);
        });
    }

}