gts/AppServer_Java/src/com/google/appengine/api/users/dev/LocalUserService.java at c228aaaf29fe70eabbb2abe77bcdcd893b60f324 · AppScale/gts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
package com.google.appengine.api.users.dev;


import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.Map;

import com.google.appengine.tools.development.AbstractLocalRpcService;
import com.google.appengine.tools.development.LocalRpcService;
import com.google.appengine.tools.development.LocalServiceContext;
import com.google.appengine.tools.development.ServiceProvider;
import com.google.appengine.tools.resources.ResourceLoader;
import com.google.apphosting.api.UserServicePb;


@ServiceProvider(LocalRpcService.class)
public final class LocalUserService extends AbstractLocalRpcService
{
    /*
     * AppScale - every method in this class has been modified
     */
    private static final String LOGIN_URL    = "/_ah/login";
    public static final String  PACKAGE      = "user";
    private static final String LOGIN_SERVER = System.getProperty("LOGIN_SERVER");

    public static final String OAUTH_CONSUMER_KEY_PROPERTY = "oauth.consumer_key";
    public static final String OAUTH_EMAIL_PROPERTY = "oauth.email";
    public static final String OAUTH_USER_ID_PROPERTY = "oauth.user_id";
    public static final String OAUTH_AUTH_DOMAIN_PROPERTY = "oauth.auth_domain";
    public static final String OAUTH_IS_ADMIN_PROPERTY = "oauth.is_admin";
    private String oauthConsumerKey = "example.com";
    private String oauthEmail = "example@example.com";
    private String oauthUserId = "0";
    private String oauthAuthDomain = "gmail.com";
    private boolean oauthIsAdmin = false;
    private final String NGINX_ADDR = "NGINX_ADDR";
    private final String DASHBOARD_HTTPS_PORT = "1443";

    public UserServicePb.CreateLoginURLResponse createLoginURL( LocalRpcService.Status status, UserServicePb.CreateLoginURLRequest request )
    {
        UserServicePb.CreateLoginURLResponse response = new UserServicePb.CreateLoginURLResponse();
        String destinationUrl = request.getDestinationUrl();
        if(destinationUrl != null && destinationUrl.startsWith("/"))
        {
            String nginxPort = ResourceLoader.getNginxPort();
            destinationUrl = "http://" + System.getProperty(NGINX_ADDR) + ":" + nginxPort + destinationUrl;
        }

        response.setLoginUrl(LOGIN_URL + "?continue=" + encode(destinationUrl));
        return response;
    }

    public UserServicePb.CreateLogoutURLResponse createLogoutURL( LocalRpcService.Status status, UserServicePb.CreateLogoutURLRequest request )
    {
        UserServicePb.CreateLogoutURLResponse response = new UserServicePb.CreateLogoutURLResponse();
        String nginxPort = ResourceLoader.getNginxPort();
        String redirect_url = "https://" + LOGIN_SERVER + ":" + DASHBOARD_HTTPS_PORT + "/logout?continue=http://" + LOGIN_SERVER + ":" + nginxPort;
        response.setLogoutUrl(redirect_url);

        return response;
    }

    public UserServicePb.CheckOAuthSignatureResponse checkOAuthSignature( LocalRpcService.Status status, UserServicePb.CheckOAuthSignatureRequest request )
    {
        UserServicePb.CheckOAuthSignatureResponse response = new UserServicePb.CheckOAuthSignatureResponse();
        response.setOauthConsumerKey(this.oauthConsumerKey);
        return response;
    }

    public UserServicePb.GetOAuthUserResponse getOAuthUser( LocalRpcService.Status status, UserServicePb.GetOAuthUserRequest request )
    {
        UserServicePb.GetOAuthUserResponse response = new UserServicePb.GetOAuthUserResponse();
        response.setEmail(this.oauthEmail);
        response.setUserId(this.oauthUserId);
        response.setAuthDomain(this.oauthAuthDomain);
        response.setIsAdmin(this.oauthIsAdmin);

        return response;
    }

    public String getPackage()
    {
        return "user";
    }

    public void init( LocalServiceContext context, Map<String, String> properties )
    {
      String oauthConsumerKeyProp = (String)properties.get("oauth.consumer_key");
      if (oauthConsumerKeyProp != null) {
        this.oauthConsumerKey = oauthConsumerKeyProp;
      }
      String oauthEmailProp = (String)properties.get("oauth.email");
      if (oauthEmailProp != null) {
        this.oauthEmail = oauthEmailProp;
      }
      String oauthUserIdProp = (String)properties.get("oauth.user_id");
      if (oauthUserIdProp != null) {
        this.oauthUserId = oauthUserIdProp;
      }
      String oauthAuthDomainProp = (String)properties.get("oauth.auth_domain");
      if (oauthAuthDomainProp != null) {
        this.oauthAuthDomain = oauthAuthDomainProp;
      }
      String oauthIsAdminProp = (String)properties.get("oauth.is_admin");
      if (oauthIsAdminProp != null)
        this.oauthIsAdmin = Boolean.valueOf(oauthIsAdminProp).booleanValue();
    }

    public void start()
    {}

    public void stop()
    {}

    private static String encode( String url )
    {
        try
        {
            return URLEncoder.encode(url, "UTF-8");
        }
        catch (UnsupportedEncodingException ex)
        {
            throw new RuntimeException("Could not find UTF-8 encoding", ex);
        }
    }
}