#!/usr/bin/env bash
set -Eeuo pipefail
#
# Usage ./update.sh [version(multiple)...]
#

development_version=main
development_version_real=12.1

defaultSuite='noble'
declare -A suites=(
	[10.5]='focal'
	[10.6]='jammy'
	[10.11]='jammy'
)

declare -A suffix=(
	['focal']='ubu2004'
	['jammy']='ubu2204'
	['noble']='ubu2404'
)

#declare -A dpkgArchToBashbrew=(
#	[amd64]='amd64'
#	[armel]='arm32v5'
#	[armhf]='arm32v7'
#	[arm64]='arm64v8'
#	[i386]='i386'
#	[ppc64el]='ppc64le'
#	[s390x]='s390x'
#)

# For testing with https://downloads.dev.mariadb.org/rest-api
typeset -r DOWNLOADS_REST_API="https://downloads.mariadb.org/rest-api"

update_version()
{
	local dir=$version$ubi
	if [ ! -d "$dir" ]; then
		echo "Directory $dir missing"
		return
	fi
	echo "$version: $mariaVersion($ubi) ($releaseStatus)"

	if [ -z "$ubi" ]; then
		suite="${suites[$version]:-$defaultSuite}"
		fullVersion=1:${mariaVersion}+maria~${suffix[${suite}]}
	else
		suite=
		fullVersion=$mariaVersion
		if [[ $version = 10.* ]]; then
			sed -e '/character-set-collations/d' docker.cnf > "$dir/docker.cnf"
		else
			sed -e '/collation-server/d' docker.cnf > "$dir/docker.cnf"
			if [[ $version != 11.4 ]]; then
				sed -i -e '/character-set-collations/d' "$dir/docker.cnf"
				sed -i -e '/character-set/d' "$dir/docker.cnf"
			fi
		fi
		sed -e "s!%%MARIADB_VERSION%%!${version%-*}!" MariaDB-ubi.repo > "$dir"/MariaDB.repo
	fi

	if [[ $version = 10.[234]* ]]; then
		arches="amd64 arm64v8 ppc64le"
	else
		arches="amd64 arm64v8 ppc64le s390x"
	fi

	cp "Dockerfile${ubi}.template" "${dir}/Dockerfile"

	cp docker-entrypoint.sh healthcheck.sh "$dir/"
	chmod a+x "$dir"/healthcheck.sh
	sed -i \
		-e 's!%%MARIADB_VERSION%%!'"$fullVersion"'!g' \
		-e 's!%%MARIADB_VERSION_BASIC%%!'"$mariaVersion"'!g' \
		-e 's!%%MARIADB_MAJOR%%!'"${version%-ubi}"'!g' \
		-e 's!%%MARIADB_RELEASE_STATUS%%!'"$releaseStatus"'!g' \
		-e 's!%%MARIADB_SUPPORT_TYPE%%!'"$supportType"'!g' \
		-e 's!%%SUITE%%!'"$suite"'!g' \
		-e 's!%%ARCHES%%! '"$arches"'!g' \
		"$dir/Dockerfile"

	sed -i \
		-e 's!%%MARIADB_VERSION_BASIC%%!'"$mariaVersion"'!g' \
		"$dir/docker-entrypoint.sh"

	vmin=${version%-ubi}
	# Start using the new executable names
	case "$vmin" in
		10.5)
			sed -i -e '/--old-mode/d' \
				-e '/--skip-ssl/d' \
				-e 's/mariadb-upgrade\([^_"]\)/mysql_upgrade\1/' \
				-e 's/mariadb-dump/mysqldump/' \
				-e 's/mariadb-admin/mysqladmin/' \
				-e 's/\bmariadb --protocol\b/mysql --protocol/' \
				-e 's/mariadb-install-db/mysql_install_db/g' \
				-e 's/--mariadbd/--mysqld/' \
				-e 's/mariadb-tzinfo-to-sql/mysql_tzinfo_to_sql/' \
				-e '0,/#ENDOFSUBSTITUTIONS/s/mariadbd/mysqld/g' \
				-e '/memory\.pressure/,+7d' "$dir/docker-entrypoint.sh"
			sed -i -e '/--skip-ssl/d' \
				-e '0,/#ENDOFSUBSTITUTIONS/s/\tmariadb/\tmysql/' "$dir/healthcheck.sh"
			sed -i -e '/^CMD/s/mariadbd/mysqld/' \
				-e 's/ && userdel.*//' \
				"$dir/Dockerfile"
			sed -i -e 's/mariadb_upgrade_info/mysql_upgrade_info/' \
				"$dir/docker-entrypoint.sh" "$dir/healthcheck.sh"
			;;
		10.6*)
			sed -i -e '/memory\.pressure/,+7d' \
				-e 's/--mariadbd/--mysqld/' \
				"$dir/docker-entrypoint.sh"
			sed -i -e '/--skip-ssl/d' "$dir/docker-entrypoint.sh" "$dir/healthcheck.sh"
			sed -i -e 's/mariadb_upgrade_info/mysql_upgrade_info/' \
				"$dir/docker-entrypoint.sh" "$dir/healthcheck.sh"
			sed -i -e 's/ && userdel.*//' \
				"$dir/Dockerfile"
			;;
		10.11)
			sed -i -e 's/mariadb_upgrade_info/mysql_upgrade_info/' \
				-e '/--skip-ssl/d' \
				"$dir/docker-entrypoint.sh" "$dir/healthcheck.sh"
			# quoted $ intentional
			# shellcheck disable=SC2016
			sed -i -e '/^ARG MARIADB_MAJOR/d' \
				-e '/^ENV MARIADB_MAJOR/d' \
				-e 's/-\$MARIADB_MAJOR//' \
				-e 's/ && userdel.*//' \
				"$dir/Dockerfile"
			;;
		*)
			# quoted $ intentional
			# shellcheck disable=SC2016
			sed -i -e '/^ARG MARIADB_MAJOR/d' \
				-e '/^ENV MARIADB_MAJOR/d' \
				-e 's/-\$MARIADB_MAJOR//' \
				"$dir/Dockerfile"
			if [ "$vmin" = 11.2 ]; then
				sed -i -e '/--skip-ssl/d' \
				       	"$dir/docker-entrypoint.sh" "$dir/healthcheck.sh"
				sed -i -e 's/ && userdel.*//' \
					"$dir/Dockerfile"
			fi
			sed -i -e 's/ \/[^ ]*50-mysqld_safe.cnf//' "$dir/Dockerfile"
			;&
	esac

	if [ -z "$suite" ]; then
		base=ubi9
	else
		base=ubuntu:$suite
	fi
	# Add version to versions.json
	versionJson="$(jq -e \
		--arg milestone "${version%%-*}" --arg milestoneversion "${version}${ubi}" --arg version "$mariaVersion" --arg fullVersion "$fullVersion" --arg releaseStatus "$releaseStatus" --arg supportType "$supportType" --arg base "$base" --arg arches "${arches# }" \
		'.[$milestoneversion] = {"milestone": $milestone, "version": $version, "fullVersion": $fullVersion, "releaseStatus": $releaseStatus, "supportType": $supportType, "base": $base, "arches": $arches|split(" ")}' versions.json)"
	printf '%s\n' "$versionJson" > versions.json
}

update_version_array()
{
	c0=$(( $1 - 2 ))
	c1=$(( $1 - 1 ))
	version=${release[$c0]}
	if [ ! -d "$version" ]; then
		echo >&2 "warning: no rule for $version"
		return
	fi
	mariaversion

	releaseStatus=${release[$c1]}

	case "$releaseStatus" in
		Alpha | Beta | Gamma | RC | Stable ) ;; # sanity check
		*) echo >&2 "error: unexpected 'release status' value for $mariaVersion: $releaseStatus"; ;;
	esac

	supportType=$2

	update_version
	ubi=-ubi update_version
}

mariaversion()
{
	# version hacks because our $DOWNLOADS_REST_API
	# seems to never be right on release and has unfinshed suppport
	# for rolling release versions.
	#if [ "$version" = 11.4 ]; then
	#	mariaVersion=11.4.7;
	#	return
	#fi
	mariaVersion=$(curl -fsSL "$DOWNLOADS_REST_API/mariadb/${version%-*}" \
		| jq -r 'first(.releases[] | .release_id | select(. | test("[0-9]+.[0-9]+.[0-9]+$")))')
	mariaVersion=${mariaVersion//\"}
}

all()
{
	printf '%s\n' "{}" > versions.json

	readarray -O 0 -c 3 -C update_version_array -t release <<< "$(curl -fsSL "$DOWNLOADS_REST_API/mariadb/" \
		| jq -r '.major_releases[] | [ .release_id ], [ .release_status ], [ .release_support_type ]  | @tsv')"
}

in_development()
{
	releaseStatus=Alpha
	supportType=Unknown
	version=$development_version
	mariaVersion=${development_version_real}.0
	update_version
}


if [ $# -eq 0 ]; then
	ubi=
	in_development
	ubi=-ubi in_development
	all
	exit 0
fi

versions=( "$@" )

for version in "${versions[@]}"; do
	version="${version%/}"
	if [ "${version#*-}" = "ubi" ]; then
		ubi=-ubi
		version=${version%-ubi}
	else
		ubi=
	fi

	if [ "${version%-*}" == $development_version ]; then
		in_development
		continue
	fi
	if [ ! -d "$version" ]; then
		version=${version%.[[:digit:]]*}
	else
		mariaversion
	fi
	readarray -t release <<< "$(curl -fsSL "$DOWNLOADS_REST_API/mariadb/" \
		| jq -r --arg version "${version%-*}" '.major_releases[] | select(.release_id == $version) | [ .release_status ] , [ .release_support_type ] | @tsv')"

	case "$version" in
	11.7)
		releaseStatus=${release[0]:-Stable}
		supportType=${release[1]:-Short Term Support}
		;;
        11.6)
		releaseStatus=${release[0]:-Stable}
		supportType=${release[1]:-Short Term Support}
		;;
	*)
		releaseStatus=${release[0]:-Unknown}
		supportType=${release[1]:-Unknown}
		;;
	esac

	update_version
done