# vim: ts=8:sts=8:sw=8:noexpandtab

# This file is part of python-markups module
# License: BSD
# Copyright: (C) Dmitry Shachnev, 2012-2014

import markups.common as common
from markups.abstract import AbstractMarkup, ConvertedMarkup
from distutils.version import LooseVersion
from os.path import abspath, dirname, join
from tempfile import mkdtemp
from shutil import rmtree


class SphinxConfig(object):
	graphviz_dot = 'dot'
	graphviz_dot_args = []
	graphviz_output_format = 'svg'
	language = None


class SphinxBuilder(object):
	config = SphinxConfig()
	imagedir = "images"

	def __init__(self):
		self.outdir = mkdtemp(prefix="pymarkups-")
		self.imgpath = join(self.outdir, self.imagedir)

	def __del__(self):
		rmtree(self.outdir)


class SphinxEnvironment(object):
	config = SphinxConfig()

	def __init__(self, filename):
		self._dir = abspath(dirname(filename)) if filename else None

	def relfn2path(self, filename):
		if self._dir and not filename.startswith("/"):
			return None, join(self._dir, filename)
		return None, filename

	def note_dependency(self, filename):
		pass


class ReStructuredTextMarkup(AbstractMarkup):
	"""Markup class for reStructuredText language.
	Inherits :class:`~markups.abstract.AbstractMarkup`.

	:param settings_overrides: optional dictionary of overrides for the
	                           `Docutils settings`_
	:type settings_overrides: dict

	.. _`Docutils settings`: http://docutils.sourceforge.net/docs/user/config.html
	"""
	name = 'reStructuredText'
	attributes = {
		common.LANGUAGE_HOME_PAGE: 'http://docutils.sourceforge.net/rst.html',
		common.MODULE_HOME_PAGE: 'http://docutils.sourceforge.net/',
		common.SYNTAX_DOCUMENTATION: 'http://docutils.sourceforge.net/docs/ref/rst/restructuredtext.html'
	}

	file_extensions = ('.rst', '.rest')
	default_extension = '.rst'

	@staticmethod
	def available():
		try:
			import docutils
		except ImportError:
			return False
		return LooseVersion(docutils.__version__) >= LooseVersion('0.13')

	def __init__(self, filename=None, settings_overrides=None):
		self.overrides = settings_overrides or {}
		self.overrides.update({
			'math_output': 'MathJax',
			'syntax_highlight': 'short',
		})
		AbstractMarkup.__init__(self, filename)
		from docutils.core import publish_parts
		self._publish_parts = publish_parts
		self._register_sphinx_directives()

	def _register_sphinx_directives(self):
		from docutils.parsers.rst import directives
		from docutils.writers._html_base import HTMLTranslator
		from docutils.frontend import Values

		try:
			from sphinx.ext.graphviz import Graphviz, GraphvizSimple, html_visit_graphviz
		except ImportError:
			pass
		else:
			directives.register_directive('graphviz', Graphviz)
			directives.register_directive('graph', GraphvizSimple)
			directives.register_directive('digraph', GraphvizSimple)
			HTMLTranslator.visit_graphviz = html_visit_graphviz
			HTMLTranslator.builder = SphinxBuilder()
			Values.env = SphinxEnvironment(self.filename)

	def convert(self, text):
		parts = self._publish_parts(text, source_path=self.filename,
			writer_name='html', settings_overrides=self.overrides)

		# Determine head
		head = parts['head']

		# Determine body
		body = parts['html_body']

		# Determine title
		title = parts['title']

		# Determine stylesheet
		origstyle = parts['stylesheet']
		# Cut off <style> and </style> tags
		stylestart = '<style type="text/css">'
		stylesheet = ''
		if stylestart in origstyle:
			stylesheet = origstyle[origstyle.find(stylestart)+25:origstyle.rfind('</style>')]
		stylesheet += common.get_pygments_stylesheet('.code')

		return ConvertedReStructuredText(head, body, title, stylesheet)


class ConvertedReStructuredText(ConvertedMarkup):

	def __init__(self, head, body, title, stylesheet):
		ConvertedMarkup.__init__(self, body, title, stylesheet)
		self.head = head

	def get_javascript(self, webenv=False):
		start_position = self.head.find('<script ')
		end_position = self.head.rfind('</script>')
		if start_position >= 0 and end_position >= 0:
			mjurl = self.head[start_position:end_position+9]+'\n'
			javascript = mjurl.replace(common.MATHJAX_WEB_URL,
			                           common.get_mathjax_url(webenv))
		else:
			javascript = ''

		return javascript

@@ -368,16 +368,32 @@
  rmtree%0A
+import warnings%0A
 %0A%0Aclass 
@@ -747,16 +747,84 @@
 utdir)%0A%0A
+%09def warn(self, message):%0A%09%09warnings.warn(message, RuntimeWarning)%0A%0A
 %0Aclass S

Fix AttributeError when dot is not available