package org.codehaus.mojo.wagon.shared;
   
   /*
    * Licensed to the Apache Software Foundation (ASF) under one or more contributor license
    * agreements. See the NOTICE file distributed with this work for additional information regarding
    * copyright ownership. The ASF licenses this file to you under the Apache License, Version 2.0 (the
    * "License"); you may not use this file except in compliance with the License. You may obtain a
    * copy of the License at
    *
   * http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software distributed under the License
   * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
   * or implied. See the License for the specific language governing permissions and limitations under
   * the License.
   */
  
  import java.io.File;
  import java.io.FileInputStream;
  import java.io.FileOutputStream;
  import java.io.IOException;
  import java.util.zip.ZipEntry;
  import java.util.zip.ZipOutputStream;
  
  import org.apache.maven.plugin.logging.Log;
  import org.apache.maven.shared.model.fileset.FileSet;
  import org.apache.maven.shared.model.fileset.util.FileSetManager;
  import org.apache.maven.wagon.CommandExecutor;
  import org.apache.maven.wagon.UnsupportedProtocolException;
  import org.apache.maven.wagon.Wagon;
  import org.apache.maven.wagon.WagonException;
  import org.codehaus.plexus.util.IOUtil;
  import org.codehaus.plexus.util.StringUtils;
  
  /**
   * @plexus.component role="org.codehaus.mojo.wagon.shared.WagonUpload" role-hint="default"
   */
  
  public class DefaultWagonUpload implements WagonUpload {
      public void upload(Wagon wagon, FileSet fileset, Log logger) throws WagonException {
  
          FileSetManager fileSetManager = new FileSetManager(logger, logger.isDebugEnabled());
  
          String[] files = fileSetManager.getIncludedFiles(fileset);
  
          String url = wagon.getRepository().getUrl() + "/";
  
          if (files.length == 0) {
              logger.info("Nothing to upload.");
              return;
          }
  
          for (int i = 0; i < files.length; ++i) {
              String relativeDestPath = StringUtils.replace(files[i], "\\", "/");
  
              if (!StringUtils.isBlank(fileset.getOutputDirectory())) {
                  relativeDestPath = fileset.getOutputDirectory() + "/" + relativeDestPath;
              }
  
              File source = new File(fileset.getDirectory(), files[i]);
  
              logger.info("Uploading " + source + " to " + url + relativeDestPath + " ...");
  
              wagon.put(source, relativeDestPath);
          }
  
      }
  
      @Override
      public void upload(Wagon wagon, FileSet fileset, boolean optimize, Log logger) throws WagonException, IOException {
          if (!optimize) {
              upload(wagon, fileset, logger);
              return;
          }
  
          if (!(wagon instanceof CommandExecutor)) {
              throw new UnsupportedProtocolException("Wagon " + wagon.getRepository().getProtocol()
                      + " does not support optimize upload");
          }
  
          logger.info("Uploading " + fileset);
  
          File zipFile;
          zipFile = File.createTempFile("wagon", ".zip");
  
          try {
              FileSetManager fileSetManager = new FileSetManager(logger, logger.isDebugEnabled());
              String[] files = fileSetManager.getIncludedFiles(fileset);
  
              if (files.length == 0) {
                  logger.info("Nothing to upload.");
                  return;
              }
  
              logger.info("Creating " + zipFile + " ...");
              createZip(files, zipFile, fileset.getDirectory());
  
              String remoteFile = zipFile.getName();
              String remoteDir = fileset.getOutputDirectory();
             if (!StringUtils.isBlank(remoteDir)) {
                 remoteFile = remoteDir + "/" + remoteFile;
             }
 
             logger.info("Uploading " + zipFile + " to " + wagon.getRepository().getUrl() + "/" + remoteFile + " ...");
             wagon.put(zipFile, remoteFile);
 
             // We use the super quiet option here as all the noise seems to kill/stall the connection
             String command = "unzip -o -qq -d " + remoteDir + " " + remoteFile;
 
             try {
                 logger.info("Remote: " + command);
                 ((CommandExecutor) wagon).executeCommand(command);
             } finally {
                 command = "rm -f " + remoteFile;
                 logger.info("Remote: " + command);
 
                 ((CommandExecutor) wagon).executeCommand(command);
             }
 
         } finally {
             zipFile.delete();
         }
 
     }
 
     private static void createZip(String[] files, File zipName, String basedir) throws IOException {
         ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(zipName));
 
         try {
             for (int i = 0; i < files.length; i++) {
                 String file = files[i];
 
                 file = file.replace('\\', '/');
 
                 writeZipEntry(zos, new File(basedir, file), file);
             }
         } finally {
             IOUtil.close(zos);
         }
     }
 
     private static void writeZipEntry(ZipOutputStream jar, File source, String entryName) throws IOException {
         byte[] buffer = new byte[1024];
 
         int bytesRead;
 
         FileInputStream is = new FileInputStream(source);
 
         try {
             ZipEntry entry = new ZipEntry(entryName);
 
             jar.putNextEntry(entry);
 
             while ((bytesRead = is.read(buffer)) != -1) {
                 jar.write(buffer, 0, bytesRead);
             }
         } finally {
             is.close();
         }
     }
 
 }