// getConnectionInfiniteCookie
public static String getConnectionInfiniteCookie(URLConnection urlConnection) {
Map<String, List<String>> headers = urlConnection.getHeaderFields();
Set<Map.Entry<String, List<String>>> entrySet = headers.entrySet();
for (Map.Entry<String, List<String>> entry : entrySet) {
String headerName = entry.getKey();
if (headerName != null && headerName.equals("Set-Cookie")) {
List<String> headerValues = entry.getValue();
for (String value : headerValues) {
if (value.contains("infinitecookie")) {
int equalsLoc = value.indexOf("=");
int semicolonLoc = value.indexOf(";");
return value.substring(equalsLoc + 1, semicolonLoc);
}
}
}
}
return null;
} // TESTED
// postToRestfulApi -
// Note: params in the addr field need to be URLEncoded
private String postToRestfulApi(
String addr, String data, HttpServletRequest request, HttpServletResponse response) {
if (localCookie) CookieHandler.setDefault(cm);
String result = "";
try {
URLConnection connection = new URL(API_ROOT + addr).openConnection();
String cookieVal = getBrowserInfiniteCookie(request);
if (cookieVal != null) {
connection.addRequestProperty("Cookie", "infinitecookie=" + cookieVal);
connection.setDoInput(true);
}
connection.setDoOutput(true);
connection.setRequestProperty("Accept-Charset", "UTF-8");
// Post JSON string to URL
OutputStream os = connection.getOutputStream();
byte[] b = data.getBytes("UTF-8");
os.write(b);
// Receive results back from API
InputStream is = connection.getInputStream();
result = IOUtils.toString(is, "UTF-8");
String newCookie = getConnectionInfiniteCookie(connection);
if (newCookie != null && response != null) {
setBrowserInfiniteCookie(response, newCookie, request.getServerPort());
}
} catch (Exception e) {
// System.out.println("Exception: " + e.getMessage());
}
return result;
} // TESTED
// callRestfulApi - Calls restful API and returns results as a string
public String callRestfulApi(
String addr, HttpServletRequest request, HttpServletResponse response) {
if (localCookie) CookieHandler.setDefault(cm);
try {
ByteArrayOutputStream output = new ByteArrayOutputStream();
URL url = new URL(API_ROOT + addr);
URLConnection urlConnection = url.openConnection();
String cookieVal = getBrowserInfiniteCookie(request);
if (cookieVal != null) {
urlConnection.addRequestProperty("Cookie", "infinitecookie=" + cookieVal);
urlConnection.setDoInput(true);
urlConnection.setDoOutput(true);
urlConnection.setRequestProperty("Accept-Charset", "UTF-8");
}
IOUtils.copy(urlConnection.getInputStream(), output);
String newCookie = getConnectionInfiniteCookie(urlConnection);
if (newCookie != null && response != null) {
setBrowserInfiniteCookie(response, newCookie, request.getServerPort());
}
return output.toString();
} catch (IOException e) {
System.out.println(e.getMessage());
return null;
}
} // TESTED