#!/bin/bash

# Get KStars version from argument, default to stable if not provided
KSTARS_VERSION=${1:-stable}

# Define variables
REPO_URL="https://github.com/ikarustechnologies/indi-firmware.git"
REPO_DIR="$HOME/.indi-firmware"
UDEV_RULES_DIR="/lib/udev/rules.d"
FIRMWARE_DIR="/lib/firmware"
QHY_FIRMWARE_DIR="/lib/firmware/qhy"
FXLOAD_HELPER="/usr/local/sbin/indi-fxload"

# Detect operating system information for package installation
if [ -f /etc/os-release ]; then
    . /etc/os-release
else
    echo "Error: Could not detect operating system information."
    exit 1
fi

# Install flatpak if not already installed based on distribution
echo "Checking for and installing flatpak..."

if ! command -v flatpak &> /dev/null; then
    case "$ID" in
        debian|ubuntu)
            echo "Detected Debian/Ubuntu. Installing flatpak using apt..."
            sudo apt install -y flatpak || { echo "Error: apt install flatpak failed"; exit 1; }
            ;;
        fedora)
            echo "Detected Fedora. Installing flatpak using dnf..."
            sudo dnf install -y flatpak || { echo "Error: dnf install flatpak failed"; exit 1; }
            ;;
        arch)
            echo "Detected Arch Linux. Installing flatpak using pacman..."
            sudo pacman -Syu --noconfirm flatpak || { echo "Error: pacman install flatpak failed"; exit 1; }
            ;;
        *)
            echo "Error: Unsupported distribution '$ID' for automatic Flatpak installation."
            echo "Please install flatpak manually and rerun this script."
            exit 1
            ;;
    esac
    echo "Flatpak installed."
else
    echo "Flatpak is already installed."
fi

echo "Checking for and installing fxload (required for QHYCCD firmware)..."
if ! command -v fxload &> /dev/null && [ ! -x /sbin/fxload ]; then
    case "$ID" in
        debian|ubuntu)
            echo "Detected Debian/Ubuntu. Installing fxload using apt..."
            sudo apt install -y fxload || { echo "Error: apt install fxload failed"; exit 1; }
            ;;
        fedora)
            echo "Detected Fedora. Installing fxload using dnf..."
            sudo dnf install -y fxload || { echo "Error: dnf install fxload failed"; exit 1; }
            ;;
        arch)
            echo "Detected Arch Linux. Installing fxload using pacman..."
            sudo pacman -Syu --noconfirm fxload || { echo "Error: pacman install fxload failed"; exit 1; }
            ;;
        *)
            echo "Warning: Unsupported distribution '$ID' for automatic fxload installation."
            echo "Please install the 'fxload' package manually and rerun this script."
            ;;
    esac
    echo "fxload installed."
else
    echo "fxload is already installed."
fi

# Check if the repository directory exists. If it does, pull; otherwise, clone.
if [ -d "$REPO_DIR" ]; then
  echo "Repository directory '$REPO_DIR' already exists. Pulling latest changes..."
  cd "$REPO_DIR" || { echo "Error: Could not navigate to $REPO_DIR"; exit 1; }
  # Use to potentially suppress GPG-related advice messages
  git pull || { echo "Error: Git pull failed"; exit 1; }
  cd - || { echo "Error: Could not return to previous directory"; exit 1; }
else
  echo "Cloning repository '$REPO_URL'..."
  # Use to potentially suppress GPG-related advice messages
  git clone "$REPO_URL" "$REPO_DIR" || { echo "Error: Git clone failed"; exit 1; }
fi

# Create destination directories if they don't exist
echo "Creating destination directories: $UDEV_RULES_DIR, $QHY_FIRMWARE_DIR..."
sudo mkdir -p "$UDEV_RULES_DIR" || { echo "Error: Could not create directory $UDEV_RULES_DIR"; exit 1; }
sudo mkdir -p "$QHY_FIRMWARE_DIR" || { echo "Error: Could not create directory $QHY_FIRMWARE_DIR"; exit 1; }
echo "Destination directories created."

# Copy udev rules files
echo "Copying udev rules files to $UDEV_RULES_DIR..."

sudo cp -v "$REPO_DIR"/indi/drivers/auxiliary/99-indi_auxiliary.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-indi_auxiliary.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi/drivers/video/80-dbk21-camera.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 80-dbk21-camera.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/indi-ffmv/99-fireflymv.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-fireflymv.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/indi-nightscape/99-nightscape.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-nightscape.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/indi-armadillo-platypus/99-armadilloplatypus.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-armadilloplatypus.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libpktriggercord/src/pentax.rules "$UDEV_RULES_DIR"/ || { echo "Error copying pentax.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libpktriggercord/src/samsung.rules "$UDEV_RULES_DIR"/ || { echo "Error copying samsung.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/indi-qsi/99-qsi.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-qsi.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/indi-sx/99-sx.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-sx.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libapogee/99-apogee.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-apogee.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/indi-dsi/99-meadedsi.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-meadedsi.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libfishcamp/99-fishcamp.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-fishcamp.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libfli/99-fli.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-fli.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/indi-gphoto/85-disable-dslr-automout.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 85-disable-dslr-automout.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libqhy/85-qhyccd.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 85-qhyccd.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libasi/99-asi.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-asi.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libmeadecam/99-meadecam.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-meadecam.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/indi-mgen/80-LacertaMgen.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 80-LacertaMgen.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/indi-orion-ssg3/99-orionssg3.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-orionssg3.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libricohcamerasdk/99-pentax.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-pentax.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libtscam/99-tscam.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-tscam.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libatik/99-atik.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-atik.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libastroasis/99-astroasis.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-astroasis.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/obsolete/95-ssag.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 95-ssag.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libmallincam/99-mallincam.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-mallincam.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libstarshootg/99-starshootg.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-starshootg.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libinovasdk/99-inovaplx.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-inovaplx.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libsbig/51-sbig-debian.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 51-sbig-debian.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libomegonprocam/99-omegonprocam.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-omegonprocam.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libsvbonycam/99-svbonycam.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-svbonycam.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libaltaircam/99-altaircam.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-altaircam.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libmicam/99-miccd.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-miccd.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libtoupcam/99-toupcam.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-toupcam.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libnncam/99-nncam.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-nncam.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libbressercam/99-bressercam.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-bressercam.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libsvbony/90-svbonyusb.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 90-svbonyusb.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libplayerone/99-player_one_astronomy.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-player_one_astronomy.rules"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libogmacam/99-ogmacam.rules "$UDEV_RULES_DIR"/ || { echo "Error copying 99-ogmacam.rules"; exit 1; }
echo "Udev rules files copied."

# Install fxload compatibility helper
echo "Installing fxload helper script to $FXLOAD_HELPER..."
sudo install -m 0755 "$REPO_DIR"/scripts/indi-fxload.sh "$FXLOAD_HELPER" || { echo "Error installing fxload helper script"; exit 1; }

# Copy individual firmware files
echo "Copying individual firmware files to $FIRMWARE_DIR..."

sudo cp -v "$REPO_DIR"/indi-3rdparty/libfishcamp/gdr_usb.hex "$FIRMWARE_DIR"/ || { echo "Error copying gdr_usb.hex"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libfishcamp/Guider_mono_rev16_intel.srec "$FIRMWARE_DIR"/ || { echo "Error copying Guider_mono_rev16_intel.srec"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libsbig/sbigucam.hex "$FIRMWARE_DIR"/ || { echo "Error copying sbigucam.hex"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libsbig/sbiglcam.hex "$FIRMWARE_DIR"/ || { echo "Error copying sbiglcam.hex"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libsbig/sbigfcam.hex "$FIRMWARE_DIR"/ || { echo "Error copying sbigfcam.hex"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libsbig/sbigpcam.hex "$FIRMWARE_DIR"/ || { echo "Error copying sbigpcam.hex"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/libsbig/stfga.bin "$FIRMWARE_DIR"/ || { echo "Error copying stfga.bin"; exit 1; }
sudo cp -v "$REPO_DIR"/indi-3rdparty/indi-dsi/meade-deepskyimager.hex "$FIRMWARE_DIR"/ || { echo "Error copying meade-deepskyimager.hex"; exit 1; }
echo "Individual firmware files copied."

# Copy QHY firmware files
echo "Copying QHY firmware files to $QHY_FIRMWARE_DIR..."

sudo cp -v "$REPO_DIR"/indi-3rdparty/libqhy/firmware/* "$QHY_FIRMWARE_DIR"/ || { echo "Error copying QHY firmware files"; exit 1; }
echo "QHY firmware files copied."

# Reload udev rules after copying
echo "Reloading udev rules..."
sudo udevadm control --reload-rules || { echo "Error reloading udev rules"; exit 1; }
echo "Udev rules reloaded."

# Add Flathub remote if it doesn't exist
echo "Adding Flathub remote if it doesn't exist..."
flatpak remote-add --user --if-not-exists flathub https://flathub.org/repo/flathub.flatpakrepo || { echo "Error adding Flathub remote"; exit 1; }
echo "Flathub remote added."

# Add StellarMate Hub remote with no GPG verification
echo "Adding kstars-nightly hub remote if it doesn't exist..."
flatpak remote-add --user --if-not-exists kstars-nightly https://origin.cdn.kde.org/flatpak/kstars-nightly/kstars-nightly.flatpakrepo || { echo "Error adding kstars-nightly remote"; exit 1; }
echo "kstars-nightly remote added."

# Install KDE Platform and SDK runtimes
echo "Installing KDE Platform and SDK runtimes..."
flatpak install --user flathub org.kde.Platform//6.10 -y || { echo "Error installing KDE Platform runtime"; exit 1; }
flatpak install --user flathub org.kde.Sdk//6.10 -y || { echo "Error installing KDE SDK runtime"; exit 1; }
echo "KDE Platform and SDK runtimes installed."

# Install KStars Flatpak package
echo "Installing KStars Flatpak package version ${KSTARS_VERSION}..."
if [ "${KSTARS_VERSION}" = "stable" ]; then
    flatpak install --user flathub org.kde.kstars//stable -y || { echo "Error installing KStars Flatpak ${KSTARS_VERSION}"; }
elif [ "${KSTARS_VERSION}" = "nightly" ]; then
    flatpak install --user kstars-nightly org.kde.kstars//master -y || { echo "Error installing KStars Flatpak ${KSTARS_VERSION}"; }
fi
flatpak update --user org.kde.kstars -y || { echo "Error updating KStars Flatpak ${KSTARS_VERSION}"; exit 1; }
flatpak override --user --filesystem=host org.kde.kstars
flatpak override --user --device=all org.kde.kstars
flatpak override --user --talk-name=org.freedesktop.secrets org.kde.kstars
flatpak override --user --talk-name=org.kde.kwalletd6 org.kde.kstars
echo "KStars Flatpak version ${KSTARS_VERSION} installed."

echo "Script finished."
echo " "
echo "================================================================"
echo "================================================================"
echo " "
echo " "
echo "                   Installation completed "
echo "                For running the Kstars, run: "
if [ "${KSTARS_VERSION}" = "stable" ]; then
    echo "       flatpak run --user org.kde.kstars//stable "
elif [ "${KSTARS_VERSION}" = "nightly" ]; then
    echo "       flatpak run --user org.kde.kstars//master "
fi
echo " "
echo " "
echo "================================================================"
echo "================================================================"