--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/WebBrowser/GreaseMonkey/GreaseMonkeyDownloader.py Sun Feb 21 19:54:14 2016 +0100
@@ -0,0 +1,180 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2012 - 2016 Detlev Offenbach <detlev@die-offenbachs.de>
+#
+
+"""
+Module implementing the downloader for GreaseMonkey scripts.
+"""
+
+from __future__ import unicode_literals
+
+import os
+
+from PyQt5.QtCore import pyqtSignal, QObject, QSettings, QRegExp, QUrl
+from PyQt5.QtWidgets import QDialog
+from PyQt5.QtNetwork import QNetworkReply
+
+from E5Gui import E5MessageBox
+
+from WebBrowser.WebBrowserWindow import WebBrowserWindow
+from WebBrowser.Network.FollowRedirectReply import FollowRedirectReply
+
+
+class GreaseMonkeyDownloader(QObject):
+ """
+ Class implementing the downloader for GreaseMonkey scripts.
+ """
+ finished = pyqtSignal()
+
+ def __init__(self, request, manager):
+ """
+ Constructor
+
+ @param request reference to the request object (QNetworkRequest)
+ @param manager reference to the GreaseMonkey manager
+ (GreaseMonkeyManager)
+ """
+ super(GreaseMonkeyDownloader, self).__init__()
+
+ self.__manager = manager
+
+ self.__reply = FollowRedirectReply(
+ request.url(), WebBrowserWindow.networkAccessManager())
+ self.__reply.finished.connect(self.__scriptDownloaded)
+
+ self.__fileName = ""
+ self.__requireUrls = []
+
+ def __scriptDownloaded(self):
+ """
+ Private slot to handle the finished download of a script.
+ """
+ if self.sender() != self.__reply:
+ self.finished.emit()
+ return
+
+ response = bytes(self.__reply.readAll()).decode()
+
+ if self.__reply.error() == QNetworkReply.NoError and \
+ "// ==UserScript==" in response:
+ from WebBrowser.Tools import WebBrowserTools
+ filePath = os.path.join(
+ self.__manager.scriptsDirectory(),
+ WebBrowserTools.getFileNameFromUrl(self.__reply.url()))
+ self.__fileName = WebBrowserTools.ensureUniqueFilename(filePath)
+
+ try:
+ f = open(self.__fileName, "w", encoding="utf-8")
+ except (IOError, OSError) as err:
+ E5MessageBox.critical(
+ None,
+ self.tr("GreaseMonkey Download"),
+ self.tr(
+ """<p>The file <b>{0}</b> could not be opened"""
+ """ for writing.<br/>Reason: {1}</p>""").format(
+ self.__fileName, str(err)))
+ self.finished.emit()
+ return
+ f.write(response)
+ f.close()
+
+ settings = QSettings(
+ os.path.join(self.__manager.requireScriptsDirectory(),
+ "requires.ini"),
+ QSettings.IniFormat)
+ settings.beginGroup("Files")
+
+ rx = QRegExp("@require(.*)\\n")
+ rx.setMinimal(True)
+ rx.indexIn(response)
+
+ for i in range(1, rx.captureCount() + 1):
+ url = rx.cap(i).strip()
+ if url and not settings.contains(url):
+ self.__requireUrls.append(QUrl(url))
+
+ self.__reply.deleteLater()
+ self.__reply = None
+
+ self.__downloadRequires()
+
+ def __requireDownloaded(self):
+ """
+ Private slot to handle the finished download of a required script.
+ """
+ if self.sender() != self.__reply:
+ self.finished.emit()
+ return
+
+ response = bytes(self.__reply.readAll()).decode()
+
+ if self.__reply.error() == QNetworkReply.NoError and response:
+ from WebBrowser.Tools import WebBrowserTools
+ filePath = os.path.join(self.__manager.requireScriptsDirectory(),
+ "require.js")
+ fileName = WebBrowserTools.ensureUniqueFilename(filePath, "{0}")
+
+ try:
+ f = open(fileName, "w", encoding="utf-8")
+ except (IOError, OSError) as err:
+ E5MessageBox.critical(
+ None,
+ self.tr("GreaseMonkey Download"),
+ self.tr(
+ """<p>The file <b>{0}</b> could not be opened"""
+ """ for writing.<br/>Reason: {1}</p>""").format(
+ fileName, str(err)))
+ self.finished.emit()
+ return
+ f.write(response)
+ f.close()
+
+ settings = QSettings(
+ os.path.join(self.__manager.requireScriptsDirectory(),
+ "requires.ini"),
+ QSettings.IniFormat)
+ settings.beginGroup("Files")
+ settings.setValue(self.__reply.originalUrl().toString(), fileName)
+
+ self.__reply.deleteLater()
+ self.__reply = None
+
+ self.__downloadRequires()
+
+ def __downloadRequires(self):
+ """
+ Private slot to initiate the download of required scripts.
+ """
+ if self.__requireUrls:
+ self.__reply = FollowRedirectReply(
+ self.__requireUrls.pop(0),
+ WebBrowserWindow.networkAccessManager())
+ self.__reply.finished.connect(self.__requireDownloaded)
+ else:
+ from .GreaseMonkeyScript import GreaseMonkeyScript
+ deleteScript = True
+ script = GreaseMonkeyScript(self.__manager, self.__fileName)
+
+ if script.isValid():
+ if not self.__manager.containsScript(script.fullName()):
+ from .GreaseMonkeyAddScriptDialog import \
+ GreaseMonkeyAddScriptDialog
+ dlg = GreaseMonkeyAddScriptDialog(self.__manager, script)
+ deleteScript = dlg.exec_() != QDialog.Accepted
+ else:
+ E5MessageBox.information(
+ None,
+ self.tr("GreaseMonkey Download"),
+ self.tr(
+ """<p><b>{0}</b> is already installed.</p>""")
+ .format(script.name()))
+
+ if deleteScript:
+ try:
+ os.remove(self.__fileName)
+ except (IOError, OSError):
+ # ignore
+ pass
+
+ self.finished.emit()
