#!/bin/bash
set -euo pipefail

HOG_PATH="/usr/local/bin/hogger"

GREEN='\033[0;32m'
NC='\033[0m' # No Color
CONFIG_FILE="/etc/hogger_settings.ini"

# Function to check and install Xcode Command Line Tools
install_xcode_clt() {
    if xcode-select --print-path &>/dev/null; then
        echo "Xcode Command Line Tools are already installed!"
        return
    fi

    xcode-select --install &>/dev/null

    echo "Installing Xcode Command Line Tools..."

    while true; do
        if xcode-select --print-path &>/dev/null; then
            echo "Installation completed!"
            return
        fi

        if ! pgrep "Install Command Line Developer Tools" &>/dev/null; then
            echo "Installation was canceled or failed!"
            exit 1
        fi

        sleep 5
    done
}

# Call the function to ensure Xcode Command Line Tools are installed
install_xcode_clt

# Checks if the hog is already running
hog_running_output=$(launchctl list | grep io.green-coding.hogger || echo "")

if [[ ! -z "$hog_running_output" ]]; then
    launchctl bootout system /Library/LaunchDaemons/io.green-coding.hogger.plist
    rm -f /tmp/latest_release.zip
fi

###
# Downloads and moves the code
###

ZIP_LOCATION=$(curl -s https://api.github.com/repos/green-coding-solutions/hog/releases/latest | grep -o 'https://[^"]*/hog_power_logger.zip')
curl -fLo /tmp/latest_release.zip $ZIP_LOCATION

if [[ -z "/tmp/latest_release.zip" ]]; then
    echo "Error: Could not fetch the ZIP URL from the GitHub API."
    exit 1
fi

mkdir -p "$HOG_PATH"

unzip -o -u /tmp/latest_release.zip -d "$HOG_PATH"
rm /tmp/latest_release.zip

chmod 755 "$HOG_PATH"
chmod -R 755 "$HOG_PATH"
chmod +x "$HOG_PATH"/power_logger.py

###
# Writing the config file
###

cat "$HOG_PATH/settings.ini" > "$CONFIG_FILE"

if [[ -t 0 ]]; then
    read -p "In order for the app to work with all features, please allow us to upload some data. [Y/n]: " upload_data
    upload_data=${upload_data:-Y}
    upload_data=$(echo "$upload_data" | tr '[:upper:]' '[:lower:]')
else
    upload_data="n"
fi

if [[ $upload_data == "y" || $upload_data == "yes" ]]; then
    sed -i '' "s|^upload_data =.*|upload_data = true|" "$CONFIG_FILE"
else
    sed -i '' "s|^upload_data =.*|upload_data = false|" "$CONFIG_FILE"
fi

echo "Installation complete. Configuration updated at $CONFIG_FILE. You might need to edit it manually."


###
# Setting up the background demon
###

mv -f "$HOG_PATH/io.green-coding.hogger.plist" /Library/LaunchDaemons/io.green-coding.hogger.plist

sed -i '' "s|PATH_PLEASE_CHANGE|$HOG_PATH|g" /Library/LaunchDaemons/io.green-coding.hogger.plist

chown root:wheel /Library/LaunchDaemons/io.green-coding.hogger.plist
chmod 644 /Library/LaunchDaemons/io.green-coding.hogger.plist

launchctl bootstrap system /Library/LaunchDaemons/io.green-coding.hogger.plist

echo -e "${GREEN}Successfully installed the Power Hog Demon!${NC}"