Ticket #6396: reload-plugins-0.1.patch

new file src/org/openstreetmap/josm/actions/ReloadPluginsAction.java

@@ +1 @@
+// License: GPL. For details, see LICENSE file.
+package org.openstreetmap.josm.actions;
+
+import static org.openstreetmap.josm.tools.I18n.tr;
+
+import java.awt.event.ActionEvent;
+import java.awt.event.KeyEvent;
+
+import org.openstreetmap.josm.plugins.PluginHandler;
+import org.openstreetmap.josm.tools.Shortcut;
+
+public class ReloadPluginsAction extends JosmAction {
+
+    public ReloadPluginsAction() {
+        super(tr("Reload plugins"), null, tr("Reload plugins"), Shortcut.registerShortcut("system:reload-plugins",
+                tr("Reload plugins"), KeyEvent.VK_F5, Shortcut.GROUP_DIRECT, Shortcut.SHIFT_DEFAULT), true);
+    }
+
+    @Override
+    public void actionPerformed(ActionEvent e) {
+        PluginHandler.reloadPlugins();
+    }
+}

src/org/openstreetmap/josm/gui/MainMenu.java

@@ -52 +52 @@
 import org.openstreetmap.josm.actions.OpenFileAction;
 import org.openstreetmap.josm.actions.OpenLocationAction;
 import org.openstreetmap.josm.actions.OrthogonalizeAction;
+import org.openstreetmap.josm.actions.ReloadPluginsAction;
 import org.openstreetmap.josm.actions.OrthogonalizeAction.Undo;
 import org.openstreetmap.josm.actions.PasteAction;
 import org.openstreetmap.josm.actions.PasteTagsAction;
@@ -359 +360 @@
         current.setAccelerator(Shortcut.registerShortcut("system:help", tr("Help"), KeyEvent.VK_F1,
                 Shortcut.GROUP_DIRECT).getKeyStroke());
         add(helpMenu, about);
+        add(helpMenu, new ReloadPluginsAction());
 
         new PresetsMenuEnabler(presetsMenu).refreshEnabled();
     }

src/org/openstreetmap/josm/gui/MapFrame.java

@@ -290 +290 @@
             throw new IllegalArgumentException("MapMode action must be subclass of MapMode");
     }
 
+    public void removeMapMode(IconToggleButton mode) {
+        toolBarActions.remove(mode);
+        toolGroup.remove(mode);
+        mapModes.remove(mode.getAction());
+    }
+
     /**
      * Fires an property changed event "visible".
      */

src/org/openstreetmap/josm/plugins/Plugin.java

@@ -72 +72 @@
         this.info = info;
     }
 
+    public void preReloadCleanup() {
+        System.out.println("no cleanup for this class");
+    }
+
     /**
      * @return The directory for the plugin to store all kind of stuff.
      */

src/org/openstreetmap/josm/plugins/PluginHandler.java

@@ -13 +13 @@
 import java.awt.event.ActionEvent;
 import java.io.File;
 import java.io.FilenameFilter;
+import java.lang.reflect.Method;
 import java.net.URL;
 import java.net.URLClassLoader;
 import java.util.ArrayList;
@@ -57 +58 @@
 import org.openstreetmap.josm.gui.preferences.PreferenceSettingFactory;
 import org.openstreetmap.josm.gui.progress.NullProgressMonitor;
 import org.openstreetmap.josm.gui.progress.ProgressMonitor;
-import org.openstreetmap.josm.io.remotecontrol.RemoteControl;
 import org.openstreetmap.josm.tools.CheckParameterUtil;
 import org.openstreetmap.josm.tools.GBC;
 import org.openstreetmap.josm.tools.ImageProvider;
@@ -429 +429 @@
     }
 
     /**
-     * Creates a class loader for loading plugin code.
-     *
-     * @param plugins the collection of plugins which are going to be loaded with this
-     * class loader
-     * @return the class loader
+     * Helper to extract the jar paths of the plugins
      */
-    public static ClassLoader createClassLoader(Collection<PluginInformation> plugins) {
+    private static URL[] createJarURLs(Collection<PluginInformation> plugins) {
         // iterate all plugins and collect all libraries of all plugins:
         List<URL> allPluginLibraries = new LinkedList<URL>();
         File pluginDir = Main.pref.getPluginsDirectory();
@@ -451 +447 @@
 
         // create a classloader for all plugins:
         URL[] jarUrls = new URL[allPluginLibraries.size()];
-        jarUrls = allPluginLibraries.toArray(jarUrls);
+        return allPluginLibraries.toArray(jarUrls);
+    }
+    /**
+     * Creates a class loader for loading plugin code.
+     *
+     * @param plugins the collection of plugins which are going to be loaded with this
+     * class loader
+     * @return the class loader
+     */
+    public static ClassLoader createClassLoader(Collection<PluginInformation> plugins) {
+        URL[] jarUrls = createJarURLs(plugins);
         URLClassLoader pluginClassLoader = new URLClassLoader(jarUrls, Main.class.getClassLoader());
         return pluginClassLoader;
     }
 
+    private static String getPackagePartOfClassName(String name) {
+        for (int i = name.length() - 1; i >= 0; i--) {
+            if (name.charAt(i) == '.')
+                return name.substring(0, i);
+        }
+        return "";
+    }
+
+    private static boolean isSubPackageOf(String sub, String parent) {
+        return sub.startsWith(parent);
+    }
+
+    /**
+     * Creates a class loader that will reload the supplied plugins
+     *
+     * The plugins should not define classes in packages above the package of the 'Plugin-Class' (defined in MANIFEST)
+     * ie. all classes must belong to a package matching: <plugin-class-package>.*
+     */
+    public static ClassLoader createClassReloader(Collection<PluginInformation> plugins) {
+        final Set<String> packageList = new HashSet<String>();
+        for (PluginInformation pi : plugins) {
+            packageList.add(getPackagePartOfClassName(pi.className));
+        }
+
+        ClassLoader deceiving = new ClassLoader(Main.class.getClassLoader()) {
+            // Overriding loadClass/1 does not seem to work ...
+            // Are the 'fullClassName' guaranteed to be full?
+            @Override
+            protected Class<?> loadClass(String fullClassName, boolean resolve) throws ClassNotFoundException {
+                String packageOfClass = getPackagePartOfClassName(fullClassName);
+                boolean hide = false;
+                for(String pkg : packageList) {
+                    if(isSubPackageOf(packageOfClass, pkg)) {
+                        hide = true;
+                        break;
+                    }
+                }
+                if(hide) {
+                    System.out.println(fullClassName);
+                    return null;
+                } else {
+                    // is it possible that super.loadClass/2 calls this.loadClass/2 again, causing infinite recursion?
+                    return super.loadClass(fullClassName, resolve);
+                }
+            }
+        };
+        URL[] jarUrls = createJarURLs(plugins);
+        return new URLClassLoader(jarUrls, deceiving);
+    }
+
+    // TODO: MANIFEST property instead? This is simpler for everyone though?
+    private static boolean supportReload(PluginProxy pluginProxy) {
+        try {
+            Method m = pluginProxy.plugin.getClass().getMethod("preReloadCleanup");
+            return !m.getDeclaringClass().equals(Plugin.class);
+        } catch (NoSuchMethodException e) {
+            return false;
+        }
+    }
+
+    // FIXME: fucked if p1 depends on p2 and only p2 is reloadable
+    // Should be OK to let 'sources' keep the old class loader since the new class loader is prepended?
+    //   Somewhat complicated if we should filter out the reloaded jars from the old loader...
+    // TODO: This does not check preconditions for the plugins, assuming they are fulfilled as the plugins are already loaded. This is of course not 100% correct.
+    // TODO: add progressmonitor support
+    // TODO: do we need to respect the loadearly/late flags?
+    // TODO: seems to be quick enough, but is it problematic to to in a worker thread?
+    /**
+     * Reloads all loaded plugins implementing 'preReloadCleanup'
+     */
+    public static void reloadPlugins() {
+        //        ProgressMonitor monitor = null;
+        //        if (monitor == null) {
+        //            monitor = NullProgressMonitor.INSTANCE;
+        //        }
+        List<PluginInformation> plugins = new ArrayList<PluginInformation>();
+        for (Iterator<PluginProxy> iter = pluginList.iterator(); iter.hasNext();) {
+            PluginProxy pp = iter.next();
+            if (supportReload(pp)) {
+                pp.preReloadCleanup();
+                plugins.add(pp.getPluginInformation());
+                iter.remove();
+            }
+        }
+        if (plugins.isEmpty())
+            return;
+        // safest to not have a to-be-reloaded map mode active
+        Main.map.selectSelectTool(false);
+        //        try {
+
+        //            List<PluginInformation> plugins = PluginHandler.buildListOfPluginsToLoad(null,monitor.createSubTaskMonitor(1, false));
+
+        //            monitor.beginTask(tr("Loading plugins ..."));
+        //            monitor.subTask(tr("Checking plugin preconditions..."));
+        //            Collection<PluginInformation> toLoad = preproccessPluginList(null, plugins);
+
+        ClassLoader cl = createClassReloader(plugins);
+
+        loadPluginsNoCheck(null, plugins, null, cl);
+        for (PluginProxy plugin : pluginList) {
+            plugin.mapFrameInitialized(Main.map, Main.map);
+        }
+        //        PluginHandler.notifyMapFrameChanged(Main.map, Main.map);
+        //        } finally {
+        //            monitor.finishTask();
+        //        }
+    }
+
     /**
      * Loads and instantiates the plugin described by <code>plugin</code> using
      * the class loader <code>pluginClassLoader</code>.
@@ -493 +607 @@
      * @param plugins the list of plugins
      * @param monitor the progress monitor. Defaults to {@see NullProgressMonitor#INSTANCE} if null.
      */
-    public static void loadPlugins(Window parent,Collection<PluginInformation> plugins, ProgressMonitor monitor) {
+    public static void loadPlugins(Window parent, Collection<PluginInformation> plugins, ProgressMonitor monitor) {
         if (monitor == null) {
             monitor = NullProgressMonitor.INSTANCE;
         }
         try {
             monitor.beginTask(tr("Loading plugins ..."));
             monitor.subTask(tr("Checking plugin preconditions..."));
-            List<PluginInformation> toLoad = new LinkedList<PluginInformation>();
-            for (PluginInformation pi: plugins) {
-                if (checkLoadPreconditions(parent, plugins, pi)) {
-                    toLoad.add(pi);
-                }
-            }
-            // sort the plugins according to their "staging" equivalence class. The
-            // lower the value of "stage" the earlier the plugin should be loaded.
-            //
-            Collections.sort(
-                    toLoad,
-                    new Comparator<PluginInformation>() {
-                        public int compare(PluginInformation o1, PluginInformation o2) {
-                            if (o1.stage < o2.stage) return -1;
-                            if (o1.stage == o2.stage) return 0;
-                            return 1;
-                        }
-                    }
-            );
+            Collection<PluginInformation> toLoad = preproccessPluginList(parent, plugins);
             if (toLoad.isEmpty())
                 return;
+            loadPluginsNoCheck(parent, toLoad, monitor, PluginHandler.createClassLoader(toLoad));
+        } finally {
+            monitor.finishTask();
+        }
+    }
 
-            ClassLoader pluginClassLoader = createClassLoader(toLoad);
-            sources.add(0, pluginClassLoader);
-            monitor.setTicksCount(toLoad.size());
-            for (PluginInformation info : toLoad) {
+    public static void loadPluginsNoCheck(Window parent, Collection<PluginInformation> plugins,
+            ProgressMonitor monitor, ClassLoader classLoader) {
+        sources.add(0, classLoader);
+        if (monitor != null)
+            monitor.setTicksCount(plugins.size());
+        for (PluginInformation info : plugins) {
+            if (monitor != null)
                 monitor.setExtraText(tr("Loading plugin ''{0}''...", info.name));
-                loadPlugin(parent, info, pluginClassLoader);
+            else
+                System.out.println(tr("Loading plugin ''{0}''...", info.name));
+            loadPlugin(parent, info, classLoader);
+            if (monitor != null)
                 monitor.worked(1);
+        }
+    }
+
+    private static Collection<PluginInformation> preproccessPluginList(Window parent,
+            Collection<PluginInformation> plugins) {
+        List<PluginInformation> toLoad = new LinkedList<PluginInformation>();
+        for (PluginInformation pi : plugins) {
+            if (checkLoadPreconditions(parent, plugins, pi)) {
+                toLoad.add(pi);
             }
-        } finally {
-            monitor.finishTask();
         }
+        // sort the plugins according to their "staging" equivalence class. The
+        // lower the value of "stage" the earlier the plugin should be loaded.
+        //
+        Collections.sort(toLoad, new Comparator<PluginInformation>() {
+            public int compare(PluginInformation o1, PluginInformation o2) {
+                if (o1.stage < o2.stage)
+                    return -1;
+                if (o1.stage == o2.stage)
+                    return 0;
+                return 1;
+            }
+        });
+        return toLoad;
     }
 
     /**

src/org/openstreetmap/josm/plugins/PluginProxy.java

@@ -1 +1 @@
 // License: GPL. Copyright 2007 by Immanuel Scholz and others
 package org.openstreetmap.josm.plugins;
 
+import java.lang.reflect.Method;
 import java.util.List;
 
 import org.openstreetmap.josm.gui.MapFrame;
@@ -33 +34 @@
         }
     }
 
+    @Override public void preReloadCleanup() {
+        try {
+            Method m = plugin.getClass().getMethod("preReloadCleanup");
+            m.invoke(plugin);
+        } catch (NoSuchMethodException e) {
+        } catch (Exception e) {
+            BugReportExceptionHandler.handleException(new PluginException(this, getPluginInformation().name, e));
+        }
+    }
+
     @Override public PreferenceSetting getPreferenceSetting() {
         try {
             return (PreferenceSetting)plugin.getClass().getMethod("getPreferenceSetting").invoke(plugin);