package com.atlassian.sal.core.upgrade;
   
   import com.atlassian.plugin.Plugin;
   import com.atlassian.sal.api.message.Message;
   import com.atlassian.sal.api.pluginsettings.PluginSettings;
   import com.atlassian.sal.api.upgrade.PluginUpgradeTask;
   import com.atlassian.sal.core.message.DefaultMessage;
   import org.slf4j.Logger;
   import org.slf4j.LoggerFactory;
  
  import java.io.PrintWriter;
  import java.io.StringWriter;
  import java.util.ArrayList;
  import java.util.Arrays;
  import java.util.Collection;
  import java.util.Collections;
  import java.util.Comparator;
  import java.util.List;
  
  /**
   * Performs an upgrade of a plugin.  Originally copied from Confluence's AbstractUpgradeManager.
   */
  public class PluginUpgrader {
  
      public static final String BUILD = ":build";
      protected List<PluginUpgradeTask> upgradeTasks;
      private static final Logger log = LoggerFactory.getLogger(PluginUpgrader.class);
      protected Plugin plugin;
      protected PluginSettings pluginSettings;
      protected List<Message> errors = new ArrayList<>();
  
      private static final Comparator<PluginUpgradeTask> UPGRADE_TASK_COMPARATOR = (t1, t2) -> {
          if (t1 == null) {
              return -1;
          }
          if (t2 == null) {
              return 1;
          }
          if (t1.getBuildNumber() > t2.getBuildNumber()) {
              return 1;
          }
  
          return (t1.getBuildNumber() < t2.getBuildNumber()) ? -1 : 0;
      };
  
      protected PluginUpgrader(final Plugin plugin, final PluginSettings pluginSettings, final List<PluginUpgradeTask> upgradeTasks) {
          this.plugin = plugin;
          this.pluginSettings = pluginSettings;
          this.upgradeTasks = new ArrayList<>(upgradeTasks);
          Collections.sort(this.upgradeTasks, UPGRADE_TASK_COMPARATOR);
      }
  
      protected List<Message> upgrade() {
          if (needUpgrade()) {
              doUpgrade();
          }
          return errors;
      }
  
      protected void doUpgrade() {
          try {
              log.info("Upgrading plugin " + plugin.getKey());
              for (final PluginUpgradeTask upgradeTask : upgradeTasks) {
  
                  if (upgradeTask.getBuildNumber() <= getDataBuildNumber()) {
                      // Current buildnumber of data is higher or equal to this upgrade task. No need to run
                      continue;
                  }
  
                  final Collection<Message> messages = upgradeTask.doUpgrade();
  
                  if (messages == null || messages.isEmpty()) {
                      upgradeTaskSucceeded(upgradeTask);
                  } else {
                      upgradeTaskFailed(upgradeTask, messages);
                  }
              }
          } catch (final Throwable e) {
              // stringify the stacktrace.
              StringWriter sw = new StringWriter();
              e.printStackTrace(new PrintWriter(sw));
  
              // use a message here intentionally instead of key. it won't get translated.
              errors.add(new DefaultMessage("Unexpected exception caught during plugin upgrade: " + sw.toString()));
              log.error("Upgrade failed: " + e.getMessage(), e);
          } finally {
              postUpgrade();
          }
      }
  
  
      protected void upgradeTaskSucceeded(final PluginUpgradeTask upgradeTask) {
          setDataBuildNumber(upgradeTask.getBuildNumber());
          log.info("Upgraded plugin " + upgradeTask.getPluginKey() + " to version " + upgradeTask.getBuildNumber() + " - " + upgradeTask.getShortDescription());
      }
  
      protected void upgradeTaskFailed(final PluginUpgradeTask upgradeTask, final Collection<Message> messages) {
          errors.addAll(messages);
          final StringBuilder msg = new StringBuilder();
         msg.append("Plugin upgrade failed for ").append(upgradeTask.getPluginKey());
         msg.append(" to version ").append(upgradeTask.getBuildNumber());
         msg.append(" - ").append(upgradeTask.getShortDescription());
         msg.append("\n");
         for (final Message message : messages) {
             msg.append("\t* ").append(message.getKey()).append(" ").append(Arrays.toString(message.getArguments()));
         }
 
         log.warn(msg.toString());
     }
 
     protected List<Message> getErrors() {
         return errors;
     }
 
     protected boolean needUpgrade() {
         final PluginUpgradeTask lastUpgradeTask = this.upgradeTasks.get(this.upgradeTasks.size() - 1);
         final int dataBuildNumber = getDataBuildNumber();
         final int lastUpgradeTaskBuildNumber = lastUpgradeTask.getBuildNumber();
         log.debug("Plugin: {}, current version: {}, highest upgrade task found: {}.",
                 plugin.getKey(), dataBuildNumber, lastUpgradeTaskBuildNumber);
         return lastUpgradeTaskBuildNumber > dataBuildNumber;
     }
 
 
     /**
      * This is the build number of the current version that the user is running under.
      */
     protected int getDataBuildNumber() {
         final String val = (String) pluginSettings.get(plugin.getKey() + BUILD);
         if (val != null) {
             return Integer.parseInt(val);
         } else {
             return 0;
         }
     }
 
     protected void setDataBuildNumber(final int buildNumber) {
         pluginSettings.put(plugin.getKey() + BUILD, String.valueOf(buildNumber));
     }
 
     protected void postUpgrade() {
         log.info("Plugin {}  upgrade completed. Current version is: {}", plugin.getKey(), getDataBuildNumber());
     }
 }