source: josm/trunk/src/org/openstreetmap/josm/gui/dialogs/properties/AbstractCopyAction.java@ 15813

// License: GPL. For details, see LICENSE file.
package org.openstreetmap.josm.gui.dialogs.properties;

import java.awt.event.ActionEvent;
import java.util.Arrays;
import java.util.Collection;
import java.util.Objects;
import java.util.function.IntFunction;
import java.util.function.Supplier;
import java.util.stream.Collectors;
import java.util.stream.Stream;

import javax.swing.AbstractAction;
import javax.swing.JTable;

import org.openstreetmap.josm.data.osm.Tagged;
import org.openstreetmap.josm.gui.datatransfer.ClipboardUtils;

/**
 * Super class of all copy actions from tag table.
 * @since 13521
 */
public abstract class AbstractCopyAction extends AbstractAction {

    private final JTable tagTable;
    private final IntFunction<String> keySupplier;
    private final Supplier<Collection<? extends Tagged>> objectSupplier;

    /**
     * Constructs a new {@code AbstractCopyAction}.
     * @param tagTable the tag table
     * @param keySupplier a supplier which returns the selected key for a given row index
     * @param objectSupplier a supplier which returns the selected tagged object(s)
     */
    public AbstractCopyAction(JTable tagTable, IntFunction<String> keySupplier, Supplier<Collection<? extends Tagged>> objectSupplier) {
        this.tagTable = Objects.requireNonNull(tagTable);
        this.keySupplier = Objects.requireNonNull(keySupplier);
        this.objectSupplier = Objects.requireNonNull(objectSupplier);
    }

    protected abstract Collection<String> getString(Tagged p, String key);

    protected Stream<String> valueStream() {
        int[] rows = tagTable.getSelectedRows();
        Collection<? extends Tagged> sel = objectSupplier.get();
        if (rows.length == 0 || sel == null || sel.isEmpty()) return Stream.empty();
        return Arrays.stream(rows)
                .mapToObj(keySupplier)
                .flatMap(key -> sel.stream().map(p -> getString(p, key)))
                .filter(Objects::nonNull)
                .flatMap(Collection::stream)
                .distinct();
    }

    @Override
    public void actionPerformed(ActionEvent ae) {
        final String values = valueStream()
                .sorted()
                .collect(Collectors.joining("\n"));
        if (!values.isEmpty()) {
            ClipboardUtils.copyString(values);
        }
    }
}