pf2_downloader.py

"""
Master module. Runs basic checks and then offloads all
of the real work to functions defined in other files.
"""
import os
import traceback
import ctypes
from platform import system
from shutil import which
from subprocess import run
import sys
from sys import argv, exit, stdin
import gettext
from gettext import gettext as _
from rich import print
import gui
import downloads
import setup
import troubleshoot
import vars
import versions
import selfupdate

# PyInstaller offers no native way to select which application you use for the console.
# Instead, it uses the system default, which is cmd.exe at time of writing.
# This hack checks if Windows Terminal is installed. If it is, and if the application
# is launched with cmd.exe instead, it relaunches the application in WT instead.
if not vars.SCRIPT_MODE and system() == 'Windows':
    if which('wt') is not None and os.environ.get("WT_SESSION") is None:
        run(['wt', argv[0]], check=True)
        exit()

# Disable QuickEdit so the process doesn't pause when clicked
if not vars.SCRIPT_MODE and system() == 'Windows':
    kernel32 = ctypes.windll.kernel32
    kernel32.SetConsoleMode(kernel32.GetStdHandle(-10), (0x4|0x80|0x20|0x2|0x10|0x1|0x00|0x100))

def sanity_check() -> None:
    """
    This is mainly for Linux, because it's easy to launch it by double-clicking it, which would
    run it in the background and not show any output. PyInstaller has no way to force a terminal
    open for this on Linux. We could implement something similar to what we do to force using WT,
    but it's not a priority right now since Linux users can figure out how to use the terminal.
    """
    if not stdin or not stdin.isatty():
        print(_("Looks like we're running in the background. We don't want that, so we're exiting."))
        exit(1)

if sys.stdout.encoding == 'ascii':
    sys.stdout.reconfigure(encoding='utf-8')
if sys.stderr.encoding == 'ascii':
    sys.stderr.reconfigure(encoding='utf-8')

if os.getenv('LANG') is None:
    import locale
    lang, enc = locale.getlocale()
    if lang is not None:
        os.environ['LANG'] = lang

if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'):
    gettext.bindtextdomain('pf2-downloader', os.path.abspath(os.path.join(os.path.dirname(__file__), 'locale')))
else:
    gettext.bindtextdomain('pf2-downloader', 'locale')
gettext.textdomain('pf2-downloader')

def wizard() -> None:
    try:
        sanity_check()
        if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'):
            selfupdate.check_downloader_update()
        setup.setup_binaries()
        setup.setup_path(False)
        versions.get_version_list()

        # Check if the game is already installed, for the purposes of running update_version_file() safely
        if os.path.exists(vars.INSTALL_PATH + '/pf2/gameinfo.txt'):
            vars.INSTALLED = True
            #versions.update_version_file()

        # All of the choice logic is handled in this function directly.
        gui.main_menu()
    
    except Exception as ex:
        if ex is not SystemExit:
            traceback.print_exc()
            print(_("[italic magenta]----- Exception details above this line -----"))
            print(_("[bold red]:warning: The program has failed. Please report this to \nhttps://github.com/Pre-Fortress-2/PF2Downloader/issues :warning:[/bold red]"))
            if os.environ.get("WT_SESSION"):
                print(_("[bold]You are safe to close this window."))
            else:
                input(_("Press Enter to exit."))
            exit(1)

def manual_script() -> None:
    try:
        if sys.argv[1] == "--help":
            print(_(vars.HELP_MENU))
            exit(0)

        if sys.argv[1] == "--install":
            if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'):
                selfupdate.check_downloader_update()
            setup.setup_path_script()
            setup.setup_binaries()

            if os.path.exists(vars.INSTALL_PATH + '/pf2/gameinfo.txt'):
                vars.INSTALLED = True

            if vars.INSTALLED:
                gui.message(_("Pre-Fortress 2 is already installed. Assuming a reinstallation."))
            downloads.install()
            troubleshoot.apply_blacklist()
            print(_("The installation has successfully completed. Remember to restart Steam!"))
            exit(0)
        elif sys.argv[1] == "--update":
            if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'):
                selfupdate.check_downloader_update()
            setup.setup_path_script()
            setup.setup_binaries()

            if os.path.exists(vars.INSTALL_PATH + '/pf2/gameinfo.txt'):
                vars.INSTALLED = True

            if not vars.INSTALLED:
                print(_("Pre-Fortress 2 isn't installed, cannot do an update. Consider using --install instead."))
                exit(1)
            else:
                vars.INSTALLED = versions.update_version_file()
                if versions.check_for_updates() == 'reinstall':
                    downloads.install()
                    troubleshoot.apply_blacklist()
                else:
                    downloads.update()
                print(_("The update has successfully completed."))
                exit(0)
        else:
            print(_("Unrecognised command. Try --help"))
            exit(1)

    except Exception as ex:
        if ex is not SystemExit:
            traceback.print_exc()
            print(_("[italic magenta]----- Exception details above this line -----"))
            print(_("[bold red]:warning: The program has failed. Please report this to \nhttps://github.com/Pre-Fortress-2/PF2Downloader/issues:warning :warning:[/bold red]"))
            exit(1)

if vars.SCRIPT_MODE:
    manual_script()
else:
    wizard()