org.codehaus.mojo.wagon.shared.DefaultWagonUpload Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wagon-maven-plugin Show documentation
Maven 2 plugin that can be used to access various operations on a given URL using a supported maven wagon. Supports recursive upload, download, and list directory content functionality. The Kuali version of this plugin exists, because the Maven version ignores the filePermissions tag from settings.xml when uploading artifacts to a repository.
There is a newer version: 1.0.5
Show newest version
/**
 * Copyright 2008-2012 The Kuali Foundation
 *
 * Licensed under the Educational Community 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.opensource.org/licenses/ecl2.php
 *
 * 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.
 */
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();
        }
    }

}