#!/bin/bash

echo "Starting up Aria's Creations tools installer"

# Check if the script is running as root
if [ "$(id -u)" -eq 0 ]; then
    echo "Running as root. No need for sudo or password."
    SUDO_CMD=""
else
    echo "Not running as root. Sudo and password required."
    PASS=$(zenity --password --title="SHT Installer" --timeout=120)

    # Check if the password input was canceled or failed
    if [ -z "$PASS" ]; then
        echo "No password entered. Exiting installer."
        exit 1
    fi

    # Test password to check if it's correct
    echo -ne "$PASS" | sudo -S echo "Escalation successful" &>/dev/null
    if [ $? -ne 0 ]; then
        echo "Password incorrect. Exiting installer."
        exit 1
    fi

    SUDO_CMD="echo -ne \"$PASS\" | sudo -S"
fi

# Download the tools and put them in place
$SUDO_CMD wget https://ci.zontreck.com/job/Projects/job/Dart/job/SimpleHelperTools/job/main/lastSuccessfulBuild/artifact/dart/out/nbt2snbt-linux-x64 -O /usr/bin/nbt2snbt

$SUDO_CMD wget https://ci.zontreck.com/job/Projects/job/Dart/job/SimpleHelperTools/job/main/lastSuccessfulBuild/artifact/dart/out/pause-linux-x64 -O /usr/bin/pause

$SUDO_CMD wget https://ci.zontreck.com/job/Projects/job/Dart/job/SimpleHelperTools/job/main/lastSuccessfulBuild/artifact/dart/out/snbt2nbt-linux-x64 -O /usr/bin/snbt2nbt

$SUDO_CMD wget https://ci.zontreck.com/job/Projects/job/Dart/job/SimpleHelperTools/job/main/lastSuccessfulBuild/artifact/dart/out/dbikc-linux-x64 -O /usr/bin/dbikc

$SUDO_CMD wget https://ci.zontreck.com/job/Projects/job/Dart/job/SimpleHelperTools/job/main/lastSuccessfulBuild/artifact/dart/out/uuidgen-linux-x64 -O /usr/bin/uuidgen

$SUDO_CMD wget https://ci.zontreck.com/job/Projects/job/Dart/job/SimpleHelperTools/job/main/lastSuccessfulBuild/artifact/dart/out/sleep-linux-x64 -O /usr/bin/vsleep

$SUDO_CMD wget https://ci.zontreck.com/job/Projects/job/Dart/job/SimpleHelperTools/job/main/lastSuccessfulBuild/artifact/dart/out/mkfsreport-linux-x64 -O /usr/bin/mkfsreport

# Set executable permissions
$SUDO_CMD chmod +x /usr/bin/{nbt2snbt,snbt2nbt,pause,mkfsreport,dbikc,vsleep}

echo "Installation Completed"
vsleep 5