path: root/config/hypr/scripts/Tak0-Autodispatch.sh

#!/usr/bin/env bash
#
# ─────────────────────────────────────────────────────────────────────────────
# Tak0-Autodispatch.sh
# ─────────────────────────────────────────────────────────────────────────────
#
# 🇬🇧 ENGLISH
# -----------------------------------------------------------------------------
# This script is an "authoritative spawn dispatcher" for Hyprland.
#
# Its purpose is to FORCE all windows belonging to a single application launch
# (main window + helpers + Electron / Steam child processes)
# onto a specific workspace.
#
# It explicitly ignores:
#   • spawn race conditions
#   • delayed window creation
#   • detached helper processes
#   • Electron / Chromium / Steam madness
#
# Typical use cases:
#   • Launch Steam / Discord / browsers without window leakage
#   • Prevent apps from spawning on the currently focused workspace
#   • Control applications that completely ignore static windowrules
#
# Invocation:
#   ./Tak0-Autodispatch.sh <workspace> [rule ...] -- <command>
#
# Important notes:
#   • All window rules are TEMPORARY
#   • No permanent pollution of Hyprland configuration
#
# -----------------------------------------------------------------------------
# 🇺🇦 УКРАЇНСЬКА
# -----------------------------------------------------------------------------
# Цей скрипт — це "авторитарний диспетчер запуску" для Hyprland.
#
# Його задача — ГАРАНТОВАНО відправити ВСІ вікна одного запуску програми
# (основне вікно + helper-и + Electron / Steam дочірні процеси)
# на вказаний workspace.
#
# Скрипту байдуже на:
#   • race conditions
#   • затримки створення вікон
#   • відʼєднані helper-процеси
#   • Electron / Chromium / Steam безумство
#
# Типові сценарії використання:
#   • Запуск Steam / Discord / браузерів без витоку вікон
#   • Заборона спавну на поточному workspace
#   • Контроль програм, які ігнорують static windowrules
#
# Запуск:
#   ./Tak0-Autodispatch.sh <workspace> [rule ...] -- <command>
#
# Важливо:
#   • Всі rules — тимчасові
#   • Глобальна конфігурація Hyprland НЕ псується
#
# ─────────────────────────────────────────────────────────────────────────────
#
# REQUIREMENTS / ВИМОГИ:
#   - hyprctl   → runtime control of Hyprland
#   - jq        → JSON client parsing
#   - pgrep/ps  → process tree inspection
#

set -u

LOGFILE="$(dirname "$0")/dispatch.log"

# ─────────────────────────────────────────────────────────────────────────────
# 0️⃣ ARGUMENT PARSING
# ─────────────────────────────────────────────────────────────────────────────
#
# EN:
#   $1            → target workspace
#   Next args     → optional capture rules (windowrulev2 syntax)
#   "--"          → argument separator
#   After "--"    → command to execute (verbatim)
#
# UA:
#   $1            → цільовий workspace
#   Далі          → capture rules (сумісні з windowrulev2)
#   "--"          → роздільник аргументів
#   Після "--"    → команда запуску (як є)
#

TARGET_WS="$1"
shift || true

CAPTURE_RULES=()
while [[ "${1-}" != "--" && -n "${1-}" ]]; do
    CAPTURE_RULES+=("$1")
    shift || break
done

if [[ "${1-}" == "--" ]]; then
    shift
fi

CMD="$*"

if [[ -z "$TARGET_WS" || -z "$CMD" ]]; then
    echo "Usage: $0 <workspace> [rule rule ...] -- <command>" >>"$LOGFILE"
    exit 1
fi

echo "=== Deploy '$CMD' → WS $TARGET_WS @ $(date) ===" >>"$LOGFILE"

# ─────────────────────────────────────────────────────────────────────────────
# 1️⃣ HYPRLAND READINESS GATE
# ─────────────────────────────────────────────────────────────────────────────
#
# EN:
#   Hyprland may not be fully initialized during early autostart.
#   hyprctl silently fails if called too early.
#
# UA:
#   Під час раннього автозапуску Hyprland може бути ще не готовий.
#   hyprctl у такому випадку тихо фейлиться.
#

for _ in {1..50}; do
    hyprctl -j monitors >/dev/null 2>&1 && break
    sleep 0.1
done

# ─────────────────────────────────────────────────────────────────────────────
# 2️⃣ CLEANUP GUARANTEE
# ─────────────────────────────────────────────────────────────────────────────
#
# EN:
#   Ensures that ALL temporary rules are removed
#   even on crash, SIGTERM, or user interruption.
#
# UA:
#   Гарантує прибирання ВСІХ тимчасових правил
#   навіть у разі крешу, SIGTERM або Ctrl+C.
#

cleanup() {
    echo "Cleanup: removing temporary capture rules and initialWorkspace at $(date)" >>"$LOGFILE"

    hyprctl keyword windowrulev2 "unset, initialClass:.*" >>"$LOGFILE" 2>&1 || true

    for RULE in "${CAPTURE_RULES[@]}"; do
        echo "Cleanup: removing temporary capture rule: $RULE" >>"$LOGFILE"
        hyprctl keyword windowrulev2 "unset, $RULE" >>"$LOGFILE" 2>&1 || true
    done
}

trap cleanup EXIT INT TERM ERR

# ─────────────────────────────────────────────────────────────────────────────
# 3️⃣ ULTRA-EARLY GLOBAL CAPTURE (NUCLEAR OPTION)
# ─────────────────────────────────────────────────────────────────────────────
#
# EN:
#   Temporarily forces ALL windows (initialClass:.*)
#   onto the target workspace.
#
#   Protects against ultra-fast helpers:
#     • gpu-process
#     • renderer
#     • steamwebhelper
#
# UA:
#   Тимчасово заганяє АБСОЛЮТНО всі вікна
#   на цільовий workspace.
#
#   Рятує від ультрашвидких helper-ів.
#

echo "Applying temporary initialWorkspace capture (initialClass:.*)" >>"$LOGFILE"
hyprctl keyword windowrulev2 \
    "initialWorkspace $TARGET_WS silent, initialClass:.*" \
    >>"$LOGFILE" 2>&1 || true

# ─────────────────────────────────────────────────────────────────────────────
# 3️⃣.1 OPTIONAL CLASS-BASED PRE-CAPTURE
# ─────────────────────────────────────────────────────────────────────────────
#
# EN:
#   Additional precision rules.
#   Useful for Electron / Steam multi-process hell.
#
# UA:
#   Додаткові class-based правила.
#   Підвищують точність для Electron / Steam.
#

for RULE in "${CAPTURE_RULES[@]}"; do
    echo "Applying temporary capture rule: $RULE" >>"$LOGFILE"
    hyprctl keyword windowrulev2 \
        "initialWorkspace $TARGET_WS silent, $RULE" \
        >>"$LOGFILE" 2>&1 || true
done

# ─────────────────────────────────────────────────────────────────────────────
# 4️⃣ APPLICATION LAUNCH
# ─────────────────────────────────────────────────────────────────────────────
#
# EN:
#   bash -c allows aliases, env vars, wrappers.
#   ROOT_PID is the root of process lineage.
#
# UA:
#   bash -c дозволяє aliases, env vars та wrappers.
#   ROOT_PID — корінь дерева процесів.
#

bash -c "$CMD" &
ROOT_PID=$!
echo "Root PID: $ROOT_PID" >>"$LOGFILE"

# Resolve canonical process name
APP_NAME=""
for _ in {1..20}; do
    if [[ -r "/proc/$ROOT_PID/comm" ]]; then
        APP_NAME="$(tr -d '\0' < /proc/$ROOT_PID/comm 2>/dev/null || true)"
        break
    fi
    sleep 0.05
done

if [[ -z "$APP_NAME" ]]; then
    read -r -a __toks <<< "$CMD"
    APP_NAME="$(basename "${__toks[0]}")"
fi

echo "App gate name: $APP_NAME" >>"$LOGFILE"

sleep 1.5

# Release the nuclear option ASAP
echo "Releasing ultra-early wide capture" >>"$LOGFILE"
hyprctl keyword windowrulev2 "unset, initialClass:.*" >>"$LOGFILE" 2>&1 || true

# ─────────────────────────────────────────────────────────────────────────────
# 5️⃣ SUPERVISION LOOP (AUTHORITATIVE PHASE)
# ─────────────────────────────────────────────────────────────────────────────
#
# EN:
#   This loop:
#     • scans ALL Hyprland clients
#     • matches PID lineage
#     • matches detached helpers
#     • matches class rules
#
# UA:
#   Цей цикл:
#     • читає ВСІ клієнти Hyprland
#     • звіряє PID дерево
#     • ловить відʼєднані helper-и
#     • застосовує class fallback
#

get_descendants() {
    local root="$1"
    local all=("$root")
    local changed=1

    while (( changed )); do
        changed=0
        for p in "${all[@]}"; do
            for c in $(pgrep -P "$p" 2>/dev/null || true); do
                if [[ ! " ${all[*]} " =~ " $c " ]]; then
                    all+=("$c")
                    changed=1
                fi
            done
        done
    done

    echo "${all[@]}"
}

pid_matches_app() {
    local pid="$1"
    local comm
    comm="$(ps -p "$pid" -o comm= 2>/dev/null)" || return 1
    [[ "$comm" == "$APP_NAME" || "$comm" == "$APP_NAME"* ]]
}

END_TIME=$((SECONDS + 20))
declare -A SEEN

while (( SECONDS < END_TIME )); do
    PIDS="$(get_descendants "$ROOT_PID")"

    while IFS=$'\t' read -r PID ADDR CLASS; do
        MATCH=0

        for TPID in $PIDS; do
            [[ "$PID" == "$TPID" ]] && MATCH=1 && break
        done

        pid_matches_app "$PID" && MATCH=1

        for RULE in "${CAPTURE_RULES[@]}"; do
            if [[ "$RULE" =~ class:\^\((.*)\)\$ ]]; then
                [[ "$CLASS" =~ ${BASH_REMATCH[1]} ]] && MATCH=1
            fi
        done

        if (( MATCH )) && [[ -z "${SEEN[$ADDR]-}" ]]; then
            echo "Placing window $ADDR (pid $PID, class $CLASS) → WS $TARGET_WS" >>"$LOGFILE"
            hyprctl dispatch movetoworkspacesilent \
                "$TARGET_WS,address:$ADDR" >>"$LOGFILE" 2>&1 || true
            SEEN[$ADDR]=1
        fi
    done < <(hyprctl clients -j | jq -r '.[] | [.pid, .address, .class] | @tsv')

    sleep 0.01
done

echo "=== Deploy finished: '$CMD' ===" >>"$LOGFILE"
exit 0