package fr.ifremer.quadrige3.ui.swing.synchro.action;
   
   /*-
    * #%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.ZipUtils;
  import fr.ifremer.quadrige3.core.service.ClientServiceLocator;
  import fr.ifremer.quadrige3.core.service.http.HttpService;
  import fr.ifremer.quadrige3.ui.swing.synchro.SynchroUIHandler;
  import org.apache.commons.io.FileUtils;
  import org.apache.commons.io.FilenameUtils;
  import org.apache.commons.lang3.StringUtils;
  import org.apache.http.client.methods.HttpGet;
  
  import java.io.*;
  
  import static org.nuiton.i18n.I18n.t;
  
  /**
   * <p>Abstract AbstractDownloadAction class.</p>
   *
   * @author Ludovic Pecquot <ludovic.pecquot@e-is.pro>
   */
  public abstract class AbstractDownloadAction extends AbstractSynchroAction {
  
      private boolean downloaded;
      private boolean toUncompress;
      private File resultDir = null;
  
      /**
       * <p>getTargetDirectory.</p>
       *
       * @return a {@link File} object.
       */
      public abstract File getTargetDirectory();
  
      /**
       * <p>getUrlPath.</p>
       *
       * @return a {@link String} object.
       */
      public abstract String getUrlPath();
  
      /**
       * <p>Constructor for AbstractDownloadAction.</p>
       *
       * @param handler a {@link SynchroUIHandler} object.
       */
      public AbstractDownloadAction(SynchroUIHandler handler) {
          super(handler);
  
          toUncompress = true;
      }
  
      /**
       * <p>Setter for the field <code>toUncompress</code>.</p>
       *
       * @param toUncompress a boolean.
       */
      public void setToUncompress(boolean toUncompress) {
          this.toUncompress = toUncompress;
      }
  
      /**
       * {@inheritDoc}
       */
      @Override
      public boolean initAction() {
          super.initAction();
  
          getHandler().showProgressCard();
  
          downloaded = false;
          getModel().setWorkingDirectory(null);
  
          return true;
      }
  
      /**
      * {@inheritDoc}
      */
     @Override
     public void doAction() throws Exception {
 
         HttpService httpService = ClientServiceLocator.instance().getHttpService();
         String fileName = getModel().getTransferFilename();
 
         if (StringUtils.isBlank(fileName)) {
             throw new IOException("can not get the file name to download");
         }
 
         File downloadedFile = new File(getConfig().getTempDirectory(), fileName);
 
         getModel().getProgressionModel().setMessage(t("quadrige3.synchro.progress.download"));
 
         // download the zip file
         HttpGet downloadHttpGet = new HttpGet(httpService.getPathUri(getUrlPath(), fileName));
         httpService.executeDownloadFileRequest(getContext().getAuthenticationInfo(), downloadHttpGet, getModel().getProgressionModel(), downloadedFile);
 
         // Compute the target directory for user
         File userDir = getTargetDirectory();
         FileUtils.forceMkdir(userDir);
 
         if (toUncompress && ZipUtils.isZipFile(downloadedFile)) {
 
             // Unzip it
             resultDir = new File(userDir, FilenameUtils.getBaseName(fileName));
             getModel().getProgressionModel().setCurrent(0);
             getModel().getProgressionModel().setMessage(t("quadrige3.synchro.progress.uncompress"));
             ZipUtils.uncompressFileToPath(downloadedFile.toPath(), resultDir.toPath(), getModel().getProgressionModel(), true);
 
         } else {
 
             // move it
             resultDir = userDir;
             FileUtils.moveFileToDirectory(downloadedFile, resultDir, true);
         }
 
         downloaded = true;
 
     }
 
     /**
      * {@inheritDoc}
      */
     @Override
     public void doneAction() {
         updateModel();
     }
 
     /**
      * {@inheritDoc}
      */
     @Override
     public void failedAction(Throwable ex) {
         updateModel();
         super.failedAction(ex);
     }
 
     private void updateModel() {
         getModel().setWorkingDirectory((downloaded && resultDir != null && resultDir.isDirectory()) ? resultDir : null);
     }
 
 }