##############################################################################
#
# Copyright (c) 2004-2006 TINY SPRL. (http://tiny.be) All Rights Reserved.
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsability of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# garantees and support are strongly adviced to contract a Free Software
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
##############################################################################


import xml.dom.minidom
import os, time
import ir, netsvc, sql_db
import osv
from interface import  report_rml
import re
import tools

import copy

parents = {
	'tr':1,
	'li':1,
	'story': 0,
	'section': 0
}

#
# Context: {'node': node.dom}
#
class browse_record_list(list):
	def __init__(self, lst, context):
		super(browse_record_list, self).__init__(lst)
		self.parents = copy.copy(parents)
		self.context = context

	def __getattr__(self, name):
		res = browse_record_list([getattr(x,name) for x in self], self.context)
		return res

	def __str__(self):
		return "browse_record_list("+str(len(self))+")"

	def repeatIn(self, name):
		print 'Decrecated ! Use repeatIn(object_list, \'variable\')'
		node = self.context['_node']
		node.data = ''
		while True:
			if not node.parentNode:
				break
			node = node.parentNode
			if node.nodeType == node.ELEMENT_NODE and node.localName in parents:
				break
		parent_node = node
		if not len(self):
			return None
		nodes = [(0,node)]
		for i in range(1,len(self)):
			newnode = parent_node.cloneNode(1)
			n = parent_node.parentNode
			n.insertBefore(newnode, parent_node)
			nodes.append((i,newnode))
		for i,node in nodes:
			self.context[name] = self[i]
			self.context['_self']._parse_node(node)
		return None

class rml_parse(object):
	localcontext = {}
	def __init__(self, cr,uid, localcontext):
		self.cr = cr
		self.uid = uid
		self.regex = re.compile('\[\[(.+?)\]\]')
		self.context = {}
		self.localcontext = localcontext
		self._node = None
		self.uniq = 1

	def repeatIn(self, lst, name):
		node = self._node
		node.data = ''
		while True:
			if not node.parentNode:
				break
			node = node.parentNode
			if node.nodeType == node.ELEMENT_NODE and node.localName in parents:
				break
		pp = node.parentNode
		ns = node.nextSibling
		pp.removeChild(node)
		self._node = pp

		if not len(lst):
			return None
		nodes = []
		for i in range(len(lst)):
			newnode = node.cloneNode(1)
			if ns:
				pp.insertBefore(newnode, ns)
			else:
				pp.appendChild(newnode)
			nodes.append((i,newnode))
		for i,node in nodes:
			self.node_context[node] = {name:lst[i]}
		return None

	def _eval(self, expr):
		try:
			pool = osv.osv.osv_pools
			self.context['repeatIn'] = self.repeatIn
			self.context['objects'] = self.objs
			self.context['data'] = self.data
			self.context['user'] = pool.get('res.users').browse(self.cr, self.uid, self.uid)
			self.context.update(self.localcontext)
			res = eval(expr, self.context)
			if res is False or res is None:
				res = ''
		except Exception,e:
			print e.args
			print dir(e)
			res = 'Error'
		return res

	def _find_parent(self, node):
		while True:
			if not node.parentNode:
				return False
			node = node.parentNode
			if node.nodeType == node.ELEMENT_NODE and node.localName in parents:
				break
		return node

	def _parse_text(self, text, level=[]):
		res = self.regex.findall(text)
		todo = []
		for key in res:
			newtext = self._eval(key)
			uniq = self.uniq
			self.uniq+=1
			for i in range(len(level)):
				if isinstance(newtext,browse_record_list) or (type(newtext)==type([])):
					newtext = newtext[level[i]]
			if isinstance(newtext,browse_record_list) or (type(newtext)==type([])):
				todo.append((key, newtext))
			else:
				if type(newtext) not in (type(''), type(u'')):
					newtext = str(newtext)
				text = text.replace('[['+key+']]', newtext.decode('utf8'))
		self._node.data = text
		if len(todo):
			for key, newtext in todo:
				parent_node = self._find_parent(self._node)
				assert parents.get(parent_node.localName,False), 'No parent node found !'
				nodes = [parent_node]
				for i in range(len(newtext)-1):
					newnode = parent_node.cloneNode(1)
					if parents.get(parent_node.localName,False):
						n = parent_node.parentNode
						parent_node.parentNode.insertAfter(newnode, parent_node)
						nodes.append(newnode)
			return False
		return text

	def _parse_node(self):
		level = []
		while True:
			if self._node.hasChildNodes():
				self._node = self._node.firstChild
			elif self._node.nextSibling:
				self._node = self._node.nextSibling
			else:
				while self._node and not self._node.nextSibling:
					self._node = self._node.parentNode
				if not self._node:
					break
				self._node = self._node.nextSibling
			if self._node in self.node_context:
				self.context.update(self.node_context[self._node])
			if (self._node.nodeType == self._node.CDATA_SECTION_NODE) or (self._node.nodeType == self._node.TEXT_NODE):
				self._parse_text(self._node.data,level)
		return True

	def _find_node(self, node, localname):
		if node.localName==localname:
			return node
		for tag in node.childNodes:
			if tag.nodeType==tag.ELEMENT_NODE:
				found = self._find_node(tag, localname)
				if found:
					return found
		return False

	def _add_header(self, node):
		rml_head = tools.file_open('custom/corporate_rml_header.rml').read()
		head_dom = xml.dom.minidom.parseString(rml_head)
		node2 = head_dom.documentElement
		for tag in node2.childNodes:
			if tag.nodeType==tag.ELEMENT_NODE:
				found = self._find_node(node, tag.localName)
				if found:
					found.parentNode.replaceChild(tag, found)
		return True

	def _parse(self, rml_dom, objs, data, context):
		self.context = context
		self.node_context = {}
		self.data = data
		self.objs = objs
		self.dom = rml_dom
		self._node = self.dom.documentElement
		self._add_header(self._node)

		self._parse_node()
		res = self.dom.documentElement.toxml('utf-8')
		return res

#
# Not thread independant !!!
# Bad concurrency, to recode !
#
class report_sxw(report_rml):
	localcontext = {}
	def __init__(self, name, table, rml):
		report_rml.__init__(self, name, table, rml, '')
		self.bin_datas = {}
		self.name = name

	def create(self, uid, ids, data, context={}):
		rml = file(self.tmpl).read()

		cr = sql_db.db.cursor()
		self.uid = uid
		self.cr = cr
		self.datas = data
		self.ids = ids
		self.pool = osv.osv.osv_pools
		objs = osv.osv.osv_pools.get(self.table).browse(cr, uid, ids, list_class=browse_record_list, context=context)
		rml_parser = rml_parse(cr, uid, self.localcontext)
		
		rml_dom = xml.dom.minidom.parseString(rml)
		
		service = netsvc.LocalService("object_proxy")

		# translate a node and its children
		def translate(de, lang):
			for n in [i for i in de.childNodes if (i.nodeType == i.ELEMENT_NODE)]:
				for m in [j for j in n.childNodes if (j.nodeType == j.TEXT_NODE)]:
					string = m.data.strip().replace('\n',' ')
					string_list = [s.strip() for s in re.split('\[\[.+?\]\]', string)]
					for source_string in string_list:
						if len(source_string):
							translated_string = service.execute(uid, 'ir.translation', '_get_source', self.name2, 'rml', lang, source_string)
							if translated_string:
								m.data = m.data.replace(source_string, translated_string.decode('utf8'))
				translate(n, lang)
				
		# translate the RML file
		if context.get('lang', False):
			translate(rml_dom.documentElement, context['lang'])

		rml2 = rml_parser._parse(rml_dom, objs, data, context)
		cr.close()

		create_doc = self.generators[data.get('report_type','pdf')]
		pdf = create_doc(rml2)
		return pdf