James Moger
2014-04-14 d3b1e34c7428fbe4ef3abeee9eaef0f885c3f19b
src/main/java/com/gitblit/manager/PluginManager.java
@@ -80,6 +80,7 @@
   public PluginManager(IRuntimeManager runtimeManager) {
      File dir = runtimeManager.getFileOrFolder(Keys.plugins.folder, "${baseFolder}/plugins");
      dir.mkdirs();
      this.runtimeManager = runtimeManager;
      this.pf4j = new DefaultPluginManager(dir);
   }
@@ -127,6 +128,29 @@
      PluginState state = pf4j.startPlugin(pluginId);
      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
@@ -229,11 +253,11 @@
   }
   @Override
   public synchronized boolean refreshRegistry() {
   public synchronized boolean refreshRegistry(boolean verifyChecksum) {
      String dr = "http://gitblit.github.io/gitblit-registry/plugins.json";
      String url = runtimeManager.getSettings().getString(Keys.plugins.registry, dr);
      try {
         File file = download(url, true);
         File file = download(url, verifyChecksum);
         if (file != null && file.exists()) {
            URL selfUrl = new URL(url.substring(0, url.lastIndexOf('/')));
            // replace ${self} with the registry url
@@ -260,7 +284,7 @@
      File[] files = folder.listFiles(jsonFilter);
      if (files == null || files.length == 0) {
         // automatically retrieve the registry if we don't have a local copy
         refreshRegistry();
         refreshRegistry(true);
         files = folder.listFiles(jsonFilter);
      }
@@ -295,6 +319,7 @@
            map.put(reg.id, reg);
         }
      }
      for (PluginWrapper pw : pf4j.getPlugins()) {
         String id = pw.getDescriptor().getPluginId();
         PluginVersion pv = pw.getDescriptor().getVersion();
@@ -319,9 +344,9 @@
   }
   @Override
   public synchronized PluginRegistration lookupPlugin(String idOrName) {
   public synchronized PluginRegistration lookupPlugin(String pluginId) {
      for (PluginRegistration reg : getRegisteredPlugins()) {
         if (reg.id.equalsIgnoreCase(idOrName) || reg.name.equalsIgnoreCase(idOrName)) {
         if (reg.id.equalsIgnoreCase(pluginId)) {
            return reg;
         }
      }
@@ -329,8 +354,8 @@
   }
   @Override
   public synchronized PluginRelease lookupRelease(String idOrName, String version) {
      PluginRegistration reg = lookupPlugin(idOrName);
   public synchronized PluginRelease lookupRelease(String pluginId, String version) {
      PluginRegistration reg = lookupPlugin(pluginId);
      if (reg == null) {
         return null;
      }