source: josm/trunk/src/org/openstreetmap/josm/actions/search/SelectionWebsiteLoader.java@ 2695

// License: GPL. Copyright 2007 by Immanuel Scholz and others
/**
 *
 */
package org.openstreetmap.josm.actions.search;

import static org.openstreetmap.josm.tools.I18n.tr;

import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLConnection;
import java.util.Collection;
import java.util.LinkedList;
import java.util.Map;

import javax.swing.JOptionPane;

import org.openstreetmap.josm.Main;
import org.openstreetmap.josm.data.osm.OsmPrimitive;
import org.openstreetmap.josm.gui.PleaseWaitRunnable;
import org.openstreetmap.josm.io.OsmIdReader;
import org.openstreetmap.josm.io.OsmTransferException;
import org.openstreetmap.josm.io.ProgressInputStream;
import org.xml.sax.SAXException;

public class SelectionWebsiteLoader extends PleaseWaitRunnable {
    public final URL url;
    public Collection<OsmPrimitive> sel;
    private final SearchAction.SearchMode mode;
    private OsmIdReader idReader = new OsmIdReader();
    public SelectionWebsiteLoader(String urlStr, SearchAction.SearchMode mode) {
        super(tr("Load Selection"));
        this.mode = mode;
        URL u = null;
        try {u = new URL(urlStr);} catch (MalformedURLException e) {}
        this.url = u;
    }
    @Override protected void realRun() {
        progressMonitor.setTicksCount(2);
        sel = mode != SearchAction.SearchMode.remove ? new LinkedList<OsmPrimitive>() : Main.main.getCurrentDataSet().allNonDeletedPrimitives();
        try {
            URLConnection con = url.openConnection();
            progressMonitor.subTask(tr("Contact {0}...", url.getHost()));
            InputStream in = new ProgressInputStream(con, progressMonitor.createSubTaskMonitor(1, true));
            progressMonitor.subTask(tr("Downloading..."));
            Map<Long, String> ids = idReader.parseIds(in);
            for (OsmPrimitive osm : Main.main.getCurrentDataSet().allNonDeletedPrimitives()) {
                if (ids.containsKey(osm.getId()) && osm.getClass().getName().toLowerCase().endsWith(ids.get(osm.getId()))) {
                    if (mode == SearchAction.SearchMode.remove) {
                        sel.remove(osm);
                    } else {
                        sel.add(osm);
                    }
                }
            }
            progressMonitor.worked(1);
        } catch (IOException e) {
            e.printStackTrace();
            JOptionPane.showMessageDialog(
                    Main.parent,
                    tr("Could not read from URL: \"{0}\"",url),
                    tr("Error"),
                    JOptionPane.ERROR_MESSAGE
            );
        } catch (SAXException e) {
            e.printStackTrace();
            JOptionPane.showMessageDialog(
                    Main.parent,
                    tr("Parsing error in URL: \"{0}\"",url),
                    tr("Error"),
                    JOptionPane.ERROR_MESSAGE
            );
        } catch(OsmTransferException e) {
            e.printStackTrace();
            if (e.getCause() != null) {
                if (e.getCause() instanceof IOException ) {
                    JOptionPane.showMessageDialog(
                            Main.parent, tr("Could not read from URL: \"{0}\"",url),
                            tr("Error"), JOptionPane.ERROR_MESSAGE);
                } else if (e.getCause() instanceof SAXException) {
                    JOptionPane.showMessageDialog(Main.parent,tr("Parsing error in URL: \"{0}\"",url),
                            tr("Error"), JOptionPane.ERROR_MESSAGE);
                }
            } else {
                JOptionPane.showMessageDialog(Main.parent,tr("Error while communicating with server.",url),
                        tr("Error"), JOptionPane.ERROR_MESSAGE);
            }

        }
    }
    @Override protected void cancel() {
        sel = null;
        idReader.cancel();
    }
    @Override protected void finish() {
        if (sel != null) {
            Main.main.getCurrentDataSet().setSelected(sel);
        }
    }
}