killswitch-switcher

#!/bin/bash

CONFIG_DIR="$HOME/.config/killswitch"
CONFIG_FILE="$CONFIG_DIR/config.sh"
STATE_DIR="$HOME/.local/state/killswitch"
NEXT_SESSION_FILE="$STATE_DIR/next-session"
CURRENT_SESSION_FILE="$STATE_DIR/current-session"

if [ -f "$CONFIG_FILE" ]; then
    source "$CONFIG_FILE"
else
    notify-send "Killswitch" "Config not found: $CONFIG_FILE"
    exit 1
fi

mkdir -p "$STATE_DIR"

export DISPLAY="${DISPLAY:-:0}"
export XDG_RUNTIME_DIR="${XDG_RUNTIME_DIR:-/run/user/$(id -u)}"

declare -A name_to_id
menu_items=""

for var in $(compgen -v | grep '^session_.*_name$'); do
    id=$(echo "$var" | sed -e 's/^session_//' -e 's/_name$//')
    name="${!var}"
    name_to_id["$name"]="$id"
    if [ -n "$menu_items" ]; then
        menu_items="$menu_items\n$name"
    else
        menu_items="$name"
    fi
done

if [ -n "$ROFI_THEME" ]; then
    choice=$(echo -e "$menu_items" | rofi -dmenu -p "Switch Mode:" -theme "$ROFI_THEME" -no-custom)
else
    choice=$(echo -e "$menu_items" | rofi -dmenu -p "Switch Mode:" -no-custom)
fi

if [ -n "$choice" ] && [ -n "${name_to_id["$choice"]}" ]; then
    target_id="${name_to_id["$choice"]}"
    
    echo "$target_id" > "$NEXT_SESSION_FILE"
    notify-send "Session Switcher" "Switching to $choice..." -t 2000
    
    if [ -f "$CURRENT_SESSION_FILE" ]; then
        CURRENT_SESSION=$(cat "$CURRENT_SESSION_FILE")

        kill_func="session_${CURRENT_SESSION}_kill"
        if type "$kill_func" >/dev/null 2>&1; then
            $kill_func
            exit 0
        fi
    fi

    if pgrep -x "gamescope" > /dev/null; then
        pkill -TERM gamescope
        systemctl --user stop gamescope-session-plus@steam 2>/dev/null || true
    elif [ -n "$HYPRLAND_INSTANCE_SIGNATURE" ]; then
        hyprctl dispatch exit
    else
        pkill -u $USER niri 2>/dev/null || true
        pkill -u $USER gamescope 2>/dev/null || true
    fi
fi