org.datacleaner.widgets.ResourceSelector Maven / Gradle / Ivy

Go to download
/**
 * DataCleaner (community edition)
 * Copyright (C) 2014 Neopost - Customer Information Management
 *
 * 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;

import java.awt.FlowLayout;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.swing.Box;
import javax.swing.JComponent;
import javax.swing.filechooser.FileFilter;

import org.apache.metamodel.util.Resource;
import org.datacleaner.configuration.DataCleanerConfiguration;
import org.datacleaner.panels.DCPanel;
import org.datacleaner.user.UserPreferences;
import org.datacleaner.util.convert.ResourceConverter;
import org.datacleaner.util.convert.ResourceConverter.ResourceStructure;
import org.datacleaner.util.convert.ResourceConverter.ResourceTypeHandler;

/**
 * A widget which allows the user to select/enter a {@link Resource} to use for
 * some file-like operation.
 */
public class ResourceSelector extends DCPanel implements ResourceTypePresenter {

    private static final long serialVersionUID = 1L;

    private final DCComboBox _resourceTypeComboBox;
    private final ResourceConverter _resourceConverter;
    private final UserPreferences _userPreferences;
    private final Map> _resourceTypePresenters;
    private boolean _openMode;
    private ResourceTypePresenter _currentPresenter;

    public ResourceSelector(DataCleanerConfiguration configuration, UserPreferences userPreferences, boolean openMode) {
        this(new ResourceConverter(configuration), userPreferences, openMode);
    }

    public ResourceSelector(ResourceConverter resourceConverter, UserPreferences userPreferences, boolean openMode) {
        _resourceConverter = resourceConverter;
        _userPreferences = userPreferences;
        _openMode = openMode;
        _resourceTypePresenters = new HashMap<>();

        final List schemes = new ArrayList<>();
        final Collection> resourceTypeHandlers = resourceConverter.getResourceTypeHandlers();
        for (ResourceTypeHandler resourceTypeHandler : resourceTypeHandlers) {
            final String scheme = resourceTypeHandler.getScheme();
            schemes.add(scheme);

            final ResourceTypePresenter presenter = createResourceTypePresenter(scheme, resourceTypeHandler);
            _resourceTypePresenters.put(scheme, presenter);
        }

        _resourceTypeComboBox = new DCComboBox<>(schemes);
        _resourceTypeComboBox.addListener(new DCComboBox.Listener() {
            @Override
            public void onItemSelected(String item) {
                onSchemeSelected(item);
            }
        });

        setScheme("file");

        setLayout(new FlowLayout(Alignment.LEFT.getFlowLayoutAlignment(), 0, 0));
        add(_resourceTypeComboBox);
        add(Box.createHorizontalStrut(4));
        for (ResourceTypePresenter presenter : getResourceTypePresenters()) {
            add(presenter.getWidget());
        }
    }

    protected ResourceTypePresenter createResourceTypePresenter(String scheme,
            ResourceTypeHandler resourceTypeHandler) {
        final ResourceTypePresenter presenter;
        switch (scheme) {
        case "file":
            presenter = new FilenameTextField(_userPreferences.getConfiguredFileDirectory(), _openMode);
            break;
        case "hdfs":
            presenter = new HdfsResourceTypePresenter();
            break;
        default:
            presenter = new TextFieldResourceTypePresenter(resourceTypeHandler);
            break;
        }
        return presenter;
    }

    public void setScheme(String scheme) {
        _resourceTypeComboBox.setSelectedItem(scheme);
    }

    private void onSchemeSelected(String item) {
        final Collection> presenters = _resourceTypePresenters.values();
        for (ResourceTypePresenter presenter : presenters) {
            presenter.getWidget().setVisible(false);
        }
        final ResourceTypePresenter presenter = _resourceTypePresenters.get(item);
        presenter.getWidget().setVisible(true);
        updateUI();
        _currentPresenter = presenter;
    }

    public void setResource(Resource resource) {
        final String stringRepresentation = _resourceConverter.toString(resource);
        final ResourceStructure structure = _resourceConverter.parseStructure(stringRepresentation);
        final String scheme = structure.getScheme();
        setScheme(scheme);

        // we need to do this ugly cast in order to call setResource(...) with a
        // generic argument
        @SuppressWarnings("unchecked")
        final ResourceTypePresenter presenter = (ResourceTypePresenter) _currentPresenter;
        presenter.setResource(resource);
    }

    public Resource getResource() {
        if (_currentPresenter == null) {
            return null;
        }
        return _currentPresenter.getResource();
    }

    public ResourceTypePresenter getResourceTypePresenter(String scheme) {
        return _resourceTypePresenters.get(scheme);
    }

    public Collection> getResourceTypePresenters() {
        return _resourceTypePresenters.values();
    }

    @Override
    public JComponent getWidget() {
        return this;
    }

    @Override
    public void addListener(Listener listener) {
        final Collection> presenters = getResourceTypePresenters();
        for (ResourceTypePresenter presenter : presenters) {
            presenter.addListener(listener);
        }
    }

    @Override
    public void removeListener(Listener listener) {
        final Collection> presenters = getResourceTypePresenters();
        for (ResourceTypePresenter presenter : presenters) {
            presenter.removeListener(listener);
        }
    }

    @Override
    public void addChoosableFileFilter(FileFilter fileFilter) {
        final Collection> presenters = getResourceTypePresenters();
        for (ResourceTypePresenter presenter : presenters) {
            presenter.addChoosableFileFilter(fileFilter);
        }
    }

    @Override
    public void removeChoosableFileFilter(FileFilter fileFilter) {
        final Collection> presenters = getResourceTypePresenters();
        for (ResourceTypePresenter presenter : presenters) {
            presenter.removeChoosableFileFilter(fileFilter);
        }
    }

    @Override
    public void setSelectedFileFilter(FileFilter fileFilter) {
        final Collection> presenters = getResourceTypePresenters();
        for (ResourceTypePresenter presenter : presenters) {
            presenter.setSelectedFileFilter(fileFilter);
        }
    }

    public void setResourcePath(String path) {
        final Resource resource = _resourceConverter.fromString(Resource.class, path);
        if (resource != null) {
            setResource(resource);
        }
    }

    public String getResourcePath() {
        final Resource resource = getResource();
        if (resource == null) {
            return null;
        }
        return _resourceConverter.toString(resource);
    }

    public void setCurrentPresenter(ResourceTypePresenter currentPresenter) {
        _currentPresenter = currentPresenter;
    }

    public void setVisibleResourceTypeCombox(boolean isVisible) {
        _resourceTypeComboBox.setVisible(isVisible);
    }

}