package fr.ifremer.quadrige2.ui.swing.common.synchro.action;
   
   /*-
    * #%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 fr.ifremer.quadrige2.ui.swing.common.synchro.SynchroUIHandler;
  import org.apache.commons.io.FileUtils;
  import org.apache.commons.io.FilenameUtils;
  import org.apache.commons.lang3.StringUtils;
  import org.apache.http.HttpException;
  import org.apache.http.HttpStatus;
  import org.apache.http.client.CredentialsProvider;
  import org.apache.http.client.methods.CloseableHttpResponse;
  import org.apache.http.client.methods.HttpGet;
  import org.apache.http.client.utils.URIBuilder;
  import org.apache.http.impl.client.CloseableHttpClient;
  import org.apache.http.impl.client.HttpClients;
  import org.nuiton.util.ZipUtil;
  
  import java.io.*;
  import java.net.URI;
  
  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 {
 
         CredentialsProvider credentialsProvider = getCredentialsProvider();
 
         try (CloseableHttpClient httpClient = HttpClients.custom().setDefaultRequestConfig(getRequestConfig())
                 .setDefaultCredentialsProvider(credentialsProvider)
                 .build()) {
 
             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);
             if (!downloadedFile.createNewFile()) {
                 throw new IOException(String.format("file %s already exists", downloadedFile.getAbsolutePath()));
             }
 
             getModel().getProgressionModel().setMessage(t("quadrige2.synchro.progress.download"));
 
             // download the zip file
             URI downloadUri = new URIBuilder(getAppendedPath(getUrlPath(), fileName)).build();
             HttpGet downloadHttpGet = new HttpGet(downloadUri);
 
             try (CloseableHttpResponse response = httpClient.execute(downloadHttpGet);
                  FileOutputStream fileOutputStream = new FileOutputStream(downloadedFile)) {
 
                 if (response.getStatusLine().getStatusCode() != HttpStatus.SC_OK) {
                     throw new HttpException(
                             String.format("error while downloading file '%s'; server responds: %s", fileName, response.getStatusLine().getReasonPhrase()));
                 }
 
                 downloadWithProgression(response.getEntity().getContent(), response.getEntity().getContentLength(), fileOutputStream);
             }
 
             getModel().getProgressionModel().setMessage(t("quadrige2.synchro.progress.uncompress"));
             getModel().getProgressionModel().setIndeterminate(true);
 
             // Compute the target directory for user
             File userDir = getTargetDirectory();
             FileUtils.forceMkdir(userDir);
 
             if (toUncompress && ZipUtil.isZipFile(downloadedFile)) {
 
                 // Unzip it
                 resultDir = new File(userDir, FilenameUtils.getBaseName(fileName));
                 ZipUtil.uncompress(downloadedFile, resultDir);
                 downloadedFile.delete();
             } else {
 
                 // move it
                 resultDir = userDir;
                 FileUtils.moveFileToDirectory(downloadedFile, resultDir, true);
             }
 
             downloaded = true;
 
         }
     }
 
     private void downloadWithProgression(InputStream input, long inputSize, OutputStream output) throws IOException {
         if (input == null || output == null) {
             return;
         }
 
         try (BufferedInputStream bis = new BufferedInputStream(input);
              BufferedOutputStream bos = new BufferedOutputStream(output);) {
 
             // initialize progression
             getModel().getProgressionModel().setSize(inputSize);
 
             long count = 0L;
             int bytesReaded;
             byte[] bytes = new byte[4096];
 
             while ((bytesReaded = bis.read(bytes)) != -1) {
 
                 // copy
                 bos.write(bytes, 0, bytesReaded);
 
                 // progresion increment
                 count += bytesReaded;
                 getModel().getProgressionModel().setCurrent(count);
 
             }
 
         }
     }
 
     /** {@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);
     }
 
 }