#!/bin/bash
# Dune Weaver Touch - One-Command Installer
# This script sets up everything needed to run Dune Weaver Touch on boot

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
ACTUAL_USER="${SUDO_USER:-$USER}"
USER_HOME=$(eval echo ~$ACTUAL_USER)

# Detect Raspberry Pi model
PI_MODEL=$(cat /proc/device-tree/model 2>/dev/null | tr -d '\0' || echo "unknown")
IS_PI5=false
if [[ "$PI_MODEL" == *"Pi 5"* ]]; then
    IS_PI5=true
fi

echo "🎯 Dune Weaver Touch - Complete Installation"
echo "============================================="
echo "App directory: $SCRIPT_DIR"
echo "User: $ACTUAL_USER"
echo "Detected: $PI_MODEL"
echo ""

# Check if running as root
if [ "$EUID" -ne 0 ]; then
    echo "❌ This installer must be run with sudo privileges"
    echo ""
    echo "Usage: sudo ./install.sh"
    echo ""
    exit 1
fi

echo "🔧 Installing system components..."
echo ""

# Function to install system scripts
install_scripts() {
    echo "📄 Installing system scripts..."
    
    local scripts=("screen-on" "screen-off" "touch-monitor")
    
    for script in "${scripts[@]}"; do
        local source_path="$SCRIPT_DIR/scripts/$script"
        local target_path="/usr/local/bin/$script"
        
        if [ -f "$source_path" ]; then
            cp "$source_path" "$target_path"
            chmod 755 "$target_path"
            chown root:root "$target_path"
            echo "   ✅ $script → $target_path"
        else
            echo "   ⚠️  $script not found at $source_path"
        fi
    done
    
    echo "   📄 System scripts installed"
}

# Function to setup systemd service
setup_systemd() {
    echo "🚀 Setting up systemd service..."

    # Update paths in the service file
    sed "s|/home/pi/dune-weaver-touch|$SCRIPT_DIR|g" "$SCRIPT_DIR/dune-weaver-touch.service" > /tmp/dune-weaver-touch.service
    sed -i "s|User=pi|User=$ACTUAL_USER|g" /tmp/dune-weaver-touch.service
    sed -i "s|Group=pi|Group=$ACTUAL_USER|g" /tmp/dune-weaver-touch.service

    # Copy service file
    cp /tmp/dune-weaver-touch.service /etc/systemd/system/

    # Enable service
    systemctl daemon-reload
    systemctl enable dune-weaver-touch.service

    echo "   🚀 Systemd service installed and enabled"
}

# Function to configure boot settings for DSI display
configure_boot_settings() {
    echo "🖥️  Configuring boot settings for DSI display..."

    local CONFIG_FILE="/boot/firmware/config.txt"
    # Fallback to old path if new path doesn't exist
    [ ! -f "$CONFIG_FILE" ] && CONFIG_FILE="/boot/config.txt"

    if [ ! -f "$CONFIG_FILE" ]; then
        echo "   ⚠️  config.txt not found, skipping boot configuration"
        return
    fi

    # Backup config.txt
    cp "$CONFIG_FILE" "${CONFIG_FILE}.backup.$(date +%Y%m%d_%H%M%S)"
    echo "   ✅ Backed up config.txt"

    # Remove old/conflicting KMS settings
    sed -i '/dtoverlay=vc4-fkms-v3d/d' "$CONFIG_FILE"
    sed -i '/dtoverlay=vc4-xfkms-v3d/d' "$CONFIG_FILE"

    if [ "$IS_PI5" = true ]; then
        # Pi 5: KMS is enabled by default, gpu_mem is ignored (dedicated VRAM)
        echo "   ℹ️  Pi 5 detected - KMS enabled by default, skipping vc4-kms-v3d overlay"
        echo "   ℹ️  Pi 5 has dedicated VRAM, skipping gpu_mem setting"
    else
        # Pi 3/4: Add full KMS if not present
        if ! grep -q "dtoverlay=vc4-kms-v3d" "$CONFIG_FILE"; then
            # Find [all] section or add at end
            if grep -q "^\[all\]" "$CONFIG_FILE"; then
                sed -i '/^\[all\]/a dtoverlay=vc4-kms-v3d' "$CONFIG_FILE"
            else
                echo -e "\n[all]\ndtoverlay=vc4-kms-v3d" >> "$CONFIG_FILE"
            fi
            echo "   ✅ Enabled full KMS (vc4-kms-v3d) for eglfs support"
        else
            echo "   ℹ️  Full KMS already enabled"
        fi

        # Add GPU memory if not present (only for Pi 3/4 with shared memory)
        if ! grep -q "gpu_mem=" "$CONFIG_FILE"; then
            echo "gpu_mem=128" >> "$CONFIG_FILE"
            echo "   ✅ Set GPU memory to 128MB"
        else
            echo "   ℹ️  GPU memory already configured"
        fi
    fi

    # Disable splash screens for cleaner boot (all Pi models)
    if ! grep -q "disable_splash=1" "$CONFIG_FILE"; then
        echo "disable_splash=1" >> "$CONFIG_FILE"
        echo "   ✅ Disabled rainbow splash"
    else
        echo "   ℹ️  Rainbow splash already disabled"
    fi

    echo "   🖥️  Boot configuration updated"
}

# Function to setup touch rotation via udev rule
setup_touch_rotation() {
    echo "👆 Setting up touchscreen rotation..."

    local UDEV_RULE_FILE="/etc/udev/rules.d/99-ft5x06-rotate.rules"

    # Create udev rule for FT5x06 touch controller (180° rotation)
    cat > "$UDEV_RULE_FILE" << 'EOF'
# Rotate FT5x06 touchscreen 180 degrees using libinput calibration matrix
# Matrix format: a b c d e f 0 0 1
# For 180° rotation: -1 0 1  0 -1 1  0 0 1
# This inverts both X and Y axes (equivalent to 180° rotation)
SUBSYSTEM=="input", KERNEL=="event*", ATTRS{name}=="*generic ft5x06*", \
  ENV{LIBINPUT_CALIBRATION_MATRIX}="-1 0 1  0 -1 1  0 0 1"
EOF

    chmod 644 "$UDEV_RULE_FILE"
    echo "   ✅ Touch rotation udev rule created: $UDEV_RULE_FILE"

    # Reload udev rules
    udevadm control --reload-rules
    udevadm trigger
    echo "   ✅ Udev rules reloaded"

    echo "   👆 Touch rotation configured (180°)"
}

# Function to hide mouse cursor
hide_mouse_cursor() {
    echo "🖱️  Configuring mouse cursor hiding..."

    # Install unclutter for hiding mouse cursor when idle
    echo "   📦 Installing unclutter..."
    apt install -y unclutter > /dev/null 2>&1

    # Create autostart directory if it doesn't exist
    local AUTOSTART_DIR="$USER_HOME/.config/autostart"
    mkdir -p "$AUTOSTART_DIR"
    chown -R "$ACTUAL_USER:$ACTUAL_USER" "$USER_HOME/.config"

    # Create unclutter autostart entry
    cat > "$AUTOSTART_DIR/unclutter.desktop" << 'EOF'
[Desktop Entry]
Type=Application
Name=Unclutter
Comment=Hide mouse cursor when idle
Exec=unclutter -idle 0.1 -root
Hidden=false
NoDisplay=false
X-GNOME-Autostart-enabled=true
EOF

    chown "$ACTUAL_USER:$ACTUAL_USER" "$AUTOSTART_DIR/unclutter.desktop"

    echo "   🖱️  Mouse cursor hiding configured"
}

# Function to setup kiosk optimizations
setup_kiosk_optimizations() {
    echo "🖥️  Setting up kiosk optimizations..."

    local CMDLINE_FILE="/boot/firmware/cmdline.txt"
    [ ! -f "$CMDLINE_FILE" ] && CMDLINE_FILE="/boot/cmdline.txt"

    # Determine DSI connector name based on Pi model
    # Pi 5 uses DSI-2 (separate DRM device), Pi 3/4 use DSI-1
    local DSI_CONNECTOR="DSI-1"
    if [ "$IS_PI5" = true ]; then
        DSI_CONNECTOR="DSI-2"
        echo "   ℹ️  Pi 5 detected - using DSI-2 connector"
    fi

    # Configure cmdline.txt for display and boot
    if [ -f "$CMDLINE_FILE" ]; then
        cp "$CMDLINE_FILE" "${CMDLINE_FILE}.backup.$(date +%Y%m%d_%H%M%S)"

        # Add video parameter for DSI display with rotation
        # Check for any existing DSI video configuration
        if ! grep -q "video=DSI-[0-9]:800x480@60,rotate=180" "$CMDLINE_FILE"; then
            sed -i "s/$/ video=${DSI_CONNECTOR}:800x480@60,rotate=180/" "$CMDLINE_FILE"
            echo "   ✅ DSI display configuration added (${DSI_CONNECTOR}:800x480@60, rotated 180°)"
        else
            echo "   ℹ️  DSI display configuration already present"
        fi

        # Add quiet splash for cleaner boot
        if ! grep -q "quiet splash" "$CMDLINE_FILE"; then
            sed -i 's/$/ quiet splash/' "$CMDLINE_FILE"
            echo "   ✅ Boot splash enabled"
        else
            echo "   ℹ️  Boot splash already enabled"
        fi
    fi

    # Note about auto-login - let user configure manually
    echo "   ℹ️  Auto-login configuration skipped (manual setup recommended)"

    echo "   🖥️  Kiosk optimizations applied"
}

# Function to setup Python virtual environment and install dependencies
setup_python_environment() {
    echo "🐍 Setting up Python virtual environment..."

    # Install system Qt6 and PySide6 packages for full eglfs support
    echo "   📦 Installing system Qt6 and PySide6 packages..."
    apt update
    apt install -y \
        python3-pyside6.qtcore \
        python3-pyside6.qtgui \
        python3-pyside6.qtqml \
        python3-pyside6.qtquick \
        python3-pyside6.qtquickcontrols2 \
        python3-pyside6.qtquickwidgets \
        python3-pyside6.qtwebsockets \
        python3-pyside6.qtnetwork \
        qml6-module-qtquick \
        qml6-module-qtquick-controls \
        qml6-module-qtquick-layouts \
        qml6-module-qtquick-window \
        qml6-module-qtquick-dialogs \
        qml6-module-qt-labs-qmlmodels \
        qt6-virtualkeyboard-plugin \
        qml6-module-qtquick-virtualkeyboard \
        qt6-base-dev \
        qt6-declarative-dev \
        libqt6opengl6 \
        libqt6core5compat6 \
        libqt6network6 \
        libqt6websockets6 > /dev/null 2>&1

    echo "   ✅ System Qt6/PySide6 packages installed"

    # Create virtual environment with system site packages
    if [ ! -d "$SCRIPT_DIR/venv" ]; then
        echo "   📦 Creating virtual environment with system site packages..."
        python3 -m venv --system-site-packages "$SCRIPT_DIR/venv" || {
            echo "   ⚠️  Could not create virtual environment. Installing python3-venv..."
            apt install -y python3-venv python3-full
            python3 -m venv --system-site-packages "$SCRIPT_DIR/venv"
        }
    else
        echo "   ℹ️  Virtual environment already exists"
    fi

    # Install non-Qt dependencies from requirements.txt
    echo "   📦 Installing Python dependencies..."
    "$SCRIPT_DIR/venv/bin/python" -m pip install --upgrade pip > /dev/null 2>&1
    "$SCRIPT_DIR/venv/bin/pip" install -r "$SCRIPT_DIR/requirements.txt" > /dev/null 2>&1

    # Change ownership to the actual user (not root)
    chown -R "$ACTUAL_USER:$ACTUAL_USER" "$SCRIPT_DIR/venv"

    echo "   🐍 Python virtual environment ready"
}

# Main installation process
echo "Starting complete installation..."
echo ""

# Install everything
setup_python_environment
install_scripts
setup_systemd
configure_boot_settings
setup_touch_rotation
hide_mouse_cursor
setup_kiosk_optimizations

echo ""
echo "🎉 Installation Complete!"
echo "========================"
echo ""
echo "📟 Detected: $PI_MODEL"
if [ "$IS_PI5" = true ]; then
    echo "   └─ Using Pi 5 optimized settings (DSI-2, dedicated VRAM)"
fi
echo ""
echo "✅ Python virtual environment created at: $SCRIPT_DIR/venv"
echo "✅ System scripts installed in /usr/local/bin/"
echo "✅ Systemd service configured for auto-start"
echo "✅ Mouse cursor hiding configured (Qt + unclutter)"
echo "✅ Kiosk optimizations applied"
echo ""
echo "🔧 Service Management:"
echo "   Start now:  sudo systemctl start dune-weaver-touch"
echo "   Stop:       sudo systemctl stop dune-weaver-touch"
echo "   Status:     sudo systemctl status dune-weaver-touch"
echo "   Logs:       sudo journalctl -u dune-weaver-touch -f"
echo ""
echo "🚀 Next Steps:"
echo "   1. ${YELLOW}⚠️  REBOOT REQUIRED${NC} for config.txt changes (vc4-kms-v3d) to take effect"
echo "   2. Configure auto-login (recommended for kiosk mode):"
echo "      sudo ./setup-autologin.sh    (automated setup)"
echo "      OR manually: sudo raspi-config → System Options → Boot/Auto Login → Desktop Autologin"
echo "   3. After reboot, the app will start automatically on boot via systemd service"
echo "   4. Check the logs if you encounter any issues: sudo journalctl -u dune-weaver-touch -f"
echo ""
echo "💡 To start the service now without rebooting:"
echo "   sudo systemctl start dune-weaver-touch"
echo ""
echo "🛠️  For development/testing (run manually):"
echo "   cd $SCRIPT_DIR"
echo "   ./run.sh"
echo ""
echo "⚙️  To change boot/login settings later:"
echo "   sudo ./configure-boot.sh"
echo ""

# Ask if user wants to start now
read -p "🤔 Would you like to start the service now? (y/N): " -n 1 -r
echo
if [[ $REPLY =~ ^[Yy]$ ]]; then
    echo "🚀 Starting Dune Weaver Touch service..."
    systemctl start dune-weaver-touch
    sleep 2
    systemctl status dune-weaver-touch --no-pager -l
    echo ""
    echo "✅ Service started! Check the status above."
fi

echo ""
echo "🎯 Installation completed successfully!"