source: josm/trunk/src/org/openstreetmap/josm/data/oauth/SignpostAdapters.java@ 9669

// License: GPL. For details, see LICENSE file.
package org.openstreetmap.josm.data.oauth;

import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.Map;

import oauth.signpost.AbstractOAuthConsumer;
import oauth.signpost.AbstractOAuthProvider;

import org.openstreetmap.josm.tools.HttpClient;

/**
 * Adapters to make {@code oauth.signpost} work with {@link HttpClient}.
 */
public final class SignpostAdapters {

    private SignpostAdapters() {
    }

    public static class OAuthProvider extends AbstractOAuthProvider {

        public OAuthProvider(String requestTokenEndpointUrl, String accessTokenEndpointUrl, String authorizationWebsiteUrl) {
            super(requestTokenEndpointUrl, accessTokenEndpointUrl, authorizationWebsiteUrl);
        }

        @Override
        protected HttpRequest createRequest(String endpointUrl) throws Exception {
            return new HttpRequest(HttpClient.create(new URL(endpointUrl), "GET"));
        }

        @Override
        protected HttpResponse sendRequest(oauth.signpost.http.HttpRequest request) throws Exception {
            return new HttpResponse(((HttpRequest) request).request.connect());
        }

        @Override
        protected void closeConnection(oauth.signpost.http.HttpRequest request, oauth.signpost.http.HttpResponse response) throws Exception {
            if (response != null) {
                ((HttpResponse) response).response.disconnect();
            } else if (request != null) {
                ((HttpRequest) request).request.disconnect();
            }
        }
    }

    public static class OAuthConsumer extends AbstractOAuthConsumer {

        public OAuthConsumer(String consumerKey, String consumerSecret) {
            super(consumerKey, consumerSecret);
        }

        @Override
        protected HttpRequest wrap(Object request) {
            return new HttpRequest(((HttpClient) request));
        }
    }

    private static class HttpRequest implements oauth.signpost.http.HttpRequest {
        private final HttpClient request;

        HttpRequest(HttpClient request) {
            this.request = request;
        }

        @Override
        public void setHeader(String name, String value) {
            request.setHeader(name, value);
        }

        @Override
        public String getMethod() {
            return request.getRequestMethod();
        }

        @Override
        public String getRequestUrl() {
            return request.getURL().toExternalForm();
        }

        @Override
        public String getContentType() {
            return request.getRequestHeader("Content-Type");
        }

        @Override
        public String getHeader(String name) {
            return request.getRequestHeader(name);
        }

        @Override
        public InputStream getMessagePayload() throws IOException {
            return null;
        }

        @Override
        public void setRequestUrl(String url) {
            throw new IllegalStateException();
        }

        @Override
        public Map<String, String> getAllHeaders() {
            throw new IllegalStateException();
        }

        @Override
        public Object unwrap() {
            throw new IllegalStateException();
        }
    }

    private static class HttpResponse implements oauth.signpost.http.HttpResponse {
        private final HttpClient.Response response;

        HttpResponse(HttpClient.Response response) {
            this.response = response;
        }

        @Override
        public int getStatusCode() throws IOException {
            return response.getResponseCode();
        }

        @Override
        public String getReasonPhrase() throws Exception {
            return response.getResponseMessage();
        }

        @Override
        public InputStream getContent() throws IOException {
            return response.getContent();
        }

        @Override
        public Object unwrap() {
            throw new IllegalStateException();
        }
    }

}