import time import requests from os import path from tkinter import ttk, messagebox, IntVar, Toplevel, N, S, W, E, END from modules.dark_titlebar import dark_title_bar from modules.utils import configGet, configSet, use_dark_mode from modules.logger import logger class WindowConfig(Toplevel): def __init__(self, parent): super().__init__(parent) self.window_width = 430 self.window_height = 247 self.screen_width = self.winfo_screenwidth() self.screen_height = self.winfo_screenheight() self.center_x = int(self.screen_width/2 - self.window_width / 2) self.center_y = int(self.screen_height/2 - self.window_height / 2) self.title("Configuration") self.geometry(f'{self.window_width}x{self.window_height}+{self.center_x}+{self.center_y}') self.resizable(False, False) self.iconbitmap(path.join("assets", "favicon.ico")) self.focus_set() if use_dark_mode() is True: dark_title_bar(self, "dark") self.columnconfigure(0, weight=1) self.columnconfigure(1, weight=3) # Address self.address_label = ttk.Label(self, text="Address:") self.address_label.grid(column=0, row=0, sticky=W, padx=9, pady=9) self.address_entry = ttk.Entry(self) self.address_entry.grid(column=1, row=0, sticky=N+S+E+W, padx=9, pady=9) if configGet("address") is not None: self.address_entry.insert(0, configGet("address")) # ======= # API Key self.apikey_label = ttk.Label(self, text="API key:") self.apikey_label.grid(column=0, row=1, sticky=W, padx=9, pady=9) self.apikey_entry = ttk.Entry(self) self.apikey_entry.grid(column=1, row=1, sticky=N+S+E+W, padx=9, pady=9) if configGet("apikey") is not None: self.apikey_entry.insert(0, configGet("apikey")) # ======= # Self-signed if configGet("allow_self_signed") is not None: self.self_signed_check_bool = IntVar(value=int(configGet("allow_self_signed"))) else: self.self_signed_check_bool = IntVar() self.self_signed_check = ttk.Checkbutton(self, text="Allow self-signed certificates", variable=self.self_signed_check_bool) self.self_signed_check.grid(column=1, row=2, sticky=W, padx=9, pady=9) # =========== # Saves location self.saves_location_label = ttk.Label(self, text="Saves location:") self.saves_location_label.grid(column=0, row=3, sticky=W, padx=9, pady=9) self.saves_location_button = ttk.Button(self, text="Browse", width=6) self.saves_location_button.grid(column=1, row=3, sticky=W, padx=9, pady=9) # ============== self.login_button = ttk.Button(self, text="Log in", style="Accent.TButton", width=10, command=self.validate_configuration) self.login_button.grid(column=1, row=4, sticky=E, padx=9, pady=9) def validate_configuration(self): if len(self.address_entry.get()) > 0: if self.address_entry.get().endswith("/"): self.address_text = self.address_entry.get() self.address_entry.delete(0, END) self.address_entry.insert(0, self.address_text[:-1]) try: requests.get(self.address_entry.get()+"/check", verify=not bool(self.self_signed_check_bool.get())) except (requests.exceptions.InvalidURL, requests.exceptions.InvalidSchema, requests.exceptions.MissingSchema) as exp: logger.error(f"Could not validate '{self.address_entry.get()}' due to {exp}") messagebox.showerror(title="Invalid address", message="Address entered does not seem to be correct one. Please provide API address starting with http:// or https:// \n\nFor example:\n- https://your-api.com:8043\n- https://your-api.com") return except (requests.exceptions.SSLError): logger.error(f"SSL certificate of '{self.address_entry.get()}' does not seem to be valid or is self-signed") messagebox.showerror(title="Invalid SSL", message="SSL certificate seems to be invalid or self-signed and thus won't be trusted. You can overwrite this by checking 'Allow self-signed certificates'.") return except Exception as exp: logger.error(f"Could not reach '{self.address_entry.get()}' due to {exp}") messagebox.showerror(title="Connection error", message="Address entered does not seem to be reachable. Please make sure you've entered the correct API address.") return response_apikey = requests.get(self.address_entry.get()+"/apikey", headers={"apikey": self.apikey_entry.get()}, verify=not bool(self.self_signed_check_bool.get())) if response_apikey.status_code != 200: logger.error(f"API key seems to be invalid. API returned {response_apikey.status_code} as a status code") messagebox.showerror(title="Invalid apikey", message="API key provided does not seem to be valid. Please check for any mistakes and if none found - take a look at the docs to learn how to generate one.") return configSet(["address"], self.address_entry.get()) configSet(["apikey"], self.apikey_entry.get()) configSet(["allow_self_signed"], bool(self.self_signed_check_bool.get())) messagebox.showinfo(title="Configuration completed", message="Your client is now configured and ready to use!") self.destroy()