1 # Copyright (C) 2015 Ipsilon project Contributors, for licensee 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 lasso import ServerAddProviderFailedError
12 class Saml2RestBase(RestProviderBase):
14 The root for REST pages.
16 Add new REST classes to this via add_subtree().
19 def __init__(self, site, config):
20 super(Saml2RestBase, self).__init__(site, config)
25 def mount(self, page):
27 self.add_subtree('SPS', SPS(self._site, self))
28 page.add_subtree(self.name, self)
31 class SPS(RestProviderBase):
33 REST interface for Service Providers
36 def __init__(self, site, parent):
37 super(SPS, self).__init__(site, parent)
40 self.backurl = parent.url
41 self.url = '%s/SPS' % (parent.url,)
45 Return the identity provider object
47 return self._site[FACILITY].available[self.parent.plugin_name]
49 def _get_sp(self, *args, **kwargs):
51 If PATH_INFO contains a value then get that value as the name of
52 the SP, otherwise return a list of all available SPs.
59 idp = self.__get_idp()
63 if instance is not None:
64 data = idp.get_data(name='name', value=instance)
66 return rest_error(404, 'Provider %s not found' % instance)
67 idval = data.keys()[0]
68 data = idp.get_data(idval=idval)
72 for idval in data.keys():
73 result = dict(provider=data[idval].get('name'),
74 metadata=data[idval].get('metadata'),)
75 results.append(result)
77 return dict(result=results)
80 def GET(self, *args, **kwargs):
81 return self._get_sp(*args, **kwargs)
84 def POST(self, *args, **kwargs):
85 cherrypy.response.status = 201
88 return rest_error(400, 'Invalid arguments. Found %d'
89 ' there should be one.')
91 metadata = kwargs.get('metadata')
93 obj = self._site[FACILITY].available[self.parent.plugin_name]
95 spc = ServiceProviderCreator(obj)
96 sp = spc.create_from_buffer(name, metadata)
97 except (InvalidProviderId, ServerAddProviderFailedError) as e:
99 return rest_error(400, str(e))
100 except Exception, e: # pylint: disable=broad-except
102 return rest_error(500, "Failed to create Service Provider")
104 obj.admin.add_sp(name, sp)
106 # Added. Now fetch and return the SP data
107 return self._get_sp(name)