#!/bin/bash
# SteamOS desktop extras that cachyos-vapor doesn't ship, taken from the
# newest Valve steamdeck-kde-presets package: "Add to Steam" (file
# right-click menu and launcher), Nested Desktop (Plasma as a game inside
# gaming mode), the "Return to Gaming Mode" icon, the Steam Keyboard window
# rule. Part of the SteamOS theme component
# (lib/vapor-theme.sh). System files go to /usr/local (nothing
# package-owned); per-user settings go through the undo journal.
# Sourced by steamify.sh; not meant to be run on its own.

NESTED_DIR=/usr/local/share/applications/steam/holo-nested-desktop
EXTRAS_FILES=(
    /usr/local/bin/holo-add-to-steam
    /usr/local/bin/holo-nested-desktop
    /usr/local/share/kio/servicemenus/steam.desktop
    /usr/local/share/plasma/kickeractions/steam.desktop
    /usr/local/share/icons/hicolor/scalable/actions/gaming-return.svg
    "$NESTED_DIR"
)
# Group name for our rule in kwinrulesrc; KWin accepts any unique id.
STEAM_KEYBOARD_RULE=cachyos-gamescope-boot-steam-keyboard

fetch_valve_presets() {
    # fetch_valve_presets <dir>: download and extract the newest
    # steamdeck-kde-presets (from the newest SteamOS release repo, jupiter-3.9,
    # jupiter-3.10, ...) into <dir>.
    local dir="$1" pkg
    pkg="$(valve_fetch_pkg jupiter steamdeck-kde-presets "$dir")" || return 1
    tar -I unzstd -xf "$pkg" -C "$dir" usr/bin usr/share ||
        { err "Extracting $(basename "$pkg") failed."; return 1; }
}

extras_enable() {
    info "Installing SteamOS desktop extras (Add to Steam, Nested Desktop, icons)..."
    local pkg
    for pkg in curl zstd kdialog; do
        # holo-add-to-steam and holo-nested-desktop show errors with kdialog.
        pacman -Q "$pkg" >/dev/null 2>&1 && continue
        sudo pacman -S --needed --noconfirm "$pkg" && state_set theme "installed_$pkg" 1
    done

    local tmp_dir src
    make_tmpdir tmp_dir
    if ! fetch_valve_presets "$tmp_dir"; then
        rm -rf "$tmp_dir"
        return 1
    fi
    src="$tmp_dir/usr"
    sudo install -D -m 755 "$src/bin/holo-add-to-steam" /usr/local/bin/holo-add-to-steam
    sudo install -D -m 755 "$src/bin/holo-nested-desktop" /usr/local/bin/holo-nested-desktop
    sudo install -D -m 644 "$src/share/kio/servicemenus/steam.desktop" /usr/local/share/kio/servicemenus/steam.desktop
    sudo install -D -m 644 "$src/share/plasma/kickeractions/steam.desktop" /usr/local/share/plasma/kickeractions/steam.desktop
    # gaming-return.svg is a symlink in the package; SteamOS uses it for the
    # Return to Gaming Mode shortcut on non-Deck hardware.
    sudo install -D -m 644 "$src/share/icons/hicolor/scalable/actions/steam-gaming-return.svg" \
        /usr/local/share/icons/hicolor/scalable/actions/gaming-return.svg
    # Nested Desktop with its Steam library art; the entry points at the
    # art under /usr/share, which is where Valve puts it.
    sudo rm -rf "$NESTED_DIR"
    sudo mkdir -p "$(dirname "$NESTED_DIR")"
    sudo cp -r "$src/share/applications/steam/holo-nested-desktop" "$NESTED_DIR"
    sudo sed -i "s|/usr/share/applications/steam/holo-nested-desktop|$NESTED_DIR|" "$NESTED_DIR/holo-nested-desktop.desktop"

    rm -rf "$tmp_dir"

    sudo gtk-update-icon-cache -q -f -t /usr/local/share/icons/hicolor 2>/dev/null || true
    kbuildsycoca6 >/dev/null 2>&1 || true
    set_shortcut_icon gaming-return

    # Keep Steam's on-screen keyboard (Steam + X) above other windows and out
    # of the taskbar, like SteamOS's kwinrulesrc.
    local rules rule="$STEAM_KEYBOARD_RULE"
    rules="$(kreadconfig6 --file kwinrulesrc --group General --key rules)"
    if [[ ",$rules," != *",$rule,"* ]]; then
        kset theme kwinrulesrc "$rule" Description "Window settings for Steam Keyboard"
        kset theme kwinrulesrc "$rule" above true
        kset theme kwinrulesrc "$rule" aboverule 2
        kset theme kwinrulesrc "$rule" skiptaskbar true
        kset theme kwinrulesrc "$rule" skiptaskbarrule 2
        kset theme kwinrulesrc "$rule" title "Steam Keyboard"
        kset theme kwinrulesrc "$rule" titlematch 2
        kset theme kwinrulesrc "$rule" type 16
        kset theme kwinrulesrc "$rule" typerule 2
        kset theme kwinrulesrc "$rule" wmclass steam
        kset theme kwinrulesrc "$rule" wmclasscomplete true
        kset theme kwinrulesrc "$rule" wmclassmatch 2
        kset theme kwinrulesrc General rules "${rules:+$rules,}$rule"
        kset theme kwinrulesrc General count "$(( $(kreadconfig6 --file kwinrulesrc --group General --key count --default 0) + 1 ))"
        qdbus6 org.kde.KWin /KWin org.kde.KWin.reconfigure >/dev/null 2>&1 || true
    fi
}

extras_disable() {
    sudo rm -rf "${EXTRAS_FILES[@]}"
    sudo gtk-update-icon-cache -q -f -t /usr/local/share/icons/hicolor 2>/dev/null || true
    kbuildsycoca6 >/dev/null 2>&1 || true
    set_shortcut_icon steam
    local pkg
    for pkg in kdialog zstd curl; do
        [[ -n "$(state_get theme "installed_$pkg")" ]] && sudo pacman -R --noconfirm "$pkg" >/dev/null 2>&1
    done
    # The keyboard rule is reverted by the theme's journal.
    return 0
}