# TigerVNC Auto-Setup Script for Headless Raspberry Pis

#!/bin/bash
# TigerVNC Auto-Setup Script for Headless Raspberry Pis
# Assumes Raspberry Pi OS with Desktop installed
# Run as the user who will be logging in (not root)

set -e

# 0. Clean up zombie VNC sessions if they exist
echo "Cleaning up any previous VNC sessions..."
tigervncserver -kill :1 2>/dev/null || true
rm -f ~/.vnc/*.pid ~/.vnc/*.log ~/.vnc/*.lock

# 1. Install TigerVNC packages
sudo apt update
sudo apt install -y tigervnc-standalone-server tigervnc-common

# 2. Set VNC password (interactive)
echo "Set VNC password for the current user:"
tigervncpasswd

# 3. Create VNC config
echo "Creating VNC config..."
mkdir -p ~/.vnc
cat > ~/.vnc/config <<EOF
localhost=no
geometry=1280x800
EOF

# 4. Create xstartup script
cat > ~/.vnc/xstartup <<'EOF'
#!/bin/sh
xrdb $HOME/.Xresources
startlxde &
EOF
chmod +x ~/.vnc/xstartup

# 5. Create systemd user service
mkdir -p ~/.config/systemd/user
cat > ~/.config/systemd/user/vncserver@.service <<EOF
[Unit]
Description=Start TigerVNC server at startup for display %%i
After=graphical.target

[Service]
Type=forking
ExecStartPre=/bin/sleep 5
ExecStart=/usr/bin/tigervncserver :%%i
ExecStop=/usr/bin/tigervncserver -kill :%%i
Restart=on-failure

[Install]
WantedBy=default.target
EOF

# 6. Enable lingering so user services run at boot
sudo loginctl enable-linger "$USER"