dcs-retribution/qt_ui/windows/preferences/QLiberationPreferences.py

import os

from PySide2.QtGui import Qt
from PySide2.QtWidgets import (
    QComboBox,
    QFileDialog,
    QFrame,
    QGridLayout,
    QLabel,
    QLineEdit,
    QMessageBox,
    QPushButton,
    QVBoxLayout,
)

from qt_ui import liberation_install, liberation_theme
from qt_ui.liberation_theme import THEMES, get_theme_index, set_theme_index


class QLiberationPreferences(QFrame):
    def __init__(self):
        super(QLiberationPreferences, self).__init__()
        self.saved_game_dir = ""
        self.dcs_install_dir = ""
        self.install_dir_ignore_warning = False

        self.dcs_install_dir = liberation_install.get_dcs_install_directory()
        self.saved_game_dir = liberation_install.get_saved_game_dir()

        self.edit_dcs_install_dir = QLineEdit(self.dcs_install_dir)
        self.edit_saved_game_dir = QLineEdit(self.saved_game_dir)

        self.edit_dcs_install_dir.setMinimumWidth(300)
        self.edit_saved_game_dir.setMinimumWidth(300)

        self.browse_saved_game = QPushButton("Browse...")
        self.browse_saved_game.clicked.connect(self.on_browse_saved_games)
        self.browse_install_dir = QPushButton("Browse...")
        self.browse_install_dir.clicked.connect(self.on_browse_installation_dir)
        self.themeSelect = QComboBox()
        [self.themeSelect.addItem(y["themeName"]) for x, y in THEMES.items()]

        self.initUi()

    def initUi(self):
        main_layout = QVBoxLayout()
        layout = QGridLayout()
        layout.addWidget(
            QLabel("<strong>DCS saved game directory:</strong>"),
            0,
            0,
            alignment=Qt.AlignLeft,
        )
        layout.addWidget(self.edit_saved_game_dir, 1, 0, alignment=Qt.AlignRight)
        layout.addWidget(self.browse_saved_game, 1, 1, alignment=Qt.AlignRight)
        layout.addWidget(
            QLabel("<strong>DCS installation directory:</strong>"),
            2,
            0,
            alignment=Qt.AlignLeft,
        )
        layout.addWidget(self.edit_dcs_install_dir, 3, 0, alignment=Qt.AlignRight)
        layout.addWidget(self.browse_install_dir, 3, 1, alignment=Qt.AlignRight)
        layout.addWidget(QLabel("<strong>Theme (Requires Restart)</strong>"), 4, 0)
        layout.addWidget(self.themeSelect, 4, 1, alignment=Qt.AlignRight)
        self.themeSelect.setCurrentIndex(get_theme_index())

        main_layout.addLayout(layout)
        main_layout.addStretch()

        self.setLayout(main_layout)

    def on_browse_saved_games(self):
        saved_game_dir = str(
            QFileDialog.getExistingDirectory(self, "Select DCS Saved Game Directory")
        )
        if saved_game_dir:
            self.saved_game_dir = saved_game_dir
            self.edit_saved_game_dir.setText(saved_game_dir)

    def on_browse_installation_dir(self):
        install_dir = str(
            QFileDialog.getExistingDirectory(self, "Select DCS Installation Directory")
        )
        if install_dir:
            self.dcs_install_dir = install_dir
            self.edit_dcs_install_dir.setText(install_dir)

    def apply(self):

        print("Applying changes")
        self.saved_game_dir = self.edit_saved_game_dir.text()
        self.dcs_install_dir = self.edit_dcs_install_dir.text()
        set_theme_index(self.themeSelect.currentIndex())

        if not os.path.isdir(self.saved_game_dir):
            error_dialog = QMessageBox.critical(
                self,
                "Wrong DCS Saved Games directory.",
                self.saved_game_dir + " is not a valid directory",
                QMessageBox.StandardButton.Ok,
            )
            error_dialog.exec_()
            return False

        if self.install_dir_ignore_warning and self.dcs_install_dir == "":
            warning_dialog = QMessageBox.warning(
                self,
                "The DCS Installation directory was not set",
                "You set an empty DCS Installation directory! "
                "<br/><br/>Without this directory, DCS Liberation can not replace the MissionScripting.lua for you and will not work properly. "
                "In this case, you need to edit the MissionScripting.lua yourself. The easiest way to do it is to replace the original file (&lt;dcs_installation_directory&gt;/Scripts/MissionScripting.lua) with the file in dcs-liberation distribution (&lt;dcs_liberation_installation&gt;/resources/scripts/MissionScripting.lua)."
                "<br/><br/>You can find more information on how to manually change this file in the Liberation Wiki (Page: Dedicated Server Guide) on GitHub.</p>"
                "<br/><br/>Are you sure that you want to leave the installation directory empty?"
                "<br/><br/><strong>This is only recommended for expert users!</strong>",
                QMessageBox.StandardButton.Yes,
                QMessageBox.StandardButton.No,
            )
            if warning_dialog == QMessageBox.No:
                return False
        elif not os.path.isdir(self.dcs_install_dir):
            error_dialog = QMessageBox.critical(
                self,
                "Wrong DCS installation directory.",
                self.dcs_install_dir
                + " is not a valid directory. DCS Liberation requires the installation directory to replace the MissionScripting.lua"
                "<br/><br/>If you ignore this Error, DCS Liberation can not work properly and needs your attention. "
                "In this case, you need to edit the MissionScripting.lua yourself. The easiest way to do it is to replace the original file (&lt;dcs_installation_directory&gt;/Scripts/MissionScripting.lua) with the file in dcs-liberation distribution (&lt;dcs_liberation_installation&gt;/resources/scripts/MissionScripting.lua)."
                "<br/><br/>You can find more information on how to manually change this file in the Liberation Wiki (Page: Dedicated Server Guide) on GitHub.</p>"
                "<br/><br/><strong>This is only recommended for expert users!</strong>",
                QMessageBox.StandardButton.Ignore,
                QMessageBox.StandardButton.Ok,
            )
            if error_dialog == QMessageBox.Ignore:
                self.install_dir_ignore_warning = True
            return False
        elif not os.path.isdir(
            os.path.join(self.dcs_install_dir, "Scripts")
        ) and os.path.isfile(os.path.join(self.dcs_install_dir, "bin", "DCS.exe")):
            error_dialog = QMessageBox.critical(
                self,
                "Wrong DCS installation directory.",
                self.dcs_install_dir + " is not a valid DCS installation directory",
                QMessageBox.StandardButton.Ok,
            )
            error_dialog.exec_()
            return False

        liberation_install.setup(self.saved_game_dir, self.dcs_install_dir)
        liberation_install.save_config()
        liberation_theme.save_theme_config()
        return True