import os
import sys

from PyQt4.QtCore import *
from PyQt4.QtGui import *
from PyQt4.QtXml import *

from sface.config import config
from sface.screens.sfascreen import SfaScreen

class nodeData(QVariant):
    def __init__(self, *args, **kws):
        QVariant.__init__(self, *args, **kws)
        self.type = None

    def setType(self, typ):
        self.type = typ

    def getType(self):
        return self.type

class DomModel(QAbstractItemModel):
    def __init__(self, document, parent = 0):
        QAbstractItemModel.__init__(self, parent)
        self.domDocument = document
        # one of the children of the rootItem is the 'xml' thing.
        # here I delete it.
        childList = document.childNodes()
        for i in range(childList.count()):
            currElem = childList.item(i)
            if (currElem.nodeType() == QDomNode.ProcessingInstructionNode):
                document.removeChild(currElem)
                break
        self.rootItem = DomItem(document, 0);

    def data(self, index, role = Qt.DisplayRole):
        # for interesting nodes, returns a dict wrapped into a QVariant.
        if not index.isValid():
            return QVariant()
        if role != Qt.DisplayRole:
            return QVariant()
        node = index.internalPointer().node()
        attributeMap = node.attributes()

        col = index.column()
        if col == 0:
            if node.nodeType() == QDomNode.ElementNode:
                qslist = QStringList()
                for i in range(attributeMap.count()):
                    attr = attributeMap.item(i)
                    elem = ' %s="%s"' % (attr.nodeName(), attr.nodeValue())
                    qslist.append(elem)
                ElemNameAndAtts = '%s%s'% (node.nodeName(), qslist.join(' '))
                answer = nodeData(ElemNameAndAtts)
                answer.setType('element')
                return answer
            elif node.nodeType() == QDomNode.AttributeNode:
                return QVariant()
            elif node.nodeType() == QDomNode.TextNode:
                answer = nodeData(node.nodeValue())
                answer.setType('text')
                return answer
            elif node.nodeType() == QDomNode.CDATASectionNode:
                return QString('unsupported node type')
            elif node.nodeType() == QDomNode.EntityReferenceNode:
                return QString('unsupported node type')
            elif node.nodeType() == QDomNode.EntityNode:
                return QString('unsupported node type')
            elif node.nodeType() == QDomNode.ProcessingInstructionNode:
                return QVariant()
            elif node.nodeType() == QDomNode.CommentNode:
                answer = nodeData(node.nodeValue())
                answer.setType('comment')
                return answer
            elif node.nodeType() == QDomNode.DocumentNode:
                return QString('unsupported node type')
            elif node.nodeType() == QDomNode.DocumentTypeNode:
                return QString('unsupported node type')
            elif node.nodeType() == QDomNode.DocumentFragmentNode:
                return QString('unsupported node type')
            elif node.nodeType() == QDomNode.NotationNode:
                return QString('unsupported node type')
            elif node.nodeType() == QDomNode.BaseNode:
                return QString('unsupported node type')
            elif node.nodeType() == QDomNode.CharacterDataNode:
                return QString('unsupported node type')
            else:
                return QVariant()
        else:
            return QVariant()

    def flags(self, index):
        if not index.isValid():
            return Qt.ItemIsEnabled
        return Qt.ItemIsEnabled | Qt.ItemIsSelectable
        
    def headerData(self, section, orientation, role):
        return QVariant()

    def index(self, row, column, parent=None):
        if not parent or not parent.isValid():
            parentItem = self.rootItem
        else:
            parentItem = parent.internalPointer()

        childItem = parentItem.child(row)
        # childItem would be None to say "false"?
        if childItem:
            return self.createIndex(row, column, childItem)
        else:
            return QModelIndex()

    def parent(self, child):
        if not child.isValid():
            return QModelIndex()
        childItem = child.internalPointer()
        parentItem = childItem.parent()
        
        if not parentItem or parentItem == self.rootItem:
            return QModelIndex()
        return self.createIndex(parentItem.row(), 0, parentItem)

    def rowCount(self, parent=None):
        if not parent or not parent.isValid():
            parentItem = self.rootItem
        else:
            parentItem = parent.internalPointer()

        return parentItem.node().childNodes().count()

    def columnCount(self, parent):
        # just one column we'll print tag name (and attributes) or the
        # tag content
        return 1


class DomItem:
    # wrapper around PyQt4.QtXml.QDomNode it keeps an hash of
    # childrens for performance reasons

    def __init__(self, node, row, parent = 0):
        # node is of type PyQt4.QtXml.QDomNode
        self.domNode = node
        self.parentItem = parent
        self.rowNumber = row
        self.childItems = {}

    def child(self, i):
        if i in self.childItems:
            return self.childItems[i]
        if i >= 0 and i < self.domNode.childNodes().count():
            childNode = self.domNode.childNodes().item(i)
            childItem = DomItem(childNode, i, self)
            self.childItems[i] = childItem
            return childItem
        return None
            
    def parent(self):
        return self.parentItem

    def node(self):
        return self.domNode

    def row(self):
        return self.rowNumber

class XmlView(QTreeView):
    def __init__(self, parent=None):
        QTreeView.__init__(self, parent)

        delegate = XmlDelegate(self)
        delegate.insertNodeDelegate('element', ElemNodeDelegate())
        delegate.insertNodeDelegate('text', TextNodeDelegate())
        delegate.insertNodeDelegate('comment', CommentNodeDelegate())
        self.setItemDelegate(delegate)

        self.setAnimated(True)
        self.setItemsExpandable(True)
        self.setRootIsDecorated(True)
        self.setHeaderHidden(True)
        self.setAttribute(Qt.WA_MacShowFocusRect, 0)
        self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)

class XmlDelegate(QItemDelegate):
    
    def __init__(self, parent=None):
        QAbstractItemDelegate.__init__(self, parent)
        self.delegates = {}

    def insertNodeDelegate(self, nodeType, delegate):
        delegate.setParent(self)
        self.delegates[nodeType] = delegate

    def removeNodeDelegate(self, nodeType, delegate):
        if nodeType in self.delegates:
            del self.delegates[nodeType]
    
    def paint(self, painter, option, index):
        nodeData = index.model().data(index)
        nodeType = nodeData.getType()
        delegate = self.delegates.get(nodeType)
        if delegate is not None:
            delegate.paint(painter, option, index)
        else:
            QItemDelegate.paint(self, painter, option, index)

    def sizeHint(self, option, index):
        fm = option.fontMetrics
        nodeData = index.model().data(index)
        nodeType = nodeData.getType()
        text = nodeData.toString()
        if nodeType == 'element' or nodeType == 'comment':
            numlines = 1
        elif nodeType == 'text':
            nl = text.count('\n')
            numlines = nl if nl > 0 else 1
        else:
            numlines = 1
        document = QTextDocument()
        document.setDefaultFont(option.font)
        document.setHtml(text)
        # the +5 is for margin. The +4 is voodoo;
        # fm.height just give it too small.
        return QSize(document.idealWidth() + 5, (fm.height() + 4) * numlines)    

class ElemNodeDelegate(QAbstractItemDelegate):
    def paint(self, painter, option, index): 
        palette = QApplication.palette()
        document = QTextDocument()
        document.setDefaultFont(option.font)
        nonHighGlobPattern = '&lt;<b><font color="#b42be2">%s</font></b>%s&gt;'
        nonHighAttPattern = ' <b>%s</b>="<font color="#1e90ff">%s</font>"'
        highGlobPattern = '&lt;<b>%s</b>%s&gt;'
        highAttPattern = ' <b>%s</b>="%s"'
        def getHtmlText(plainText, globPattern, attPattern):
            tmp = plainText.split(' ', 1)
            elemName = tmp[0]
            AttListHtml = ''
            if len(tmp) > 1:
                # many elems don't have atts...
                attList = tmp[1].split()
                for att in attList:
                    tmp = att.split('=')
                    attName = tmp[0]
                    attValue = tmp[1][1:-1]
                    AttListHtml += (attPattern % (attName, attValue))
            html = (globPattern % (elemName, AttListHtml))
            return html
        def colorize(color, text):
            return '<font color=' + color + '>' + text + '</font>'
        nodeData = index.model().data(index)
        nodeType = nodeData.getType()
        # Uff... QString Vs string...
        text = str(nodeData.toString())
        if option.state & QStyle.State_Selected:
            htmlText = colorize(palette.highlightedText().color().name(),
                                getHtmlText(text, highGlobPattern, highAttPattern))
            document.setHtml(QString(htmlText))
        else:
            htmlText = getHtmlText(text, nonHighGlobPattern, nonHighAttPattern)
            document.setHtml(QString(htmlText))
        color = palette.highlight().color() \
            if option.state & QStyle.State_Selected \
            else palette.base().color()
        painter.save()
        # voodoo: if not highlighted, filling the rect
        # with the base color makes no difference
        painter.fillRect(option.rect, color)
        painter.translate(option.rect.x(), option.rect.y())
        document.drawContents(painter)
        painter.restore()

class TextNodeDelegate(QAbstractItemDelegate):
    def paint(self, painter, option, index): 
        palette = QApplication.palette()
        document = QTextDocument()
        document.setDefaultFont(option.font)
        def verbatimize(text):
            text.replace('\n', '<br>')
            return '<pre>' + text + '</pre'
        def colorize(color, text):
            return '<font color=' + color + '>' + text + '</font>'
        nodeData = index.model().data(index)
        nodeType = nodeData.getType()
        text = nodeData.toString()
        if option.state & QStyle.State_Selected:
            htmlText = colorize(palette.highlightedText().color().name(),
                                verbatimize(text))
            document.setHtml(QString(htmlText))
        else:
            htmlText = verbatimize(text)
            document.setHtml(QString(htmlText))
        color = palette.highlight().color() \
            if option.state & QStyle.State_Selected \
            else palette.base().color()
        painter.save()
        # voodoo: if not highlighted, filling the rect
        # with the base color makes no difference
        painter.fillRect(option.rect, color)
        painter.translate(option.rect.x(), option.rect.y())
        document.drawContents(painter)
        painter.restore()

class CommentNodeDelegate(QAbstractItemDelegate):
    def paint(self, painter, option, index): 
        paint(self, painter, option, index)

def paint(self, painter, option, index):
    text = index.model().data(index).property('content').toString()
    palette = QApplication.palette()
    document = QTextDocument()
    document.setDefaultFont(option.font)
    if option.state & QStyle.State_Selected:
        rx = QRegExp(QString('<font .*>'))
        rx.setMinimal(True)
        # If selected, I remove the <font color="..."> by hand,
        # and give the highlight color
        document.setHtml(QString("<font color=%1>%2</font>") \
                             .arg(palette.highlightedText().color().name())\
                             .arg(text.replace(rx, QString('')).
                                  replace(QString('</font>'),QString(''))))
    else:
        document.setHtml(text)
    color = palette.highlight().color() \
        if option.state & QStyle.State_Selected \
        else palette.base().color()
    painter.save()
    # voodoo: if not highlighted, filling the rect
    # with the base color makes no difference
    painter.fillRect(option.rect, color)
    painter.translate(option.rect.x(), option.rect.y())
    document.drawContents(painter)
    painter.restore()