#!/usr/bin/env python

#

# Copyright 2007 Google Inc.

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

#     http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

#

"""Trivial implementation of the UserService."""

import os

import urllib

import urlparse

from google.appengine.api import apiproxy_stub

from google.appengine.api import user_service_pb

_DEFAULT_LOGIN_URL = 'https://www.google.com/accounts/Login?continue=%s'

_DEFAULT_LOGOUT_URL = 'https://www.google.com/accounts/Logout?continue=%s'

class UserServiceStub(apiproxy_stub.APIProxyStub):

  """Trivial implementation of the UserService."""

  def __init__(self,

               login_url=_DEFAULT_LOGIN_URL,

               logout_url=_DEFAULT_LOGOUT_URL,

               service_name='user'):

    """Initializer.

    Args:

      login_url: String containing the URL to use for logging in.

      logout_url: String containing the URL to use for logging out.

      service_name: Service name expected for all calls.

    Note: Both the login_url and logout_url arguments must contain one format

    parameter, which will be replaced with the continuation URL where the user

    should be redirected after log-in or log-out has been completed.

    """

    super(UserServiceStub, self).__init__(service_name)

    self.__num_requests = 0

    self._login_url = login_url

    self._logout_url = logout_url

    os.environ['AUTH_DOMAIN'] = 'gmail.com'

  def num_requests(self):

    return self.__num_requests

  def _Dynamic_CreateLoginURL(self, request, response):

    """Trivial implementation of UserService.CreateLoginURL().

    Args:

      request: the URL to redirect to after login; a base.StringProto

      response: the login URL; a base.StringProto

    """

    self.__num_requests += 1

    response.set_login_url(

        self._login_url %

        urllib.quote(self._AddHostToContinueURL(request.destination_url())))

  def _Dynamic_CreateLogoutURL(self, request, response):

    """Trivial implementation of UserService.CreateLogoutURL().

    Args:

      request: the URL to redirect to after logout; a base.StringProto

      response: the logout URL; a base.StringProto

    """

    self.__num_requests += 1

    response.set_logout_url(

        self._logout_url %

        urllib.quote(self._AddHostToContinueURL(request.destination_url())))

  def _AddHostToContinueURL(self, continue_url):

    """Adds the request host to the continue url if no host is specified.

    Args:

      continue_url: the URL which may or may not have a host specified

    Returns:

      string

    """

    (protocol, host, path, parameters, query, fragment) = urlparse.urlparse(continue_url, 'http')

    if host:

      return continue_url

    host = os.environ['SERVER_NAME']

    if os.environ['SERVER_PORT'] != '80':

      host = host + ":" + os.environ['SERVER_PORT']

    if path == '':

      path = '/'

    return urlparse.urlunparse(

      (protocol, host, path, parameters, query, fragment))