add-listings.py

#!/usr/bin/env python

import re, fileinput

regex = re.compile('<!--{{{(?P<file>.*):(?P<start>\d+)(?:-(?P<end>\d+))?(?:%(?P<lang>.*))?}}}-->')

for line in fileinput.input():
	m = regex.match(line)

	# Pass the line through if it didn't match the special comment syntax
	if m is None:
		print line,

	# Otherwise, replace it with lines of code from the program
	else:
		with open(m.group('file')) as f:
			lines = f.readlines()

		start = int(m.group('start')) - 1
		if m.group('end') is not None:
			end = int(m.group('end'))
		else:
			end = start + 1

		lines = lines[start:end]

		# Some of these mimetypes are iffy...
		if m.group('lang') is None:
			mimetype = 'x-csrc'
		elif m.group('lang') == 'c':
			mimetype = 'x-csrc'
		elif m.group('lang') == 'autoconf':
			mimetype = 'm4'
		elif m.group('lang') == 'automake':
			mimetype = 'x-automake'
		elif m.group('lang') == 'desktop':
			mimetype = 'x-desktop'
		else:
			mimetype = 'plain'

		# Remove any indentation common to the whole block
		while all([l.startswith(' ') for l in lines if l != '\n']) \
			or all([l.startswith('\t') for l in lines if l != '\n']):
			lines = [(l[1:] if l != '\n' else l) for l in lines]

		print '<listing>'
		print '<title><file href="../{0}">{0}</file></title>'.format(m.group('file'))
		print '<code mime="text/{0}"><![CDATA['.format(mimetype)
		for l in lines:
			print l,
		print ']]></code>'
		print '</listing>'