--- /dev/null
+#!/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)
finally:
if con:
con.close()
+
+ def get_data(self, plugin):
+ con = None
+ rows = []
+ try:
+ con = sqlite3.connect(self._admin_dbname)
+ cur = con.cursor()
+ cur.execute("CREATE TABLE IF NOT EXISTS " +
+ plugin + "_data (id INTEGER, name TEXT, value TEXT)")
+ cur.execute("SELECT * FROM " + plugin + "_data")
+ rows = cur.fetchall()
+ con.commit()
+ except sqlite3.Error, e:
+ if con:
+ con.rollback()
+ cherrypy.log.error("Failed to load %s data: [%s]" % (plugin, e))
+ finally:
+ if con:
+ con.close()
+
+ data = dict()
+ for row in rows:
+ if row[0] not in data:
+ data[row[0]] = dict()
+
+ item = data[row[0]]
+ if row[1] in item:
+ if item[row[1]] is list:
+ item[row[1]].append(row[2])
+ else:
+ v = item[row[1]]
+ item[row[1]] = [v, row[2]]
+ else:
+ item[row[1]] = row[2]
+
+ return data
+
+ def save_data(self, plugin, data):
+ SELECT = "SELECT name, value FROM %s_data WHERE id=?" % plugin
+ UPDATE = "UPDATE %s_data SET value=? WHERE id=? AND name=?" % plugin
+ INSERT = "INSERT INTO %s_data VALUES(?,?,?)" % plugin
+ con = None
+ try:
+ con = sqlite3.connect(self._admin_dbname)
+ cur = con.cursor()
+ for idval in data:
+ curvals = dict()
+ for row in cur.execute(SELECT, (idval,)):
+ curvals[row[0]] = row[1]
+
+ datum = data[idval]
+ for name in datum:
+ if name in curvals:
+ cur.execute(UPDATE, (datum[name], idval, name))
+ else:
+ cur.execute(INSERT, (idval, name, datum[name]))
+
+ con.commit()
+ except sqlite3.Error, e:
+ if con:
+ con.rollback()
+ cherrypy.log.error("Failed to store %s data: [%s]" % (plugin, e))
+ raise
+ finally:
+ if con:
+ con.close()