Upgrade plugin command

James Moger

2014-04-13 74f3d92cca058f5c297e5b6032b5e41c935a5504

Upgrade plugin command

 src/main/java/com/gitblit/manager/GitblitManager.java

@@ -1260,6 +1260,11 @@
    }

    @Override
    public boolean upgradePlugin(String pluginId, String url, boolean verifyChecksum) throws IOException {
        return pluginManager.upgradePlugin(pluginId, url, verifyChecksum);
    }

    @Override
    public List<PluginRegistration> getRegisteredPlugins() {
        return pluginManager.getRegisteredPlugins();
    }

 src/main/java/com/gitblit/manager/IPluginManager.java

@@ -132,6 +132,17 @@
    boolean installPlugin(String url, boolean verifyChecksum) throws IOException;

    /**
     * Upgrade the install plugin from the specified url.
     *
     * @param pluginId
     * @param url
     * @param verifyChecksum
     * @return true if the upgrade has been successful
     * @throws IOException
     */
    boolean upgradePlugin(String pluginId, String url, boolean verifyChecksum) throws IOException;

    /**
     * The list of all registered plugins.
     *
     * @return a list of registered plugins

 src/main/java/com/gitblit/manager/PluginManager.java

@@ -130,6 +130,29 @@
        return PluginState.STARTED.equals(state);
    }

    public synchronized boolean upgradePlugin(String pluginId, String url, boolean verifyChecksum) throws IOException {
        // ensure we can download the update BEFORE we remove the existing one
        File file = download(url, verifyChecksum);
        if (file == null || !file.exists()) {
            logger.error("Failed to download plugin {}", url);
            return false;
        }

        if (deletePlugin(pluginId)) {
            String newPluginId = pf4j.loadPlugin(file);
            if (StringUtils.isEmpty(newPluginId)) {
                logger.error("Failed to load plugin {}", file);
                return false;
            }

            PluginState state = pf4j.startPlugin(newPluginId);
            return PluginState.STARTED.equals(state);
        } else {
            logger.error("Failed to delete plugin {}", pluginId);
        }
        return false;
    }

    @Override
    public synchronized boolean disablePlugin(String pluginId) {
        return pf4j.disablePlugin(pluginId);
@@ -296,6 +319,7 @@
                map.put(reg.id, reg);
            }
        }

        for (PluginWrapper pw : pf4j.getPlugins()) {
            String id = pw.getDescriptor().getPluginId();
            PluginVersion pv = pw.getDescriptor().getVersion();

 src/main/java/com/gitblit/models/PluginRegistry.java

@@ -62,7 +62,7 @@
    }



    public static enum InstallState {

        NOT_INSTALLED, INSTALLED, CAN_UPDATE, UNKNOWN

        NOT_INSTALLED, INSTALLED, UPDATE_AVAILABLE, UNKNOWN

    }



    /**

@@ -131,7 +131,7 @@
            case -1:

                return InstallState.UNKNOWN;

            case 1:

                return InstallState.CAN_UPDATE;

                return InstallState.UPDATE_AVAILABLE;

            default:

                return InstallState.INSTALLED;

            }


 src/main/java/com/gitblit/transport/ssh/commands/PluginDispatcher.java

@@ -15,6 +15,7 @@
 */
package com.gitblit.transport.ssh.commands;

import java.io.IOException;
import java.util.Collections;
import java.util.List;

@@ -56,6 +57,7 @@
        register(user, RefreshPlugins.class);
        register(user, AvailablePlugins.class);
        register(user, InstallPlugin.class);
        register(user, UpgradePlugin.class);
        register(user, UninstallPlugin.class);
    }

@@ -444,7 +446,7 @@

            List<PluginRegistration> list;
            if (updates) {
                list = gitblit.getRegisteredPlugins(InstallState.CAN_UPDATE);
                list = gitblit.getRegisteredPlugins(InstallState.UPDATE_AVAILABLE);
            } else {
                list = gitblit.getRegisteredPlugins();
            }
@@ -527,13 +529,51 @@
                        throw new Failure(1, String.format("Failed to install %s", urlOrIdOrName));
                    }
                }
            } catch (Exception e) {
            } catch (IOException e) {
                log.error("Failed to install " + urlOrIdOrName, e);
                throw new Failure(1, String.format("Failed to install %s", urlOrIdOrName), e);
            }
        }
    }

    @CommandMetaData(name = "upgrade", description = "Upgrade a plugin")
    public static class UpgradePlugin extends PluginCommand {

        @Argument(index = 0, required = true, metaVar = "<ID>|<INDEX>", usage = "the plugin to upgrade")
        protected String id;

        @Option(name = "--version", usage = "The specific version to install")
        private String version;

        @Option(name = "--noverify", usage = "Disable checksum verification")
        private boolean disableChecksum;

        @Override
        public void run() throws Failure {
            IGitblit gitblit = getContext().getGitblit();
            PluginWrapper pluginWrapper = getPlugin(id);
            if (pluginWrapper == null) {
                throw new UnloggedFailure("Invalid plugin specified!");
            }

            PluginRelease pv = gitblit.lookupRelease(pluginWrapper.getPluginId(), version);
            if (pv == null) {
                throw new Failure(1,  String.format("Plugin \"%s\" is not in the registry!", pluginWrapper.getPluginId()));
            }

            try {
                if (gitblit.upgradePlugin(pluginWrapper.getPluginId(), pv.url, !disableChecksum)) {
                    stdout.println(String.format("Upgraded %s", pluginWrapper.getPluginId()));
                } else {
                    throw new Failure(1, String.format("Failed to upgrade %s", pluginWrapper.getPluginId()));
                }
            } catch (IOException e) {
                log.error("Failed to upgrade " + pluginWrapper.getPluginId(), e);
                throw new Failure(1, String.format("Failed to upgrade %s", pluginWrapper.getPluginId()), e);
            }
        }
    }

    @CommandMetaData(name = "uninstall", aliases = { "rm", "del" }, description = "Uninstall a plugin")
    public static class UninstallPlugin extends PluginCommand {