AddDirectoryAction.java

/*
 * Copyright (C) 2014 Brockmann Consult GmbH (info@brockmann-consult.de)
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the Free
 * Software Foundation; either version 3 of the License, or (at your option)
 * any later version.
 * 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 General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, see http://www.gnu.org/licenses/
 */

package org.esa.snap.ui.product;

import com.bc.ceres.binding.ValidationException;
import com.jidesoft.swing.FolderChooser;
import org.esa.snap.core.util.PropertyMap;
import org.esa.snap.core.util.SystemUtils;
import org.esa.snap.ui.AppContext;
import org.esa.snap.ui.ModalDialog;

import javax.swing.AbstractAction;
import javax.swing.JFileChooser;
import java.awt.Window;
import java.awt.event.ActionEvent;
import java.io.File;

/**
 * @author Thomas Storm
 */
class AddDirectoryAction extends AbstractAction {

    private final String lastDirProperty;
    private boolean recursive;
    private AppContext appContext;
    private InputListModel listModel;
    private String defaultPattern;

    AddDirectoryAction(AppContext appContext, InputListModel listModel, boolean recursive, String lastDirProperty, String defaultPattern) {
        this(recursive, lastDirProperty);
        this.appContext = appContext;
        this.listModel = listModel;
        if (defaultPattern == null) {
            this.defaultPattern = recursive ? "*.dim" : "*";
        }
        this.defaultPattern = defaultPattern;
    }

    private AddDirectoryAction(boolean recursive, String lastDirProperty) {
        this("Add directory" + (recursive ? " recursively" : "(s)") + "...", lastDirProperty);
        this.recursive = recursive;
    }

    private AddDirectoryAction(String title, String lastDirProperty) {
        super(title);
        this.lastDirProperty = lastDirProperty;
    }

    @Override
    public void actionPerformed(ActionEvent e) {
        final FolderChooser folderChooser = new FolderChooser();

        final PropertyMap preferences = appContext.getPreferences();
        String lastDir = preferences.getPropertyString(lastDirProperty, SystemUtils.getUserHomeDir().getPath());
        if (lastDir != null) {
            folderChooser.setCurrentDirectory(new File(lastDir));
        }
        folderChooser.setMultiSelectionEnabled(!recursive);

        final Window parent = appContext.getApplicationWindow();

        final int result = folderChooser.showOpenDialog(parent);
        if (result != JFileChooser.APPROVE_OPTION) {
            return;
        }

        final FileSelectionPatternDialog dialog = new FileSelectionPatternDialog(defaultPattern, parent);
        if (dialog.show() != ModalDialog.ID_OK) {
            return;
        }
        final String pattern = dialog.getPattern();

        File[] selectedDirs;
        if (recursive) {
            File selectedDir = folderChooser.getSelectedFolder();
            lastDir = selectedDir.getAbsolutePath();
            selectedDir = new File(selectedDir, "**");
            selectedDir = new File(selectedDir, pattern);
            selectedDirs = new File[]{selectedDir};
        } else {
            final File[] selectedPaths = folderChooser.getSelectedFiles();
            if (selectedPaths.length > 0) {
                lastDir = selectedPaths[0].getAbsolutePath();
            }
            selectedDirs = new File[selectedPaths.length];
            for (int i = 0; i < selectedPaths.length; i++) {
                File selectedFile = selectedPaths[i];
                selectedDirs[i] = new File(selectedFile, pattern);
            }
        }
        preferences.setPropertyString(lastDirProperty, lastDir);
        try {
            listModel.addElements(selectedDirs);
        } catch (ValidationException ve) {
            // not expected to ever come here
            appContext.handleError("Invalid input path", ve);
        }
    }

}