1 files changed, 181 insertions, 0 deletions
diff --git a/rpkid/rpki-confgen b/rpkid/rpki-confgen
new file mode 100755
index 00000000..9913a1e5
--- /dev/null
+++ b/rpkid/rpki-confgen
@@ -0,0 +1,181 @@
+#!/usr/bin/env python
+
+import sys
+import getopt
+import textwrap
+
+from lxml.etree import Element, SubElement, ElementTree
+
+space2 = " " * 2
+space4 = " " * 4
+space6 = " " * 6
+space8 = " " * 8
+
+text_wrapper = textwrap.TextWrapper()
+conf_wrapper = textwrap.TextWrapper(initial_indent = "# ", subsequent_indent = "# ")
+item_wrapper = textwrap.TextWrapper(initial_indent = space2, subsequent_indent = space2)
+xml6_wrapper = textwrap.TextWrapper(initial_indent = space6, subsequent_indent = space6)
+xml8_wrapper = textwrap.TextWrapper(initial_indent = space8, subsequent_indent = space8)
+
+class Option(object):
+
+  def __init__(self, name, value, doc):
+    self.name = name
+    self.value = value
+    self.doc = doc
+
+  @property
+  def width(self):
+    return len(self.name)
+
+  def to_xml(self):
+    x = Element("option", name = self.name, value = self.value)
+    for d in self.doc:
+      SubElement(x, "doc").text = "\n" + xml8_wrapper.fill(d) + "\n" + space6
+    return x
+
+  def to_wiki(self, f):
+    f.write("\n%s::" % self.name)
+    for d in self.doc:
+      f.write("\n%s\n" % item_wrapper.fill(d))
+
+  def to_conf(self, f, width):
+    for d in self.doc:
+      f.write("\n%s\n" % conf_wrapper.fill(d))
+    f.write("\n%-*s = %s\n" % (width, self.name, self.value))
+
+class Section(object):
+
+  def __init__(self, name):
+    self.name = name
+    self.doc = []
+    self.options = []
+
+  @property
+  def width(self):
+    return max(o.width for o in self.options)
+
+  @classmethod
+  def from_xml(cls, elt):
+    self = cls(name = elt.get("name"))
+    for x in elt.iterchildren("doc"):
+      self.doc.append(" ".join(x.text.split()))
+    for x in elt.iterchildren("option"):
+      self.options.append(Option(name = x.get("name"), value = x.get("value"),
+                                 doc = [" ".join(d.text.split())
+                                        for d in x.iterchildren("doc")]))
+    return self
+
+  def to_xml(self):
+    x = Element("section", name = self.name)
+    for d in self.doc:
+      SubElement(x, "doc").text = "\n" + xml6_wrapper.fill(d) + "\n" + space4
+    x.extend(o.to_xml() for o in self.options)
+    return x
+
+  def to_wiki(self, f):
+    f.write('''\
+{{{
+#!comment
+THIS PAGE WAS GENERATED AUTOMATICALLY, DO NOT EDIT.
+
+Generated from ''' + ident + '''
+by                 $Id$
+}}}
+
+= ![''' + self.name + '''] section =
+''')
+    for d in self.doc:
+      f.write("\n%s\n" % text_wrapper.fill(d))
+    for o in self.options:
+      o.to_wiki(f)
+
+  def to_conf(self, f, width):
+    f.write('''
+################################################################################
+
+[''' + self.name + ''']
+''')
+    for d in self.doc:
+      f.write("\n%s\n" % conf_wrapper.fill(d))
+    for o in self.options:
+      o.to_conf(f, width)
+
+sections = []
+section_map = None
+option_map = None
+ident = None
+
+opts, argv = getopt.getopt(sys.argv[1:], "",
+                           ["read-xml=", "write-xml=",
+                            "write-wiki=", "write-conf=",
+                            "set=", "autoconf"])
+for o, a in opts:
+
+  if o == "--read-xml":
+    option_map = None
+    root = ElementTree(file = a).getroot()
+    ident = root.get("ident")
+    sections.extend(Section.from_xml(x) for x in root)
+    option_map = {}
+    section_map = {}
+    for section in sections:
+      if section.name in section_map:
+        sys.exit("Duplicate section %s" % section.name)
+      section_map[section.name] = section
+      for option in section.options:
+        name = (section.name, option.name)
+        if name in option_map:
+          sys.exit("Duplicate option %s::%s" % name)
+        option_map[name] = option
+
+  elif o == "--set":
+    try:
+      name, value = a.split("=", 1)
+      section, option = name.split("::")
+    except ValueError:
+      sys.exit("Couldn't parse --set specification \"%s\"" % a)
+    name = (section, option)
+    if name not in option_map:
+      sys.exit("Couldn't find option %s::%s" % name)
+    option_map[name].value = value
+
+  elif o == "--autoconf":
+    try:
+      import rpki.autoconf
+      for option in section_map["autoconf"].options:
+        try:
+          option.value = getattr(rpki.autoconf, option.name)
+        except AttributeError:
+          pass
+    except ImportError:
+      sys.exit("rpki.autoconf module is not available")
+    except KeyError:
+      sys.exit("Couldn't find autoconf section")
+
+  elif o == "--write-xml":
+    x = Element("configuration", ident = ident)
+    x.extend(s.to_xml() for s in sections)
+    ElementTree(x).write(a, pretty_print = True, encoding = "us-ascii")
+
+  elif o == "--write-wiki":
+    with open(a, "w") as f:
+      for i, section in enumerate(sections):
+        if i:
+          f.write("\f\n")
+        section.to_wiki(f)
+
+  elif o == "--write-conf":
+    with open(a, "w") as f:
+      f.write('''\
+# Automatically generated.  Edit as needed, but be careful of overwriting.
+#
+# Generated from ''' + ident + '''
+# by             $Id$
+''')
+      width = max(s.width for s in sections)
+      for section in sections:
+        section.to_conf(f, width)
+
+if argv:
+  sys.exit("Unexpected arguments %s" % argv)