#!/bin/bash

# Exit on error
set -e

# Change to the root directory of the project
cd "$(dirname "$0")/.."

# Temporary pkg-config shim (Fedora 40+: only webkit2gtk-4.1.pc exists; some tooling still asks for 4.0)
WAILS_PKGCFG_SHIM=""
cleanup_pkgconfig_shim() {
    if [ -n "$WAILS_PKGCFG_SHIM" ] && [ -d "$WAILS_PKGCFG_SHIM" ]; then
        rm -rf "$WAILS_PKGCFG_SHIM"
    fi
}
trap cleanup_pkgconfig_shim EXIT

APP_NAME="wis-free-v3"
BUILD_DIR="build/bin"
EXECUTABLE="$BUILD_DIR/$APP_NAME"

echo "========================================"
echo "   wis-free-v3 - Linux Build Script"
echo "========================================"
echo ""

VERSION_FILE="$(dirname "$0")/VERSION"
APP_VERSION="dev"
if [ -f "$VERSION_FILE" ]; then
    APP_VERSION="$(head -n1 "$VERSION_FILE" | tr -d '\r\n')"
    APP_VERSION="$(echo "$APP_VERSION" | sed 's/^[[:space:]]*//;s/[[:space:]]*$//')"
fi
if [ -z "$APP_VERSION" ]; then
    APP_VERSION="dev"
fi
echo "[0/3] App version: $APP_VERSION"
echo ""

# Function to check if a command exists
command_exists() {
    command -v "$1" >/dev/null 2>&1
}

# 1. Check for basic tools
if ! command_exists go; then
    echo "[ERROR] Go is not installed. Please install Go 1.23+."
    exit 1
fi

if ! command_exists wails; then
    echo "[INFO] Wails CLI not found. Installing..."
    go install github.com/wailsapp/wails/v2/cmd/wails@latest
    export PATH=$PATH:$(go env GOPATH)/bin
fi

# 2. Check for Linux dependencies
echo "[1/3] Checking system dependencies..."
MISSING_DEPS=0

# 2a. GNOME tray icon support check (before build — avoids launching app if this will fail)
# AppIndicator icons need the GNOME Shell extension on GNOME desktop environments.
# Other DEs (KDE Plasma, XFCE, Cinnamon, etc.) usually support them natively.
if [ "$XDG_CURRENT_DESKTOP" = "GNOME" ] || [ "$XDG_CURRENT_DESKTOP" = "ubuntu:GNOME" ]; then
    if ! gnome-extensions show appindicator@altamirano.fabio &>/dev/null; then
        echo ""
        echo "==============================================================="
        echo "  ERROR: GNOME AppIndicator extension is not installed"
        echo "==============================================================="
        echo ""
        echo "  GNOME does not show system tray icons without this extension."
        echo "  The app will run but the tray icon will be invisible."
        echo ""
        if command_exists dnf; then
            echo "  Install it with:"
            echo ""
            echo "    sudo dnf install gnome-shell-extension-appindicator"
            echo ""
        elif command_exists apt-get; then
            echo "  Install it with:"
            echo ""
            echo "    sudo apt install gnome-shell-extension-appindicator"
            echo ""
        else
            echo "  Install the 'gnome-shell-extension-appindicator' package"
            echo "  for your distribution."
            echo ""
        fi
        echo "  Then log out and back in, and enable it in:"
        echo "    Settings > Extensions > AppIndicator and KStatusNotifierItem Support"
        echo ""
        echo "==============================================================="
        exit 1
    fi
fi

DEPS=("gcc" "pkg-config")

for dep in "${DEPS[@]}"; do
    if ! command_exists $dep; then
        echo "[WARNING] Missing basic build tool: $dep"
        MISSING_DEPS=1
    fi
done

# We can't easily check C headers, but we can try to find them with pkg-config
# (Fedora often ships webkit2gtk-4.1.pc; Debian/Ubuntu often use webkit2gtk-4.0.pc)
webkit2_ok() {
    pkg-config --exists webkit2gtk-4.0 2>/dev/null && return 0
    pkg-config --exists webkit2gtk-4.1 2>/dev/null && return 0
    return 1
}

# Tray: Debian uses ayatana-appindicator3; Fedora package names differ but .pc is often the same
appindicator_ok() {
    pkg-config --exists ayatana-appindicator3-0.1 2>/dev/null && return 0
    pkg-config --exists appindicator3-0.1 2>/dev/null && return 0
    return 1
}

if command_exists pkg-config; then
    if ! pkg-config --exists gtk+-3.0 || ! webkit2_ok; then
        echo "[WARNING] Missing Wails dependencies (GTK3 / WebKit2GTK)."
        MISSING_DEPS=1
    fi
    if ! pkg-config --exists x11 xtst xcb xkbcommon-x11; then
        echo "[WARNING] Missing gohook dependencies (X11 / Xtst / Xcb / Xkbcommon)."
        MISSING_DEPS=1
    fi
    if ! pkg-config --exists alsa; then
        echo "[WARNING] Missing audio dependencies (ALSA)."
        MISSING_DEPS=1
    fi
    if ! appindicator_ok; then
        echo "[WARNING] Missing systray dependencies (appindicator / ayatana)."
        MISSING_DEPS=1
    fi
fi

if [ $MISSING_DEPS -eq 1 ]; then
    echo ""
    echo "It looks like you are missing some required libraries."
    echo ""
    echo "Wayland note: global hotkeys use the XDG GlobalShortcuts portal when WAYLAND_DISPLAY"
    echo "or XDG_SESSION_TYPE=wayland is set (xdg-desktop-portal + a supporting compositor, e.g. KDE Plasma)."
    echo "Override: WISFREE_USE_X11_HOTKEY=1 forces X11 grabs (needs XWayland);"
    echo "WISFREE_USE_PORTAL_HOTKEY=1 forces the portal on X11 sessions for testing."
    echo ""
    
    DEBIAN_DEPS="build-essential pkg-config libgtk-3-dev libwebkit2gtk-4.0-dev libx11-dev libx11-xcb-dev libxtst-dev libasound2-dev libayatana-appindicator3-dev libxkbcommon-x11-dev"
    # Runtime niceties (optional): libnotify-bin — status toasts; playerctl — pause media while recording
    DEBIAN_RUNTIME_OPT="libnotify-bin playerctl"
    # Fedora 40+: WebKit2GTK 4.0 packages are gone; use 4.1 + Wails -tags webkit2_41 (see wails build below).
    # pkgconf-pkg-config provides `pkg-config` on Fedora.
    FEDORA_DEPS="gcc gcc-c++ make pkgconf-pkg-config gtk3-devel webkit2gtk4.1-devel libX11-devel libxcb-devel libXtst-devel alsa-lib-devel libayatana-appindicator-gtk3-devel libxkbcommon-x11-devel"
    # Same as FEDORA_DEPS but classic libappindicator (some spins/repos lack Ayatana -devel)
    FEDORA_DEPS_ALT="gcc gcc-c++ make pkgconf-pkg-config gtk3-devel webkit2gtk4.1-devel libX11-devel libxcb-devel libXtst-devel alsa-lib-devel libappindicator-gtk3-devel libxkbcommon-x11-devel"
    FEDORA_RUNTIME_OPT="libnotify playerctl xdg-desktop-portal"
    ARCH_DEPS="base-devel pkgconf gtk3 webkit2gtk libx11 libxtst alsa-lib libayatana-appindicator libxkbcommon-x11"
    ARCH_RUNTIME_OPT="libnotify playerctl"
    
    echo "The full list of dependencies needed:"
    echo "  [Ubuntu/Debian]: sudo apt update && sudo apt install -y $DEBIAN_DEPS"
    echo "  [Ubuntu/Debian] optional: sudo apt install -y $DEBIAN_RUNTIME_OPT"
    echo "  [Fedora]:        sudo dnf install -y $FEDORA_DEPS"
    echo "                   (if Ayatana devel is missing, use libappindicator-gtk3-devel — see FEDORA_DEPS_ALT in this script)"
    echo "  [Fedora] optional: sudo dnf install -y $FEDORA_RUNTIME_OPT"
    echo "  [Arch Linux]:    sudo pacman -S $ARCH_DEPS"
    echo "  [Arch Linux] optional: sudo pacman -S $ARCH_RUNTIME_OPT"
    echo ""
    
    if command_exists apt-get; then
        read -p "Would you like to automatically install missing dependencies now? (Requires sudo) (y/N): " INSTALL_DEPS
        if [[ "$INSTALL_DEPS" == "y" || "$INSTALL_DEPS" == "Y" ]]; then
            echo "Installing dependencies..."
            sudo apt update && sudo apt install -y $DEBIAN_DEPS
        else
            echo "Skipping installation."
            read -p "Press Enter to attempt build anyway, or Ctrl+C to cancel..."
        fi
    elif command_exists dnf; then
        read -p "Would you like to automatically install missing dependencies with dnf now? (Requires sudo) (y/N): " INSTALL_DEPS
        if [[ "$INSTALL_DEPS" == "y" || "$INSTALL_DEPS" == "Y" ]]; then
            echo "Installing dependencies (Fedora)..."
            if ! sudo dnf install -y $FEDORA_DEPS; then
                echo "[INFO] Retrying dnf with libappindicator-gtk3-devel instead of Ayatana..."
                sudo dnf install -y $FEDORA_DEPS_ALT
            fi
        else
            echo "Skipping installation."
            read -p "Press Enter to attempt build anyway, or Ctrl+C to cancel..."
        fi
    else
        read -p "Press Enter to attempt build anyway, or Ctrl+C to cancel..."
    fi
fi

# 3. Build the application
echo "[2/3] Building with Wails..."

# Pre-emptively fix npm bin permissions if they got messed up (common issue on some systems)
if [ -d "frontend/node_modules/.bin" ]; then
    chmod +x frontend/node_modules/.bin/* 2>/dev/null || true
fi

# Fedora 40+ / rolling: only webkit2gtk-4.1 is shipped. Wails v2 uses -tags webkit2_41 for that API.
# Also symlink webkit2gtk-4.0.pc -> 4.1.pc so CGO / other deps that still probe "webkit2gtk-4.0" resolve.
WAILS_WEBKIT_TAGS=()
if pkg-config --exists webkit2gtk-4.1 2>/dev/null; then
    WAILS_WEBKIT_TAGS=(-tags webkit2_41)
fi

if ! pkg-config --exists webkit2gtk-4.0 2>/dev/null && pkg-config --exists webkit2gtk-4.1 2>/dev/null; then
    WEBKIT41_PC=""
    for dir in \
        $(printf '%s' "${PKG_CONFIG_PATH:-}" | tr ':' '\n') \
        /usr/lib64/pkgconfig \
        /usr/lib/pkgconfig \
        /usr/local/lib64/pkgconfig \
        /usr/local/lib/pkgconfig; do
        [ -z "$dir" ] && continue
        [ -f "$dir/webkit2gtk-4.1.pc" ] || continue
        WEBKIT41_PC="$dir/webkit2gtk-4.1.pc"
        break
    done
    if [ -n "$WEBKIT41_PC" ]; then
        WAILS_PKGCFG_SHIM=$(mktemp -d "${TMPDIR:-/tmp}/wails-pkgcfg-shim.XXXXXX")
        ln -sf "$WEBKIT41_PC" "$WAILS_PKGCFG_SHIM/webkit2gtk-4.0.pc"
        export PKG_CONFIG_PATH="$WAILS_PKGCFG_SHIM${PKG_CONFIG_PATH:+:}${PKG_CONFIG_PATH:-}"
        echo "[INFO] Using PKG_CONFIG_PATH shim: webkit2gtk-4.0.pc -> $(basename "$WEBKIT41_PC")"
    fi
fi

wails build -platform linux/amd64 -clean "${WAILS_WEBKIT_TAGS[@]}" -ldflags "-X main.AppVersion=${APP_VERSION}"

if [ ! -f "$EXECUTABLE" ]; then
    echo "[ERROR] Build failed. Binary not found at $EXECUTABLE."
    exit 1
fi

# wails build sometimes auto-launches the binary on Linux — kill it so it
# doesn't block the terminal while we ask about installation.
pkill -f "$EXECUTABLE" 2>/dev/null || true

echo "[3/3] Build successful!"
echo "      Output: $EXECUTABLE"
echo ""

# 4. Optional Installation
read -p "Would you like to install it globally to /usr/local/bin and add a desktop shortcut? (y/n): " INSTALL
if [[ "$INSTALL" == "y" || "$INSTALL" == "Y" ]]; then
    echo "Installing..."
    
    # Needs sudo
    sudo cp "$EXECUTABLE" "/usr/local/bin/$APP_NAME"
    sudo chmod +x "/usr/local/bin/$APP_NAME"
    
    # Create Desktop shortcut
    DESKTOP_FILE="/usr/share/applications/$APP_NAME.desktop"
    
    # Try to grab the icon from the Wails build directory if available
    ICON_PATH="/usr/share/pixmaps/$APP_NAME.png"
    if [ -f "build/appicon.png" ]; then
        sudo cp "build/appicon.png" "$ICON_PATH"
    fi
    
    # Absolute Exec path: some desktop environments do not put /usr/local/bin on PATH
    # for .desktop launches, so "Exec=wis-free-v3" can fail with no visible error.
    cat << EOF > /tmp/$APP_NAME.desktop
[Desktop Entry]
Type=Application
Name=WIS Free V3
Comment=Voice Dictation App
Exec=/usr/local/bin/$APP_NAME
TryExec=/usr/local/bin/$APP_NAME
Icon=$APP_NAME
StartupWMClass=$APP_NAME
Terminal=false
Categories=Utility;Audio;
EOF

    sudo mv /tmp/$APP_NAME.desktop "$DESKTOP_FILE"
    sudo chmod 644 "$DESKTOP_FILE"
    
    echo ""
    echo "Installation complete! You can now launch 'WIS Free V3' from your app launcher,"
    echo "or by typing '$APP_NAME' in your terminal."
else
    echo "Skipping installation. You can run the app directly via: ./$EXECUTABLE"
fi