1 files changed, 137 insertions, 0 deletions
diff --git a/core/src/main/java/net/oauth/client/httpclient4/HttpMethodResponse.java b/core/src/main/java/net/oauth/client/httpclient4/HttpMethodResponse.java
new file mode 100755
index 0000000..1f7b747
--- /dev/null
+++ b/core/src/main/java/net/oauth/client/httpclient4/HttpMethodResponse.java
@@ -0,0 +1,137 @@
+/*
+ * Copyright 2008 Sean Sullivan
+ *
+ * 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 net.oauth.client.httpclient4;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.URL;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import net.oauth.OAuth;
+import net.oauth.client.ExcerptInputStream;
+import net.oauth.http.HttpMessage;
+import net.oauth.http.HttpResponseMessage;
+import org.apache.http.Header;
+import org.apache.http.HttpEntityEnclosingRequest;
+import org.apache.http.HttpResponse;
+import org.apache.http.client.methods.HttpRequestBase;
+
+/**
+ * An HttpResponse, encapsulated as an OAuthMessage.
+ * 
+ * This class relies on <a href="http://hc.apache.org">Apache HttpClient</a>
+ * version 4.
+ * 
+ * @author Sean Sullivan
+ * @hide
+ */
+public class HttpMethodResponse extends HttpResponseMessage
+{
+
+    /**
+     * Construct an OAuthMessage from the HTTP response, including parameters
+     * from OAuth WWW-Authenticate headers and the body. The header parameters
+     * come first, followed by the ones from the response body.
+     */
+    public HttpMethodResponse(HttpRequestBase request, HttpResponse response, byte[] requestBody,
+            String requestEncoding) throws IOException
+    {
+        super(request.getMethod(), new URL(request.getURI().toString()));
+        this.httpRequest = request;
+        this.httpResponse = response;
+        this.requestBody = requestBody;
+        this.requestEncoding = requestEncoding;
+        this.headers.addAll(getHeaders());
+    }
+
+    private final HttpRequestBase httpRequest;
+    private final HttpResponse httpResponse;
+    private final byte[] requestBody;
+    private final String requestEncoding;
+
+    @Override
+    public int getStatusCode()
+    {
+        return httpResponse.getStatusLine().getStatusCode();
+    }
+
+    @Override
+    public InputStream openBody() throws IOException
+    {
+        return httpResponse.getEntity().getContent();
+    }
+
+    private List<Map.Entry<String, String>> getHeaders()
+    {
+        List<Map.Entry<String, String>> headers = new ArrayList<Map.Entry<String, String>>();
+        Header[] allHeaders = httpResponse.getAllHeaders();
+        if (allHeaders != null) {
+            for (Header header : allHeaders) {
+                headers.add(new OAuth.Parameter(header.getName(), header.getValue()));
+            }
+        }
+        return headers;
+    }
+
+    /** Return a complete description of the HTTP exchange. */
+    @Override
+    public void dump(Map<String, Object> into) throws IOException
+    {
+        super.dump(into);
+        {
+            StringBuilder request = new StringBuilder(httpRequest.getMethod());
+            request.append(" ").append(httpRequest.getURI().getPath());
+            String query = httpRequest.getURI().getQuery();
+            if (query != null && query.length() > 0) {
+                request.append("?").append(query);
+            }
+            request.append(EOL);
+            for (Header header : httpRequest.getAllHeaders()) {
+                request.append(header.getName()).append(": ").append(header.getValue()).append(EOL);
+            }
+            if (httpRequest instanceof HttpEntityEnclosingRequest) {
+                HttpEntityEnclosingRequest r = (HttpEntityEnclosingRequest) httpRequest;
+                long contentLength = r.getEntity().getContentLength();
+                if (contentLength >= 0) {
+                    request.append("Content-Length: ").append(contentLength).append(EOL);
+                }
+            }
+            request.append(EOL);
+            if (requestBody != null) {
+                request.append(new String(requestBody, requestEncoding));
+            }
+            into.put(REQUEST, request.toString());
+        }
+        {
+            StringBuilder response = new StringBuilder();
+            String value = httpResponse.getStatusLine().toString();
+            response.append(value).append(EOL);
+            for (Header header : httpResponse.getAllHeaders()) {
+                String name = header.getName();
+                value = header.getValue();
+                response.append(name).append(": ").append(value).append(EOL);
+            }
+            response.append(EOL);
+            if (body != null) {
+                response.append(new String(((ExcerptInputStream) body).getExcerpt(),
+                        getContentCharset()));
+            }
+            into.put(HttpMessage.RESPONSE, response.toString());
+        }
+    }
+}