1 # Copyright (C) 2015 Ipsilon project Contributors, for license see COPYING
4 from ipsilon.providers.common import RestProviderBase
5 from ipsilon.providers.common import FACILITY
6 from ipsilon.rest.common import rest_error, jsonout
7 from ipsilon.providers.saml2.provider import ServiceProviderCreator
8 from ipsilon.providers.saml2.provider import InvalidProviderId
9 from ipsilon.util.page import admin_protect
10 from lasso import ServerAddProviderFailedError
13 class Saml2RestBase(RestProviderBase):
15 The root for REST pages.
17 Add new REST classes to this via add_subtree().
20 def __init__(self, site, config):
21 super(Saml2RestBase, self).__init__(site, config)
26 def mount(self, page):
28 self.add_subtree('SPS', SPS(self._site, self))
29 page.add_subtree(self.name, self)
32 class SPS(RestProviderBase):
34 REST interface for Service Providers
37 def __init__(self, site, parent):
38 super(SPS, self).__init__(site, parent)
41 self.backurl = parent.url
42 self.url = '%s/SPS' % (parent.url,)
46 Return the identity provider object
48 return self._site[FACILITY].available[self.parent.plugin_name]
50 def _get_sp(self, *args, **kwargs):
52 If PATH_INFO contains a value then get that value as the name of
53 the SP, otherwise return a list of all available SPs.
60 idp = self.__get_idp()
64 if instance is not None:
65 data = idp.get_data(name='name', value=instance)
67 return rest_error(404, 'Provider %s not found' % instance)
68 idval = data.keys()[0]
69 data = idp.get_data(idval=idval)
73 for idval in data.keys():
74 result = dict(provider=data[idval].get('name'),
75 metadata=data[idval].get('metadata'),)
76 results.append(result)
78 return dict(result=results)
82 def GET(self, *args, **kwargs):
83 return self._get_sp(*args, **kwargs)
87 def POST(self, *args, **kwargs):
88 cherrypy.response.status = 201
91 return rest_error(400, 'Invalid arguments. Found %d'
92 ' there should be one.')
94 metadata = kwargs.get('metadata')
96 obj = self._site[FACILITY].available[self.parent.plugin_name]
98 spc = ServiceProviderCreator(obj)
99 sp = spc.create_from_buffer(name, metadata)
100 except (InvalidProviderId, ServerAddProviderFailedError) as e:
102 return rest_error(400, str(e))
103 except Exception, e: # pylint: disable=broad-except
105 return rest_error(500, "Failed to create Service Provider")
107 obj.admin.add_sp(name, sp)
109 # Added. Now fetch and return the SP data
110 return self._get_sp(name)