hammer.py

import os
import sys

import valvefgd
from PyQt5 import QtCore, QtWidgets

from QtPyHammer.ui.core import MainWindow
from QtPyHammer.ui.user_preferences.theme import load_theme


# TODO: session restore / shutdown warning (unsaved data will be lost etc.)

def load_ini(ini):
    # perhaps return an encapsulated QSettings
    # with more direct access to variables (convert from default string etc.)
    # it would also be handy to have defaults saved in the code, so we can restore
    return QtCore.QSettings(ini, QtCore.QSettings.IniFormat)


class QtPyHammerApp(QtWidgets.QApplication):
    # code anywhere in QtPyHammer can access the app by calling:
    # -- QtWidgets.QApplication.instance()
    def __init__(self, argv):
        super(QtWidgets.QApplication, self).__init__(argv)
        self.folder = os.path.dirname(__file__)
        self.preferences = load_ini("configs/preferences.ini")
        game = self.preferences.value("Game", "Team Fortress 2")
        self.game_config = load_ini(f"configs/games/{game}.ini")
        self.hotkeys = load_ini("configs/controls/hammer.ini")
        self.themes = dict()
        # ^ {theme_name: theme}
        for filename in os.listdir("configs/themes/"):
            theme_name = filename.rpartition(".")[0]  # filename without extention
            self.themes[theme_name] = load_theme(f"configs/themes/{filename}")
        theme = self.preferences.value("Theme", "default")
        if theme not in self.themes:
            theme = "default"
        self.setStyle(QtWidgets.QStyleFactory.create("Fusion"))
        self.setPalette(self.themes[theme])
        if sys.platform == "win32":
            reg = QtCore.QSettings(r"HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Themes\Personalize",
                                   QtCore.QSettings.NativeFormat)
            if reg.value("AppsUseLightTheme") == 0:  # windows system darkmode
                dark_theme = f"{theme}_dark"
                if dark_theme not in self.themes:
                    dark_theme = "default_dark"
                self.setPalette(self.themes[dark_theme])
                self.setStyleSheet("QToolTip { color: #ffffff; background-color: #2a82da; border: 1px solid white; }")
                # ^ TODO: allow themes to include .css files
                # -- and have them locked to individual widgets?
        fgd_file = self.game_config.value("Hammer/GameData0")
        self.fgd = valvefgd.parser.FgdParse(fgd_file)
        # TODO: check gameinfo.txt for extra content paths
        # -- allow the user to load custom content from their own folders
        # -- then copy that content to tf/custom/{vmf.filename} on compile
        # -- as well as preparing .qc files etc. for models & textures made in-editor


if __name__ == "__main__":
    app = QtPyHammerApp([])  # sys.argv is for filenames only
    window = MainWindow()
    window.showMaximized()
    for filename in sys.argv[1:]:
        window.open(filename)
    sys.exit(app.exec())