ServiceFilterResponseImpl.java

/*
Copyright (c) Microsoft
All Rights Reserved
Apache 2.0 License
 
   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.
 
See the Apache Version 2.0 License for specific language governing permissions and limitations under the License.
 */
/*
 * ServiceFilterResponseImpl.java
 */

package com.azure.webapi;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;

import org.apache.http.Header;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.StatusLine;

import com.microsoft.aad.test.todoapi.Constants;

/**
 * ServiceFilterResponse implementation
 */
public class ServiceFilterResponseImpl implements ServiceFilterResponse {
	/**
	 * The original response
	 */
	private HttpResponse mResponse;

	/**
	 * The response content
	 */
	private byte[] mResponseContent;

	/**
	 * Constructor
	 * 
	 * @param response
	 *            The request's response
	 * @throws IOException
	 * @throws IllegalStateException
	 */
	public ServiceFilterResponseImpl(HttpResponse response)
			throws IllegalStateException, IOException {
		mResponse = response;
		mResponseContent = null;

		// Get the response's content
		HttpEntity entity = mResponse.getEntity();
		if (entity != null) {
			InputStream instream = entity.getContent();
			
			ByteArrayOutputStream out = new ByteArrayOutputStream();
			byte[] buffer = new byte[1024];
			int length;

			while ((length = instream.read(buffer)) != -1) out.write(buffer, 0, length);
			instream.close();

			mResponseContent = out.toByteArray();;
		} else {
			mResponseContent = null;
		}
	}

	@Override
	public Header[] getHeaders() {
		return mResponse.getAllHeaders();
	}

	@Override
	public String getContent() {
		if (mResponseContent != null) {
			String responseContent = null;
			try {
				responseContent = new String(mResponseContent, Constants.UTF8_ENCODING);
			} catch (UnsupportedEncodingException e) {
			}
			return responseContent;
		} else {
			return null;
		}
	}
	
	@Override
	public byte[] getRawContent() {
		return mResponseContent;
	}

	@Override
	public StatusLine getStatus() {
		return mResponse.getStatusLine();
	}
}