#!/usr/bin/env bash
set -e

RED='\033[0;31m'
GREEN='\033[0;32m'
ORANGE='\033[0;33m'
NC='\033[0m'

DESTINATION_PATH="/usr/local/bin"
REPO_URL="https://raw.githubusercontent.com/v4n00/h2mm-cli/refs/heads/master"

function log() {
	local type="$1"
	shift
	case "$type" in
		INFO)
			echo -e "$*" >&2
			;;
		ERROR)
			echo -e "${RED}[ERROR]${NC} $*" >&2
			;;
		PROMPT)
			echo -ne "$*" >&2
			;;
	esac
}

# --- Main ---

log INFO "Running h2mm CLI Installer (https://github.com/v4n00/h2mm-cli)"

# breaking changes hash table
breaking_changes_patches=(
    ["2"]='sed -i "s/^\([0-9]\+\),/\1,ENABLED,/" "$1/mods.csv"'
    ["3"]='sed -i "1 i\\3" "$1/mods.csv"'
	["4"]='tmp_file=$(mktemp) && awk '\''BEGIN {FS=OFS=","} NR==1 {print 4; next} {print NR-1, $2, $3, $4, $5}'\'' "$1/mods.csv" > "$tmp_file" && tee "$1/mods.csv" < "$tmp_file" > /dev/null && rm "$tmp_file"'
	["5"]='sed -i "s/^\([0-9]\+\),\(.*\),\(.*\),\(.*\)/\1,\2,\3,,,,\4/" "$1/mods.csv"; sed -i "1 s/4/5/" "$1/mods.csv"'
	["6"]='sed -i "s/^\([0-9]\+\),\(.*\),\(.*\),\(.*\),\(.*\),\(.*\),\(.*\)/\1,\2,\3,\4,\6,\7/" "$1/mods.csv"; sed -i "1 s/5/6/" "$1/mods.csv"'
)

# notify if update is happening
installed_version=""
latest_version=""
if [[ -x "$(command -v h2mm)" ]]; then
    installed_version=$(h2mm --version)

	# if installed version isn't x.x.x crash
	if [[ ! "$installed_version" =~ ^[0-9]+\.[0-9]+\.[0-9]+$ ]]; then
		log ERROR "Installed version is not in the correct format."
		log ERROR "h2mm is installed here -> $(which h2mm)"
		log ERROR "Delete the script file and retry the install script, any mods installed will not be lost."
		log INFO 'Download using the command: bash -c "$(curl -fsSL https://raw.githubusercontent.com/v4n00/h2mm-cli/refs/heads/master/install.sh)"'
		exit 1
	fi

    latest_version=$(curl -sS "$REPO_URL"/version)
    if [[ "$latest_version" == "$installed_version" ]]; then
        log INFO "You are reinstalling version ${GREEN}$installed_version${NC}."
    else
        log INFO "You are upgrading from ${ORANGE}$installed_version${NC} -> ${GREEN}$latest_version${NC}."
    fi
fi

# if steam os, set destination path to ~/.local/bin
steamos=false

if [[ -f "/etc/os-release" ]]; then
	if grep -q "ID=steamos" /etc/os-release; then
		steamos=true
	fi
fi

if [[ $steamos == true ]]; then
    # steam os
    DESTINATION_PATH="$HOME/.local/bin"
    log INFO "Detected SteamOS, setting installation path to ${ORANGE}$DESTINATION_PATH${NC}."

    mkdir -p "$DESTINATION_PATH"

    # check if ~/.local/bin is in PATH
    if [[ ":$PATH:" != *":$HOME/.local/bin:"* ]]; then
        # add ~/.local/bin to PATH
        log INFO "Installing the script on SteamOS means adding ${ORANGE}$DESTINATION_PATH${NC} to your \$PATH."
        log INFO "If you're using a different shell than bash (the default), you may need to add it manually."

		log PROMPT "Do you want to automatically add ${ORANGE}$DESTINATION_PATH${NC} to your \$PATH in ~/.bashrc? (Y/n): "
        IFS= read -e response
        if [[ "$response" == "y" || "$response" = "Y" || -z "$response" ]]; then
            echo "export PATH=\"\$HOME/.local/bin:\$PATH\"" >> "$HOME/.bashrc"
			[[ $? -ne 0 ]] && { log ERROR "Failed to add $DESTINATION_PATH to \$PATH in ~/.bashrc." ; exit 1; }

			source "$HOME/.bashrc"
			export PATH="$HOME/.local/bin:$PATH" # fallback kinda in case sourcing fails

            log INFO "Added ${ORANGE}$DESTINATION_PATH${NC} to your \$PATH in ~/.bashrc."
        fi
    fi
else
    # not steam os
    # set another path if needed
	log PROMPT "Install the script to ${ORANGE}$DESTINATION_PATH${NC} or specify another path (must be included in \$PATH)? (Y/path): "
    IFS= read -e response

    if [[ "$response" != "y" && "$response" != "Y" && -n "$response" ]]; then
        DESTINATION_PATH="$response"
        [[ ! -d "$DESTINATION_PATH" ]] && { log ERROR "Path $DESTINATION_PATH does not exist." ; exit 1; }
    fi
fi

# handle breaking changes
installed_major=$(echo "$installed_version" | awk -F. '{print $2}')
latest_major=$(echo "$latest_version" | awk -F. '{print $2}')

if [[ $latest_major -gt $installed_major ]]; then
	log INFO ""
	log INFO "${GREEN}IMPORTANT${NC}: Major version upgrade detected. Check out the changelogs here -> https://github.com/v4n00/h2mm-cli/releases"
	log INFO "The script will proceed to upgrade the database file. Creating a backup in case anything goes wrong."

	# make backup
	h2mm export

	# check if game directory is in ~/.config/h2mm/h2path
	if [[ -f "$HOME/.config/h2mm/h2path" ]]; then
		game_dir=$(cat "$HOME/.config/h2mm/h2path")
		[[ ! -d "$game_dir" ]] && { log ERROR "Helldivers 2 data directory is not valid: $game_dir." ; exit 1; }
	else
		log INFO "Searching for the Helldivers 2 data directory... (10 seconds timeout)"
		game_dir=$(timeout 10 find "$HOME" -type d -path "*/Steam/steamapps/common/Helldivers\ 2/data" 2>/dev/null | head -n 1)
	fi

	# if not found, prompt user
	if [[ -z "$game_dir" ]]; then
		# if not found, ask user for the directory
		log INFO "Could not find the Helldivers 2 data directory automatically."
		log PROMPT "Please enter the path to the Helldivers 2 data directory: "
		IFS= read -e game_dir; unset IFS
		game_dir="$(substitute_home "$game_dir")"

		[[ ! -d "$game_dir" ]] && { log ERROR "Provided path is not a valid directory."; exit 1; }
	else
		# confirm with the user that the directory is ok
		log INFO "Found Helldivers 2 data directory: $game_dir"
		log PROMPT "Is this the correct directory? (Y/n): "
		read confirm

		if [[ "$confirm" != "y" && "$confirm" != "Y" && "$confirm" != "" ]]; then
			log PROMPT "Please enter the path to the Helldivers 2 data directory: "
			IFS= read -e game_dir; unset IFS
			game_dir="$(substitute_home "$game_dir")"

			[[ ! -d "$game_dir" ]] && { log ERROR "Provided path is not a valid directory."; exit 1; }
		fi
	fi

	[[ ! -f "$game_dir/mods.csv" ]] && { log ERROR "mods.csv not found in $game_dir." ; exit 1; }

	# iterate from installed major number to latest major number
	for ((i = installed_major + 1; i <= latest_major; i++)); do
		if [[ -n "${breaking_changes_patches[$i]}" ]]; then
			# apply breaking changes patch
			eval $(echo "${breaking_changes_patches[$i]}" | sed "s:\$1:$game_dir:g")
		else
			log INFO "No breaking changes for version $i."
			continue
		fi

		if [[ $? -ne 0 ]]; then
			log ERROR "Failed to apply breaking changes patch for version $i. Do you want to continue? (Y/n): "
			read -er response

			[[ "$response" != "y" && "$response" != "Y" && -n "$response" ]] && { log INFO "Exiting." ; exit 1; }
		else
			log INFO "Version upgrade fix ${GREEN}successfully${NC} applied for version $i."
		fi
	done
	log INFO ""
fi

# install
log INFO "Installing h2mm to ${ORANGE}$DESTINATION_PATH${NC}."

# check if we need sudo based on destination path
SUDO_CMD=""
if [[ ! -w "$DESTINATION_PATH" || ( -f "$DESTINATION_PATH/h2mm" && ! -w "$DESTINATION_PATH/h2mm" ) ]]; then
    SUDO_CMD="sudo"
    log INFO "Destination path ${RED}requires${NC} elevated permissions, using sudo."
fi

$SUDO_CMD curl "$REPO_URL"/h2mm --output "$DESTINATION_PATH/h2mm"
$SUDO_CMD chmod +x "$DESTINATION_PATH/h2mm"
log INFO ""

[[ ! -x "$(command -v h2mm)" ]] && { log ERROR "Installation failed. Mod manager was not found in \$PATH." ; exit 1; }

log INFO "Helldivers 2 Mod Manager CLI ${GREEN}successfully${NC} installed."
log INFO "${GREEN}IMPORTANT${NC}: To install mods, you need to have installed:"
log INFO " -> \"unzip\" package for .zip archives"
log INFO " -> \"unarchiver\" package for .rar archives"
log INFO "If you do not know how to install these packages, please search for your linux distro on how to install packages."
log INFO ""
log INFO "Use the mod manager by running 'h2mm' in your terminal."
log INFO "View the changelogs: https://github.com/v4n00/h2mm-cli/releases/latest"
log INFO "Check out the ${ORANGE}Nexus Mods integration${NC} by running 'h2mm nexus-setup'."
log INFO "Made with <3 by v4n and contributors."