diff -r 3fc8dfeb6ebe -r b99e7fd55fd3 src/eric7/WebBrowser/GreaseMonkey/GreaseMonkeyDownloader.py
--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/src/eric7/WebBrowser/GreaseMonkey/GreaseMonkeyDownloader.py Thu Jul 07 11:23:56 2022 +0200
@@ -0,0 +1,155 @@
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2012 - 2022 Detlev Offenbach <detlev@die-offenbachs.de>
+#
+
+"""
+Module implementing the downloader for GreaseMonkey scripts.
+"""
+
+import os
+import pathlib
+
+from PyQt6.QtCore import pyqtSignal, pyqtSlot, QObject, QSettings
+from PyQt6.QtNetwork import QNetworkReply, QNetworkRequest
+
+from WebBrowser.WebBrowserWindow import WebBrowserWindow
+
+
+class GreaseMonkeyDownloader(QObject):
+ """
+ Class implementing the downloader for GreaseMonkey scripts.
+
+ @signal finished(fileName) emitted to indicate the end of a script download
+ (str)
+ @signal error() emitted to indicate a script download error
+ """
+ finished = pyqtSignal(str)
+ error = pyqtSignal()
+
+ DownloadMainScript = 1
+ DownloadRequireScript = 2
+
+ def __init__(self, url, manager, mode):
+ """
+ Constructor
+
+ @param url URL to download script from
+ @type QUrl
+ @param manager reference to the GreaseMonkey manager
+ @type GreaseMonkeyManager
+ @param mode download mode
+ @type int (one of DownloadMainScript, DownloadRequireScript)
+ """
+ super().__init__()
+
+ self.__manager = manager
+
+ self.__reply = WebBrowserWindow.networkManager().get(
+ QNetworkRequest(url))
+ if mode == GreaseMonkeyDownloader.DownloadMainScript:
+ self.__reply.finished.connect(self.__scriptDownloaded)
+ else:
+ self.__reply.finished.connect(self.__requireDownloaded)
+
+ self.__fileName = ""
+
+ def updateScript(self, fileName):
+ """
+ Public method to set the file name for the script to be downloaded.
+
+ @param fileName file name for the script
+ @type str
+ """
+ self.__fileName = fileName
+
+ @pyqtSlot()
+ def __scriptDownloaded(self):
+ """
+ Private slot to handle the finished download of a script.
+ """
+ self.deleteLater()
+ self.__reply.deleteLater()
+
+ if self.__reply.error() != QNetworkReply.NetworkError.NoError:
+ self.error.emit()
+ return
+
+ response = bytes(self.__reply.readAll()).decode()
+
+ if "// ==UserScript==" not in response:
+ self.error.emit()
+ return
+
+ if not self.__fileName:
+ from WebBrowser.Tools import WebBrowserTools
+ filePath = os.path.join(
+ self.__manager.scriptsDirectory(),
+ WebBrowserTools.getFileNameFromUrl(self.__reply.url()))
+ self.__fileName = WebBrowserTools.ensureUniqueFilename(filePath)
+
+ try:
+ with open(self.__fileName, "w", encoding="utf-8") as f:
+ f.write(response)
+ except OSError:
+ self.error.emit()
+ return
+
+ self.finished.emit(self.__fileName)
+
+ @pyqtSlot()
+ def __requireDownloaded(self):
+ """
+ Private slot to handle the finished download of a required script.
+ """
+ self.deleteLater()
+ self.__reply.deleteLater()
+
+ if self.__reply.error() != QNetworkReply.NetworkError.NoError:
+ self.error.emit()
+ return
+
+ response = bytes(self.__reply.readAll()).decode()
+
+ if not response:
+ self.error.emit()
+ return
+
+ settings = QSettings(
+ os.path.join(self.__manager.requireScriptsDirectory(),
+ "requires.ini"),
+ QSettings.Format.IniFormat)
+ settings.beginGroup("Files")
+
+ if not self.__fileName:
+ self.__fileName = settings.value(
+ self.__reply.request().url().toString())
+ if not self.__fileName:
+ name = (
+ pathlib.Path(self.__reply.request().url().path()).name
+ )
+ if not name:
+ name = "require.js"
+ elif not name.endswith(".js"):
+ name += ".js"
+ filePath = os.path.join(
+ self.__manager.requireScriptsDirectory(), name)
+ from WebBrowser.Tools import WebBrowserTools
+ self.__fileName = WebBrowserTools.ensureUniqueFilename(
+ filePath, "{0}")
+ if not pathlib.Path(self.__fileName).is_absolute():
+ self.__fileName = os.path.join(
+ self.__manager.requireScriptsDirectory(),
+ self.__fileName)
+
+ try:
+ with open(self.__fileName, "w", encoding="utf-8") as f:
+ f.write(response)
+ except OSError:
+ self.error.emit()
+ return
+
+ settings.setValue(self.__reply.request().url().toString(),
+ pathlib.Path(self.__fileName).name)
+
+ self.finished.emit(self.__fileName)
