RDtop/install.sh

#!/bin/bash
set -euo pipefail

#######################################
# RDtop — Universal RustDesk Headless Installer
# One command: curl -fsSL https://git.softuniq.eu/NW/RDtop/raw/branch/main/install.sh | sudo bash
#######################################

ARCH=$(uname -m)
case "$ARCH" in
  x86_64)  RUSTDESK_ARCH="x86_64";  DEB_ARCH="amd64" ;;
  aarch64) RUSTDESK_ARCH="aarch64"; DEB_ARCH="arm64" ;;
  armv7l)  RUSTDESK_ARCH="armv7";   DEB_ARCH="armhf" ;;
  *) echo "ERROR: Unsupported architecture: $ARCH"; exit 1 ;;
esac

RUSTDESK_VERSION="1.4.6"
RUSTDESK_URL="https://github.com/rustdesk/rustdesk/releases/download/${RUSTDESK_VERSION}/rustdesk-${RUSTDESK_VERSION}-${RUSTDESK_ARCH}.deb"
TMP_DEB="/tmp/rustdesk-${RUSTDESK_VERSION}-${DEB_ARCH}.deb"
LOG="/tmp/rdtop-install.log"

log() { echo "[$(date '+%F %T')] $1" | tee -a "$LOG"; }

echo ""
echo "============================================"
echo "  RDtop — RustDesk Headless Installer"
echo "  Arch: $ARCH | Package: ${RUSTDESK_ARCH}.deb"
echo "============================================"
echo ""

# --- 1. Install dependencies (verbose on fresh machine) ---
log "[2/8] Installing dependencies..."
export DEBIAN_FRONTEND=noninteractive

if command -v apt-get >/dev/null 2>&1; then
    apt-get update -qq || true
    # Core packages — must succeed
    apt-get install -y -qq curl wget ca-certificates gnupg lsb-release >/dev/null 2>&1 || {
        log "  CRITICAL: Failed to install base tools (curl/wget)"
        exit 1
    }
    # Xorg + dummy driver
    apt-get install -y -qq xserver-xorg-core xserver-xorg-video-dummy x11-xserver-utils >/dev/null 2>&1 || true
    # VA/VDPAU libs (RustDesk needs these)
    apt-get install -y -qq libva2 libvdpau1 libva-drm2 libva-x11-2 >/dev/null 2>&1 || true
    # Polkit + inhibit (for preventing sleep)
    apt-get install -y -qq policykit-1 dbus dbus-x11 >/dev/null 2>&1 || true
elif command -v dnf >/dev/null 2>&1; then
    dnf install -y curl wget xorg-x11-server-Xorg xorg-x11-drv-dummy polkit dbus >/dev/null 2>&1 || true
elif command -v pacman >/dev/null 2>&1; then
    pacman -Sy --noconfirm curl wget xorg-server xf86-video-dummy polkit dbus >/dev/null 2>&1 || true
fi

# --- 2. Disable sleep / suspend / screensaver (critical for headless) ---
log "[3/8] Disabling sleep, suspend, hibernate, screensaver..."

# systemd — mask sleep targets
systemctl mask sleep.target suspend.target hibernate.target hybrid-sleep.target 2>/dev/null || true

# logind — prevent idle sleep
if [ -f /etc/systemd/logind.conf ]; then
    cp /etc/systemd/logind.conf /etc/systemd/logind.conf.bak.$(date +%s) 2>/dev/null || true
    cat > /etc/systemd/logind.conf <<'EOF'
[Login]
HandleLidSwitch=ignore
HandleLidSwitchExternalPower=ignore
HandleLidSwitchDocked=ignore
IdleAction=ignore
EOF
    systemctl restart systemd-logind 2>/dev/null || true
    log "  systemd-logind: sleep disabled"
fi

# GNOME / gsettings — disable screensaver and lock (if available)
if command -v gsettings >/dev/null 2>&1; then
    gsettings set org.gnome.desktop.screensaver lock-enabled false 2>/dev/null || true
    gsettings set org.gnome.desktop.screensaver idle-activation-enabled false 2>/dev/null || true
    gsettings set org.gnome.desktop.session idle-delay 0 2>/dev/null || true
    gsettings set org.gnome.desktop.screensaver lock-active false 2>/dev/null || true
    log "  GNOME screensaver: disabled"
fi

# X11 — disable screen blanking
if command -v xset >/dev/null 2>&1; then
    xset s off 2>/dev/null || true
    xset -dpms 2>/dev/null || true
    xset s noblank 2>/dev/null || true
fi

# --- 3. Detect display environment ---
log "[4/8] Detecting display environment..."
HAS_DISPLAY=false
HAS_XORG=false

if [ -n "${DISPLAY:-}" ] && xdpyinfo >/dev/null 2>&1; then
    HAS_DISPLAY=true
    log "  Active display detected: $DISPLAY"
fi

if pgrep -x "Xorg" >/dev/null 2>&1 || pgrep -x "X" >/dev/null 2>&1; then
    HAS_XORG=true
    log "  Xorg already running"
fi

# --- 4. Setup dummy driver (force DUMMY0 as primary) ---
log "[5/8] Setting up dummy driver + forcing DUMMY0 primary..."
mkdir -p /etc/X11/xorg.conf.d/

# Remove conflicting configs
rm -f /etc/X11/xorg.conf.d/20-intel-virtual.conf 2>/dev/null || true
rm -f /etc/X11/xorg.conf.d/90-fallback.conf 2>/dev/null || true

# Create dummy config with explicit primary output
cat > /etc/X11/xorg.conf.d/20-dummy-headless.conf <<'XORG'
Section "Device"
    Identifier "DummyHeadless"
    Driver "dummy"
    VideoRam 256000
EndSection

Section "Monitor"
    Identifier "DummyMonitor"
    HorizSync 28-80
    VertRefresh 48-75
EndSection

Section "Screen"
    Identifier "DummyScreen"
    Device "DummyHeadless"
    Monitor "DummyMonitor"
    DefaultDepth 24
    SubSection "Display"
        Depth 24
        Modes "1920x1080" "1280x720"
        Virtual 1920 1080
    EndSubSection
EndSection

Section "ServerLayout"
    Identifier "DummyLayout"
    Screen 0 "DummyScreen" 0 0
    Option "AutoAddGPU" "false"
EndSection
XORG
chmod 644 /etc/X11/xorg.conf.d/20-dummy-headless.conf

# --- 5. Start Xorg if headless ---
if [ "$HAS_DISPLAY" = false ] && [ "$HAS_XORG" = false ]; then
    log "[6/8] Starting Xorg dummy display on :0..."
    rm -f /tmp/.X11-unix/X0 /tmp/.X0-lock 2>/dev/null || true
    mkdir -p /tmp/.X11-unix
    chmod 1777 /tmp/.X11-unix
    
    # Start Xorg with dummy config (explicit config file)
    if [ "$EUID" -eq 0 ]; then
        nohup Xorg :0 -config /etc/X11/xorg.conf.d/20-dummy-headless.conf -nolisten tcp >/tmp/xorg-dummy.log 2>&1 &
    else
        log "  WARNING: Not root, trying Xvfb..."
        apt-get install -y -qq xvfb >/dev/null 2>&1 || true
    fi
    sleep 4
    
    # Verify
    if [ -S /tmp/.X11-unix/X0 ]; then
        log "  Xorg dummy display :0 created"
    else
        log "  WARNING: Xorg socket not found, trying Xvfb..."
        pkill -x Xorg 2>/dev/null || true
        rm -f /tmp/.X11-unix/X0 /tmp/.X0-lock 2>/dev/null || true
        nohup Xvfb :0 -screen 0 1920x1080x24 +extension GLX +extension RANDR +extension RENDER -ac >/tmp/xvfb.log 2>&1 &
        sleep 3
    fi
    
    # Setup xauth
    if [ "$EUID" -eq 0 ]; then
        xauth -f /root/.Xauthority add :0 . $(mcookie) 2>/dev/null || true
        chmod 600 /root/.Xauthority 2>/dev/null || true
    fi
else
    log "  Step 6 skipped: existing display detected"
fi

export DISPLAY=:0
export XAUTHORITY="${XAUTHORITY:-/root/.Xauthority}"

# Force DUMMY0 as primary BEFORE RustDesk starts
log "  Forcing DUMMY0 as primary output..."
if command -v xrandr >/dev/null 2>&1; then
    # Wait for xrandr to see outputs
    for i in 1 2 3 4 5; do
        if xrandr >/dev/null 2>&1; then
            # Check if DUMMY0 exists
            if xrandr | grep -q "^DUMMY0"; then
                xrandr --output DUMMY0 --mode "1920x1080" --primary 2>/dev/null || true
                log "    DUMMY0 set to primary 1920x1080"
                break
            fi
            # Check if VIRTUAL1 exists (intel fallback)
            if xrandr | grep -q "^VIRTUAL1"; then
                xrandr --newmode "1920x1080_60" 173.00 1920 2048 2248 2576 1080 1083 1088 1120 2>/dev/null || true
                xrandr --addmode VIRTUAL1 "1920x1080_60" 2>/dev/null || true
                xrandr --output VIRTUAL1 --mode "1920x1080_60" --primary 2>/dev/null || true
                log "    VIRTUAL1 set to primary 1920x1080"
                break
            fi
        fi
        sleep 1
    done
fi

# --- 6. Download & install RustDesk (with fallback) ---
log "[7/8] Downloading RustDesk ${RUSTDESK_VERSION}..."

# Try .deb from GitHub
if command -v curl >/dev/null 2>&1; then
    curl -fsSL -o "$TMP_DEB" "$RUSTDESK_URL" 2>/dev/null || true
elif command -v wget >/dev/null 2>&1; then
    wget -q -O "$TMP_DEB" "$RUSTDESK_URL" 2>/dev/null || true
fi

INSTALL_OK=false

if [ -f "$TMP_DEB" ] && [ -s "$TMP_DEB" ]; then
    log "  Installing from .deb..."
    # Force install even if dependencies missing (fix with apt-get after)
    dpkg --force-depends --force-confnew -i "$TMP_DEB" >/dev/null 2>&1 || true
    # Fix any missing deps
    apt-get install -f -y -qq >/dev/null 2>&1 || true
    rm -f "$TMP_DEB"
    
    if [ -f /usr/share/rustdesk/rustdesk ]; then
        INSTALL_OK=true
    fi
fi

# Fallback: install from repo if .deb failed or not downloaded
if [ "$INSTALL_OK" = false ]; then
    log "  .deb failed or not available, trying repository install..."
    apt-get install -y -qq rustdesk 2>/dev/null || true
    
    # Check again
    if [ -f /usr/share/rustdesk/rustdesk ]; then
        INSTALL_OK=true
    fi
fi

if [ "$INSTALL_OK" = false ]; then
    log "ERROR: RustDesk installation failed. Check /tmp/rdtop-install.log"
    exit 1
fi

log "  RustDesk installed: $(/usr/share/rustdesk/rustdesk --version 2>/dev/null || echo 'unknown')"

# --- 7. Configure RustDesk for headless (no monitor selection, survive lock) ---
log "[8/8] Configuring RustDesk for headless + sleep-safe..."

mkdir -p ~/.config/rustdesk

# Write config: public rendezvous, auto-display selection disabled (DUMMY0 is primary)
cat > ~/.config/rustdesk/RustDesk2.toml <<'EOF'
rendezvous_server = 'rs-ny.rustdesk.com:21116'
nat_type = 1
serial = 0
unlock_pin = ''

[options]
local-ip-addr = 'auto'
hw-codec = 'N'
EOF
chmod 600 ~/.config/rustdesk/RustDesk2.toml

# Kill any old RustDesk
pkill -f "/usr/share/rustdesk/rustdesk" 2>/dev/null || true
sleep 2

# Start RustDesk with all env vars
export DISPLAY=:0
export XAUTHORITY=/root/.Xauthority
export LIBVA_DRIVER_NAME=none
export VDPAU_DRIVER=none
export LIBGL_ALWAYS_SOFTWARE=1
export LIBGL_ALWAYS_INDIRECT=1

nohup /usr/share/rustdesk/rustdesk --server >/tmp/rustdesk-server.log 2>&1 &
sleep 5

# Verify running
if ! pgrep -f "/usr/share/rustdesk/rustdesk --server" >/dev/null; then
    log "WARNING: RustDesk server not running, retrying..."
    sleep 3
    nohup /usr/share/rustdesk/rustdesk --server >/tmp/rustdesk-server.log 2>&1 &
    sleep 3
fi

# Get credentials
RUSTDESK_ID=$(/usr/share/rustdesk/rustdesk --get-id 2>/dev/null || echo "N/A")

# Get or set password
RUSTDESK_PASS=""
if [ -f ~/.config/rustdesk/RustDesk.toml ]; then
    RUSTDESK_PASS=$(python3 -c "
import sys
try:
    with open(sys.argv[1]) as f:
        content = f.read()
        for line in content.split('\n'):
            if 'password' in line and '=' in line:
                val = line.split('=', 1)[1].strip().strip(\"'\").strip('\"')
                if val: print(val); break
except: pass
" ~/.config/rustdesk/RustDesk.toml 2>/dev/null || true)
fi

if [ -z "$RUSTDESK_PASS" ]; then
    RUSTDESK_PASS="<пароль в GUI: rustdesk → Настройки → Безопасность>"
fi

# Summary
echo ""
echo "============================================"
echo "         RustDesk Ready!"
echo "============================================"
echo ""
printf "  %-14s %s\n" "ID:" "$RUSTDESK_ID"
printf "  %-14s %s\n" "Password:" "$RUSTDESK_PASS"
printf "  %-14s %s\n" "Display:" ":0 (DUMMY0 primary 1920x1080)"
printf "  %-14s %s\n" "Sleep/Lock:" "disabled (systemd masked)"
printf "  %-14s %s\n" "Version:" "$RUSTDESK_VERSION"
printf "  %-14s %s\n" "Arch:" "$ARCH"
echo ""
echo "  Connect:      rustdesk $RUSTDESK_ID"
echo "  Status:       systemctl status display-manager"
echo "  Logs:         /tmp/rdtop-install.log"
echo "  RustDesk:     /tmp/rustdesk-server.log"
echo "  Xorg:         /tmp/xorg-dummy.log"
echo ""
echo "============================================"

log "Installation complete. ID: $RUSTDESK_ID"

# Post-install: show monitor status
if command -v xrandr >/dev/null 2>&1; then
    echo ""
    echo "--- Current monitor status ---"
    xrandr --listmonitors 2>/dev/null || echo "xrandr not available"
fi

# --- 8. Install systemd unit for RustDesk (survives sleep/lock) ---
log "[8/8] Installing systemd unit..."

# Write unit inline
cat > /etc/systemd/system/rustdesk-headless.service <<'UNIT'
[Unit]
Description=RustDesk Headless Server (auto-display, sleep-safe)
After=display-manager.service graphical.target network.target
Wants=display-manager.service

[Service]
Type=simple
User=root
Environment="DISPLAY=:0"
Environment="XAUTHORITY=/root/.Xauthority"
Environment="LIBVA_DRIVER_NAME=none"
Environment="VDPAU_DRIVER=none"
Environment="LIBGL_ALWAYS_SOFTWARE=1"
Environment="LIBGL_ALWAYS_INDIRECT=1"

ExecStartPre=/bin/bash -c 'for i in 1 2 3 4 5 6 7 8 9 10; do if xrandr >/dev/null 2>&1; then if xrandr | grep -q "^DUMMY0"; then xrandr --output DUMMY0 --mode "1920x1080" --primary 2>/dev/null || true; exit 0; fi; fi; sleep 1; done'

ExecStart=/usr/share/rustdesk/rustdesk --server
Restart=always
RestartSec=10
StartLimitInterval=60
StartLimitBurst=3

[Install]
WantedBy=multi-user.target
UNIT

systemctl daemon-reload
systemctl enable rustdesk-headless.service

# Stop the nohup process and start via systemd
pkill -f "/usr/share/rustdesk/rustdesk --server" 2>/dev/null || true
sleep 2
systemctl start rustdesk-headless.service

sleep 3
if systemctl is-active rustdesk-headless.service >/dev/null 2>&1; then
    log "  systemd unit: rustdesk-headless.service ACTIVE"
else
    log "  WARNING: rustdesk-headless.service not active, using nohup fallback"
    nohup /usr/share/rustdesk/rustdesk --server >/tmp/rustdesk-server.log 2>&1 &
fi