#!/usr/bin/env bash
# TA-ProxMenu preloader

source /opt/idssys/ta-proxmenu/inc/runtime-config.inc
TAPM_ENSURE_RUNTIME_CONFIG || exit 1

source /opt/idssys/ta-proxmenu/inc/git-update.inc
source /opt/idssys/ta-proxmenu/inc/cluster-update.inc

[ "${2:-}" != "q" ] && source /opt/idssys/ta-proxmenu/colors.inc
source /opt/idssys/ta-proxmenu/defaults.inc
TAPM_FLEET_START "$VERS"
trap 'TAPM_FLEET_FINISH "$?"' EXIT

UPDATE_REPOSITORY() {
	local repository="$1"
	local branch="$2"
	local label="$3"
	local current_branch
	local state

	if [[ ! -d "${repository}/.git" ]]; then
		echo -e "${idsCL[Red]}${label} is not a Git repository.${idsCL[Default]}"
		return 1
	fi

	current_branch="$(git -C "$repository" branch --show-current 2>/dev/null)"
	if [[ -z "$current_branch" ]]; then
		echo -e "${idsCL[Red]}${label} is in a detached HEAD state; update skipped.${idsCL[Default]}"
		return 1
	fi
	if [[ "$current_branch" != "$branch" ]]; then
		echo -e "${idsCL[Red]}${label} is on '${current_branch}', not '${branch}'; update skipped.${idsCL[Default]}"
		return 1
	fi
	if TAPM_GIT_WORKTREE_DIRTY "$repository"; then
		echo -e "${idsCL[LightYellow]}${label} has local changes; update skipped to preserve them.${idsCL[Default]}"
		git -C "$repository" status --short
		return 1
	fi

	if ! TAPM_GIT_FETCH_BRANCH "$repository" "$branch" 30 >/dev/null 2>&1; then
		echo -e "${idsCL[Red]}Could not fetch branch '${branch}' for ${label}.${idsCL[Default]}"
		return 1
	fi

	state="$(TAPM_GIT_BRANCH_STATE "$repository" "$branch")"
	case "$state" in
		current)
			echo -e "${idsCL[Green]}${label} is current (${branch}).${idsCL[Default]}"
			return 0
			;;
		behind)
			echo -en "${idsCL[LightCyan]}Updating ${label} (${branch})...${idsCL[Default]}"
			if TAPM_GIT_FAST_FORWARD "$repository" "$branch" >/dev/null 2>&1; then
				echo -e " ${idsCL[Green]}Done${idsCL[Default]}"
				return 0
			fi
			echo -e " ${idsCL[Red]}Failed; local files were preserved.${idsCL[Default]}"
			return 1
			;;
		ahead)
			echo -e "${idsCL[LightYellow]}${label} is ahead of origin; its local commits were preserved.${idsCL[Default]}"
			return 1
			;;
		diverged)
			echo -e "${idsCL[LightYellow]}${label} has diverged from origin; automatic update was refused.${idsCL[Default]}"
			return 1
			;;
		*)
			echo -e "${idsCL[Red]}Could not determine the update state for ${label}.${idsCL[Default]}"
			return 1
			;;
	esac
}

SWITCH_TAPM_BRANCH() {
	local target_branch="$1"
	local repository='/opt/idssys/ta-proxmenu'
	local current_branch
	local target_state=''

	if ! git check-ref-format --branch "$target_branch" >/dev/null 2>&1; then
		echo -e "${idsCL[Red]}'${target_branch}' is not a valid Git branch name.${idsCL[Default]}"
		return 1
	fi
	if [[ ! -d "${repository}/.git" ]]; then
		echo -e "${idsCL[Red]}TA-ProxMenu is not a Git repository.${idsCL[Default]}"
		return 1
	fi

	current_branch="$(git -C "$repository" branch --show-current 2>/dev/null)"
	if [[ -z "$current_branch" ]]; then
		echo -e "${idsCL[Red]}TA-ProxMenu is in a detached HEAD state; branch switching was refused.${idsCL[Default]}"
		return 1
	fi
	if [[ "$target_branch" == "$current_branch" ]]; then
		echo -e "${idsCL[Green]}TA-ProxMenu is already using '${target_branch}'.${idsCL[Default]}"
		return 0
	fi
	if TAPM_GIT_WORKTREE_DIRTY "$repository"; then
		echo -e "${idsCL[LightYellow]}TA-ProxMenu has local changes; branch switching was refused to preserve them.${idsCL[Default]}"
		git -C "$repository" status --short
		return 1
	fi

	echo -e "${idsCL[LightCyan]}Fetching TA-ProxMenu branch '${target_branch}'...${idsCL[Default]}"
	if ! TAPM_GIT_FETCH_BRANCH "$repository" "$target_branch" 30 >/dev/null 2>&1; then
		echo -e "${idsCL[Red]}Could not fetch branch '${target_branch}' from origin.${idsCL[Default]}"
		return 1
	fi

	if git -C "$repository" show-ref --verify --quiet "refs/heads/${target_branch}"; then
		target_state="$(
			TAPM_GIT_RELATION "$repository" "refs/heads/${target_branch}" \
				"refs/remotes/origin/${target_branch}"
		)" || {
			echo -e "${idsCL[Red]}Could not compare local and remote '${target_branch}'.${idsCL[Default]}"
			return 1
		}
		case "$target_state" in
			ahead)
				echo -e "${idsCL[LightYellow]}Local branch '${target_branch}' has commits not on origin; switching was refused.${idsCL[Default]}"
				return 1
				;;
			diverged)
				echo -e "${idsCL[LightYellow]}Local branch '${target_branch}' has diverged from origin; switching was refused.${idsCL[Default]}"
				return 1
				;;
		esac

		git -C "$repository" switch "$target_branch" >/dev/null || return 1
	else
		git -C "$repository" switch --create "$target_branch" \
			--track "origin/${target_branch}" >/dev/null || return 1
	fi

	if [[ "$target_state" == "behind" ]] &&
		! TAPM_GIT_FAST_FORWARD "$repository" "$target_branch" >/dev/null; then
		echo -e "${idsCL[Red]}Could not fast-forward '${target_branch}'; no commits were discarded.${idsCL[Default]}"
		return 1
	fi

	rm -f /var/cache/ta-proxmenu/update-status 2>/dev/null || true
	echo -e "${idsCL[Green]}TA-ProxMenu is now using branch '${target_branch}'. Loading its menu...${idsCL[Default]}"
	exec /opt/idssys/ta-proxmenu/run.sh
}

INSTALL_LOCAL_UPDATES() {
	local current_branch
	local update_failed=0

	echo -e "${idsCL[LightCyan]}Checking for updates...${idsCL[Default]}"

	current_branch="$(git -C /opt/idssys/ta-proxmenu branch --show-current)"
	if [ -z "$current_branch" ]; then
		echo -e "${idsCL[Red]}TA-ProxMenu is in a detached HEAD state; update skipped${idsCL[Default]}"
		update_failed=1
	else
		UPDATE_REPOSITORY /opt/idssys/ta-proxmenu "$current_branch" "TA-ProxMenu" ||
			update_failed=1
	fi

	rm -f /var/cache/ta-proxmenu/update-status 2>/dev/null || true

	if (( update_failed == 0 )); then
		source /opt/idssys/ta-proxmenu/defaults.inc
		echo -e "${idsCL[Green]}Update check complete. Installed version: ${VERS}${idsCL[Default]}"
		return 0
	fi

	return 1
}

case "${1:-}" in
	update|u)
		if [[ "${2:-}" == "--local-only" ]]; then
			INSTALL_LOCAL_UPDATES
		else
			INSTALL_CLUSTER_UPDATES
		fi
		exit $?
		;;
	main)
		SWITCH_TAPM_BRANCH "$1"
		exit $?
		;;
	tapm)
		exit 0
		;;
esac

/opt/idssys/ta-proxmenu/proxmenu-scripts.sh \
	"${1:-}" "${2:-}" "${3:-}" "${4:-}"

exit $?