main.py

import os
import sys
import shutil
import signal
from loguru import logger

from services.paths import (
    get_user_config_dir,
    get_config_path,
    get_suits_config_path,
    get_user_state_dir,
    resolve_style_file,
)

def _on_signal(*_):
    sys.exit(0)

signal.signal(signal.SIGTERM, _on_signal)
signal.signal(signal.SIGINT, _on_signal)

# Configure clean production logging (INFO default, DEBUG with AGILITY_DEBUG=1 or --debug)
log_level = "DEBUG" if os.getenv("AGILITY_DEBUG", "").lower() in ("1", "true", "yes") or "--debug" in sys.argv else "INFO"
logger.remove()
logger.add(sys.stderr, level=log_level, format="<green>{time:YYYY-MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan> - <level>{message}</level>")

def seed_user_environment():
    user_dir = get_user_config_dir()
    cfg_dir = os.path.join(user_dir, "config")
    style_dir = os.path.join(user_dir, "style")
    state_dir = get_user_state_dir()

    os.makedirs(cfg_dir, exist_ok=True)
    os.makedirs(style_dir, exist_ok=True)
    os.makedirs(state_dir, exist_ok=True)

    # Ensure core configuration files are present
    get_config_path("config.json")
    get_suits_config_path()

    # Seed baseline stylesheets into user style dir if missing
    for css_file in ["borders.css", "fonts.css", "colors.css"]:
        user_css = os.path.join(style_dir, css_file)
        if not os.path.exists(user_css):
            default_css = resolve_style_file(css_file)
            if os.path.exists(default_css) and default_css != user_css:
                try:
                    shutil.copy2(default_css, user_css)
                except Exception:
                    pass

    # Clean up stale legacy component stylesheets in user style dir that shadow updated system stylesheets
    # Keep user customization files: borders, fonts, colors, agility-shell-colors, or custom*
    whitelist = {"borders.css", "fonts.css", "colors.css", "agility-shell-colors.css", "custom.css"}
    if os.path.isdir(style_dir):
        for fname in os.listdir(style_dir):
            if fname.endswith(".css") and fname not in whitelist and not fname.startswith("custom"):
                try:
                    os.remove(os.path.join(style_dir, fname))
                except Exception:
                    pass

    # Migrate existing widget settings if needed
    user_settings = os.path.join(user_dir, "widget_settings.json")
    if not os.path.exists(user_settings):
        legacy_qs = os.path.expanduser("~/.config/quickshell/widget_settings.json")
        repo_dir = os.path.dirname(os.path.abspath(__file__))
        default_qs = os.path.join(repo_dir, "quickshell", "agility", "widget_settings.json")
        src_settings = legacy_qs if os.path.exists(legacy_qs) else default_qs
        if os.path.exists(src_settings):
            try:
                shutil.copy2(src_settings, user_settings)
            except Exception:
                pass

seed_user_environment()

import utils.fabric_compat
import bar
import services.singletons as singletons
from setproctitle import setproctitle
from fabric import Application
from services.wallpaper import WallpaperService
from services.style import StyleService
from services.awe_service import AweService
from utils.sounds import play_sound
setproctitle("agility-shell")

app = Application("agility-shell")

singletons.style_service = StyleService(app)

singletons.style_service.reload()

bar_manager = bar.initialise_bars()
singletons.bar_manager = bar_manager

wallpaper_service = WallpaperService.get_instance()
# wallpaper_service.set_bar_manager(bar_manager)

from services.suits_service import SuitsService, suits_service
profile_service = singletons.profile_service

AweService.get_instance().init_startup()

play_sound("session-start")
app.run()