From 8bca77a1738f0badfbcf42357b675d7dce2da61a Mon Sep 17 00:00:00 2001
From: Patrick Uiterwijk <puiterwijk@redhat.com>
Date: Dec 23 2015 13:39:53 +0000
Subject: Add tool to create a configuration file from database


This tool allows the user to migrate the configuration from
a database to a configuration file.

Fixes #64
Signed-off-by: Patrick Uiterwijk <puiterwijk@redhat.com>
Reviewed-by: Simo Sorce <simo@redhat.com>

---

diff --git a/contrib/fedora/ipsilon.spec b/contrib/fedora/ipsilon.spec
index ecaffc2..2031da5 100644
--- a/contrib/fedora/ipsilon.spec
+++ b/contrib/fedora/ipsilon.spec
@@ -240,6 +240,7 @@ mkdir -p %{buildroot}%{_localstatedir}/cache/ipsilon
 install -d -m 0700 %{buildroot}%{_sharedstatedir}/ipsilon
 install -d -m 0700 %{buildroot}%{_sysconfdir}/ipsilon
 mv %{buildroot}/%{_bindir}/ipsilon %{buildroot}/%{_libexecdir}
+mv %{buildroot}/%{_bindir}/ipsilon-db2conf %{buildroot}/%{_sbindir}
 mv %{buildroot}/%{_bindir}/ipsilon-server-install %{buildroot}/%{_sbindir}
 mv %{buildroot}/%{_bindir}/ipsilon-upgrade-database %{buildroot}/%{_sbindir}
 mv %{buildroot}%{_defaultdocdir}/%{name} %{buildroot}%{_defaultdocdir}/%{name}-%{version}
@@ -288,6 +289,7 @@ fi
 %{python2_sitelib}/ipsilon/tools/files.py*
 
 %files
+%{_sbindir}/ipsilon-db2conf
 %{_sbindir}/ipsilon-server-install
 %{_sbindir}/ipsilon-upgrade-database
 %{_datadir}/ipsilon/templates/install/*.conf
diff --git a/ipsilon/install/ipsilon-db2conf b/ipsilon/install/ipsilon-db2conf
new file mode 100755
index 0000000..4d9760a
--- /dev/null
+++ b/ipsilon/install/ipsilon-db2conf
@@ -0,0 +1,99 @@
+#!/usr/bin/python
+#
+# Copyright (C) 2015 Ipsilon project Contributors, for license see COPYING
+
+import argparse
+import cherrypy
+from ConfigParser import RawConfigParser
+import logging
+import os
+import sys
+import unicodedata
+
+from ipsilon.util.data import AdminStore
+import ipsilon.util.sessions
+
+logger = logging.getLogger(__name__)
+
+default_sections = ['info_config',
+                    'login_config',
+                    'provider_config']
+
+
+def sanitize_value(value):
+    # This clears all non-printable characters, like linebreaks and such
+    return "".join(ch
+                   for ch
+                   in value
+                   if unicodedata.category(ch)[0] != "C")
+
+
+def config_to_file(config, output):
+    parser = RawConfigParser()
+    for section in config:
+        parser.add_section(section)
+        for module in config[section]:
+            for option in config[section][module]:
+                name = '%s %s' % (module, option)
+                value = config[section][module][option]
+                value = sanitize_value(value)
+                parser.set(section, name, value)
+
+    parser.write(output)
+
+
+def get_config(cfgfile):
+    cherrypy.lib.sessions.SqlSession = ipsilon.util.sessions.SqlSession
+    cherrypy.config.update(cfgfile)
+
+    adminstore = AdminStore()
+    admin_config = adminstore.load_config()
+    for option in admin_config:
+        cherrypy.config[option] = admin_config[option]
+
+    config = {}
+
+    for section in default_sections:
+        print 'Handling section %s' % section
+        config[section] = adminstore.load_options(section)
+        plugins = config[section].get('global',
+                                      {}).get('enabled',
+                                              '').split(',')
+        for plugin in plugins:
+            if plugin:
+                data_name = '%s_data' % plugin
+                plugin_data = adminstore.get_unique_data(data_name)
+                config[data_name] = plugin_data
+
+    return config
+
+if __name__ == '__main__':
+    def_logger = logging.getLogger()
+    ch = logging.StreamHandler(sys.stdout)
+    ch.setLevel(logging.ERROR)
+    def_logger.addHandler(ch)
+
+    parser = argparse.ArgumentParser(description='This tool produces a config '
+                                                 + 'file from a configuration '
+                                                 + 'database')
+    parser.add_argument('cfgfile', help='Path to the ipsilon configuration',
+                        default='/etc/ipsilon/idp/ipsilon.conf')
+    parser.add_argument('output', help='Where to store the generated'
+                                       + 'configuration file',
+                        default='configuration.conf')
+    parser.add_argument('--force', help='Overwrite existing config file',
+                        action='store_true')
+
+    args = parser.parse_args()
+
+    if os.path.exists(args.output) and not args.force:
+        print 'Output file %s already exists. Please delete it or use --force' \
+              % args.output
+        sys.exit(1)
+
+    config = get_config(args.cfgfile)
+
+    with open(args.output, 'w') as output:
+        config_to_file(config, output)
+
+    print 'Configuration file %s written' % args.output
diff --git a/setup.py b/setup.py
index 073ad4f..a91356c 100755
--- a/setup.py
+++ b/setup.py
@@ -50,6 +50,7 @@ setup(
                    glob('templates/admin/providers/*.html')),
                  ],
     scripts = ['ipsilon/ipsilon',
+               'ipsilon/install/ipsilon-db2conf',
                'ipsilon/install/ipsilon-upgrade-database',
                'ipsilon/install/ipsilon-server-install',
                'ipsilon/install/ipsilon-client-install']