import logging import os.path import sys from typing import Dict, List from PyQt6 import QtWidgets from PyQt6.QtCore import Qt from PyQt6.QtGui import QStandardItemModel, QStandardItem from window.mod_manager_window_export import Ui_MainWindow class MainWindow(QtWidgets.QMainWindow): def __init__(self, parent, parent_logger: logging.Logger, version: str = ""): super(MainWindow, self).__init__() self.parent = parent self.__logger = logging.getLogger("MainWindow") for handler in parent_logger.handlers: self.__logger.addHandler(handler) self.__logger.setLevel(parent_logger.level) self.ui = Ui_MainWindow() self.ui.setupUi(self) self.setWindowTitle(f"LC Mod Manager {version}") self.ui.actionAdd_new_mod.triggered.connect(self.on_action_add_new_mod) self.ui.actionrefresh_mods.triggered.connect(self.on_action_refresh_mods) self.ui.actionCheck_for_Updates.triggered.connect(self.on_action_check_for_updates) self.ui.actionSet_game_folder.triggered.connect(self.on_action_set_game_folder) self.ui.actionRemove_ALL_manager_files.triggered.connect(self.on_action_remove_all_files) self.ui.DeleteModFilesButton.pressed.connect(self.on_pressed_delete_mod_files_button) self.ui.ApplyChangesButton.pressed.connect(self.on_pressed_apply_changes_button) self.ui.DisacrdChangesButton.pressed.connect(self.on_pressed_discard_changes_button) self.ui.TODOButton.hide() self.ui.ApplyChangesButton.hide() self.ui.DisacrdChangesButton.hide() self.ui.DeleteModFilesButton.hide() self.ui.actionCheck_for_Updates.setDisabled(True) def set_available_mods(self, available_mods: Dict[str, str]): """ Sets the given mods as the list of available mods :param available_mods: :return: """ item_model = QStandardItemModel(self.ui.AvailableModsList) for mod_name in available_mods.keys(): item = QStandardItem(mod_name + f" ({available_mods[mod_name]['version']})") item.setFlags(Qt.ItemFlag.ItemIsUserCheckable | Qt.ItemFlag.ItemIsEnabled) item.setData(Qt.CheckState.Unchecked, Qt.ItemDataRole.CheckStateRole) item.setData(mod_name, Qt.ItemDataRole.UserRole) item_model.appendRow(item) if self.parent.is_mod_installed(mod_name): item.setCheckState(Qt.CheckState.Checked) item_model.itemChanged.connect(self.on_available_mod_item_changed) self.ui.AvailableModsList.setModel(item_model) def set_installed_mods(self, installed_mods: Dict[str, str]): """ Sets the given mods as the list of installed mods :param installed_mods: string list of all the mods as to be written to the list :return: """ item_model = QStandardItemModel(self.ui.InstalledModsListView) for mod_name in installed_mods.keys(): if mod_name in self.parent.available_mods.keys(): mod_version = self.parent.available_mods[mod_name]['version'] else: mod_version = "Not Tracked" item = QStandardItem(mod_name + f" ({mod_version})") item.setFlags(Qt.ItemFlag.ItemIsUserCheckable | Qt.ItemFlag.ItemIsEnabled) item.setData(Qt.CheckState.Unchecked, Qt.ItemDataRole.CheckStateRole) item.setData(mod_name, Qt.ItemDataRole.UserRole) item_model.appendRow(item) item.setCheckState(Qt.CheckState.Checked) item_model.itemChanged.connect(self.on_installed_mod_item_changed) self.ui.InstalledModsListView.setModel(item_model) # UI Callback functions ## Actions def on_action_add_new_mod(self): self.__logger.debug("Action: \"add new mod\" triggered!") dialog = QtWidgets.QFileDialog(self, "Select Lethal Company mod") dialog.setFileMode(QtWidgets.QFileDialog.FileMode.ExistingFiles) result = dialog.getOpenFileName(filter='ZIP (*.zip)') self.__logger.debug(f"user selected \"{result[0]}\"") if not os.path.isfile(result[0]): dialog = QtWidgets.QMessageBox() dialog.setWindowTitle("Not a file") dialog.setInformativeText(f"The given file \"{result}\" did not look like a file!") dialog.setIcon(QtWidgets.QMessageBox.Icon.Warning) dialog.exec() if not self.parent.is_valid_mod_file(result[0]): dialog = QtWidgets.QMessageBox() dialog.setWindowTitle("Not a valid mod file") dialog.setInformativeText( f"The given file \"{result}\" did not look like a mod file. Is the manifest.json present?") dialog.setIcon(QtWidgets.QMessageBox.Icon.Warning) dialog.exec() self.parent.add_mod_file(result[0]) def on_action_refresh_mods(self): self.__logger.debug("Action: \"refresh mods\" triggered!") self.parent.index_stored_mods() self.parent.index_installed_mods() def on_action_check_for_updates(self): self.__logger.debug("Action: \"check for updates\" triggered!") def on_action_set_game_folder(self): self.__logger.debug("Action: \"set game folder\" triggered!") dialog = QtWidgets.QFileDialog(self, "Select Lethal Company folder") dialog.setFileMode(QtWidgets.QFileDialog.FileMode.Directory) dialog.setOptions(QtWidgets.QFileDialog.Option.ShowDirsOnly) result = dialog.getExistingDirectory() dir_accepted = self.parent.set_game_folder(result) while not dir_accepted: dialog = QtWidgets.QMessageBox() dialog.setWindowTitle("Invalid game path") dialog.setInformativeText(f"The given path \"{result}\" did not look like the Lethal Company game folder!\n" f"If you can't find it. Try using steam and select \"browse local files\"!") dialog.setIcon(QtWidgets.QMessageBox.Icon.Warning) dialog.exec() dialog = QtWidgets.QFileDialog(self, "Select Lethal Company folder") dialog.setFileMode(QtWidgets.QFileDialog.FileMode.Directory) dialog.setOptions(QtWidgets.QFileDialog.Option.ShowDirsOnly) result = dialog.getExistingDirectory() dir_accepted = self.parent.set_game_folder(result) def on_action_remove_all_files(self): self.__logger.debug("Action: \"remove all manager files\" triggered!") self.parent.nuke_manager_files() ## Buttons def on_pressed_delete_mod_files_button(self): self.__logger.debug("Pressed button: \"Delete Mod Files\"") raise NotImplementedError def on_pressed_apply_changes_button(self): self.__logger.debug("Pressed button: \"Apply Changes\"") raise NotImplementedError def on_pressed_discard_changes_button(self): self.__logger.debug("Pressed button: \"Discard Changes\"") raise NotImplementedError def on_available_mod_item_changed(self, item: QStandardItem): self.__logger.debug(f"Available Mod list item \"{item.text()}\" changed to {item.checkState()}") mod_name = item.data(Qt.ItemDataRole.UserRole) if item.checkState() == Qt.CheckState.Checked: self.parent.install_mod(mod_name) elif item.checkState() == Qt.CheckState.Unchecked: self.parent.uninstall_mod(mod_name) def on_installed_mod_item_changed(self, item: QStandardItem): self.__logger.debug(f"Installed Mod list item \"{item.text()}\" changed to {item.checkState()}") mod_name = item.data(Qt.ItemDataRole.UserRole) if item.checkState() == Qt.CheckState.Unchecked: # mod should be uninstalled accepted = QtWidgets.QMessageBox.question(self, "Really uninstall mod?", f"Do you really want to uninstall the mod \"{mod_name}\"?\n" "This could lead to permanent data loss if it wasn't tracked!") if accepted: self.parent.uninstall_mod(mod_name) else: return