package javango.test;
import java.util.HashMap;
import java.util.Map;
import com.google.inject.Inject;
import com.google.inject.Injector;
import javango.contrib.sessions.Session;
import javango.core.User;
import javango.core.servers.RequestProcessor;
import javango.http.HttpException;
import javango.http.HttpRequest;
import javango.http.HttpResponse;
public class Client {
// TODO Probably should allow Guice to get this instance, giving ability for a singleton or an object per requst
RequestProcessor requestProcessor;
HttpRequest request;
User user;
Session session = new TestSession();
@Inject
public Client(Injector injector) throws Exception {
requestProcessor = injector.getInstance(RequestProcessor.class);
}
public HttpResponse get(String url) throws HttpException {
String[] urlAndParams = url.split("\\?");
if (request != null) request.close();
request = new HttpRequest();
request.setContext("/");
request.setUser(user);
request.setPath(urlAndParams[0]);
request.setMethod("GET");
request.setSession(session);
Map<String, String[]> parameters = new HashMap<String, String[]>();
if (urlAndParams.length > 1) {
String[] params = urlAndParams[1].split("&");
for(String param : params) {
String[] value = param.split("=");
parameters.put(value[0], new String[]{value[1]});
}
}
request.setParameterMap(parameters);
return requestProcessor.service(request);
}
public HttpResponse post(String url) throws HttpException {
return post(url, new HashMap<String, String[]>());
}
public HttpResponse post(String url, Map<String, String[]> postData) throws HttpException {
String[] urlAndParams = url.split("\\?");
if (request != null) request.close();
request = new HttpRequest();
request.setContext("/");
request.setUser(user);
request.setPath(urlAndParams[0]);
request.setMethod("POST");
request.setSession(session);
if (urlAndParams.length > 1) {
String[] params = urlAndParams[1].split("&");
for(String param : params) {
String[] value = param.split("=");
postData.put(value[0], new String[]{value[1]});
}
}
request.setParameterMap(postData);
return requestProcessor.service(request);
}
public void close() throws HttpException {
if (request != null) request.close();
request = null;
}
public User getUser() {
return user;
}
public void setUser(User user) {
this.user = user;
}
public Session getSession() {
return session;
}
public void setSession(Session session) {
this.session = session;
}
}