# -*- coding: UTF-8 -*-


# Lasso Simulator
# By: Emmanuel Raviart <eraviart@entrouvert.com>
#
# Copyright (C) 2004 Entr'ouvert
# http://lasso.entrouvert.org
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA


import lasso

from websimulator import *


class Provider(WebSite):
    httpResponseHeaders = WebSite.httpResponseHeaders.copy()
    httpResponseHeaders.update({
        'Liberty-Enabled': 'LIBV=urn:liberty:iff:2003-08,http://projectliberty.org/specs/v1',
        })
    lassoServerDump = None
    providerId = None # The Liberty providerID of this web site
    sessionTokensByNameIdentifier = None
    userIdsByNameIdentifier = None

    def __init__(self, internet, url):
        WebSite.__init__(self, internet, url)
        self.userIdsByNameIdentifier = {}
        self.sessionTokensByNameIdentifier = {}

    def getLassoServer(self):
        return lasso.Server.new_from_dump(self.lassoServerDump)

    def getSessionFromNameIdentifier(self, nameIdentifier):
        sessionToken = self.sessionTokensByNameIdentifier.get(nameIdentifier, None)
        if sessionToken is None:
            # The user has no federation on this site or has no authentication assertion for this
            # federation.
            return None
        return self.sessions.get(sessionToken, None)

    def getUserFromNameIdentifier(self, nameIdentifier):
        userId = self.userIdsByNameIdentifier.get(nameIdentifier, None)
        if userId is None:
            # The user has no federation on this site.
            return None
        return self.users.get(userId, None)