mirror of
https://github.com/Melon-Bread/RetroUFO
synced 2024-11-25 00:38:33 -05:00
248 lines
8.5 KiB
Python
Executable File
248 lines
8.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Grabs the latest version of every libretro core from the build bot.
|
|
"""
|
|
|
|
__author__ = "Melon Bread"
|
|
__version__ = "0.8.0"
|
|
__license__ = "MIT"
|
|
|
|
import os
|
|
import platform
|
|
import sys
|
|
import time
|
|
import zipfile
|
|
from shutil import rmtree
|
|
from urllib.request import urlretrieve
|
|
|
|
from PySide2.QtCore import QThread, Signal
|
|
from PySide2.QtWidgets import (QApplication, QCheckBox, QComboBox, QDialog,
|
|
QFileDialog, QLineEdit, QPushButton, QTextEdit,
|
|
QVBoxLayout, QMessageBox)
|
|
|
|
URL = 'https://buildbot.libretro.com/nightly'
|
|
|
|
# These are the default core locations with normal RetroArch installs based off of 'retroarch.default.cfg`
|
|
CORE_LOCATION = {
|
|
'linux': '{}/.config/retroarch/cores'.format(os.path.expanduser('~')),
|
|
'apple/osx': '/Applications/RetroArch.app/Contents/Resources/cores', # macOS
|
|
'windows': '{}/AppData/Roaming/RetroArch/cores'.format(os.path.expanduser('~'))
|
|
}
|
|
|
|
|
|
class GrabThread(QThread):
|
|
add_to_log = Signal(str)
|
|
|
|
def __init__(self, _platform, _architecture, _location):
|
|
QThread.__init__(self)
|
|
self.platform = _platform
|
|
self.architecture = _architecture
|
|
self.location = _location
|
|
|
|
def __del__(self):
|
|
self.wait()
|
|
|
|
def run(self):
|
|
self.add_to_log.emit('~Starting UFO Grabber~\n')
|
|
self.download_cores(self.platform, self.architecture)
|
|
self.extract_cores(self.location)
|
|
pass
|
|
|
|
def download_cores(self, _platform, _architecture):
|
|
""" Downloads every core to the working directory """
|
|
|
|
cores = []
|
|
|
|
# Makes core directory to store archives if needed
|
|
if not os.path.isdir('cores'):
|
|
os.makedirs("cores")
|
|
time.sleep(1) # TODO: Do not leave this here! Find a way to wait for the core dir to be made!
|
|
|
|
# Downloads a list of all the cores available
|
|
urlretrieve(
|
|
'{}/{}/{}/latest/.index-extended'.format(
|
|
URL, _platform, _architecture), 'cores/index')
|
|
time.sleep(1) # TODO: Do not leave this here! Find a way to wait for the index to download!
|
|
self.add_to_log.emit('Obtained core index!\n')
|
|
|
|
# Adds all the core's file names to a list
|
|
core_index = open('cores/index')
|
|
|
|
for line in core_index:
|
|
file_name = line.split(' ', 2)[2:]
|
|
cores.append(file_name[0].rstrip())
|
|
core_index.close()
|
|
cores.sort()
|
|
|
|
# Downloads each core from the list
|
|
self.add_to_log.emit('Downloading Cores\n')
|
|
for core in cores:
|
|
urlretrieve(
|
|
'{}/{}/{}/latest/{}'.format(URL, _platform, _architecture,
|
|
core), 'cores/{}'.format(core))
|
|
self.add_to_log.emit('Downloaded {} ...'.format(core))
|
|
|
|
# Removes index file for easier extraction
|
|
os.remove('cores/index')
|
|
|
|
def extract_cores(self, _location):
|
|
""" Extracts each downloaded core to the RA core directory """
|
|
self.add_to_log.emit('\nExtracting all cores to: {}\n'.format(_location))
|
|
|
|
for file in os.listdir('cores'):
|
|
archive = zipfile.ZipFile('cores/{}'.format(file))
|
|
archive.extractall(_location)
|
|
self.add_to_log.emit('Extracted {} ...'.format(file))
|
|
|
|
|
|
class Form(QDialog):
|
|
def __init__(self, parent=None):
|
|
super(Form, self).__init__(parent)
|
|
self.setWindowTitle('RetroUFO')
|
|
|
|
# Create widgets
|
|
self.chkboxPlatformDetect = QCheckBox('Platform Auto-Detect')
|
|
self.chkboxPlatformDetect.setChecked(True)
|
|
self.chkboxPlatformDetect.stateChanged.connect(self.auto_detect)
|
|
|
|
self.cmbboxPlatform = QComboBox()
|
|
self.cmbboxPlatform.setEnabled(False)
|
|
self.cmbboxPlatform.setEditable(False)
|
|
self.cmbboxPlatform.addItem('Linux')
|
|
self.cmbboxPlatform.addItem('macOS')
|
|
self.cmbboxPlatform.addItem('Windows')
|
|
|
|
self.cmbboxArchitecture = QComboBox()
|
|
self.cmbboxArchitecture.setEnabled(False)
|
|
self.cmbboxArchitecture.setEditable(False)
|
|
self.cmbboxArchitecture.addItem('x86')
|
|
self.cmbboxArchitecture.addItem('x86_64')
|
|
|
|
self.chkboxLocationDetect = QCheckBox('Core Location Auto-Detect')
|
|
self.chkboxLocationDetect.setChecked(True)
|
|
self.chkboxLocationDetect.stateChanged.connect(self.auto_location)
|
|
|
|
self.leditCoreLocation = QLineEdit('')
|
|
self.leditCoreLocation.setEnabled(False)
|
|
|
|
self.btnCoreLocation = QPushButton('...')
|
|
self.btnCoreLocation.setEnabled(False)
|
|
self.btnCoreLocation.clicked.connect(self.choose_location)
|
|
|
|
self.teditLog = QTextEdit()
|
|
self.teditLog.setReadOnly(True)
|
|
|
|
self.chkboxKeepDownload = QCheckBox('Keep Downloaded Cores')
|
|
self.chkboxKeepDownload.setChecked(False)
|
|
|
|
self.btnGrabCores = QPushButton('Grab Cores')
|
|
self.btnGrabCores.clicked.connect(self.grab_cores)
|
|
|
|
# Create layout and add widgets
|
|
layout = QVBoxLayout()
|
|
layout.addWidget(self.chkboxPlatformDetect)
|
|
layout.addWidget(self.cmbboxPlatform)
|
|
layout.addWidget(self.cmbboxArchitecture)
|
|
layout.addWidget(self.chkboxLocationDetect)
|
|
layout.addWidget(self.leditCoreLocation)
|
|
layout.addWidget(self.btnCoreLocation)
|
|
layout.addWidget(self.teditLog)
|
|
layout.addWidget(self.chkboxKeepDownload)
|
|
layout.addWidget(self.btnGrabCores)
|
|
|
|
# Set dialog layout
|
|
self.setLayout(layout)
|
|
|
|
def auto_detect(self):
|
|
if self.chkboxPlatformDetect.isChecked():
|
|
self.cmbboxPlatform.setEnabled(False)
|
|
self.cmbboxArchitecture.setEnabled(False)
|
|
else:
|
|
self.cmbboxPlatform.setEnabled(True)
|
|
self.cmbboxArchitecture.setEnabled(True)
|
|
|
|
def auto_location(self):
|
|
if self.chkboxLocationDetect.isChecked():
|
|
self.leditCoreLocation.setEnabled(False)
|
|
self.btnCoreLocation.setEnabled(False)
|
|
else:
|
|
self.leditCoreLocation.setEnabled(True)
|
|
self.btnCoreLocation.setEnabled(True)
|
|
|
|
def choose_location(self):
|
|
directory = QFileDialog.getExistingDirectory(
|
|
self, 'Choose Target Location', os.path.expanduser('~'))
|
|
|
|
self.leditCoreLocation.insert(directory)
|
|
|
|
def update_log(self, _info):
|
|
self.teditLog.insertPlainText('{}\n'.format(_info))
|
|
|
|
def grab_cores(self):
|
|
""" Where the magic happens """
|
|
|
|
# TODO: Lock (disable) the UI elements while grabbing cores
|
|
|
|
platform = self.get_platform()
|
|
architecture = self.get_architecture()
|
|
location = self.get_location()
|
|
|
|
self.grab = GrabThread(platform, architecture, location)
|
|
self.grab.add_to_log.connect(self.update_log)
|
|
self.grab.start()
|
|
|
|
if not self.chkboxKeepDownload.isChecked():
|
|
self.clean_up()
|
|
|
|
def get_platform(self):
|
|
""" Gets the Platform and Architecture if not supplied """
|
|
|
|
if not self.chkboxPlatformDetect.isChecked():
|
|
if self.cmbboxPlatform.currentText() == 'macOS':
|
|
return 'apple/osx' # macOS
|
|
else:
|
|
return self.cmbboxPlatform.currentText().lower()
|
|
else:
|
|
if platform.system() == 'Linux':
|
|
return 'linux'
|
|
elif platform.system() == 'Darwin': # macOS
|
|
return 'apple/osx'
|
|
elif platform.system() == 'Windows' or 'MSYS_NT' in platform.system(): # Checks for MSYS environment as well
|
|
return 'windows'
|
|
else:
|
|
msgBox = QMessageBox.warning(self, 'Error', 'Platform not found or supported!', QMessageBox.Ok)
|
|
msgBox.exec_()
|
|
|
|
def get_architecture(self):
|
|
""" Gets the Platform and Architecture if not supplied """
|
|
|
|
if '64' in platform.architecture()[0]:
|
|
return 'x86_64'
|
|
|
|
elif '32' in platform.architecture()[0]:
|
|
return 'x86'
|
|
else:
|
|
msgBox = QMessageBox.warning(self, 'Error', 'Architecture not found or supported', QMessageBox.Ok)
|
|
msgBox.exec_()
|
|
|
|
def get_location(self):
|
|
if not self.chkboxLocationDetect.isChecked():
|
|
return self.leditCoreLocation.text()
|
|
else:
|
|
return CORE_LOCATION[self.get_platform()]
|
|
|
|
def clean_up(self):
|
|
""" Removes all the downloaded files """
|
|
if os.listdir('cores'):
|
|
rmtree('cores/')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# Create the Qt Application
|
|
app = QApplication(sys.argv)
|
|
# Create and show the form
|
|
form = Form()
|
|
form.show()
|
|
# Run the main Qt loop
|
|
sys.exit(app.exec_())
|