# -*- coding: utf-8 -*-

#******************************************************************************
#
# Metatools
# ---------------------------------------------------------
# Metadata browser/editor
#
# Copyright (C) 2011-2016 NextGIS (info@nextgis.com)
#
# This source 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 code 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.
#
# A copy of the GNU General Public License is available on the World Wide Web
# at <http://www.gnu.org/copyleft/gpl.html>. You can also obtain it by writing
# to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston,
# MA 02111-1307, USA.
#
#******************************************************************************

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

from qgis.core import *
from qgis.gui import *

import os

FORM_CLASS, _ = uic.loadUiType(os.path.join(os.path.dirname(__file__), 'ui/viewer.ui'))

from error_handler import ErrorHandler

class MetatoolsViewer(QDialog, FORM_CLASS):
  def __init__(self, parent=None):
    super(MetatoolsViewer, self).__init__(parent)
    self.setupUi(self)
    self.setWindowFlags(Qt.Window | Qt.WindowMaximizeButtonHint)

    #set browser context menu
    self.webView.setContextMenuPolicy(Qt.CustomContextMenu)
    self.webView.customContextMenuRequested.connect(self.openMenu)
    self.contextMenu=QMenu()
    self.actionCopy.activated.connect(self.slotCopy)
    self.actionPrint.activated.connect(self.slotPrint)
    self.actionCopyAll.activated.connect(self.slotCopyAll)

  def openMenu(self, position):
    self.contextMenu.clear()
    if self.webView.selectedText():
      self.contextMenu.addAction(self.actionCopy)
    self.contextMenu.addAction(self.actionCopyAll)
    self.contextMenu.addSeparator()
    self.contextMenu.addAction(self.actionPrint)

    self.contextMenu.exec_(self.webView.mapToGlobal(position))

  def slotPrint(self):
    printer = QPrinter()
    dialog = QPrintDialog(printer)
    if dialog.exec_() == QDialog.Accepted:
      self.webView.print_(printer)

  def slotCopyAll(self):
    mimeData=QMimeData()
    mimeData.setHtml(self.webView.page().mainFrame().toHtml())
    mimeData.setText(self.webView.page().mainFrame().toPlainText())
    clipboard = QApplication.clipboard()
    clipboard.setMimeData(mimeData)

  def slotCopy(self):
    if self.webView.selectedText():
      clipboard = QApplication.clipboard()
      clipboard.setText(self.webView.selectedText())

  def setContent(self, metaProvider, xsltFilePath):
    # load data
    xsltFile = QFile(xsltFilePath)
    xsltFile.open(QIODevice.ReadOnly)
    xslt = unicode(xsltFile.readAll())
    xsltFile.close()

    src = metaProvider.getMetadata()

    # translate
    qry = QXmlQuery(QXmlQuery.XSLT20)

    self.handler = ErrorHandler(self.tr("Translation error"))
    qry.setMessageHandler(self.handler)

    qry.setFocus(src)
    qry.setQuery(xslt)

    result = qry.evaluateToString()

    #workaround, for PyQt < 4.8
    #array = ""
    #buf = QBuffer(array)
    #buf.open(QIODevice.WriteOnly)
    #qry.evaluateTo(buf)
    #result = unicode(array)

    if result:
      #QXmlPattern not support CDATA section
      result = result.replace('&amp;', '&')
      result = result.replace('&gt;', '>')
      result = result.replace('&lt;', '<')

      self.webView.setHtml(result) # QString.fromUtf8(result))
      return True
    else:
      return False

  def setHtml(self, html):
    self.webView.setHtml(html)

