#!/bin/sh -i
set -e
# Install node and npm via nvm - https://github.com/nvm-sh/nvm

# Run this script like - bash script-name.sh

# Tested on Ubuntu, MacOS
# @playground

if [[ $OSTYPE == 'darwin'* ]]
then
	echo 'MacOS'
	which -s brew
	if [[ $? != 0 ]]
	then
		# Install Homebrew
		ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"
	fi
	brew install socat jq
fi

ENV_SETUP=""
CONFIG_FILE=""
PS3='Choose your environment setup: '
envsetup=("Cli-And-Anax" "CLI-Only" "CLI-In-Container" "Anax-In-Container" "Run-In-Containers" "All-In-One" "Confirm" "Quit")
select fav in "${envsetup[@]}"; do
	case $fav in
		"Cli-And-Anax")
			echo "$fav runs both CLI & Agent on host machine, choose <Confirm> to continue setup."
			ENV_SETUP=$fav
			# optionally call a function or run some code here
			;;
		"CLI-Only")
			echo "$fav install CLI only on host, this option implies agent is running in a container, choose <Confirm> to continue setup."
			ENV_SETUP=$fav
			# optionally call a function or run some code here
			;;
		"CLI-In-Container")
			echo "$fav runs CLI in a container, this option implies agent is running in a container, choose <Confirm> to continue setup."
			ENV_SETUP=$fav
			# optionally call a function or run some code here
			;;
		"Anax-In-Container")
			echo "$fav, Runs Agent in container , choose <Confirm> to continue setup."
			ENV_SETUP=$fav
			# optionally call a function or run some code here
			;;
		"Run-In-Containers")
			echo "$fav, Runs both CLI and Agent in its own container, choose <Confirm> to continue setup."
			ENV_SETUP=$fav
			# optionally call a function or run some code here
			;;
		"All-In-One")
			echo "$fav, Runs CLI, Agent & Management Hub on the same machine, choose <Confirm> to continue setup."
			ENV_SETUP=$fav
			# optionally call a function or run some code here
			;;
		"Confirm")
			if [ "${ENV_SETUP}" = "" ]
			then
				echo "You have not chosen an env setup yet."
			else
				echo "You have chosen $ENV_SETUP"
				break
			fi
			# optionally call a function or run some code here
			;;
		"Quit")
	    echo "User requested exit"
	    exit
	    ;;
    *) echo "invalid option $REPLY";;
  esac
done

PS3='Continue with setup: '
configfile=("Config-File" "Confirm" "Help" "Quit")
select fav in "${configfile[@]}"; do
	case $fav in
		"Config-File")
			echo "Please provide absolute path to configuration file, then choose <Confirm> to continue setup."
			read CONFIG_FILE
			# optionally call a function or run some code here
			;;
		"Help")
			echo "You would need to provide your configuration json file with the following info."
			echo '\033[1;33m'
			echo '{'
			echo '  "org": {'
    	echo '    "HZN_ORG_ID": "examples",'
			echo '    "HZN_DEVICE_TOKEN": "",'
			echo '    "HZN_DEVICE_ID": "device-name",'
			echo '    "HZN_EXCHANGE_USER_AUTH": "************",'
			echo '    "HZN_EXCHANGE_URL": "http://xxx.xxx.xxx.xxx:3090/v1",'
			echo '    "HZN_FSS_CSSURL": "http://xxx.xxx.xxx.xxx:9443/",'
			echo '    "HZN_AGBOT_URL": "http://xxx.xxx.xxx.xxx:3111",'
			echo '    "HZN_SDO_SVC_URL": "http://xxx.xxx.xxx.xxx:9008/api",'
			echo '    "HZN_AGENT_PORT": "8510",'
			echo '    "HZN_CSS": false,'
			echo '    "ANAX": "https://github.com/open-horizon/anax/releases/latest/download/agent-install.sh"'  
			echo '  },'
			echo '  "service": {'
			echo '    "SERVICE_NAME": "chunk-saved-model-service",'
			echo '    "SERVICE_CONTAINER_NAME": "chunk-saved-model-service",'
			echo '    "SERVICE_VERSION": "1.0.0",'
			echo '    "SERVICE_VERSION_RANGE_UPPER": "1.0.0",'
			echo '    "SERVICE_VERSION_RANGE_LOWER": "1.0.0",'
			echo '    "SERVICE_CONTAINER_CREDS": "",'
			echo '    "VOLUME_MOUNT": "/mms-shared",'
			echo '    "MMS_SHARED_VOLUME": "mms_shared_volume",'
			echo '    "MMS_OBJECT_TYPE": "chunk_object_detection",'
			echo '    "MMS_OBJECT_ID": "chunk_config.json",'
			echo '    "MMS_OBJECT_FILE": "config/config.json",'
			echo '    "MMS_CONTAINER_CREDS": "",'
			echo '    "MMS_CONTAINER_NAME": "chunk-mms-service",'
			echo '    "MMS_SERVICE_NAME": "chunk-mms-service",'
			echo '    "MMS_SERVICE_VERSION": "1.0.1",'
			echo '    "MMS_SERVICE_FALLBACK_VERSION": "1.0.0",'
			echo '    "UPDATE_FILE_NAME": "model.zip"'
			echo '  }'
			echo '  "folders": ['
			echo '    "/var/tmp/horizon/horizon1/fss-domain-socket",'
			echo '    "/var/tmp/horizon/horizon1/ess-auth",'
			echo '    "/var/tmp/horizon/horizon1/secrets",'
			echo '    "/var/tmp/horizon/horizon1/nmp"'
			echo '  ],'
			echo '  "local": {'
			echo '    "YOUR_DOCKERHUB_ID": "dockerid",'
			echo '    "DOCKER_REGISTRY": "hub.docker.com",'
			echo '    "DOCKER_TOKEN": "dckr_pat_w......"'
			echo '  },'
			echo '  "anaxInContainer": "docker run -d -t --restart always --name horizon1 --privileged -p 127.0.0.1:8081:8510 -e DOCKER_NAME=horizon1 -e HZN_VAR_RUN_BASE=/var/tmp/horizon/horizon1 -v /var/run/docker.sock:/var/run/docker.sock -v /var/horizon:/etc/default/horizon:ro -v /var/agent-install.crt:/var/agent-install.crt -v horizon1_var:/var/horizon/ -v horizon1_etc:/etc/horizon/ -v /var/tmp/horizon/horizon1:/var/tmp/horizon/horizon1 openhorizon/amd64_anax:2.30.0-952",'
			echo '  "cliInContainer": "docker run -d -it --restart always --name hzn-cli --privileged --network=\"host\" -v /var/run/docker.sock:/var/run/docker.sock -v /var/agent-install.crt:/var/agent-install.crt -e HORIZON_URL=http://localhost:8081 -e HZN_ORG_ID=${HZN_ORG_ID} -e HZN_EXCHANGE_USER_AUTH=${HZN_EXCHANGE_USER_AUTH} -e HZN_FSS_CSSURL=${HZN_FSS_CSSURL} -e HZN_EXCHANGE_URL=${HZN_EXCHANGE_URL} -e version=v2.30.0-952 playbox21/hzn-cli_amd64"'
			echo '}'
			echo '\033[0m'
			;;
		"Confirm")
			if [ "${CONFIG_FILE}" = "" ]
			then
				echo "You have not provided your configuration file."
			else
				echo "You have provided $CONFIG_FILE for this setup"
				break
			fi
			# optionally call a function or run some code here
			;;
		"Quit")
	    echo "User requested exit"
	    exit
	    ;;
    *) echo "invalid option $REPLY";;
  esac
done

if [[ -f "${HOME}/.nvm/nvm.sh" ]]; then
  export NVM_DIR="$HOME/.nvm"
  [ -s "$NVM_DIR/nvm.sh" ] && \. "$NVM_DIR/nvm.sh"  # This loads nvm
  [ -s "$NVM_DIR/bash_completion" ] && \. "$NVM_DIR/bash_completion"  # This loads nvm bash_completion
else echo "nvm not installed"
	# Define versions
	INSTALL_NODE_VER=16
	INSTALL_NVM_VER=0.39.1

	# You can pass argument to this script --version 8
	if [ "$1" = '--version' ]; then
		echo "==> Using specified node version - $2"
		INSTALL_NODE_VER=$2
	fi

	echo "==> Ensuring .bashrc exists and is writable"
	touch ~/.bashrc

	echo "==> Installing node version manager (NVM). Version $INSTALL_NVM_VER"
	# Removed if already installed
	rm -rf ~/.nvm
	# Unset exported variable
	export NVM_DIR=

	# Install nvm 
	curl -sSL https://raw.githubusercontent.com/nvm-sh/nvm/v$INSTALL_NVM_VER/install.sh --output install.sh && chmod 777 install.sh && /bin/sh install.sh
	# Make nvm command available to terminal
	source ~/.nvm/nvm.sh

	echo "==> Installing node js version $INSTALL_NODE_VER"
	nvm install $INSTALL_NODE_VER

	echo "==> Make this version system default"
	nvm alias default $INSTALL_NODE_VER
	nvm use default

	echo "==> Checking for versions"
	nvm --version

	echo "==> List installed node versions"
	nvm ls

	nvm cache clear
	echo "==> Now you're all setup and ready for development. If changes are yet to take effect in the current shell, try source ~/.bashrc or open a new shell"

fi

echo "==> Print binary paths"
which npm
which node

echo "==> Checking for versions"
node --version
npm --version

echo -e "==> Update npm to latest version, if this stuck then terminate (CTRL+C) the execution"
npm install -g npm

echo "==> Installing oh cli"
npm i -g hzn-cli

echo "==> Checking oh version"
oh --version

echo "==> Setting up hzn environment..."
sudo touch /etc/default/horizon

if [ "${ENV_SETUP}" = "Cli-And-Anax" ]
then
	echo "$ENV_SETUP, here we go."
	oh deploy autoSetup --config_file ${CONFIG_FILE}
elif [ "${ENV_SETUP}" = "CLI-Only" ]
then
	echo "$ENV_SETUP, here we go."
	oh deploy autoSetupCliOnly --config_file ${CONFIG_FILE}
elif [ "${ENV_SETUP}" = "CLI-In-Container" ]
then
	echo "$ENV_SETUP, here we go."
	oh deploy autoSetupCliInContainer --config_file ${CONFIG_FILE}
elif [ "${ENV_SETUP}" = "Anax-In-Container" ]
then
	echo "$ENV_SETUP, here we go."
	oh deploy autoSetupAnaxInContainer --config_file ${CONFIG_FILE}
elif [ "${ENV_SETUP}" = "Run-In-Containers" ]
then
	echo "$ENV_SETUP, here we go."
	oh deploy autoSetupContainer --config_file ${CONFIG_FILE}
elif [ "${ENV_SETUP}" = "All-In-One" ]
then
	echo "$ENV_SETUP, here we go."
	oh deploy autoSetupAllInOne --config_file ${CONFIG_FILE}
else
	echo "Something went wrong...$ENV_SETUP"
	break
fi