#  mistletoe-rtf: RTF output for the mistletoe Markdown parser
#  Copyright © 2019  Lee Yingtong Li (RunasSudo)
#
#  This Source Code Form is subject to the terms of the Mozilla Public
#  License, v. 2.0. If a copy of the MPL was not distributed with this
#  file, You can obtain one at http://mozilla.org/MPL/2.0/.

import mistletoe

class RTFRenderer(mistletoe.base_renderer.BaseRenderer):
	def __init__(self, *extras):
		super().__init__(*extras)
	
	def render_strong(self, token):
		return r'{{\b {}}}'.format(self.render_inner(token))
	
	def render_emphasis(self, token):
		return r'{{\i {}}}'.format(self.render_inner(token))
	
	def render_inline_code(self, token):
		return r'{{\f1 {}}}'.format(self.render_inner(token))
	
	def render_raw_text(self, token):
		return token.content.replace('\\', '\\\\').replace('{', r'\{').replace('}', r'\}')
	
	def render_strikethrough(self, token):
		return r'{{\strike {}}}'.format(self.render_inner(token))
	
	def render_image(self, token):
		raise Exception('NYI')
	
	def render_link(self, token):
		return r'{{\field{{\*\fldinst{{HYPERLINK "{target}"}}}}{{\fldrslt{{\ul\cf1 {inner}}}}}}}'.format(target=token.target, inner=self.render_inner(token))
	
	def render_auto_link(self, token):
		raise Exception('NYI')
	
	def render_escape_sequence(self, token):
		raise Exception('NYI')
	
	def render_heading(self, token):
		template = r'{{\b\fs{fontsize}{inner}\par}}'
		fontsize = {1: 32, 2: 28, 3: 24, 4: 24, 5: 24, 6: 24}[token.level]
		return template.format(fontsize=fontsize, inner=self.render_inner(token))
	
	def render_quote(self, token):
		return r'{{\li400 {}}}'.format(self.render_inner(token))
	
	def render_paragraph(self, token):
		template = r'{{{}\par}}'
		return template.format(self.render_inner(token))
	
	def render_block_code(self, token):
		return r'{{\li400\f1 {}\par}}'.format(self.render_inner(token).rstrip('\n').replace('\n', r'\line '))
	
	def render_list(self, token):
		raise Exception('NYI')
	
	def render_list_item(self, token):
		raise Exception('NYI')
	
	def render_table(self, token):
		template = r'{{\sa0\trowd{cellxs} {inner}}}'
		
		cell_width = (6*1440) / len(token.column_align) # 6 inches
		
		cellxs = [r'\cellx{}'.format(int(cell_width*(i+1))) for i in range(len(token.column_align))]
		
		if hasattr(token, 'header'):
			head_rendered = self.render_table_row(token.header)
		
		inner = self.render_inner(token)
		
		return template.format(cellxs=''.join(cellxs), inner=head_rendered+inner)
	
	def render_table_row(self, token):
		cells = [self.render(child) for child in token.children]
		return ''.join(cells) + r'\row '
	
	def render_table_cell(self, token):
		template = r'{{{inner}\intbl\cell}}'
		return template.format(inner=self.render_inner(token))
	
	def render_thematic_break(self, token):
		raise Exception('NYI')
	
	def render_line_break(self, token):
		return r'\line '
	
	def render_document(self, token):
		template = r'{{\rtf1\deff0{{\fonttbl{{\f0\froman Times;}}{{\f1\froman Courier;}}}}{{\colortbl;\red0\green0\blue0\;\red0\green0\blue255;}}\sa200\f0 {inner}}}'
		
		result_str = template.format(inner=self.render_inner(token))
		
		# Escape Unicode
		result = []
		for char in result_str:
			if ord(char) <= 0x7f:
				result.append(char)
			else:
				result.append(r'\u{}?'.format(ord(char)))
		
		return ''.join(result)