Upgrade plugin command

James Moger

2014-04-13 74f3d92cca058f5c297e5b6032b5e41c935a5504

 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 {