gui.py

import subprocess
from functools import partial
import time
import tkinter as tk
from tkinter import ttk

import requests


ENDPOINT = "http://127.0.0.1:8080"


def boot_and_exit(char_id, char_new, token):
    subprocess.Popen(["./mhf-iel.exe", str(char_id), str(int(char_new)), token])
    exit()


class LoginScreen:
    def __init__(self, root):
        self.root = root
        self.username = tk.StringVar()
        self.password = tk.StringVar()

        self.error_label = None
        self.frm = ttk.Frame(root, padding=15)
        self.frm.grid(sticky=(tk.W, tk.N, tk.E, tk.S))

        input_frm = ttk.Frame(self.frm)
        input_frm.grid()
        ttk.Label(input_frm, text="Username").grid(row=1, column=1, pady=5, padx=5)
        ttk.Entry(input_frm, textvariable=self.username).grid(row=1, column=2)
        ttk.Label(input_frm, text="Password").grid(row=2, column=1, pady=5, padx=5)
        ttk.Entry(input_frm, textvariable=self.password).grid(row=2, column=2)

        button_frame = ttk.Frame(self.frm)
        button_frame.grid(pady=(10, 0))
        ttk.Button(
            button_frame, text="Login", command=partial(self.connect, "login")
        ).grid(row=0, column=0)
        ttk.Button(
            button_frame,
            text="Create Account",
            command=partial(self.connect, "register"),
        ).grid(row=0, column=1, padx=(20, 0), sticky=tk.E)

        for child in self.frm.winfo_children():
            child.grid_configure(sticky=(tk.W, tk.E))

    def connect(self, action, *args):
        if self.error_label:
            self.error_label.destroy()

        try:
            resp = requests.post(
                f"{ENDPOINT}/{action}",
                json={"username": self.username.get(), "password": self.password.get()},
            )
            resp.raise_for_status()
        except requests.HTTPError as e:
            self.error_label = ttk.Label(
                self.frm, text=f"Unable to {action}: {resp.status_code}"
            )
            self.error_label.grid()
            return
        except Exception as e:
            self.error_label = ttk.Label(
                self.frm, text=f"Unable to connect to server: {e}"
            )
            self.error_label.grid()
            return

        data = resp.json()
        self.frm.destroy()

        CharSelectionScreen(self.root, data.get("characters", []), data["token"])


class CharSelectionScreen:
    def __init__(self, root, characters=[], token=""):
        self.root = root
        self.token = token
        self.character_id = tk.IntVar()

        self.error_label = None
        self.frm = ttk.Frame(root, padding=15)
        self.frm.grid(sticky=tk.W)

        char_frm = ttk.Frame(self.frm)
        char_frm.grid()
        for i, character in enumerate(characters, start=1):
            ttk.Label(char_frm, text=character["name"]).grid(row=i, column=1)
            ttk.Button(
                char_frm, text="Select", command=partial(self.select, character["id"])
            ).grid(row=i, column=2)

        ttk.Button(
            self.frm, text="Create New Character", command=self.create_character
        ).grid()

    def select(self, char_id, *args):
        boot_and_exit(char_id, False, self.token)

    def create_character(self, *args):
        if self.error_label:
            self.error_label.destroy()

        try:
            resp = requests.post(f"{ENDPOINT}/character", json={"token": self.token})
            resp.raise_for_status()
        except requests.HTTPError as e:
            self.error_label = ttk.Label(
                self.frm, text=f"Unable to create character: {resp.status_code}"
            )
            self.error_label.grid()
            return
        except Exception as e:
            self.error_label = ttk.Label(
                self.frm, text=f"Unable to connect to server: {e}"
            )
            self.error_label.grid()
            return

        data = resp.json()
        boot_and_exit(data["id"], True, self.token)


root = tk.Tk(className="mhf")
root.columnconfigure(0, weight=1)
root.rowconfigure(0, weight=1)
LoginScreen(root)
root.mainloop()