java/src/main/com/google/appengine/tools/development/LocalHttpRequestEnvironment.java

   1 // Copyright 2009 Google Inc. All Rights Reserved.
   2
   3 package com.google.appengine.tools.development;
   4
   5 import com.google.appengine.api.NamespaceManager;
   6 import com.google.appengine.api.users.dev.LoginCookieUtils;
   7
   8 import javax.servlet.http.HttpServletRequest;
   9
  10 /**
  11  * {@code LocalHttpRequestEnvironment} is a simple {@link LocalEnvironment} that
  12  * retrieves authentication details from the cookie that is maintained by the
  13  * stub implementation of {@link UserService}.
  14  *
  15  */
  16 public class LocalHttpRequestEnvironment extends LocalEnvironment {
  17   /**
  18    * The name of the HTTP header specifying the default namespace
  19    * for API calls.
  20    */
  21   static final String DEFAULT_NAMESPACE_HEADER = "X-AppEngine-Default-Namespace";
  22   static final String CURRENT_NAMESPACE_HEADER = "X-AppEngine-Current-Namespace";
  23   private static final String CURRENT_NAMESPACE_KEY =
  24       NamespaceManager.class.getName() + ".currentNamespace";
  25   private static final String APPS_NAMESPACE_KEY =
  26       NamespaceManager.class.getName() + ".appsNamespace";
  27
  28   private static final String USER_ID_KEY =
  29       "com.google.appengine.api.users.UserService.user_id_key";
  30   private static final String USER_ORGANIZATION_KEY =
  31       "com.google.appengine.api.users.UserService.user_organization";
  32   private static final String X_APPENGINE_QUEUE_NAME = "X-AppEngine-QueueName";
  33
  34   private final LoginCookieUtils.CookieData loginCookieData;
  35
  36   public LocalHttpRequestEnvironment(String appId, String serverName, String majorVersionId,
  37       int instance, HttpServletRequest request, Long deadlineMillis) {
  38     super(appId, serverName, majorVersionId, instance, deadlineMillis);
  39     this.loginCookieData = LoginCookieUtils.getCookieData(request);
  40     String requestNamespace = request.getHeader(DEFAULT_NAMESPACE_HEADER);
  41     if (requestNamespace != null) {
  42       attributes.put(APPS_NAMESPACE_KEY, requestNamespace);
  43     }
  44     String currentNamespace = request.getHeader(CURRENT_NAMESPACE_HEADER);
  45     if (currentNamespace != null) {
  46       attributes.put(CURRENT_NAMESPACE_KEY, currentNamespace);
  47     }
  48     if (loginCookieData != null) {
  49       attributes.put(USER_ID_KEY, loginCookieData.getUserId());
  50       attributes.put(USER_ORGANIZATION_KEY, "");
  51     }
  52     if (request.getHeader(X_APPENGINE_QUEUE_NAME) != null) {
  53       attributes.put(ApiProxyLocalImpl.IS_OFFLINE_REQUEST_KEY, Boolean.TRUE);
  54     }
  55     attributes.put(HTTP_SERVLET_REQUEST, request);
  56   }
  57
  58   @Override
  59   public boolean isLoggedIn() {
  60     return loginCookieData != null;
  61   }
  62
  63   @Override
  64   public String getEmail() {
  65     if (loginCookieData == null) {
  66       return null;
  67     } else {
  68       return loginCookieData.getEmail();
  69     }
  70   }
  71
  72   @Override
  73   public boolean isAdmin() {
  74     if (loginCookieData == null) {
  75       return false;
  76     } else {
  77       return loginCookieData.isAdmin();
  78     }
  79   }
  80 }