0template.py

# Copyright (C) 2013, Thomas Leonard
# See the README file for details, or visit http://0install.net.

from __future__ import print_function

from xml.dom import minidom, Node
import argparse
import os
import sys
import string
import time

from zeroinstall import support
from zeroinstall.injector import namespaces
from zeroinstall.injector.config import load_config

def die(msg):
	print(msg, file=sys.stderr)
	sys.exit(1)

import expand
import retrieval
import digest

version = 'git-checkout'

config = load_config()

parser = argparse.ArgumentParser(description='Fill in a 0install feed template.')
parser.add_argument('template', help='the template file to process')
parser.add_argument('substitutions', metavar='name=value', help='values to insert', nargs='*')
parser.add_argument('-o', '--output', help='output filename')
parser.add_argument('--from-feed', help='existing feed to derive template from')

args = parser.parse_args()

template = args.template

if not os.path.exists(template):
	if args.substitutions:
		die("{template} does not exist".format(template = template))
	import create
	create.create(args)
	sys.exit(0)

if args.from_feed is not None:
	die("--from-feed can only be used to create new templates, but '{template}' already exists".format(template = template))

if not template.endswith('.xml.template'):
	die("Template must be named *.xml.template, not {template}".format(template = template))
output_file_stem = template[:-13]

env = {}
for subst in args.substitutions:
	if '=' not in subst:
		die("Substitutions must be in the form name=value, not {subst}".format(subst = subst))
	name, value = subst.split('=', 1)
	if name in env:
		die("Multiple values given for {name}!".format(name = name))
	env[name] = value

# Load the template
doc = minidom.parse(args.template)

# Expand {} template strings
expand.process_doc(doc, env)

template_dir = os.path.dirname(os.path.abspath(output_file_stem))

def process_impl(elem):
	if not elem.getAttribute("released"):
		today = time.strftime("%Y-%m-%d")
		elem.setAttribute("released", today)

external_tool = os.environ.get('0TEMPLATE_EXTERNAL_TOOL', '')

# Process implementations
for impl in doc.documentElement.getElementsByTagNameNS(namespaces.XMLNS_IFACE, 'implementation'):
	process_impl(impl)
	if not external_tool:
		retrieval.process_elements(impl, impl, template_dir)
		digest.add_digests(args.template, impl, config)

def get_version(impl):
	while True:
		v = impl.getAttribute('version')
		if v: return v
		impl = impl.parentNode
		if not impl or impl.nodeType != Node.ELEMENT_NODE: die("Missing version for implementation")

impls = doc.getElementsByTagNameNS(namespaces.XMLNS_IFACE, 'implementation')
output_file = args.output if args.output is not None else output_file_stem + '-' + get_version(impls[0]) + '.xml'

print("Writing", output_file)
with open(output_file, 'wb') as stream:
	stream.write(b'<?xml version="1.0"?>\n')
	stream.write(doc.documentElement.toxml('utf-8'))
	stream.write(b'\n')

if external_tool:
	import subprocess
	external_tool_exit_code = subprocess.call([external_tool, '--add-missing', output_file])
	if external_tool_exit_code != 0:
		os.remove(output_file)
		sys.exit(external_tool_exit_code)