/*
 *  Copyright 2015 the original author or authors. 
 *  @https://github.com/scouter-project/scouter
 *
 *  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 scouter.xtra.httpclient;

import org.apache.http.Header;
import org.apache.http.HttpHost;
import org.apache.http.HttpRequest;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPut;
import org.apache.http.client.methods.HttpUriRequest;

import scouter.agent.proxy.IHttpClient;

public class HttpClient43 implements IHttpClient {
	public String getHost(Object o) {
		try {
			HttpHost host = (HttpHost) o;
			return host.toHostString();
		} catch (Exception e) {
			return null;
		}
	}

	public void addHeader(Object o, String key, String value) {
		if (o instanceof HttpRequest) {
			HttpRequest req = (HttpRequest) o;
			req.addHeader(key, value);
		}
	}

	public String getHeader(Object o, String key) {
		if (o instanceof HttpRequest) {
			HttpRequest req = (HttpRequest) o;
			Header h = req.getFirstHeader(key);
			if (h != null) {
				return h.getValue();
			}
		}
		return null;
	}

	public String getURI(Object o) {
		if (o instanceof HttpUriRequest) {
			HttpUriRequest req = (HttpUriRequest) o;
			return req.getURI().getPath();
		} else if (o instanceof HttpGet) {
			HttpGet req = (HttpGet) o;
			return req.getURI().getPath();
		} else if (o instanceof HttpPut) {
			HttpPut req = (HttpPut) o;
			return req.getURI().getPath();
		} else if (o instanceof HttpRequest) {
			HttpRequest req = (HttpRequest) o;
			return req.getRequestLine().getUri();
		}
		return o.toString();
	}
}