import pygments.formatters
import jinja2
import util
import os
import re
from . import templates

class CodeShowFormatter(pygments.formatters.html.HtmlFormatter):
	def __init__(self, **options):
		pygments.formatters.html.HtmlFormatter.__init__(self, **options)
		self.raw_path = options.get('raw_path', '')
		self.use_extensions = options.get('use_extensions', True)

	def _format_lines(self, tokensource):
		for i, t in pygments.formatters.html.HtmlFormatter._format_lines(self, tokensource):
			if i == 1:
				indent_level = len(re.match('^\s*', t).group(0).replace('\t', '    '))/2
				t = t.strip()
				t = '<p class="i%d%s">%s</p>\n' % (indent_level, len(t) == 0 and ' blank' or '', t)
			yield i, t

	def wrap(self, source, outfile):
		return self._wrap_code(source)

	def _wrap_code(self, source):
		yield 0, templates['top'].render(
			name=util.truncate(self.title),
			cssfile=self.cssfile,
			raw_path=self.raw_path,
			use_extensions=self.use_extensions)
		for i, t in source: yield i, t
		yield 0, templates['bottom'].render()