java/src/main/com/google/appengine/tools/remoteapi/HostedAppEngineClient.java

   1 // Copyright 2011 Google. All Rights Reserved.
   2 package com.google.appengine.tools.remoteapi;
   3
   4 import com.google.appengine.api.urlfetch.HTTPHeader;
   5 import com.google.appengine.api.urlfetch.HTTPMethod;
   6 import com.google.appengine.api.urlfetch.HTTPRequest;
   7 import com.google.appengine.api.urlfetch.HTTPResponse;
   8 import com.google.appengine.api.urlfetch.URLFetchService;
   9 import com.google.appengine.api.urlfetch.URLFetchServiceFactory;
  10
  11 import org.apache.commons.httpclient.Cookie;
  12 import org.apache.commons.httpclient.Header;
  13 import org.apache.commons.httpclient.HttpMethodBase;
  14
  15 import java.io.IOException;
  16 import java.net.URL;
  17 import java.util.List;
  18
  19 /**
  20  * An {@link AppEngineClient} implementation that uses {@link URLFetchService}.
  21  * This implementation must be used when the client is an App Engine container
  22  * since URLFetchService is the only way to make HTTP requests in this
  23  * environment.
  24  *
  25  */
  26 class HostedAppEngineClient extends AppEngineClient {
  27
  28   private final URLFetchService urlFetch = URLFetchServiceFactory.getURLFetchService();
  29
  30   HostedAppEngineClient(RemoteApiOptions options, List<Cookie> authCookies,
  31       String appId) {
  32     super(options, authCookies, appId);
  33   }
  34
  35   private void addCookies(HTTPRequest req) {
  36     for (Cookie cookie : getAuthCookies()) {
  37       req.addHeader(
  38           new HTTPHeader("Cookie", String.format("%s=%s", cookie.getName(), cookie.getValue())));
  39     }
  40   }
  41
  42   @Override
  43   Response get(String path) throws IOException {
  44     HTTPRequest req = new HTTPRequest(new URL(makeUrl(path)), HTTPMethod.GET);
  45     req.getFetchOptions().doNotFollowRedirects();
  46     for (String[] headerPair : getHeadersForGet()) {
  47       req.addHeader(new HTTPHeader(headerPair[0], headerPair[1]));
  48     }
  49     addCookies(req);
  50     HTTPResponse resp = urlFetch.fetch(req);
  51     return createResponse(resp);
  52   }
  53
  54   @Override
  55   Response post(String path, String mimeType, byte[] body) throws IOException {
  56     HTTPRequest req = new HTTPRequest(new URL(makeUrl(path)), HTTPMethod.POST);
  57     req.getFetchOptions().doNotFollowRedirects();
  58     for (String[] headerPair : getHeadersForPost(mimeType)) {
  59       req.addHeader(new HTTPHeader(headerPair[0], headerPair[1]));
  60     }
  61     addCookies(req);
  62     req.setPayload(body);
  63     HTTPResponse resp = urlFetch.fetch(req);
  64     return createResponse(resp);
  65   }
  66
  67   /**
  68    * A small hack to get access to
  69    * {@link HttpMethodBase#getContentCharSet(Header)} (which really ought to be
  70    * static) so that we don't need to write code to extract the charset out of
  71    * the Content-Type header.
  72    */
  73   private static class DummyMethod extends HttpMethodBase {
  74     private static final DummyMethod INSTANCE = new DummyMethod();
  75     static String getCharset(HTTPHeader header) {
  76       Header apacheHeader = new Header(header.getName(), header.getValue());
  77       return INSTANCE.getContentCharSet(apacheHeader);
  78     }
  79
  80     @Override
  81     public String getName() {
  82       return "dummy";
  83     }
  84   }
  85
  86   static Response createResponse(HTTPResponse resp) {
  87     byte[] body = resp.getContent();
  88     String charset = null;
  89     for (HTTPHeader header : resp.getHeaders()) {
  90       if (header.getName().toLowerCase().equals("content-type")) {
  91         charset = DummyMethod.getCharset(header);
  92         break;
  93       }
  94     }
  95     return new Response(resp.getResponseCode(), body, charset);
  96   }
  97
  98 }