ipsilon/providers/saml2/rest.py

   1 # Copyright (C) 2015  Ipsilon project Contributors, for licensee see COPYING
   2
   3 import cherrypy
   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
  10
  11
  12 class Saml2RestBase(RestProviderBase):
  13     """
  14     The root for REST pages.
  15
  16     Add new REST classes to this via add_subtree().
  17     """
  18
  19     def __init__(self, site, config):
  20         super(Saml2RestBase, self).__init__(site, config)
  21         self.name = 'saml2'
  22         self.cfg = config
  23         self.url = None
  24
  25     def mount(self, page):
  26         self.url = page.url
  27         self.add_subtree('SPS', SPS(self._site, self))
  28         page.add_subtree(self.name, self)
  29
  30
  31 class SPS(RestProviderBase):
  32     """
  33     REST interface for Service Providers
  34     """
  35
  36     def __init__(self, site, parent):
  37         super(SPS, self).__init__(site, parent)
  38
  39         self.parent = parent
  40         self.backurl = parent.url
  41         self.url = '%s/SPS' % (parent.url,)
  42
  43     def __get_idp(self):
  44         """
  45         Return the identity provider object
  46         """
  47         return self._site[FACILITY].available[self.parent.plugin_name]
  48
  49     def _get_sp(self, *args, **kwargs):
  50         """
  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.
  53         """
  54         if len(args) > 0:
  55             instance = args[0]
  56         else:
  57             instance = None
  58
  59         idp = self.__get_idp()
  60
  61         results = list()
  62
  63         if instance is not None:
  64             data = idp.get_data(name='name', value=instance)
  65             if len(data) == 0:
  66                 return rest_error(404, 'Provider %s not found' % instance)
  67             idval = data.keys()[0]
  68             data = idp.get_data(idval=idval)
  69         else:
  70             data = idp.get_data()
  71
  72         for idval in data.keys():
  73             result = dict(provider=data[idval].get('name'),
  74                           metadata=data[idval].get('metadata'),)
  75             results.append(result)
  76
  77         return dict(result=results)
  78
  79     @jsonout
  80     def GET(self, *args, **kwargs):
  81         return self._get_sp(*args, **kwargs)
  82
  83     @jsonout
  84     def POST(self, *args, **kwargs):
  85         cherrypy.response.status = 201
  86
  87         if len(args) != 1:
  88             return rest_error(400, 'Invalid arguments. Found %d'
  89                                    ' there should be one.')
  90         name = args[0]
  91         metadata = kwargs.get('metadata')
  92
  93         obj = self._site[FACILITY].available[self.parent.plugin_name]
  94         try:
  95             spc = ServiceProviderCreator(obj)
  96             sp = spc.create_from_buffer(name, metadata)
  97         except (InvalidProviderId, ServerAddProviderFailedError) as e:
  98             self._debug(repr(e))
  99             return rest_error(400, str(e))
 100         except Exception, e:  # pylint: disable=broad-except
 101             self._debug(repr(e))
 102             return rest_error(500, "Failed to create Service Provider")
 103
 104         obj.admin.add_sp(name, sp)
 105
 106         # Added. Now fetch and return the SP data
 107         return self._get_sp(name)