#!/usr/bin/env python

'''
By Kevin Schluff

Code from From ASPN.

Placed here by Dan.
'''

import SilverCity
import docutils.parsers.rst
import StringIO

def code_block( name, arguments, options, content, lineno,
			content_offset, block_text, state, state_machine ):
	"""
	The code-block directive provides syntax highlighting for blocks
	of code.  It is used with the the following syntax::
	
	.. code-block:: CPP
		
		#include <iostream>
		
		int main( int argc, char* argv[] )
		{
		std::cout << "Hello world" << std::endl;
		}
		
	The directive requires the name of a language supported by SilverCity
	as its only argument.  All code in the indented block following
	the directive will be colourized.  Note that this directive is only
	supported for HTML writers.
	"""
	language = arguments[0]
	try:
		module = getattr(SilverCity, language)
		generator = getattr(module, language+"HTMLGenerator")
	except AttributeError:
		error = state_machine.reporter.error( "No SilverCity lexer found "
		"for language '%s'." % language, 
		docutils.nodes.literal_block(block_text, block_text), line=lineno )
		return [error]
	io = StringIO.StringIO()
	generator().generate_html( io, '\n'.join(content) )
	html = '<div class="code-block">\n%s\n</div>\n' % io.getvalue()
	raw = docutils.nodes.raw('',html, format = 'html')
	return [raw]
	
code_block.arguments = (1,0,0)
code_block.options = {'language' : docutils.parsers.rst.directives.unchanged }
code_block.content = 1

# Simply importing this module will make the directive available.
docutils.parsers.rst.directives.register_directive('code-block', code_block)

if __name__ == "__main__":
	import docutils.core
	docutils.core.publish_cmdline(writer_name='html')