MapWithAI/src/main/java/org/openstreetmap/josm/plugins/mapwithai/actions/AddMapWithAILayerAction.java

// License: GPL. For details, see LICENSE file.
package org.openstreetmap.josm.plugins.mapwithai.actions;

import static java.util.function.Predicate.not;
import static org.openstreetmap.josm.gui.help.HelpUtil.ht;

import java.awt.event.ActionEvent;
import java.io.Serial;
import java.util.ArrayList;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ForkJoinTask;

import org.openstreetmap.josm.actions.AdaptableAction;
import org.openstreetmap.josm.actions.AddImageryLayerAction;
import org.openstreetmap.josm.actions.JosmAction;
import org.openstreetmap.josm.data.osm.DataSet;
import org.openstreetmap.josm.data.osm.OsmData;
import org.openstreetmap.josm.gui.MainApplication;
import org.openstreetmap.josm.gui.layer.OsmDataLayer;
import org.openstreetmap.josm.gui.preferences.ToolbarPreferences;
import org.openstreetmap.josm.gui.progress.NullProgressMonitor;
import org.openstreetmap.josm.gui.util.GuiHelper;
import org.openstreetmap.josm.plugins.mapwithai.backend.MapWithAIDataUtils;
import org.openstreetmap.josm.plugins.mapwithai.backend.MapWithAILayer;
import org.openstreetmap.josm.plugins.mapwithai.data.mapwithai.MapWithAIInfo;
import org.openstreetmap.josm.tools.ImageProvider;
import org.openstreetmap.josm.tools.ImageProvider.ImageSizes;
import org.openstreetmap.josm.tools.ImageResource;
import org.openstreetmap.josm.tools.JosmRuntimeException;
import org.openstreetmap.josm.tools.Logging;

/**
 * Action displayed in MapWithAI menu to add data to the MapWithAI layer.
 * Largely copied from {@link AddImageryLayerAction}.
 */
public class AddMapWithAILayerAction extends JosmAction implements AdaptableAction {
    @Serial
    private static final long serialVersionUID = 1403912860658467920L;
    private final transient MapWithAIInfo info;

    /**
     * Constructs a new {@code AddMapWithAILayerAction} for the given
     * {@code MapWithAIInfo}. If an http:// icon is specified, it is fetched
     * asynchronously.
     *
     * @param info The source info
     */
    public AddMapWithAILayerAction(MapWithAIInfo info) {
        super(info.getName(), /* ICON */"imagery_menu", info.getToolTipText(), null, true,
                ToolbarPreferences.IMAGERY_PREFIX + info.getToolbarName(), false);
        setHelpId(ht("/Preferences/Imagery"));
        this.info = info;
        installAdapters();

        // change toolbar icon from if specified
        final var icon = info.getIcon();
        if (icon != null) {
            final var future = new ImageProvider(icon).setOptional(true).getResourceAsync(result -> {
                if (result != null) {
                    GuiHelper.runInEDT(() -> result.attachImageIcon(this));
                }
            });
            try {
                future.get();
            } catch (InterruptedException e) {
                Logging.error(e);
                Thread.currentThread().interrupt();
            } catch (ExecutionException e) {
                Logging.error(e);
            }
        } else {
            try {
                final var resource = new ImageResource(
                        this.info.getSourceCategory().getIcon(ImageSizes.MENU).getImage());
                resource.attachImageIcon(this);
            } catch (JosmRuntimeException e) {
                // Eclipse doesn't like giving applications their resources...
                if (!e.getMessage().contains("failed to locate image")) {
                    throw e;
                }
            }
        }
    }

    @Override
    public void actionPerformed(ActionEvent e) {
        if (isEnabled()) {
            MainApplication.worker.execute(() -> realRun(this.info));
        }
    }

    /**
     * Run the download tasks. This should be run off of the EDT, see #23529.
     *
     * @param info The external data to download
     */
    private static void realRun(MapWithAIInfo info) {
        MapWithAILayer layer = MapWithAIDataUtils.getLayer(false);
        final DataSet ds;
        final OsmData<?, ?, ?, ?> boundsSource;
        final var dataLayer = getDataLayer();
        if (layer != null && !layer.getData().getDataSourceBounds().isEmpty()) {
            ds = layer.getDataSet();
            boundsSource = ds;
        } else if (dataLayer != null && !dataLayer.getDataSet().getDataSourceBounds().isEmpty()) {
            boundsSource = dataLayer.getDataSet();
            layer = MapWithAIDataUtils.getLayer(true);
            ds = layer.getDataSet();
        } else {
            boundsSource = null;
            ds = null;
        }
        if (boundsSource != null && ds != null) {
            final var pool = MapWithAIDataUtils.getForkJoinPool();
            final var forkJoinTasks = new ArrayList<ForkJoinTask<DataSet>>(boundsSource.getDataSourceBounds().size());
            for (var b : boundsSource.getDataSourceBounds()) {
                final var task = MapWithAIDataUtils.download(NullProgressMonitor.INSTANCE, b, info,
                        MapWithAIDataUtils.MAXIMUM_SIDE_DIMENSIONS);
                forkJoinTasks.add(task);
                pool.execute(task);
            }
            for (var task : forkJoinTasks) {
                ds.mergeFrom(task.join());
            }
        }
        if (layer != null) {
            layer.addDownloadedInfo(info);
        }
    }

    private static OsmDataLayer getDataLayer() {
        return MainApplication.getLayerManager().getLayersOfType(OsmDataLayer.class).stream()
                .filter(not(MapWithAILayer.class::isInstance)).findFirst().orElse(null);
    }

    @Override
    protected boolean listenToSelectionChange() {
        return false;
    }

    @Override
    protected void updateEnabledState() {
        setEnabled(!info.isBlacklisted() && MainApplication.getLayerManager().getActiveDataLayer() != null
                && (MapWithAIDataUtils.getLayer(false) == null
                        || !MapWithAIDataUtils.getLayer(false).hasDownloaded(info)));
    }

    @Override
    public String toString() {
        return "AddMapWithAILayerAction [info=" + info + ']';
    }
}