1 files changed, 130 insertions, 0 deletions
diff --git a/src/lib/Bcfg2/Client/Tools/Debconf.py b/src/lib/Bcfg2/Client/Tools/Debconf.py
new file mode 100644
index 000000000..784f7e9bc
--- /dev/null
+++ b/src/lib/Bcfg2/Client/Tools/Debconf.py
@@ -0,0 +1,130 @@
+"""Debconf Support for Bcfg2"""
+
+import subprocess
+import Bcfg2.Options
+import Bcfg2.Client.Tools
+
+
+class Debconf(Bcfg2.Client.Tools.Tool):
+    """Debconf Support for Bcfg2."""
+    name = 'Debconf'
+    __execs__ = ['/usr/bin/debconf-communicate', '/usr/bin/debconf-show']
+    __handles__ = [('Conf', 'debconf')]
+    __req__ = {'Conf': ['name']}
+
+    def __init__(self, config):
+        Bcfg2.Client.Tools.Tool.__init__(self, config)
+
+        #: This is the referrence to the Popen object of the
+        #: running debconf-communicate process. If this is None,
+        #: no process is runnning.
+        self.debconf = None
+
+    def _start_debconf(self):
+        if self.debconf is None:
+            self.debconf = subprocess.Popen(
+                ['/usr/bin/debconf-communicate'],
+                stdin=subprocess.PIPE, stdout=subprocess.PIPE)
+
+    def _stop_debconf(self):
+        if self.debconf is not None:
+            self.debconf.stdin.close()
+            self.debconf.stdout.close()
+            self.debconf = None
+
+    def _debconf_reply(self, msg):
+        if self.debconf is None:
+            self._start_debconf()
+
+        self.logger.debug('Debconf: %s' % msg.strip())
+        self.debconf.stdin.write(msg)
+        line = self.debconf.stdout.readline().rstrip('\n')
+        self.logger.debug('< %s' % line)
+        reply = line.split(' ', 1)
+
+        result = None
+        if len(reply) > 1:
+            result = reply[1]
+        return (reply[0] == '0', result)
+
+    def debconf_get(self, key):
+        (success, value) = self._debconf_reply('GET %s\n' % key)
+        if not success:
+            return (False, None)
+
+        (_, seen) = self._debconf_reply('FGET %s seen\n' % key)
+        return (seen == 'true', value)
+
+    def debconf_set(self, key, value):
+        (success, _) = self._debconf_reply('SET %s %s\n' % (key, value))
+        if success:
+            self._debconf_reply('FSET %s seen true\n' % key)
+
+        return success
+
+    def debconf_reset(self, key):
+        (success, _) = self._debconf_reply('RESET %s\n' % key)
+        return success
+
+    def VerifyConf(self, entry, _modlist):
+        """ Verify the given Debconf entry. """
+        if entry.get('ignore', 'false').lower() == 'true':
+            return True
+
+        (seen, current_value) = self.debconf_get(entry.get('name'))
+        if not seen:
+            current_value = '%s (not seen)' % current_value
+        entry.set('current_value', current_value)
+
+        return seen and current_value == entry.get('value')
+
+    def InstallConf(self, entry):
+        """ Install the given Debconf entry. """
+        return self.debconf_set(entry.get('name'), entry.get('value'))
+
+    def Inventory(self, structures=None):
+        try:
+            result = Bcfg2.Client.Tools.Tool.Inventory(self, structures)
+        finally:
+            self._stop_debconf()
+
+        return result
+    Inventory.__doc__ = Bcfg2.Client.Tools.Tool.Inventory.__doc__
+
+    def Install(self, entries):
+        try:
+            result = Bcfg2.Client.Tools.Tool.Install(self, entries)
+        finally:
+            self._stop_debconf()
+
+        return result
+    Install.__doc__ = Bcfg2.Client.Tools.Tool.Install.__doc__
+
+    def Remove(self, entries):
+        try:
+            for entry in entries:
+                self.debconf_reset(entry.get('name'))
+                self.modified += entry
+        finally:
+            self._stop_debconf()
+            self.extra = self.FindExtra()
+    Remove.__doc__ = Bcfg2.Client.Tools.Tool.Remove.__doc__
+
+    def FindExtra(self):
+        specified = [entry.get('name')
+                     for entry in self.getSupportedEntries()]
+        extra = dict()
+        listowners = self.cmd.run(['/usr/bin/debconf-show', '--listowners'])
+        if listowners.success:
+            owners = listowners.stdout.splitlines()
+
+            values = self.cmd.run(['/usr/bin/debconf-show'] + owners)
+            for line in values.stdout.splitlines():
+                if len(line) > 2 and line[0] == '*':
+                    (name, current_value) = line[2:].split(':', 2)
+                    if name not in specified and name not in extra:
+                        extra[name] = Bcfg2.Client.XML.Element(
+                            'Conf', name=name, type='debconf',
+                            current_value=current_value[1:])
+        return extra.values()
+    FindExtra.__doc__ = Bcfg2.Client.Tools.Tool.FindExtra.__doc__