import urllib

try:
    import json as simplejson
except ImportError:
    try:
        import simplejson
    except ImportError:
        from django.utils import simplejson

from django.conf import settings
from django.contrib.auth.models import AnonymousUser
from django.test.client import Client, RequestFactory
from django.utils.importlib import import_module
from mock import patch
from social_auth.views import complete

class DumbResponse(object):
    """
    Response from a call to, urllib2.urlopen()
    """

    def __init__(self, data_str, url=None):
        self.data_str = data_str
        self.url = url

    def read(self):
        return self.data_str


class NoBackendError(Exception):
    """
    Used when a client attempts to login with a invalid backend.
    """
    pass


class SocialClient(Client):
    """
    Test client to login/register a user
    Does so by mocking api posts/responses.

    Only supports facebook.
    """

    @patch('social_auth.backends.facebook.FacebookAuth.enabled')
    @patch('social_auth.utils.urlopen')
    def login(self, user, mock_urlopen, mock_facebook_enabled, backend='facebook'):
        """
        Login or Register a facebook user.

        If the user has never logged in then they get registered and logged in.
        If the user has already registered, then they are logged in.

        user: dict
        backend: 'facebook'

        example user:
        {
            'first_name': 'Django',
            'last_name': 'Reinhardt',
            'verified': True,
            'name': 'Django Reinhardt',
            'locale': 'en_US',
            'hometown': {
                'id': '12345678',
                'name': 'Any Town, Any State'
            },
            'expires': '4812',
            'updated_time': '2012-01-29T19:27:32+0000',
            'access_token': 'dummyToken',
            'link': 'http://www.facebook.com/profile.php?id=1234',
            'location': {
                'id': '108659242498155',
                'name': 'Chicago, Illinois'
            },
            'gender': 'male',
            'timezone': -6,
            'id': '1234',
            'email': 'user@domain.com'
        }
        """

        token = 'dummyToken'
        backends = {
            'facebook': (
                urllib.urlencode({
                    'access_token': token,
                    'expires': 3600,
                }),
                simplejson.dumps(user),
            ),

            'google': (
                simplejson.dumps({
                    "access_token": token,
                    "token_type": "Bearer",
                    "expires_in": 3600,
                }),
                simplejson.dumps(user),
            ),

            'linkedin': (
                urllib.urlencode({
                    'oauth_token': token,
                    'oauth_token_secret': token,
                    'oauth_callback_confirmed': 'true',
                    'xoauth_request_auth_url': (
                        'https://api.linkedin.com/uas/oauth/authorize'),
                    'oauth_expires_in': 3600,
                }),
                urllib.urlencode({
                    'oauth_token': token,
                    'oauth_token_secret': token,
                    'oauth_expires_in': 3600,
                    'oauth_authorization_expires_in': 3600,
                }),
                (('<?xml version="1.0" encoding="UTF-8" standalone="yes"?>\n'
                  '<person>\n'
                  '  <id>{id}</id>\n'
                  '  <email-address>{email}</email-address>\n'
                  '  <first-name>{first_name}</first-name>\n'
                  '  <last-name>{last_name}</last-name>\n'
                  '</person>\n').format(**user)),
            ),
        }

        if backend not in backends:
            raise NoBackendError("%s is not supported" % backend)

        """
        mock out urlopen
        """
        mock_urlopen.side_effect = [
            DumbResponse(r) for r in backends[backend]
        ]
        # make it work when no FACEBOOK_APP_ID declared
        mock_facebook_enabled.return_value = True
        factory = RequestFactory()
        request = factory.post('', {'code': 'dummy',
            'redirect_state': 'dummy'})

        engine = import_module(settings.SESSION_ENGINE)
        if self.session:
            request.session = self.session
        else:
            request.session = engine.SessionStore()

        request.user = AnonymousUser()
        request.session['facebook_state'] = 'dummy'

        # make it happen.
        redirect = complete(request, backend)

        request.session.save()

        # Set the cookie for this session.
        session_cookie = settings.SESSION_COOKIE_NAME
        self.cookies[session_cookie] = request.session.session_key
        cookie_data = {
            'max-age': None,
            'path': '/',
            'domain': settings.SESSION_COOKIE_DOMAIN,
            'secure': settings.SESSION_COOKIE_SECURE or None,
            'expires': None,
        }
        self.cookies[session_cookie].update(cookie_data)

        return True