#!/usr/bin/python3

# This script is used to convert .yaml files to patterns and package groups.
# Copyright (C) 2011 Marko Saukko <FIRSTNAME.LASTNAME@gmail.com>

# 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 2
# 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, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

# This script is based on MeeGo package-group scripts
# https://meego.gitorious.org/meego-os-base/package-groups/

import yaml
import sys, os
import optparse
from lxml import etree

rpm_ns="http://linux.duke.edu/metadata/rpm"
pattern_ns="http://novell.com/package/metadata/suse/pattern"
NSMAP = {None : pattern_ns, "rpm": rpm_ns}

NSMAP_GROUP = {None : pattern_ns, "rpm": rpm_ns, "patterns": pattern_ns}

def process_yaml(stream, version, release, xmlroot, nsmap_name, newobsapi):
	"Process all documents in the yaml stream and return a count of number handled"

	all_docs = yaml.load_all(stream, Loader = yaml.SafeLoader)
	
	for y in all_docs:
		# <pattern>
		proot = etree.SubElement(xmlroot, "pattern",  nsmap=nsmap_name)
		
		# <name>
		etree.SubElement(proot, "name").text = y['Name']

		# Old OBS isn't able to handle these options.
		if newobsapi:
			# <version>
			if 'Version' in y or version:
				entry = etree.SubElement(proot, "version")
				ver = "0"
				if version:
					ver = version
				else:
					ver = y['Version']

				# Set to 0 by default as that is what OBS expects.
				epoch = "0"
				if 'Epoch' in y:
					epoch = y['Epoch']

				# As above...
				rel = "0"
				if release:
					rel = release
				if 'Release' in y:
					rel =  y['Release']

				entry.set('ver', "%s" % ver)
				entry.set('epoch', "%s" % epoch)
				entry.set('rel', "%s" % rel)

			# <arch>
			if 'Arch' in y:
				etree.SubElement(proot, "arch").text = "%s" % y['Arch']

		# <summary>
		etree.SubElement(proot, "summary").text = y['Summary']
		# <description>
		etree.SubElement(proot, "description").text = y['Description']
		# <uservisible>
		etree.SubElement(proot, "uservisible")
		# <category>
		cat = etree.SubElement(proot, "category")
		cat.text = "Base Group"
		cat.set("lang", "en")

		package_keys = ['Packages','Conflicts', 'Requires', 'Recommends', 'Suggests', 'Provides', 'Obsoletes']
		for key in package_keys:
			if key not in y:
				continue

			collect = y[key]
			if key == "Packages":
				# Support obsoleted keys, this should be removed in the future
				key = "Requires"
				print ("WARNING: Oboleted key 'Packages' in .yaml please change to 'Requires'.")
			
			req = etree.SubElement(proot, "{%s}%s" % (rpm_ns,key.lower()))

			for p in collect:
				if type(p).__name__=='dict':
					print ("ERROR: Found dict and expected string value. '%s'" % (p))
					sys.exit(1)
				entry = etree.SubElement(req, "{%s}entry" %rpm_ns)

				name = p
				ver = None
				op_in  = [">=", "<=", ">",  "<",  "="]
				op_out = ["GE", "LE", "GT", "LT", "EQ"]
				opc = 0
				for op in op_in:
					if op in p:
						name, ver = p.split(op)
						break
					opc = opc + 1

				entry.set("name", name.strip())
				if ver:
					entry.set("flags", "%s" % (op_out[opc]))
					entry.set("ver", "%s" % (ver.strip()))

def create_patterns(patterns_dir, version, release, outputdir, newobsapi):
	dirlist = os.listdir(patterns_dir)
	dirlist.sort()
	for f in dirlist:
		if not f.endswith('.yaml'):
			continue
		
		stream = open("%s/%s" %(patterns_dir,f), 'r')
		xmlroot = etree.Element("temporary_root",  nsmap=NSMAP)
		
		process_yaml(stream, version, release, xmlroot, NSMAP, newobsapi)

		for pattern in xmlroot.findall("pattern"):
			
			name = pattern.find("name")
			if name == None:
				print ("Pattern didn't have name skipping.")
				continue
			output_file = "%s/%s.xml" % (outputdir,name.text.lower())
			print ("Working on %s" % (output_file))

			etree.ElementTree(pattern).write(output_file, pretty_print=True)

def merge_patterns(patterns_dir, version, release, outputdir, newobsapi):
	xmlroot = etree.Element("patterns")
	output_file = "%s/patterns.xml" % (outputdir)
	dirlist = os.listdir(patterns_dir)
	dirlist.sort()

	for f in dirlist:
		if not f.endswith('.yaml'):
			continue
		print ("Merging %s to %s." % (f,output_file))
		stream = file("%s/%s" %(patterns_dir,f), 'r')
		process_yaml(stream, version, release, xmlroot, NSMAP_GROUP, newobsapi)

	patterns = xmlroot.findall("pattern")
	xmlroot.set('count', "%d" % (len(patterns)))

	etree.ElementTree(xmlroot).write(output_file, pretty_print=True)

if __name__ == '__main__':
	parser = optparse.OptionParser()

	parser.add_option("", "--patternxml", action="store_true", dest="patternxml",
			default=False,
			help="Create separated pattern XML file for each pattern.")
	parser.add_option("", "--patternsxml", action="store_true", dest="patternsxml",
			default=False,
			help="Create merged patterns.xml from all the available patterns.")
	parser.add_option("", "--groupxml", action="store_true", dest="groupxml",
			default=False,
			help="Create group.xml.")
	parser.add_option("-p", "--patterndir", type="string", dest="patterndir",
			default=None,
			help="Directory where the pattern .yaml files are located.")
	parser.add_option("-o", "--outputdir", type="string", dest="outputdir",
			default=".",
			help="Output directory where the resulting .xml files are created.")
	parser.add_option("", "--old-obs-xml-format", action="store_false", dest="newobsapi",
			default=True,
			help="The old OBS api isn't able to handle the newer xml format.")
	parser.add_option("--version", type="string", dest="version", default=None, help="Version number")
	parser.add_option("--release", type="string", dest="release", default=None, help="Release number")
	
	(options, args) = parser.parse_args()
	
	if (options.groupxml):
		print ("ERROR: Groupxml isn't supported atm.")
		exit(1)

	if (not options.patternsxml and not options.patternxml):
		# Default to patternxml.
		options.patternxml = True
	
	if (not options.patterndir or not os.path.exists(options.patterndir)):
		print ("Error: Pattern dir '%s' doesn't exist." % (options.patterndir))
		exit(1)
	
	if options.outputdir and not os.path.exists(options.outputdir):
		os.makedirs(options.outputdir)
	
	if options.patternxml:
		create_patterns(options.patterndir, options.version, options.release, options.outputdir, options.newobsapi)

	if options.patternsxml:
		merge_patterns(options.patterndir, options.version, options.release, options.outputdir, options.newobsapi)