package fr.ifremer.quadrige3.batch.shape.action;
   
   /*-
    * #%L
    * Quadrige3 Batch :: Shape import/export
    * %%
    * Copyright (C) 2017 - 2018 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.batch.shape.config.ShapeConfigurationOption;
  import fr.ifremer.quadrige3.batch.shape.service.ImportShapeServiceImpl;
  import fr.ifremer.quadrige3.core.action.ActionUtils;
  import fr.ifremer.quadrige3.core.config.QuadrigeConfiguration;
  import fr.ifremer.quadrige3.core.dao.technical.Daos;
  import fr.ifremer.quadrige3.core.exception.QuadrigeTechnicalException;
  import org.apache.commons.collections.CollectionUtils;
  import org.apache.commons.io.FilenameUtils;
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  
  import java.io.File;
  import java.io.IOException;
  import java.nio.file.*;
  import java.util.List;
  import java.util.stream.Collectors;
  
  import static org.nuiton.i18n.I18n.t;
  
  public class ImportShapeAction {
  
      private static final Log log = LogFactory.getLog(ExportShapeAction.class);
  
      public void doAction() throws Exception {
  
          QuadrigeConfiguration config = QuadrigeConfiguration.getInstance();
          List<String> args = config.getApplicationConfig().getUnparsed();
          File inputDir = null;
          File outputDir = null;
  
          // Check connection
          ActionUtils.logConnectionProperties();
          boolean isValidConnection = Daos.isValidConnectionProperties(config.getJdbcDriver(),
                  config.getJdbcURL(),
                  config.getJdbcUsername(),
                  config.getJdbcPassword());
  
          if (!isValidConnection) {
              log.error(t("quadrige3.batches.server.connection.failed"));
              return;
          }
  
          if (CollectionUtils.size(args) == 0) {
              inputDir = config.getApplicationConfig().getOptionAsFile(ShapeConfigurationOption.INPUT_SHAPES_DIRECTORY.getKey());
              outputDir = config.getApplicationConfig().getOptionAsFile(ShapeConfigurationOption.OUTPUT_SHAPES_DIRECTORY.getKey());
          } else if (CollectionUtils.size(args) == 2) {
              inputDir = new File(args.get(0));
              outputDir = new File(args.get(1));
          }
  
          if (inputDir == null) {
              log.error(t("quadrige3.batch.shape.error.directory.input.null"));
              return;
          }
          if (!inputDir.exists()) {
              log.error(t("quadrige3.batch.shape.error.directory.input.notExists", inputDir.getAbsolutePath()));
              return;
          }
          if (!inputDir.isDirectory()) {
              log.error(t("quadrige3.batch.shape.error.directory.input.invalid", inputDir.getAbsolutePath()));
              return;
          }
  
          if (outputDir == null) {
              log.error(t("quadrige3.batch.shape.error.directory.output.null"));
              return;
          }
          if (!outputDir.exists() && !outputDir.mkdirs()) {
              log.error(t("quadrige3.batch.shape.error.directory.output.notCreated", outputDir.getAbsolutePath()));
              return;
          }
  
          // list files to import
          List<Path> inputFiles = Files.list(Paths.get(inputDir.getAbsolutePath()))
                  .filter(this::isZipFile)
                  .collect(Collectors.toList());
 
         if (CollectionUtils.isEmpty(inputFiles)) {
             log.info(t("quadrige3.batch.shape.error.directory.input.empty"));
             return;
         }
 
         // move all zip files to import into a temp directory
         Path tempDir = Paths.get(config.getTempDirectory().getAbsolutePath(), "import_" + System.currentTimeMillis());
         Files.createDirectories(tempDir);
 
         // loop into temp directory and call service for each file
         ImportShapeServiceImpl service = new ImportShapeServiceImpl();
         try {
 
             // execute validation & import
             inputFiles.forEach(file -> {
                 log.info(t("quadrige3.batch.shape.process.start", file));
                 // move to temp directory
                 Path fileToImport = moveFileToDirectory(file, tempDir);
                 service.importFromFile(file, fileToImport);
             });
 
         } finally {
 
             // move all files in temp directory to output
             Path outputPath = Paths.get(outputDir.getAbsolutePath());
             Files.list(tempDir).filter(Files::isRegularFile).forEach(file -> moveFileToDirectory(file, outputPath));
 
             // delete temp directory
             try {
                 Files.delete(tempDir);
             } catch (DirectoryNotEmptyException ignored) {
             }
 
         }
 
         log.info(t("quadrige3.batch.shape.process.end"));
     }
 
     private boolean isZipFile(Path file) {
         return Files.isRegularFile(file) && FilenameUtils.isExtension(file.getFileName().toString().toLowerCase(), "zip");
     }
 
     private static Path moveFileToDirectory(Path inputFile, Path outputDir) {
         try {
             Path outputFile = outputDir.resolve(inputFile.getFileName());
             Files.move(inputFile, outputFile, StandardCopyOption.REPLACE_EXISTING);
             return outputFile;
         } catch (IOException e) {
             throw new QuadrigeTechnicalException(t("quadrige3.batch.shape.error.file.move", inputFile, outputDir), e);
         }
     }
 
 }