org.adeptnet.maven.wagon.providers.http.HttpWagon Maven / Gradle / Ivy

Go to download
package org.adeptnet.maven.wagon.providers.http;

/*
 * 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 org.apache.maven.wagon.providers.http.AbstractHttpClientWagonFixed;
import org.apache.http.HttpEntity;
import org.apache.http.HttpException;
import org.apache.http.HttpStatus;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.maven.wagon.ResourceDoesNotExistException;
import org.apache.maven.wagon.TransferFailedException;
import org.apache.maven.wagon.authorization.AuthorizationException;
import org.apache.maven.wagon.shared.http.HtmlFileListParser;

import java.io.IOException;
import java.util.Collections;
import java.util.List;

/**
 * @author Michal Maczka
 */
public class HttpWagon extends AbstractHttpClientWagonFixed {

    @Override
    public List getFileList(String destinationDirectory)
            throws AuthorizationException, ResourceDoesNotExistException, TransferFailedException {
        return getFileList(getInitialBackoffSeconds(), destinationDirectory);
    }

    private List getFileList(int wait, String destinationDirectory)
            throws TransferFailedException, ResourceDoesNotExistException, AuthorizationException {
        if (destinationDirectory.length() > 0 && !destinationDirectory.endsWith("/")) {
            destinationDirectory += "/";
        }

        String url = getRepository().getUrl() + "/" + destinationDirectory;

        HttpGet getMethod = new HttpGet(url);

        try {
            CloseableHttpResponse response = execute(getMethod);
            try {
                int statusCode = response.getStatusLine().getStatusCode();

                fireTransferDebug(url + " - Status code: " + statusCode);

                switch (statusCode) {
                    case HttpStatus.SC_OK:
                        break;

                    case HttpStatus.SC_FORBIDDEN:
                        throw new AuthorizationException("Access denied to: " + url);

                    case HttpStatus.SC_UNAUTHORIZED:
                        throw new AuthorizationException("Not authorized.");

                    case HttpStatus.SC_PROXY_AUTHENTICATION_REQUIRED:
                        throw new AuthorizationException("Not authorized by proxy.");

                    case HttpStatus.SC_NOT_FOUND:
                        throw new ResourceDoesNotExistException("File: " + url + " does not exist");

                    case SC_TOO_MANY_REQUESTS:
                        return getFileList(backoff(wait, url), destinationDirectory);

                    //add more entries here
                    default:
                        throw new TransferFailedException(
                                "Failed to transfer file: " + url + ". Return code is: " + statusCode);
                }
                HttpEntity entity = response.getEntity();
                if (entity != null) {
                    return HtmlFileListParser.parseFileList(url, entity.getContent());
                } else {
                    return Collections.emptyList();
                }

            } finally {
                response.close();
            }
        } catch (IOException e) {
            throw new TransferFailedException("Could not read response body.", e);
        } catch (HttpException e) {
            throw new TransferFailedException("Could not read response body.", e);
        } catch (InterruptedException e) {
            throw new TransferFailedException("Unable to wait for resource.", e);
        }
    }

}