LGHUB_Profile_Editor.py

#!/usr/bin/env python3
# -------------------------------------------------------------
# Logitech G-Hub Profile Editor
# -------------------------------------------------------------
# Note: Same as "GHUB_A6IB3(A).py", renamed for distribution.

import os
import json
import re
import logging
import sqlite3
import tkinter as tk
from tkinter import ttk, filedialog, messagebox
from PIL import Image, ImageTk
import ctypes

# ---------------------------------------
# Logging
# ---------------------------------------
logging.basicConfig(
    level=logging.DEBUG,
    format='[%(levelname)s] %(asctime)s - %(message)s'
)

# ---------------------------------------
# Constants
# ---------------------------------------
CONFIG_FILENAME = "ghub_db_location_config.json"  # Updated config file name
TABLE_NAME = "DATA"
ID_COLUMN = "_id"
JSON_COLUMN = "FILE"

# ---------------------------------------
# Config Handling
# ---------------------------------------
def get_config_path():
    """
    Returns the absolute path to the config file in the same folder as this script/exe.
    """
    exe_dir = os.path.dirname(os.path.abspath(__file__))
    return os.path.join(exe_dir, CONFIG_FILENAME)

def load_or_create_config():
    """
    Tries to load the config from 'ghub_db_location_config.json'.
    If it doesn't exist, prompts the user to pick a DB path,
    then writes a new config file.
    Returns the loaded config dict with at least a "db_path" key.
    """
    config_path = get_config_path()

    if os.path.isfile(config_path):
        # Load existing config
        try:
            with open(config_path, "r", encoding="utf-8") as f:
                cfg = json.load(f)
            if "db_path" in cfg and os.path.isfile(cfg["db_path"]):
                logging.debug(f"Config loaded successfully: {cfg}")
                return cfg
            else:
                logging.debug("Config file found but db_path invalid. Prompting user.")
        except Exception as e:
            logging.warning(f"Failed to parse config file: {e}")

    # If we reach here, config is missing or db_path is invalid.
    # Prompt user to pick the DB path
    root = tk.Tk()
    root.withdraw()
    messagebox.showinfo("First Setup", "Please locate your Logitech G-Hub 'settings.db' file.")
    db_path = filedialog.askopenfilename(
        title="Select your G-Hub settings.db",
        filetypes=[("SQLite DB", "*.db"), ("All files", "*.*")]
    )
    root.destroy()

    if not db_path or not os.path.isfile(db_path):
        # User canceled or invalid file
        raise FileNotFoundError("No valid G-Hub DB path chosen. Cannot proceed.")

    # Create minimal config
    cfg = {"db_path": db_path}

    try:
        with open(config_path, "w", encoding="utf-8") as f:
            json.dump(cfg, f, indent=2)
        logging.debug(f"Created new config: {cfg}")
    except Exception as e:
        logging.warning(f"Failed to write config file: {e}")

    return cfg

def save_config(cfg):
    """
    Overwrites the entire config object to 'ghub_db_location_config.json'.
    """
    config_path = get_config_path()
    try:
        with open(config_path, "w", encoding="utf-8") as f:
            json.dump(cfg, f, indent=2)
        logging.debug("Config updated/saved.")
    except Exception as e:
        logging.error(f"Failed to save config: {e}")

# ---------------------------------------
# DB Helpers
# ---------------------------------------
def load_profiles_from_db(db_path):
    """
    Reads the 'DATA' table from G-Hub's settings.db, decodes the BLOB column as JSON,
    returns a list of dicts: { db_row_id, entire_json, profile }.
    Each 'profile' is one entry in the "applications" array.
    """
    logging.debug(f"Loading profiles from DB: {db_path}")
    conn = sqlite3.connect(db_path)
    cursor = conn.cursor()
    cursor.execute(f"SELECT {ID_COLUMN}, {JSON_COLUMN} FROM {TABLE_NAME}")
    rows = cursor.fetchall()
    conn.close()

    all_profiles = []
    for (row_id, data_blob) in rows:
        if not data_blob:
            continue
        try:
            json_str = data_blob.decode("utf-8")
            parsed_data = json.loads(json_str)
        except Exception as e:
            logging.warning(f"Failed to parse JSON row {row_id}: {e}")
            continue

        apps_section = parsed_data.get("applications")
        if isinstance(apps_section, dict):
            apps_list = apps_section.get("applications", [])
            if isinstance(apps_list, list):
                for prof in apps_list:
                    all_profiles.append({
                        "db_row_id": row_id,
                        "entire_json": parsed_data,
                        "profile": prof
                    })

    # Sort them by profile "name" alphabetically
    all_profiles.sort(key=lambda p: p["profile"].get("name", "").lower())
    logging.info(f"Loaded {len(all_profiles)} profiles across all rows.")
    return all_profiles

def save_profile_to_db(db_path, row_id, entire_json):
    """
    Writes the updated JSON (as BLOB) back to the DB, row matching row_id.
    """
    try:
        new_json_str = json.dumps(entire_json, indent=2)
        new_blob = new_json_str.encode("utf-8")
    except Exception as e:
        logging.error(f"Could not encode updated JSON: {e}")
        return

    try:
        conn = sqlite3.connect(db_path)
        cursor = conn.cursor()
        sql = f"UPDATE {TABLE_NAME} SET {JSON_COLUMN} = ? WHERE {ID_COLUMN} = ?"
        cursor.execute(sql, (new_blob, row_id))
        conn.commit()
        conn.close()
        logging.debug(f"Row {row_id} updated in DB.")
    except Exception as e:
        logging.error(f"DB update failed: {e}")

# ---------------------------------------
# Main GUI
# ---------------------------------------
class GHubEditorApp:
    def __init__(self, master, config):
        self.master = master
        master.title("G-Hub Profile Editor")

        # Minimal dark theme
        self.apply_dark_theme(master)

        # Keep config in memory so we can re-save changes (like new db_path)
        self.config = config
        self.db_path = config["db_path"]

        # Determine the icon_cache folder relative to the DB file
        db_dir = os.path.dirname(self.db_path)
        self.icon_cache_folder = os.path.join(db_dir, "icon_cache")

        # Create icon_cache folder if missing
        if not os.path.isdir(self.icon_cache_folder):
            try:
                os.makedirs(self.icon_cache_folder, exist_ok=True)
                logging.debug(f"Created icon_cache folder: {self.icon_cache_folder}")
            except Exception as e:
                logging.warning(f"Failed to create icon_cache folder: {e}")

        self.profiles = load_profiles_from_db(self.db_path)
        self.selected_profile_index = None
        self.icon_tk = None

        # --------- Menubar / Settings menu ---------
        self.menubar = tk.Menu(self.master, bg="#2a2a2a", fg="#ffffff", tearoff=False)
        self.settings_menu = tk.Menu(self.menubar, tearoff=False, bg="#2a2a2a", fg="#ffffff")
        self.settings_menu.add_command(label="Change DB Path", command=self.change_db_path)
        self.menubar.add_cascade(label="Settings", menu=self.settings_menu)
        self.master.config(menu=self.menubar)

        # --------- Left Frame for list ---------
        self.left_frame = tk.Frame(master, bg="#2a2a2a")
        self.left_frame.pack(side=tk.LEFT, fill=tk.BOTH, expand=False)

        # --------- Right Frame for details ---------
        self.right_frame = ttk.Frame(master, style="Dark.TFrame")
        self.right_frame.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)

        # Listbox (with exportselection=0)
        self.profile_list_var = tk.StringVar()
        self.profile_listbox = tk.Listbox(
            self.left_frame,
            listvariable=self.profile_list_var,
            width=40,
            height=25,
            exportselection=False,  # preserve selection
            bg="#2a2a2a",
            fg="#ffffff",
            selectbackground="#444444",
            highlightthickness=1,
            highlightcolor="#444444",
            bd=0
        )
        self.profile_listbox.bind("<<ListboxSelect>>", self.on_profile_select)

        # Scrollbar
        self.scrollbar = tk.Scrollbar(self.left_frame, orient=tk.VERTICAL, bg="#2a2a2a")
        self.scrollbar.config(command=self.profile_listbox.yview)
        self.profile_listbox.config(yscrollcommand=self.scrollbar.set)

        self.profile_listbox.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)
        self.scrollbar.pack(side=tk.LEFT, fill=tk.Y)

        # Populate list
        self.populate_list()

        # Row offset
        row_idx = 0

        # Add / Delete
        add_button = ttk.Button(self.right_frame, text="Add Entry", command=self.add_entry)
        add_button.grid(row=row_idx, column=0, padx=5, pady=5, sticky="w")

        del_button = ttk.Button(self.right_frame, text="Delete Entry", command=self.delete_entry)
        del_button.grid(row=row_idx, column=1, padx=5, pady=5, sticky="w")

        # Name
        row_idx += 1
        ttk.Label(self.right_frame, text="Profile Name:", style="Dark.TLabel").grid(
            row=row_idx, column=0, sticky="e", padx=5, pady=5
        )
        self.name_entry_var = tk.StringVar()
        self.name_entry = ttk.Entry(self.right_frame, textvariable=self.name_entry_var, width=50, style="Dark.TEntry")
        self.name_entry.grid(row=row_idx, column=1, sticky="w", padx=5, pady=5)

        # App Path
        row_idx += 1
        ttk.Label(self.right_frame, text="Application Path:", style="Dark.TLabel").grid(
            row=row_idx, column=0, sticky="e", padx=5, pady=5
        )
        self.app_path_var = tk.StringVar()
        self.app_path_entry = ttk.Entry(self.right_frame, textvariable=self.app_path_var, width=50, style="Dark.TEntry")
        self.app_path_entry.grid(row=row_idx, column=1, sticky="w", padx=5, pady=5)

        # Icon Path
        row_idx += 1
        ttk.Label(self.right_frame, text="Icon (posterPath):", style="Dark.TLabel").grid(
            row=row_idx, column=0, sticky="e", padx=5, pady=5
        )
        self.icon_path_var = tk.StringVar()
        self.icon_path_entry = ttk.Entry(self.right_frame, textvariable=self.icon_path_var, width=50, style="Dark.TEntry")
        self.icon_path_entry.grid(row=row_idx, column=1, sticky="w", padx=5, pady=5)

        ttk.Button(self.right_frame, text="Browse Icon", command=self.browse_icon).grid(
            row=row_idx, column=2, sticky="w", padx=5, pady=5
        )

        # Clear icon
        row_idx += 1
        clear_button = ttk.Button(self.right_frame, text="Clear Icon", command=self.clear_icon)
        clear_button.grid(row=row_idx, column=1, sticky="w", padx=5, pady=5)

        # Icon preview
        row_idx += 1
        self.icon_label = ttk.Label(self.right_frame, text="(No icon loaded)", style="Dark.TLabel")
        self.icon_label.grid(row=row_idx, column=0, columnspan=3, padx=5, pady=5)

        # Save changes
        row_idx += 1
        ttk.Button(self.right_frame, text="Save Changes", command=self.save_changes).grid(
            row=row_idx, column=0, columnspan=3, pady=10
        )

        # Let second column expand
        self.right_frame.grid_columnconfigure(1, weight=1)

    # -----------------------------
    # Dark Theme Setup
    # -----------------------------
    def apply_dark_theme(self, root):
        style = ttk.Style(root)
        style.theme_use("clam")
        root.configure(bg="#2a2a2a")

        # TFrame
        style.configure("Dark.TFrame", background="#2a2a2a")
        # TLabel
        style.configure("Dark.TLabel", background="#2a2a2a", foreground="#ffffff")
        # TEntry
        style.configure("Dark.TEntry", fieldbackground="#444444", foreground="#ffffff")
        # TButton
        style.configure("TButton", background="#444444", foreground="#ffffff")

    # -----------------------------
    # Menubar - Change DB Path
    # -----------------------------
    def change_db_path(self):
        """
        Opens a file dialog so the user can change the DB path.
        Updates config and reloads profiles and icon_cache_folder.
        """
        new_path = filedialog.askopenfilename(
            title="Select your G-Hub settings.db",
            filetypes=[("SQLite DB", "*.db"), ("All files", "*.*")]
        )
        if not new_path or not os.path.isfile(new_path):
            return  # user canceled or invalid

        self.config["db_path"] = new_path
        save_config(self.config)

        # Recompute icon_cache_folder
        db_dir = os.path.dirname(new_path)
        self.icon_cache_folder = os.path.join(db_dir, "icon_cache")

        if not os.path.isdir(self.icon_cache_folder):
            try:
                os.makedirs(self.icon_cache_folder, exist_ok=True)
                logging.debug(f"Created icon_cache folder: {self.icon_cache_folder}")
            except Exception as e:
                logging.warning(f"Failed to create icon_cache folder: {e}")

        # Reload everything
        self.db_path = new_path
        self.profiles = load_profiles_from_db(self.db_path)
        self.selected_profile_index = None
        self.populate_list()

        # Clear fields
        self.name_entry_var.set("")
        self.app_path_var.set("")
        self.icon_path_var.set("")
        self.icon_label.config(text="(No icon loaded)", image="")
        messagebox.showinfo("DB Path Changed", f"Now using DB:\n{new_path}")

    # -----------------------------
    # Profile List
    # -----------------------------
    def populate_list(self):
        names = [p["profile"].get("name", "(Unnamed)") for p in self.profiles]
        self.profile_list_var.set(names)

    def on_profile_select(self, event):
        try:
            idx = self.profile_listbox.curselection()[0]
        except IndexError:
            self.selected_profile_index = None
            return

        self.selected_profile_index = idx
        item = self.profiles[idx]
        prof = item["profile"]

        self.name_entry_var.set(prof.get("name", ""))
        self.app_path_var.set(prof.get("applicationPath", ""))
        self.icon_path_var.set(prof.get("posterPath", ""))

        self.load_icon_preview()

    # -----------------------------
    # Add / Delete
    # -----------------------------
    def add_entry(self):
        if not self.profiles:
            messagebox.showinfo("No DB Rows", "No existing rows found in DB to attach a new entry.")
            return

        row_id = self.profiles[0]["db_row_id"]
        entire_json = self.profiles[0]["entire_json"]
        apps_section = entire_json.get("applications", {})
        apps_list = apps_section.get("applications", [])

        new_profile = {
            "applicationId": "new-app-id",
            "applicationPath": "",
            "isCustom": True,
            "name": "New Entry",
            "posterPath": ""
        }
        apps_list.append(new_profile)
        apps_section["applications"] = apps_list
        entire_json["applications"] = apps_section

        save_profile_to_db(self.db_path, row_id, entire_json)

        # Reload
        self.profiles = load_profiles_from_db(self.db_path)
        self.populate_list()

        # Auto-select
        idx = None
        for i, p in enumerate(self.profiles):
            if p["profile"].get("name") == "New Entry":
                idx = i
                break
        if idx is not None:
            self.profile_listbox.selection_clear(0, tk.END)
            self.profile_listbox.selection_set(idx)
            self.profile_listbox.event_generate("<<ListboxSelect>>")

    def delete_entry(self):
        if self.selected_profile_index is None:
            messagebox.showwarning("No selection", "Please select a profile first.")
            return

        confirm = messagebox.askyesno("Delete Entry", "Are you sure you want to delete this entry?")
        if not confirm:
            return

        item = self.profiles[self.selected_profile_index]
        row_id = item["db_row_id"]
        entire_json = item["entire_json"]
        prof_to_delete = item["profile"]

        apps_list = entire_json.get("applications", {}).get("applications", [])
        if prof_to_delete in apps_list:
            apps_list.remove(prof_to_delete)
        entire_json["applications"]["applications"] = apps_list

        save_profile_to_db(self.db_path, row_id, entire_json)

        # Reload
        self.profiles = load_profiles_from_db(self.db_path)
        self.selected_profile_index = None
        self.populate_list()

        # Clear fields
        self.name_entry_var.set("")
        self.app_path_var.set("")
        self.icon_path_var.set("")
        self.icon_label.config(text="(No icon loaded)", image="")

    # -----------------------------
    # Icon
    # -----------------------------
    def browse_icon(self):
        if self.selected_profile_index is None:
            messagebox.showwarning("No Profile", "Select a profile first.")
            return

        file_path = filedialog.askopenfilename(
            title="Select icon file",
            filetypes=[
                ("Image Files", "*.bmp;*.ico;*.png;*.jpg;*.jpeg"),
                ("All files", "*.*")
            ]
        )
        if not file_path:
            return

        try:
            img = Image.open(file_path)
        except Exception as e:
            logging.error(f"Failed to open image {file_path}: {e}")
            messagebox.showerror("Error", f"Could not open image file:\n{e}")
            return

        item = self.profiles[self.selected_profile_index]
        prof = item["profile"]
        existing_path = prof.get("posterPath", "").strip()
        app_name = prof.get("name", "").strip() or "app_unknown"
        safe_name = re.sub(r'[^\w\s-]', '', app_name).strip().replace(' ', '_') or "icon"

        # If there's an existing file, overwrite its base. Otherwise, create new
        if existing_path:
            base, _ext = os.path.splitext(existing_path)
            final_path = base + ".bmp"
        else:
            final_path = os.path.join(self.icon_cache_folder, safe_name + ".bmp")

        # Convert to BMP
        try:
            img.save(final_path, "BMP")
        except Exception as e:
            logging.error(f"Failed to save BMP to {final_path}: {e}")
            messagebox.showerror("Error", f"Could not save BMP:\n{e}")
            return

        prof["posterPath"] = final_path
        self.icon_path_var.set(final_path)
        self.load_icon_preview()

    def clear_icon(self):
        if self.selected_profile_index is None:
            messagebox.showwarning("No Profile", "Select a profile first.")
            return
        item = self.profiles[self.selected_profile_index]
        prof = item["profile"]
        prof["posterPath"] = ""
        self.icon_path_var.set("")
        self.icon_label.config(text="(No icon loaded)", image="", compound=tk.NONE)
        logging.info("Icon cleared. posterPath is now empty.")

    def load_icon_preview(self):
        path = self.icon_path_var.get().strip()
        if not path:
            self.icon_label.config(text="(No icon loaded)", image="", compound=tk.NONE)
            return
        if not os.path.isfile(path):
            self.icon_label.config(text="(File not found)", image="", compound=tk.NONE)
            return

        try:
            img = Image.open(path)
            self.icon_tk = ImageTk.PhotoImage(img)
            self.icon_label.config(image=self.icon_tk, text="", compound=tk.NONE)
        except Exception as e:
            logging.warning(f"Failed to load image '{path}': {e}")
            self.icon_label.config(text="(Invalid image)", image="", compound=tk.NONE)

    # -----------------------------
    # Save
    # -----------------------------
    def save_changes(self):
        if self.selected_profile_index is None:
            messagebox.showwarning("No Profile", "Select a profile first.")
            return

        item = self.profiles[self.selected_profile_index]
        row_id = item["db_row_id"]
        entire_json = item["entire_json"]
        prof = item["profile"]

        prof["name"] = self.name_entry_var.get()
        prof["applicationPath"] = self.app_path_var.get()
        prof["posterPath"] = self.icon_path_var.get()

        save_profile_to_db(self.db_path, row_id, entire_json)

        # Reload
        self.profiles = load_profiles_from_db(self.db_path)
        self.populate_list()

        # Keep selection if possible
        new_name = prof["name"]
        idx = None
        for i, p in enumerate(self.profiles):
            if p["profile"].get("name") == new_name:
                idx = i
                break
        if idx is not None:
            self.profile_listbox.selection_clear(0, tk.END)
            self.profile_listbox.selection_set(idx)
            self.profile_listbox.event_generate("<<ListboxSelect>>")

        logging.info(f"Changes saved for profile '{new_name}' (row ID={row_id}).")
        messagebox.showinfo("Saved", f"Profile '{new_name}' updated.")

# ---------------------------------------
# Main
# ---------------------------------------
def main():
    # 1) Load or create config
    config = load_or_create_config()

    # 2) Launch GUI
    root = tk.Tk()
    
    # Configure dark theme for Windows titlebar
    try:
        root.tk.call('tk', 'windowingsystem')
        root.tk.call('set', '::tk::WindowingsSystem', 'win32')
        root.wm_attributes('-toolwindow', False)
        root.wm_attributes('-transparentcolor', '')
        DWMWA_USE_IMMERSIVE_DARK_MODE = 20
        root.wm_attributes('-alpha', 0.0)
        root.update()
        hwnd = ctypes.windll.user32.GetParent(root.winfo_id())
        ctypes.windll.dwmapi.DwmSetWindowAttribute(
            hwnd,
            DWMWA_USE_IMMERSIVE_DARK_MODE,
            ctypes.byref(ctypes.c_int(1)),
            ctypes.sizeof(ctypes.c_int)
        )
        root.wm_attributes('-alpha', 1.0)
    except:
        pass  # Fallback if Windows-specific calls fail

    # 3) Create app instance
    app = GHubEditorApp(root, config)
    root.mainloop()

if __name__ == "__main__":
    main()