mirror of
https://github.com/KeksPirates/SoftwareManager.git
synced 2026-08-03 17:39:42 +02:00
Vxrtrauter
121 lines
3.6 KiB
Python
121 lines
3.6 KiB
Python
from network.direct_download.handle import DirectDownloadHandle
|
|
from utils.general.shutdown import closehelper
|
|
from utils.logging.logs import consoleLog
|
|
from utils.data.state import state
|
|
from PySide6.QtCore import Qt, QTimer, QEventLoop
|
|
from PySide6 import QtWidgets
|
|
import libtorrent as lt
|
|
import subprocess
|
|
import tempfile
|
|
import hashlib
|
|
import sys
|
|
import os
|
|
|
|
|
|
def _verify_hash(file_path: str, expected_hash: str) -> bool:
|
|
sha256 = hashlib.sha256()
|
|
with open(file_path, "rb") as f:
|
|
for chunk in iter(lambda: f.read(8192), b""):
|
|
sha256.update(chunk)
|
|
return f"sha256:{sha256.hexdigest()}" == expected_hash
|
|
|
|
|
|
def download_update(assets: list):
|
|
filename = None
|
|
setup_hash = None
|
|
url = None
|
|
|
|
for asset in assets:
|
|
if "-windows-setup.exe" in asset["name"]:
|
|
filename = asset["name"]
|
|
setup_hash = asset["hash"]
|
|
url = asset["url"]
|
|
break
|
|
|
|
if not filename:
|
|
consoleLog("Error: No Windows installer found in release assets")
|
|
return
|
|
|
|
installer_path = os.path.join(tempfile.gettempdir(), filename)
|
|
|
|
progress = QtWidgets.QProgressDialog("Downloading update... (0.0 MB/s)", None, 0, 100)
|
|
progress.setWindowTitle("Updating")
|
|
progress.setWindowModality(Qt.WindowModality.ApplicationModal)
|
|
progress.setMinimumDuration(0)
|
|
progress.setAutoClose(False)
|
|
progress.setAutoReset(False)
|
|
progress.setValue(0)
|
|
progress.show()
|
|
QtWidgets.QApplication.processEvents()
|
|
|
|
original_limit = state.down_speed_limit
|
|
state.down_speed_limit = 0
|
|
handle = DirectDownloadHandle(url, filename, tempfile.gettempdir())
|
|
handle.start()
|
|
|
|
loop = QEventLoop()
|
|
timer = QTimer()
|
|
timer.setInterval(100)
|
|
|
|
def poll_progress():
|
|
status = handle.status()
|
|
|
|
if status.error:
|
|
state.down_speed_limit = original_limit
|
|
progress.close()
|
|
consoleLog(f"Update download failed: {status.error}")
|
|
timer.stop()
|
|
loop.quit()
|
|
return
|
|
|
|
if status.total_wanted > 0:
|
|
pct = int(status.total_wanted_done * 100 / status.total_wanted)
|
|
speed_mb = status.download_rate / (1024 * 1024)
|
|
progress.setValue(pct)
|
|
progress.setLabelText(f"Downloading update... ({speed_mb:.1f} MB/s)")
|
|
|
|
if status.state == lt.torrent_status.seeding:
|
|
timer.stop()
|
|
loop.quit()
|
|
|
|
if progress.wasCanceled():
|
|
state.down_speed_limit = original_limit
|
|
consoleLog("Update download cancelled by user.")
|
|
timer.stop()
|
|
loop.quit()
|
|
return
|
|
|
|
timer.timeout.connect(poll_progress)
|
|
timer.start()
|
|
loop.exec()
|
|
|
|
if progress.wasCanceled():
|
|
return
|
|
|
|
if handle.status().error:
|
|
return
|
|
|
|
state.down_speed_limit = original_limit
|
|
|
|
if setup_hash:
|
|
if _verify_hash(installer_path, setup_hash):
|
|
consoleLog(f"Successfully validated installer hash ({setup_hash})")
|
|
else:
|
|
progress.close()
|
|
consoleLog("Error: Invalid file hash, file may be corrupted")
|
|
QtWidgets.QMessageBox.critical(
|
|
None,
|
|
"Update Failed",
|
|
"Error: Invalid file hash, file may be corrupted"
|
|
)
|
|
sys.exit(0)
|
|
else:
|
|
consoleLog("Skipping hash verification (no hash found for release)")
|
|
|
|
progress.setLabelText("Installing update...")
|
|
progress.setValue(100)
|
|
QtWidgets.QApplication.processEvents()
|
|
|
|
closehelper()
|
|
subprocess.Popen([installer_path, "/VERYSILENT", "/SUPPRESSMSGBOXES", "/SP-", "/CLOSEAPPLICATIONS"])
|
|
os._exit(0) |