#!/usr/bin/python

#  Author: Joshua Brindle <jbrindle@tresys.com>

#

# Copyright (C) 2003 - 2005 Tresys Technology, LLC

#      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, version 2.

"""

	this does dstuff

"""

import sys

import getopt

import pyplate

from xml.dom.ext import *

from xml.dom.ext.reader import Sax2

def read_policy_xml(filename):

	try:

		reader = Sax2.Reader()

		doc = reader.fromStream(filename)

	except: 

		error("Error while parsing xml")

	return doc

def gen_tunable_conf(doc, file):

	for node in doc.getElementsByTagName("tunable"):

		s = string.split(node.firstChild.data, "\n")

		for line in s:

			file.write("# %s\n" % line)

		tun_name = tun_val = None

        	for (name, value) in node.attributes.items():

			if name[1] == "name":

				tun_name = value.value

			elif name[1] == "dftval":

				tun_val = value.value

			if tun_name and tun_val:

	            		file.write("%s = %s\n\n" % (tun_name, tun_val))

				tun_name = tun_val = None

def gen_module_conf(doc, file):

	for node in doc.getElementsByTagName("module"):

		for desc in node.getElementsByTagName("summary"):

			s = string.split(desc.firstChild.data, "\n")

			for line in s:

				file.write("# %s\n" % line)	

			file.write("#\n")

			for (name, value) in node.attributes.items():

				if name[1] == "name":

					file.write("# %s\n\n" % value.value)

def gen_docs(doc, file):

	try:

		bodyfile = open("templates/header.html", "r")

		intfile = open("templates/interface.html", "r")

	except:

		error("Could not open templates")

	interface_buf = None

	interface_parameters = {}

	for node in doc.getElementsByTagName("module"):

		for interface in node.getElementsByTagName("interface"):

			interface_tpl = pyplate.Template(intfile.read())

			for i,v in interface.attributes.items():

				interface_name = v

			for desc in interface.getElementsByTagName("description"):

				interface_desc = desc.firstChild.data

			for desc in interface.getElementsByTagName("securitydesc"):

				if desc:

					interface_secdesc = desc.firstChild.data

				else:

					interface_secdesc = None

			for args in interface.getElementsByTagName("parameter"):

				paramdesc = args.firstChild.data

				for i,v in interface.attributes.items():

					arg = { "name" : v,

						"desc" : paramdesc }

def error(error):

        sys.stderr.write("%s exiting for: " % sys.argv[0])

        sys.stderr.write("%s\n" % error)

        sys.stderr.flush()

        sys.exit(1)

def usage():

	sys.stdout.write("%s [-tmd] -x <xmlfile>\n\n" % sys.argv[0])

	sys.stdout.write("Options:\n")

	sys.stdout.write("-t --tunables			--	write tunable config to <file>\n")

	sys.stdout.write("-m --modules <file>		--	write module config to <file>\n")

	sys.stdout.write("-d --docs <dir>		--	write interface documentation to <dir>\n")

	sys.stdout.write("-x --xml <file>		--	filename to read xml data from\n")

try:

	opts, args = getopt.getopt(sys.argv[1:], "t:m:d:x:", ["tunables","modules","docs","xml"])

except getopt.GetoptError:

	usage()

	sys.exit(1)

tunables = modules = docs = xmlfile = None

for opt, val in opts:

	if opt in ("-t", "--tunables"):

		tunables = val

	if opt in ("-m", "--modules"):

		modules = val

	if opt in ("-d", "--docs"):

		docs = val

	if opt in ("-x", "--xml"):

		xmlfile = val

if xmlfile == None:

	usage()

	sys.exit(1)

doc = read_policy_xml(xmlfile)

if tunables:

	try:

		conf = open(tunables, 'w')

	except:

		error("Could not open tunables file for writing")

	gen_tunable_conf(doc, conf)

	conf.close()

if modules:

	try:

		conf = open(modules, 'w')

	except:

		error("Could not open modules file for writing")

	gen_module_conf(doc, conf)

	conf.close()

if docs: 

	gen_docs(doc, sys.stdout)