bt.tracker.http.CommonsHttpResponseHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of bt-http-tracker-client Show documentation

Library for integration with HTTP BitTorrent trackers

There is a newer version: 1.10

Show newest version

/*
 * Copyright (c) 2016—2017 Andrei Tomashpolskiy and individual contributors.
 *
 * Licensed 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.
 */

package bt.tracker.http;

import bt.BtException;
import bt.tracker.TrackerResponse;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.StatusLine;
import org.apache.http.client.ResponseHandler;
import org.apache.http.entity.ContentType;
import org.apache.http.util.EntityUtils;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.nio.charset.Charset;

class CommonsHttpResponseHandler implements ResponseHandler {

    private Charset defaultHttpCharset;
    private HttpResponseHandler httpResponseHandler;

    CommonsHttpResponseHandler(HttpResponseHandler httpResponseHandler) {
        this.defaultHttpCharset = Charset.forName("ISO-8859-1");
        this.httpResponseHandler = httpResponseHandler;
    }

    @Override
    public TrackerResponse handleResponse(HttpResponse response) {

        final StatusLine statusLine = response.getStatusLine();
        final HttpEntity entity = response.getEntity();
        if (statusLine.getStatusCode() >= 300) {
            try {
                EntityUtils.consume(entity);
            } catch (IOException e) {
                // do nothing...
            }
            return TrackerResponse.exceptional(new BtException(
                    "Tracker returned error (" + statusLine.getStatusCode() + ": "
                            + statusLine.getReasonPhrase() + ")"));
        }

        if (entity == null) {
            return TrackerResponse.exceptional(new BtException("Tracker response is empty"));
        } else {
            try {

                Charset charset = null;
                ContentType contentType = ContentType.get(entity);
                if (contentType != null) {
                    charset = contentType.getCharset();
                }
                if (charset == null) {
                    charset = defaultHttpCharset;
                }

                ByteArrayOutputStream bytes = new ByteArrayOutputStream();
                entity.writeTo(bytes);
                return httpResponseHandler.handleResponse(bytes.toByteArray(), charset);
            } catch (IOException e) {
                return TrackerResponse.exceptional(new BtException("Failed to read tracker response", e));
            }
        }
    }
}