Add provider plugins loader

Signed-off-by: Simo Sorce <simo@redhat.com>

1 files changed, 87 insertions, 0 deletions

diff --git a/ipsilon/providers/common.py b/ipsilon/providers/common.py
new file mode 100755
index 0000000..4599735
--- /dev/null
+++ b/ipsilon/providers/common.py
@@ -0,0 +1,87 @@
+#!/usr/bin/python
+#
+# Copyright (C) 2014  Simo Sorce <simo@redhat.com>
+#
+# see file 'COPYING' for use and warranty information
+#
+# 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 3 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, see <http://www.gnu.org/licenses/>.
+
+from ipsilon.util.plugin import PluginLoader, PluginObject
+from ipsilon.util.page import Page
+import cherrypy
+
+
+class ProviderBase(PluginObject):
+
+    def __init__(self, name, path):
+        super(ProviderBase, self).__init__()
+        self.name = name
+        self.path = path
+
+
+class ProviderPageBase(Page):
+
+    def __init__(self, site, config):
+        super(ProviderPageBase, self).__init__(site)
+        self.plugin_name = config.name
+        self.cfg = config
+
+    def GET(self, *args, **kwargs):
+        raise cherrypy.HTTPError(501)
+
+    def POST(self, *args, **kwargs):
+        raise cherrypy.HTTPError(501)
+
+    def root(self, *args, **kwargs):
+        op = getattr(self, cherrypy.request.method, self.GET)
+        if callable(op):
+            return op(*args, **kwargs)
+        else:
+            raise cherrypy.HTTPError(405)
+
+    def _debug(self, fact):
+        superfact = '%s: %s' % (self.plugin_name, fact)
+        super(ProviderPageBase, self)._debug(superfact)
+
+    def _audit(self, fact):
+        cherrypy.log('%s: %s' % (self.plugin_name, fact))
+
+
+FACILITY = 'provider_config'
+
+
+class LoadProviders(object):
+
+    def __init__(self, root, site):
+        loader = PluginLoader(LoadProviders, FACILITY, 'IdpProvider')
+        site[FACILITY] = loader.get_plugin_data()
+        providers = site[FACILITY]
+
+        available = providers['available'].keys()
+        self._debug('Available providers: %s' % str(available))
+
+        for item in providers['whitelist']:
+            self._debug('IdP Provider in whitelist: %s' % item)
+            if item not in providers['available']:
+                continue
+            self._debug('IdP Provider enabled: %s' % item)
+            providers['enabled'].append(item)
+            provider = providers['available'][item]
+            if item in providers['config']:
+                provider.set_config(providers['config'][item])
+            root.__dict__[item] = provider.get_tree(site)
+
+    def _debug(self, fact):
+        if cherrypy.config.get('debug', False):
+            cherrypy.log(fact)