#!/bin/sh
##===----------------------------------------------------------------------===##
#
# This file is Modular Inc proprietary.
#
##===----------------------------------------------------------------------===##

set -e

maybe_sudo() {
	if [ "$(whoami)" = "root" ]; then
		"$@"
	elif type sudo >/dev/null; then
		sudo -E "$@"
	else
		echo "Sorry, either root or 'sudo' is required."
		return 1
	fi
}

remote_script() {
	# Prefer curl over wget, but wget is available by default. We shouldn't need
	# any fallback path where packages are installed. Note that for older shells
	# we do not have pipefail available, therefore ensure that on failure we pipe
	# in a failure that will result in failure of the function.
	if type curl >/dev/null; then
		(curl -1sLf "$1" || echo "exit 1") | maybe_sudo bash
	elif type wget >/dev/null; then
		(wget -q -O - "$1" || echo "exit 1") | maybe_sudo bash
	else
		echo "Sorry, one of 'curl' or 'wget' is required."
		return 1
	fi
}

# The main function of the script. Wrap in a function so an imcomplete download
# will not result in a partially executed script.
__main__() {
	# If the user is using WSL1, tell them that they should use WSL2 instead.
	# "Microsoft" in the uname is captialized on WSL1 and lowercase on WSL2. This is
	# a semi-reliable way to detect WSL1.
	if [ "$(uname -r | grep -c Microsoft)" -eq 1 ]; then
		echo "It looks like you are using WSL1 which is unsupported by Modular."
		echo "We recommend switching your Linux distribution to run with WSL2 instead."
		echo "See https://learn.microsoft.com/en-us/windows/wsl/install#upgrade-version-from-wsl-1-to-wsl-2 to learn how to switch to WSL2."
		exit 1
	fi

	# Reliably detecting whether a system is Deb-based or RPM-based can be tricky,
	# as both tools can be installed on a system. Complex cases may need to fall
	# back to manual setup, however it should be the case Debian-deriviatives have
	# /etc/debian_version, while this file should not be present in
	# non-Debian-based systems.
	if [ "$(uname)" = "Linux" ] && [ -f /etc/debian_version ]; then
		remote_script "https://dl.modular.com/${URL_SLUG:-public/installer}/setup.deb.sh"
		maybe_sudo apt install -yq --reinstall modular
	elif [ "$(uname)" = "Linux" ] && type rpm >/dev/null && type dnf >/dev/null; then
		remote_script "https://dl.modular.com/${URL_SLUG:-public/installer}/setup.rpm.sh"
		maybe_sudo dnf -yq install modular ||
			(maybe_sudo dnf -q list installed modular >/dev/null 2>&1 &&
				maybe_sudo dnf -yq reinstall modular)
	elif [ "$(uname)" = "Linux" ] && type rpm >/dev/null && type yum >/dev/null; then
		remote_script "https://dl.modular.com/${URL_SLUG:-public/installer}/setup.rpm.sh"
		maybe_sudo yum -yq install modular ||
			(maybe_sudo yum -q list installed modular >/dev/null 2>&1 &&
				maybe_sudo yum -yq reinstall modular)
	elif [ "$(uname)" = "Darwin" ]; then
		UNAME_MACHINE="$(/usr/bin/uname -m)"
		if [ "${UNAME_MACHINE}" = "arm64" ]; then
			if [ -z "$(command -v brew)" ]; then
				echo "Homebrew not detected. Please install homebrew from https://brew.sh/"
				exit 1
			else
				echo "Updating Homebrew"
				brew update
				echo "Installing/upgrading Modular using Homebrew"
				if brew ls --versions modular >/dev/null; then
					# if modular is already installed, lets upgrade it.
					brew upgrade modularml/packages/modular
				else
					brew install modularml/packages/modular
				fi
			fi
		else
			echo "Sorry, modular is not currently supported on Intel based Macs. Please visit https://www.modular.com/ to learn about supported platforms."
			exit 1
		fi
	else
		echo "Sorry, this system is not recognized. Please visit https://www.modular.com/ to learn about supported platforms."
		exit 1
	fi

	if ! [ -z "$1" ]; then
		modular config-set web.id $1
	fi

	cat <<EOF
__  __           _       _
|  \/  | ___   __| |_   _| | __ _ _ __
| |\/| |/ _ \ / _\` | | | | |/ _\` | '__|
| |  | | (_) | (_| | |_| | | (_| | |
|_|  |_|\___/ \__,_|\__,_|_|\__,_|_|

Welcome to the Modular CLI!

To get started, see:
MAX  - https://modul.ar/install-max
Mojo - https://modul.ar/install-mojo

For info about this tool, type "modular --help".
EOF

}
__main__ $1 || exit 1