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(
  37       String appId, String majorVersionId, HttpServletRequest request, Long deadlineMillis) {
  38     super(appId, majorVersionId, 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   }
  56
  57   public boolean isLoggedIn() {
  58     return loginCookieData != null;
  59   }
  60
  61   public String getEmail() {
  62     if (loginCookieData == null) {
  63       return null;
  64     } else {
  65       return loginCookieData.getEmail();
  66     }
  67   }
  68
  69   public boolean isAdmin() {
  70     if (loginCookieData == null) {
  71       return false;
  72     } else {
  73       return loginCookieData.isAdmin();
  74     }
  75   }
  76 }