#!/bin/bash

OS=""
ARCH=""
BASE_URL="https://github.com/Privado-Inc/privado-cli/releases/latest/download/privado-"
REPO_URL="https://github.com/Privado-Inc/privado-cli/"

function findOS {
    if [[ "$OSTYPE" == "linux-gnu"* ]]; then
        OS="linux"
    elif [[ "$OSTYPE" == "darwin"* ]]; then
        OS="darwin"
    elif [[ "$OSTYPE" == "msys" ]]; then 
        OS="windows"
    else
	echo "Unsupported OS"
	exit 10
    fi
}

function findArch {
    ARCH_STR=$(uname -m)
    if [[ "$ARCH_STR" == "x86_64" ]]; then
	ARCH="amd64"
    elif [[ "$ARCH_STR" == "arm64" ]]; then
	ARCH="arm64"
    else
	echo "Unsupported Architecture"
        exit 10
    fi
}

function downloadAndInstallLatestVersion {
    if [[ "$OS" == "" || "$ARCH" == "" ]]; then
        echo "Unsupported OS or Arch type. Please visit $REPO_URL for more information."
        exit 1
    fi

    mkdir -p $HOME/.privado/bin
    if [[ "$OS" == "windows" ]]; then
	echo "Downloading binary:"
	curl -L "$BASE_URL$OS-$ARCH.zip" -o /tmp/privado-$OS-$ARCH.zip
	echo "Downloading binary MD5:"
	curl -L "$BASE_URL$OS-$ARCH.zip.md5" -o /tmp/privado-$OS-$ARCH.zip.md5
	MD5_ACTUAL=$(certutil -hashfile /tmp/privado-$OS-$ARCH.zip MD5 | head -n2 | tail -n1)
	MD5_EXPECTED=$(cat /tmp/privado-$OS-$ARCH.zip.md5)
    else
	echo "Downloading binary:"
	curl -L "$BASE_URL$OS-$ARCH.tar.gz" -o /tmp/privado-$OS-$ARCH.tar.gz
	echo "Downloading binary MD5:"
	curl -L "$BASE_URL$OS-$ARCH.tar.gz.md5" -o /tmp/privado-$OS-$ARCH.tar.gz.md5
	if [[ "$OS" == "linux" ]]; then
		MD5_ACTUAL=$(md5sum /tmp/privado-$OS-$ARCH.tar.gz | awk '{print $1}')
	else
		MD5_ACTUAL=$(md5 /tmp/privado-$OS-$ARCH.tar.gz | awk '{print $4}')
	fi
	MD5_EXPECTED=$(cat /tmp/privado-$OS-$ARCH.tar.gz.md5)
    fi

    if [[ "$MD5_EXPECTED" != "$MD5_ACTUAL" ]]; then
	echo "Error in downloading the file. Please retry"
	exit 10
    fi

    if [[ "$OS" == "windows" ]]; then
	unzip -o /tmp/privado-$OS-$ARCH.zip -d $HOME/.privado/bin
    elif [[ "$OS" == "darwin" ]]; then
        tar -xf /tmp/privado-$OS-$ARCH.tar.gz -C $HOME/.privado/bin
    else
	tar -xf /tmp/privado-$OS-$ARCH.tar.gz -C $HOME/.privado/bin
    fi

    WHO_AM_I=$(whoami)

    if [[ "$WHO_AM_I" == "root" ]]; then
		cp $HOME/.privado/bin/privado /usr/local/bin/privado
		chmod 755 /usr/local/bin/privado
		echo "Installation is complete! Run 'privado' to use Privado CLI"
    else
    	DEFAULT_PROFILE_PATH="$HOME/.bashrc"
    	NO_FILE=""
    	for EACH_PROFILE in ".profile" ".bash_profile" ".zprofile" ".zshrc"
    	do
      		if [[ -f $HOME/$EACH_PROFILE ]]; then
				cat $HOME/$EACH_PROFILE | grep -q "/.privado" || echo "export PATH=\$PATH:$HOME/.privado/bin" >> $HOME/$EACH_PROFILE
				NO_FILE="true"
      		fi
    	done
    
    	if [[ "$NO_FILE" == "" ]] || [[ "$OS" == "linux"  ]]; then
        	cat $DEFAULT_PROFILE_PATH | grep -q "/.privado" || echo "export PATH=\$PATH:$HOME/.privado/bin" >> $DEFAULT_PROFILE_PATH
    	fi
	
		echo "Installation is complete! Please open a new session and run 'privado' to use Privado CLI"
    fi
}

function checkDocker {
	docker ps > /dev/null 2>&1
	EXIT_CODE=$?
	if [[ "$EXIT_CODE" != "0" ]]; then
		echo "> Preflight checks failed!"
		echo "> Either Docker is not installed, not running, or you do not have appropriate permissions to use the same. Please retry this script with sudo privileges."
		exit
	fi	
}


function preFlightChecks {
	checkDocker
}

function cleanup {
	rm -f /tmp/privado-*
}

findOS
findArch
preFlightChecks
downloadAndInstallLatestVersion
cleanup