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 fr.ifremer.quadrige3.core.dao.technical.Assert;
  import fr.ifremer.quadrige3.core.service.persistence.PersistenceServiceHelper;
  import fr.ifremer.quadrige3.ui.swing.action.AbstractMainUIAction;
  import fr.ifremer.quadrige3.ui.swing.content.AbstractMainUIHandler;
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  
  import java.io.File;
  
  import static org.nuiton.i18n.I18n.t;
  
  /**
   * To import a db and use it.
   *
   * @since 1.1
   */
  public class RestoreDbAction extends AbstractMainUIAction {
  
      /**
       * Logger.
       */
      private static final Log LOG = LogFactory.getLog(RestoreDbAction.class);
  
      private boolean mustBackup;
      private BackupDbAction backupDbAction;
      private File backupFile;
      private File restoreFile;
      private String jdbcUrl;
  
      /**
       * <p>Constructor for RestoreDbAction.</p>
       *
       * @param handler a {@link AbstractMainUIHandler} object.
       */
      public RestoreDbAction(AbstractMainUIHandler handler) {
          super(handler, true);
          setActionDescription(t("quadrige3.dbManager.action.restoreDb.tip"));
      }
  
      /**
       * <p>Setter for the field <code>restoreFile</code>.</p>
       *
       * @param restoreFile a {@link File} object.
       */
      public void setRestoreFile(File restoreFile) {
          this.restoreFile = restoreFile;
      }
  
      /**
       * {@inheritDoc}
       */
      @Override
      public boolean prepareAction() throws Exception {
  
          boolean doAction = super.prepareAction();
          mustBackup = getContext().isDbExist();
  
          if (doAction) {
  
              if (mustBackup) {
  
                  displayInfoMessage(
                          t("quadrige3.dbManager.title.backup.db"),
                          t("quadrige3.dbManager.action.restoreDb.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.restoreDb.no.backup.db.choosen")
                     );
 
                     doAction = false;
                 }
 
             }
 
             if (doAction && restoreFile == null) {
 
                 // choose file to import
                 restoreFile = chooseFile(
                         t("quadrige3.dbManager.title.choose.dbImportFile"),
                         t("quadrige3.dbManager.action.chooseDbFile"),
                         "^.*\\.zip", t("quadrige3.common.file.zip")
                 );
 
                 if (restoreFile == null) {
 
                     displayWarningMessage(
                             t("quadrige3.dbManager.title.choose.dbImportFile"),
                             t("quadrige3.dbManager.action.restoreDb.no.import.file.choosen")
                     );
 
                     doAction = false;
                 }
             }
         }
         return doAction;
     }
 
     /**
      * {@inheritDoc}
      */
     @Override
     public void doAction() throws Exception {
         jdbcUrl = getConfig().getJdbcUrl();
         Assert.notNull(restoreFile);
         if (mustBackup) {
             Assert.notNull(backupDbAction);
             Assert.notNull(backupFile);
         }
 
         if (LOG.isInfoEnabled()) {
             LOG.info("Will import db: " + restoreFile);
         }
 
         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);
         }
 
         // restore database
         getProgressionUIModel().setMessage(t("quadrige3.dbManager.action.restoreDb.step.unzipArchive", restoreFile));
         PersistenceServiceHelper.restoreDatabase(restoreFile.toPath(), getProgressionUIModel());
 
         // open database
         OpenDbAction openDbAction = getActionFactory().createLogicAction(getHandler(), OpenDbAction.class);
         openDbAction.setAfterImportDb(true);
         openDbAction.prepareAction();
         getActionEngine().runInternalAction(openDbAction);
 
     }
 
     /**
      * {@inheritDoc}
      */
     @Override
     public void postSuccessAction() {
         getHandler().reloadDbManagerText();
 
         super.postSuccessAction();
 
         sendMessage(t("quadrige3.flash.info.db.imported", jdbcUrl));
 
         // make sure title is reloaded
         getHandler().changeTitle();
     }
 
     /**
      * {@inheritDoc}
      */
     @Override
     public void postFailedAction(Throwable error) {
         getHandler().reloadDbManagerText();
         super.postFailedAction(error);
     }
 
     /**
      * {@inheritDoc}
      */
     @Override
     public void releaseAction() {
         restoreFile = backupFile = null;
         backupDbAction = null;
         super.releaseAction();
     }
 
 }