#!/bin/sh
set -eu

#region logging setup
if [ "${MISE_DEBUG-}" = "true" ] || [ "${MISE_DEBUG-}" = "1" ]; then
	debug() {
		echo "$@" >&2
	}
else
	debug() {
		:
	}
fi

if [ "${MISE_QUIET-}" = "1" ] || [ "${MISE_QUIET-}" = "true" ]; then
	info() {
		:
	}
else
	info() {
		echo "$@" >&2
	}
fi

error() {
	echo "$@" >&2
	exit 1
}
#endregion

#region environment setup
get_os() {
	os="$(uname -s)"
	if [ "$os" = Darwin ]; then
		echo "macos"
	elif [ "$os" = Linux ]; then
		echo "linux"
	else
		error "unsupported OS: $os"
	fi
}

get_arch() {
	musl=""
	if type ldd >/dev/null 2>/dev/null; then
		libc=$(ldd /bin/ls | grep 'musl' | head -1 | cut -d ' ' -f1)
		if [ -n "$libc" ]; then
			musl="-musl"
		fi
	fi
	arch="$(uname -m)"
	if [ "$arch" = x86_64 ]; then
		echo "x64$musl"
	elif [ "$arch" = aarch64 ] || [ "$arch" = arm64 ]; then
		echo "arm64$musl"
	elif [ "$arch" = armv6l ]; then
		echo "armv6$musl"
	elif [ "$arch" = armv7l ]; then
		echo "armv7$musl"
	else
		error "unsupported architecture: $arch"
	fi
}

shasum_bin() {
	if command -v shasum >/dev/null 2>&1; then
		echo "shasum"
	elif command -v sha256sum >/dev/null 2>&1; then
		echo "sha256sum"
	else
		error "mise install requires shasum or sha256sum but neither is installed. Aborting."
	fi
}

get_checksum() {
	os="$(get_os)"
	arch="$(get_arch)"

	checksum_linux_x86_64="b3c4f287d0ba4593c9b70faf7a6589690fb0d7e3cf49cc520e75f2b0fcdbd7b5  ./mise-v2024.5.3-linux-x64.tar.gz"
	checksum_linux_x86_64_musl="13ffdc995e26021c4f874383e1ab377ac77f2536ff3031ac7a14c945ef5c42e0  ./mise-v2024.5.3-linux-x64-musl.tar.gz"
	checksum_linux_arm64="c44f9ef4e9be505d9f05034c36e77700827db403aef5432b14dc4c8741c6569f  ./mise-v2024.5.3-linux-arm64.tar.gz"
	checksum_linux_arm64_musl="2353c0dc9ca84bbe48125a21d1f41b2478fe5ad455eb263ee3af8ba3a7cb75b3  ./mise-v2024.5.3-linux-arm64-musl.tar.gz"
	checksum_linux_armv6="fd4dd288722d5480644f0f8d53ecf46589cc0a7d8c587c57eb8cc681a3030ac9  ./mise-v2024.5.3-linux-armv6.tar.gz"
	checksum_linux_armv6_musl="18184149d3632388c8f255f9b48774897a505ebcf4e8d86d2bb1561d4bc0d35f  ./mise-v2024.5.3-linux-armv6-musl.tar.gz"
	checksum_linux_armv7="2bd3a99b8d2059f6f5c0c2fc570dd23a60ca1b66169cf4b759a8a0a7877863c3  ./mise-v2024.5.3-linux-armv7.tar.gz"
	checksum_linux_armv7_musl="64fbf4a852d3a56ffce9439070952c4c266a48f32bf43575f55468680c742f2e  ./mise-v2024.5.3-linux-armv7-musl.tar.gz"
	checksum_macos_x86_64="b0c882039d76c2d0da6d724913417984fd2ef02602296e05bd4c2d5026358fed  ./mise-v2024.5.3-macos-x64.tar.gz"
	checksum_macos_arm64="c39d63edc38179db3b0bde2f51f9bdfb49e6526968f4aae59571d7493ca59bc9  ./mise-v2024.5.3-macos-arm64.tar.gz"

	if [ "$os" = "linux" ]; then
		if [ "$arch" = "x64" ]; then
			echo "$checksum_linux_x86_64"
		elif [ "$arch" = "x64-musl" ]; then
			echo "$checksum_linux_x86_64_musl"
		elif [ "$arch" = "arm64" ]; then
			echo "$checksum_linux_arm64"
		elif [ "$arch" = "arm64-musl" ]; then
			echo "$checksum_linux_arm64_musl"
		elif [ "$arch" = "armv6" ]; then
			echo "$checksum_linux_armv6"
		elif [ "$arch" = "armv6-musl" ]; then
			echo "$checksum_linux_armv6_musl"
		elif [ "$arch" = "armv7" ]; then
			echo "$checksum_linux_armv7"
		elif [ "$arch" = "armv7-musl" ]; then
			echo "$checksum_linux_armv7_musl"
		else
			warn "no checksum for $os-$arch"
		fi
	elif [ "$os" = "macos" ]; then
		if [ "$arch" = "x64" ]; then
			echo "$checksum_macos_x86_64"
		elif [ "$arch" = "arm64" ]; then
			echo "$checksum_macos_arm64"
		else
			warn "no checksum for $os-$arch"
		fi
	else
		warn "no checksum for $os-$arch"
	fi
}

#endregion

download_file() {
	url="$1"
	filename="$(basename "$url")"
	cache_dir="$(mktemp -d)"
	file="$cache_dir/$filename"

	info "mise: installing mise..."

	if command -v curl >/dev/null 2>&1; then
		debug ">" curl -#fLo "$file" "$url"
		curl -#fLo "$file" "$url"
	else
		if command -v wget >/dev/null 2>&1; then
			debug ">" wget -qO "$file" "$url"
			stderr=$(mktemp)
			wget -O "$file" "$url" >"$stderr" 2>&1 || error "wget failed: $(cat "$stderr")"
		else
			error "mise standalone install requires curl or wget but neither is installed. Aborting."
		fi
	fi

	echo "$file"
}

install_mise() {
	# download the tarball
	version="v2024.5.3"
	os="$(get_os)"
	arch="$(get_arch)"
	install_path="${MISE_INSTALL_PATH:-$HOME/.local/bin/mise}"
	install_dir="$(dirname "$install_path")"
	tarball_url="https://github.com/jdx/mise/releases/download/${version}/mise-${version}-${os}-${arch}.tar.gz"

	cache_file=$(download_file "$tarball_url")
	debug "mise-setup: tarball=$cache_file"

	debug "validating checksum"
	cd "$(dirname "$cache_file")" && get_checksum | "$(shasum_bin)" -c >/dev/null

	# extract tarball
	mkdir -p "$install_dir"
	rm -rf "$install_path"
	cd "$(mktemp -d)"
	tar -xzf "$cache_file"
	mv mise/bin/mise "$install_path"
	info "mise: installed successfully to $install_path"
}

after_finish_help() {
	case "${SHELL:-}" in
	*/zsh)
		info "mise: run the following to activate mise in your shell:"
		info "echo \"eval \\\"\\\$($install_path activate zsh)\\\"\" >> \"${ZDOTDIR-$HOME}/.zshrc\""
		info ""
		info "mise: this must be run in order to use mise in the terminal"
		info "mise: run \`mise doctor\` to verify this is setup correctly"
		;;
	*/bash)
		info "mise: run the following to activate mise in your shell:"
		info "echo \"eval \\\"\\\$($install_path activate bash)\\\"\" >> ~/.bashrc"
		info ""
		info "mise: this must be run in order to use mise in the terminal"
		info "mise: run \`mise doctor\` to verify this is setup correctly"
		;;
	*/fish)
		info "mise: run the following to activate mise in your shell:"
		info "echo \"$install_path activate fish | source\" >> ~/.config/fish/config.fish"
		info ""
		info "mise: this must be run in order to use mise in the terminal"
		info "mise: run \`mise doctor\` to verify this is setup correctly"
		;;
	*)
		info "mise: run \`$install_path --help\` to get started"
		;;
	esac
}

install_mise
after_finish_help