package fr.ifremer.quadrige3.ui.swing.content.db;
   
   /*-
    * #%L
    * Quadrige3 Core :: Quadrige3 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.base.Preconditions;
  import fr.ifremer.quadrige3.core.dao.technical.Daos;
  import fr.ifremer.quadrige3.core.exception.QuadrigeBusinessException;
  import fr.ifremer.quadrige3.core.service.persistence.PersistenceServiceHelper;
  import fr.ifremer.quadrige3.ui.swing.callback.DatabaseUpdaterCallBack;
  import fr.ifremer.quadrige3.ui.swing.action.AbstractMainUIAction;
  import fr.ifremer.quadrige3.ui.swing.content.AbstractMainUIHandler;
  import org.apache.commons.io.FileUtils;
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.nuiton.updater.ApplicationUpdater;
  
  import java.io.File;
  
  import static org.nuiton.i18n.I18n.t;
  
  /**
   * To install (or reinstall) a db from last network one.
   *
   * @since 2.4
   */
  public class InstallDbAction extends AbstractMainUIAction {
  
      /**
       * Logger.
       */
      private static final Log LOG = LogFactory.getLog(InstallDbAction.class);
  
      private boolean mustBackup;
      private BackupDbAction backupDbAction;
      private File backupFile;
  
      /**
       * <p>Constructor for InstallDbAction.</p>
       *
       * @param handler a {@link AbstractMainUIHandler} object.
       */
      public InstallDbAction(AbstractMainUIHandler handler) {
          super(handler, true);
          setActionDescription(t("quadrige3.dbManager.action.installDb.tip"));
      }
  
      /**
       * {@inheritDoc}
       */
      @Override
      public boolean prepareAction() throws Exception {
          boolean doAction = super.prepareAction();
          mustBackup = getContext().isDbExist();
  
          if (doAction) {
  
              // check db url is reachable
              doAction = getContext().checkUpdateReachable(getConfig().getInstallDbUrl(), true);
          }
  
          if (doAction && mustBackup) {
  
              displayInfoMessage(
                      t("quadrige3.dbManager.title.backup.db"),
                      t("quadrige3.dbManager.action.installDb.backup.db")
              );
  
              backupDbAction = getActionFactory().createLogicAction(getHandler(), BackupDbAction.class);
              backupDbAction.prepareAction();
              backupFile = backupDbAction.getBackupFile();
  
              if (backupFile == null) {
  
                  displayWarningMessage(
                          t("quadrige3.dbManager.title.backup.db"),
                          t("quadrige3.dbManager.action.installDb.no.backup.db.choosen")
                  );
  
                 doAction = false;
             }
 
         }
 
         return doAction;
     }
 
     /**
      * {@inheritDoc}
      */
     @Override
     public void doAction() {
 
         if (mustBackup) {
             Preconditions.checkNotNull(backupDbAction);
             Preconditions.checkNotNull(backupFile);
         }
 
         createProgressionUIModel();
 
         // perform backup
         if (mustBackup) {
             // execute backup action
             getActionEngine().runInternalAction(backupDbAction);
             // close database
             getContext().closePersistenceService();
             getContext().clearDbContext();
             // clean database directory
             PersistenceServiceHelper.deleteDatabaseDirectory(getProgressionUIModel());
             getContext().setDbExist(false);
         }
 
         // Make sure there is no version on DB directory.
         // This can occur during dev, when running DB in server mode
         File dbVersionFile = new File(getConfig().getDbDirectory(), Daos.DB_VERSION_FILE);
         if (dbVersionFile.exists()) {
             FileUtils.deleteQuietly(dbVersionFile);
         }
 
         // ------------------------------------------------------------------ //
         // --- install db                                                     //
         // ------------------------------------------------------------------ //
         File current = getConfig().getDataDirectory();
         String url = getConfig().getInstallDbUrl();
 
         if (LOG.isInfoEnabled()) {
             LOG.info(String.format("Try to install / update db (current data location: %s), using update url: %s", current, url));
         }
 
         File dest = new File(getConfig().getBaseDirectory(), "NEW");
 
         getProgressionUIModel().setTotal(0);
         getProgressionUIModel().setMessage(t("quadrige3.dbManager.action.upgradeDb.check"));
         DatabaseUpdaterCallBack callback
                 = new DatabaseUpdaterCallBack(this, getProgressionUIModel());
         ApplicationUpdater up = new ApplicationUpdater();
         up.update(url,
                 current,
                 dest,
                 false,
                 callback,
                 getProgressionUIModel());
 
         if (!callback.isDbInstalled()) {
             throw new QuadrigeBusinessException(t("quadrige3.dbManager.action.installDb.error"));
         }
 
         getProgressionUIModel().setMessage(t("quadrige3.dbManager.action.upgradeDb.opening"));
 
         // Migrate installed DB (Mantis #40873)
         PersistenceServiceHelper.migrateDbName();
 
         getContext().setDbExist(true);
         getContext().setDbJustInstalled(true);
 
         // ------------------------------------------------------------------ //
         // --- open db                                                        //
         // ------------------------------------------------------------------ //
         getActionEngine().runInternalAction(getHandler(), OpenDbAction.class);
     }
 
     @Override
     protected void releaseAction() {
         backupFile = null;
         backupDbAction = null;
         super.releaseAction();
     }
 
 }