Modus/config/data.py

import json
import os

import gi
from fabric.utils.helpers import get_relative_path
from gi.repository import Gdk, GLib

gi.require_version("Gtk", "3.0")

APP_NAME = "modus"
APP_NAME_CAP = "Modus"

ALLOWED_PLAYERS = ["vlc", "cmus", "firefox", "spotify", "chromium", "vivaldi", "brave"]


def parse_timeout_string(timeout_str):
    """
    Parse timeout string in format like '5s', '10m', '30s' etc.
    Returns timeout in milliseconds.
    """
    if not timeout_str or not isinstance(timeout_str, str):
        return 5000

    timeout_str = timeout_str.strip().lower()

    if timeout_str.endswith("s"):
        try:
            seconds = int(timeout_str[:-1])
            return seconds * 1000
        except ValueError:
            return 5000
    elif timeout_str.endswith("m"):
        try:
            minutes = int(timeout_str[:-1])
            return minutes * 60 * 1000
        except ValueError:
            return 5000
    else:
        try:
            seconds = int(timeout_str)
            return seconds * 1000
        except ValueError:
            return 5000


CACHE_DIR = str(GLib.get_user_cache_dir()) + f"/{APP_NAME}"

USERNAME = os.getlogin()
HOSTNAME = os.uname().nodename
HOME_DIR = os.path.expanduser("~")

CONFIG_DIR = os.path.expanduser(f"~/.config/{APP_NAME}")

screen = Gdk.Screen.get_default()
CURRENT_WIDTH = screen.get_width()
CURRENT_HEIGHT = screen.get_height()


WALLPAPERS_DIR_DEFAULT = get_relative_path("../assets/wallpapers_example")
CONFIG_FILE = get_relative_path("../config/assets/config.json")
MATUGEN_STATE_FILE = os.path.join(CONFIG_DIR, "matugen")

WORKSPACE_NUMS = False
WORKSPACE_USE_CHINESE_NUMERAL = False
WORKSPACE_DOTS = False

DOCK_THEME = "Pills"


def load_config():
    """Load the configuration from config.json"""
    config = {}

    if os.path.exists(CONFIG_FILE):
        try:
            with open(CONFIG_FILE, "r") as f:
                config = json.load(f)
        except Exception as e:
            print(f"Error loading config: {e}")

    return config


if os.path.exists(CONFIG_FILE):
    with open(CONFIG_FILE, "r") as f:
        config = json.load(f)
    WALLPAPERS_DIR = config.get("wallpapers_dir", WALLPAPERS_DIR_DEFAULT)
    DOCK_POSITION = config.get("dock_position", "Bottom")
    VERTICAL = DOCK_POSITION in ["Left", "Right"]
    TERMINAL_COMMAND = config.get("terminal_command", "kitty -e")
    DOCK_ENABLED = config.get("dock_enabled", True)
    DOCK_AUTO_HIDE = config.get("dock_auto_hide", True)
    DOCK_ALWAYS_OCCLUDED = config.get("dock_always_occluded", False)
    DOCK_THEME = config.get("dock_theme", "Pills")
    DOCK_ICON_SIZE = config.get("dock_icon_size", 28)
    WORKSPACE_NUMS = config.get("workspace_nums", False)
    WORKSPACE_USE_CHINESE_NUMERAL = config.get("workspace_use_chinese_numerals", False)
    WORKSPACE_DOTS = config.get("workspace_dots", False)
    WINDOW_SWITCHER_ITEMS_PER_ROW = config.get("window_switcher_items_per_row", 13)
    DOCK_HIDE_SPECIAL_WORKSPACE = config.get("dock_hide_special_workspace", True)
    DOCK_HIDE_SPECIAL_WORKSPACE_APPS = config.get(
        "dock_hide_special_workspace_apps", True
    )

    NOTIFICATION_TIMEOUT_STR = config.get("notification_timeout", "5s")
    NOTIFICATION_TIMEOUT = parse_timeout_string(NOTIFICATION_TIMEOUT_STR)
    NOTIFICATION_IGNORED_APPS = config.get("notification_ignored_apps", ["Hyprshot"])

    DOCK_COMPONENTS_VISIBILITY = {
        "workspace": config.get("dock_workspace_visible", True),
        "metrics": config.get("dock_metrics_visible", True),
        "battery": config.get("dock_battery_visible", True),
        "date_time": config.get("dock_date_time_visible", True),
        "controls": config.get("dock_controls_visible", True),
        "indicators": config.get("dock_indicators_visible", True),
        "notifications": config.get("dock_notifications_visible", True),
        "systray": config.get("dock_tray_visible", True),
        "applications": config.get("dock_applications_visible", True),
        "language": config.get("dock_language_visible", True),
    }

    METRICS_DISKS = config.get("metrics_disks", ["/"])
    METRICS_VISIBLE = config.get(
        "metrics_visible",
        {"cpu": True, "ram": True, "disk": True, "swap": True, "gpu": False},
    )
else:
    WALLPAPERS_DIR = WALLPAPERS_DIR_DEFAULT
    DOCK_POSITION = "Bottom"
    VERTICAL = False
    DOCK_ENABLED = True
    DOCK_ALWAYS_OCCLUDED = False
    DOCK_AUTO_HIDE = True
    TERMINAL_COMMAND = "kitty -e"
    DOCK_THEME = "Pills"
    DOCK_ICON_SIZE = 30
    WINDOW_SWITCHER_ITEMS_PER_ROW = 13
    DOCK_HIDE_SPECIAL_WORKSPACE = True
    DOCK_HIDE_SPECIAL_WORKSPACE_APPS = True

    NOTIFICATION_TIMEOUT_STR = "5s"
    NOTIFICATION_TIMEOUT = parse_timeout_string(NOTIFICATION_TIMEOUT_STR)
    NOTIFICATION_IGNORED_APPS = ["Hyprshot"]

    DOCK_COMPONENTS_VISIBILITY = {
        "workspace": True,
        "metrics": True,
        "battery": True,
        "date_time": True,
        "controls": True,
        "indicators": True,
        "notifications": True,
        "systray": True,
        "applications": True,
        "language": True,
    }

    METRICS_DISKS = ["/"]
    METRICS_VISIBLE = {
        "cpu": True,
        "ram": True,
        "disk": True,
        "swap": True,
        "gpu": False,
    }