package fr.ifremer.quadrige2.ui.swing.common.callback;
   
   /*-
    * #%L
    * Quadrige2 Core :: Quadrige2 UI Common
    * $Id:$
    * $HeadURL:$
    * %%
    * Copyright (C) 2017 Ifremer
   * %%
   * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU Affero General Public License as published by
   * the Free Software Foundation, either version 3 of the License, or
   * (at your option) any later version.
   * 
   * This program is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU General Public License for more details.
   * 
   * You should have received a copy of the GNU Affero General Public License
   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
   * #L%
   */
  
  
  import com.google.common.collect.Lists;
  import com.google.common.collect.Maps;
  import fr.ifremer.quadrige2.ui.swing.common.ApplicationUIContext;
  import fr.ifremer.quadrige2.ui.swing.common.action.AbstractAction;
  import fr.ifremer.quadrige2.ui.swing.common.model.ProgressionModel;
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.nuiton.jaxx.application.ApplicationTechnicalException;
  import org.nuiton.jaxx.application.swing.action.ApplicationActionException;
  import org.nuiton.updater.ApplicationInfo;
  import org.nuiton.updater.ApplicationUpdaterCallback;
  
  import java.util.List;
  import java.util.Map;
  import java.util.Map.Entry;
  
  import static org.nuiton.i18n.I18n.t;
  
  /**
   * CallBack to update plugins.
   *
   * @since 1.0
   */
  public class PluginsUpdaterCallBack implements ApplicationUpdaterCallback {
  
      /**
       * Logger.
       */
      private static final Log LOG = LogFactory.getLog(PluginsUpdaterCallBack.class);
  
      protected final ApplicationUIContext context;
  
      protected List<String> pluginNames;
      
      protected final ProgressionModel progressionModel;
  
      protected boolean applicationUpdated;
  
      protected final AbstractAction<?, ?, ?> action;
  
      /**
       * <p>Constructor for PluginsUpdaterCallBack.</p>
       *
       * @param action a {@link AbstractAction} object.
       * @param progressionModel a {@link ProgressionModel} object.
       */
      public PluginsUpdaterCallBack(AbstractAction<?, ?, ?> action, ProgressionModel progressionModel) {
          this.action = action;
          this.context = action.getContext();
          this.progressionModel = progressionModel;
      }
  
      /**
       * <p>setTypes.</p>
       *
       * @param types a {@link String} object.
       */
      public void setTypes(String... types) {
          this.pluginNames = Lists.newArrayList(types);
      }
      /**
       * <p>isApplicationUpdated.</p>
       *
       * @return a boolean.
       */
      public boolean isApplicationUpdated() {
          return applicationUpdated;
      }
  
      /** {@inheritDoc} */
      @Override
      public Map<String, ApplicationInfo> updateToDo(Map<String, ApplicationInfo> appToUpdate) {
  
         Map<String, ApplicationInfo> result = Maps.newHashMap();
         for (String pluginName : pluginNames) {
             ApplicationInfo info = appToUpdate.get(pluginName);
             if (info != null) {
                 result.put(info.name, info);
             }
         }
         return result;
     }
 
     /** {@inheritDoc} */
     @Override
     public void startUpdate(ApplicationInfo info) {
         progressionModel.setMessage(t("quadrige2.pluginsUpdater.startUpdate", info.name, info.newVersion));
     }
 
     /** {@inheritDoc} */
     @Override
     public void updateDone(Map<String, ApplicationInfo> appToUpdate,
         Map<String, Exception> appUpdateError) {
 
         boolean updateDone = false;
 
         // check errors
         for (final Entry<String, ApplicationInfo> entry : appToUpdate.entrySet()) {
             Exception error = appUpdateError.get(entry.getKey());
             if (error != null) {
                 throw ApplicationActionException.propagateError(
                     action, new ApplicationTechnicalException(t("quadrige2.pluginsUpdater.error", entry.getKey()), error));
             }
             ApplicationInfo info = entry.getValue();
             if (info != null) {
 
                 if (LOG.isInfoEnabled()) {
                     LOG.info(String.format(
                         "A '%s' plugin update was downloaded (oldVersion: %s, newVersion: %s), will restart application to use it", 
                         entry.getKey(), info.oldVersion, info.newVersion));
                 }
                 updateDone = true;
             }
         }
 
         if (updateDone) {
             applicationUpdated = true;
         }
     }
 
     /** {@inheritDoc} */
     @Override
     public void aborted(String propertiesURL, Exception eee) {
         if (LOG.isErrorEnabled()) {
             LOG.error("Could not update from " + propertiesURL, eee);
         }
         throw ApplicationActionException.propagateError(action, eee);
     }
 }