package ch.iterate.openstack.swift.handler;
import org.apache.http.HttpResponse;
import org.apache.http.client.ResponseHandler;
import org.apache.http.entity.ContentType;
import org.apache.http.protocol.HTTP;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.URI;
import java.nio.charset.Charset;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import ch.iterate.openstack.swift.AuthenticationResponse;
import ch.iterate.openstack.swift.Response;
import ch.iterate.openstack.swift.exception.AuthorizationException;
import ch.iterate.openstack.swift.exception.GenericException;
import ch.iterate.openstack.swift.model.Region;
import com.google.gson.JsonArray;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParseException;import com.google.gson.JsonParser;
public class AuthenticationJson20ResponseHandler implements ResponseHandler<AuthenticationResponse> {
public AuthenticationResponse handleResponse(final HttpResponse response) throws IOException {
if(response.getStatusLine().getStatusCode() == 200 ||
response.getStatusLine().getStatusCode() == 203) {
Charset charset = HTTP.DEF_CONTENT_CHARSET;
ContentType contentType = ContentType.get(response.getEntity());
if(contentType != null) {
if(contentType.getCharset() != null) {
charset = contentType.getCharset();
}
}
try {
final JsonParser parser = new JsonParser();
final JsonObject json = parser.parse(new InputStreamReader(response.getEntity().getContent(), charset)).getAsJsonObject();
final JsonObject auth = json.getAsJsonObject("access");
final JsonObject user = auth.getAsJsonObject("user");
String defaultRegion = null;
if(user.get("RAX-AUTH:defaultRegion") != null) {
defaultRegion = user.get("RAX-AUTH:defaultRegion").getAsString();
}
final String token = auth.getAsJsonObject("token").get("id").getAsString();
final JsonArray serviceCatalogs = auth.getAsJsonArray("serviceCatalog");
final Set<Region> regions = new HashSet<Region>();
final Map<String, String> cdnUrls = new HashMap<String, String>();
for(JsonElement e : serviceCatalogs) {
final JsonObject serviceCatalog = e.getAsJsonObject();
if(serviceCatalog.get("type").getAsString().equals("rax:object-cdn")) {
for(JsonElement endpoint : serviceCatalog.getAsJsonArray("endpoints")) {
String regionId = endpoint.getAsJsonObject().get("region").getAsString();
String publicUrl = endpoint.getAsJsonObject().get("publicURL").getAsString();
cdnUrls.put(regionId, publicUrl);
}
}
if(serviceCatalog.get("type").getAsString().equals("hpext:cdn")) {
for(JsonElement endpoint : serviceCatalog.getAsJsonArray("endpoints")) {
String regionId = endpoint.getAsJsonObject().get("region").getAsString();
String publicUrl = endpoint.getAsJsonObject().get("publicURL").getAsString();
cdnUrls.put(regionId, publicUrl);
}
}
if(serviceCatalog.get("type").getAsString().equals("object-store")) {
for(JsonElement endpoint : serviceCatalog.getAsJsonArray("endpoints")) {
String regionId = endpoint.getAsJsonObject().get("region").getAsString();
String publicUrl = endpoint.getAsJsonObject().get("publicURL").getAsString();
String cdnUrl = cdnUrls.containsKey(regionId) ? cdnUrls.get(regionId) : null;
regions.add(new Region(regionId, URI.create(publicUrl), cdnUrl == null ? null : URI.create(cdnUrl),
regionId.equals(defaultRegion)));
}
}
}
return new AuthenticationResponse(response, token, regions);
}
catch(JsonParseException e) {
throw new IOException(e.getMessage(), e);
}
}
else if(response.getStatusLine().getStatusCode() == 401 || response.getStatusLine().getStatusCode() == 403) {
throw new AuthorizationException(new Response(response));
}
throw new GenericException(new Response(response));
}
}