conqp.rcon/rcon/gui.py

"""GTK based GUI."""

from argparse import ArgumentParser, Namespace
from json import dump, load
from logging import DEBUG, INFO, basicConfig, getLogger
from os import getenv, name
from pathlib import Path
from socket import gaierror, timeout
from typing import Iterable, NamedTuple, Type

from gi import require_version

require_version("Gtk", "3.0")
from gi.repository import Gtk

from rcon import battleye, source
from rcon.client import BaseClient
from rcon.config import LOG_FORMAT
from rcon.exceptions import SessionTimeout, WrongPassword


__all__ = ["main"]


if name == "posix":
    CACHE_DIR = Path.home().joinpath(".cache")
elif name == "nt":
    CACHE_DIR = Path(getenv("TEMP") or getenv("TMP"))
else:
    raise NotImplementedError("Unsupported operating system.")


CACHE_FILE = CACHE_DIR.joinpath("rcongui.json")
LOGGER = getLogger("rcongui")


def get_args() -> Namespace:
    """Parse and return the command line arguments."""

    parser = ArgumentParser(description="A minimalistic, GTK-based RCON GUI.")
    parser.add_argument(
        "-B",
        "--battleye",
        action="store_true",
        help="use BattlEye RCon instead of Source RCON",
    )
    parser.add_argument(
        "-d", "--debug", action="store_true", help="print additional debug information"
    )
    parser.add_argument(
        "-t",
        "--timeout",
        type=float,
        metavar="seconds",
        help="connection timeout in seconds",
    )
    return parser.parse_args()


class RCONParams(NamedTuple):
    """Represent the RCON parameters."""

    host: str
    port: int
    passwd: str
    command: Iterable[str]


class GUI(Gtk.Window):
    """A GTK based GUI for RCON."""

    def __init__(self, args: Namespace):
        """Initialize the GUI."""
        super().__init__(title="RCON GUI")
        self.args = args

        self.set_position(Gtk.WindowPosition.CENTER)

        self.grid = Gtk.Grid()
        self.add(self.grid)

        self.host = Gtk.Entry()
        self.host.set_placeholder_text("Host")
        self.grid.attach(self.host, 0, 0, 1, 1)

        self.port = Gtk.SpinButton.new_with_range(0, 65535, 1)
        self.port.set_placeholder_text("Port")
        self.grid.attach(self.port, 1, 0, 1, 1)

        self.passwd = Gtk.Entry()
        self.passwd.set_placeholder_text("Password")
        self.passwd.set_visibility(False)
        self.grid.attach(self.passwd, 2, 0, 1, 1)

        self.command = Gtk.Entry()
        self.command.set_placeholder_text("Command")
        self.grid.attach(self.command, 0, 1, 2, 1)

        self.button = Gtk.Button(label="Run")
        self.button.connect("clicked", self.on_button_clicked)
        self.grid.attach(self.button, 2, 1, 1, 1)

        self.result = Gtk.TextView()
        self.result.set_wrap_mode(Gtk.WrapMode.WORD)
        self.result.set_property("editable", False)
        self.grid.attach(self.result, 0, 2, 2, 1)

        self.savepw = Gtk.CheckButton(label="Save password")
        self.grid.attach(self.savepw, 2, 2, 1, 1)

        self.load_gui_settings()

    @property
    def client_cls(self) -> Type[BaseClient]:
        """Return the client class."""
        return battleye.Client if self.args.battleye else source.Client

    @property
    def result_text(self) -> str:
        """Return the result text."""
        if (buf := self.result.get_buffer()) is not None:
            return buf.get_text(
                buf.get_iter_at_line(0),
                buf.get_iter_at_line(buf.get_line_count()),
                True,
            )

        return ""

    @result_text.setter
    def result_text(self, text: str):
        """Set the result text."""
        if (buf := self.result.get_buffer()) is not None:
            buf.set_text(text)

    @property
    def gui_settings(self) -> dict:
        """Return the GUI settings as a dict."""
        json = {
            "host": self.host.get_text(),
            "port": self.port.get_value_as_int(),
            "command": self.command.get_text(),
            "result": self.result_text,
            "savepw": (savepw := self.savepw.get_active()),
        }

        if savepw:
            json["passwd"] = self.passwd.get_text()

        return json

    @gui_settings.setter
    def gui_settings(self, json: dict):
        """Set the GUI settings."""
        self.host.set_text(json.get("host", ""))
        self.port.set_value(json.get("port", 0))
        self.passwd.set_text(json.get("passwd", ""))
        self.command.set_text(json.get("command", ""))
        self.result_text = json.get("result", "")
        self.savepw.set_active(json.get("savepw", False))

    def load_gui_settings(self) -> None:
        """Load the GUI settings from the cache file."""
        try:
            with CACHE_FILE.open("rb") as cache:
                self.gui_settings = load(cache)
        except FileNotFoundError:
            LOGGER.warning("Cache file not found: %s", CACHE_FILE)
        except PermissionError:
            LOGGER.error("Insufficient permissions to read: %s", CACHE_FILE)
        except ValueError:
            LOGGER.error("Cache file contains garbage: %s", CACHE_FILE)

    def save_gui_settings(self):
        """Save the GUI settings to the cache file."""
        try:
            with CACHE_FILE.open("w", encoding="utf-8") as cache:
                dump(self.gui_settings, cache, indent=2)
        except PermissionError:
            LOGGER.error("Insufficient permissions to read: %s", CACHE_FILE)

    def show_error(self, message: str):
        """Show an error message."""
        message_dialog = Gtk.MessageDialog(
            transient_for=self,
            message_type=Gtk.MessageType.ERROR,
            buttons=Gtk.ButtonsType.OK,
            text=message,
        )
        message_dialog.run()
        message_dialog.destroy()

    def run_rcon(self) -> str:
        """Return the current RCON settings."""
        with self.client_cls(
            self.host.get_text().strip(),
            self.port.get_value_as_int(),
            timeout=self.args.timeout,
            passwd=self.passwd.get_text(),
        ) as client:
            return client.run(*self.command.get_text().strip().split())

    def on_button_clicked(self, _):
        """Run the client."""
        try:
            result = self.run_rcon()
        except ValueError as error:
            self.show_error(str(error))
        except gaierror as error:
            self.show_error(error.strerror)
        except ConnectionRefusedError:
            self.show_error("Connection refused.")
        except (TimeoutError, timeout):
            self.show_error("Connection timed out.")
        except WrongPassword:
            self.show_error("Wrong password.")
        except SessionTimeout:
            self.show_error("Session timed out.")
        else:
            self.result_text = result

    def terminate(self, *args, **kwargs):
        """Save the settings and terminates the application."""
        self.save_gui_settings()
        Gtk.main_quit(*args, **kwargs)


def main() -> None:
    """Start the GUI."""

    args = get_args()
    basicConfig(format=LOG_FORMAT, level=DEBUG if args.debug else INFO)
    win = GUI(args)
    win.connect("destroy", win.terminate)
    win.show_all()
    Gtk.main()