DankMaterialShell/Services/FocusedWindowService.qml

pragma Singleton
pragma ComponentBehavior: Bound

import QtQuick
import Quickshell
import Quickshell.Io

Singleton {
    id: root

    property bool niriAvailable: false
    property string focusedAppId: ""
    property string focusedAppName: ""
    property string focusedWindowTitle: ""
    property int focusedWindowId: -1

    function loadInitialFocusedWindow() {
        if (root.niriAvailable)
            focusedWindowQuery.running = true;

    }

    function updateFocusedWindowData() {
        if (root.niriAvailable && root.focusedWindowId !== -1)
            focusedWindowQuery.running = true;
        else
            clearFocusedWindow();
    }

    function clearFocusedWindow() {
        root.focusedAppId = "";
        root.focusedAppName = "";
        root.focusedWindowTitle = "";
    }

    // Convert app_id to a more user-friendly display name
    function getDisplayName(appId) {
        if (!appId)
            return "";

        // Common app_id to display name mappings
        const appNames = {
            "com.mitchellh.ghostty": "Ghostty",
            "org.mozilla.firefox": "Firefox",
            "org.gnome.Nautilus": "Files",
            "org.gnome.TextEditor": "Text Editor",
            "com.google.Chrome": "Chrome",
            "org.telegram.desktop": "Telegram",
            "com.spotify.Client": "Spotify",
            "org.kde.konsole": "Konsole",
            "org.gnome.Terminal": "Terminal",
            "code": "VS Code",
            "code-oss": "VS Code",
            "org.mozilla.Thunderbird": "Thunderbird",
            "org.libreoffice.LibreOffice": "LibreOffice",
            "org.gimp.GIMP": "GIMP",
            "org.blender.Blender": "Blender",
            "discord": "Discord",
            "slack": "Slack",
            "zoom": "Zoom"
        };
        // Return mapped name or clean up the app_id
        if (appNames[appId])
            return appNames[appId];

        // Try to extract a clean name from the app_id
        // Remove common prefixes and make first letter uppercase
        let cleanName = appId.replace(/^(org\.|com\.|net\.|io\.)/, '').replace(/\./g, ' ').split(' ').map((word) => {
            return word.charAt(0).toUpperCase() + word.slice(1);
        }).join(' ');
        return cleanName;
    }

    Component.onCompleted: {
        // Use the availability from NiriWorkspaceService to avoid duplicate checks
        root.niriAvailable = NiriWorkspaceService.niriAvailable;
        // Connect to workspace service events
        NiriWorkspaceService.onNiriAvailableChanged.connect(() => {
            root.niriAvailable = NiriWorkspaceService.niriAvailable;
            if (root.niriAvailable)
                loadInitialFocusedWindow();

        });
        if (root.niriAvailable)
            loadInitialFocusedWindow();

    }

    // Listen to window focus changes from NiriWorkspaceService
    Connections {
        function onFocusedWindowIdChanged() {
            root.focusedWindowId = parseInt(NiriWorkspaceService.focusedWindowId) || -1;
            updateFocusedWindowData();
        }

        function onFocusedWindowTitleChanged() {
            root.focusedWindowTitle = NiriWorkspaceService.focusedWindowTitle;
        }

        target: NiriWorkspaceService
    }

    // Process to get focused window info
    Process {
        id: focusedWindowQuery

        command: ["niri", "msg", "--json", "focused-window"]
        running: false

        stdout: StdioCollector {
            onStreamFinished: {
                if (text && text.trim()) {
                    try {
                        const windowData = JSON.parse(text.trim());
                        root.focusedAppId = windowData.app_id || "";
                        root.focusedWindowTitle = windowData.title || "";
                        root.focusedAppName = getDisplayName(windowData.app_id || "");
                        root.focusedWindowId = parseInt(windowData.id) || -1;
                    } catch (e) {
                        console.warn("FocusedWindowService: Failed to parse focused window data:", e);
                        clearFocusedWindow();
                    }
                } else {
                    clearFocusedWindow();
                }
            }
        }

    }

}