#!/usr/bin/env bash

{ # Prevent script from running if partially downloaded

	set -euo pipefail

	DOTPATH=${HOME}/.dotfiles
	BRANCH=""
	YES=false
	while getopts b:y flag; do
		case "${flag}" in
		b) BRANCH=${OPTARG} ;;
		y) YES=true ;;
		*) echo "Invalid option: -${OPTARG}" && exit 1 ;;
		esac
	done

	echo -e "🥾 \033[1;34mBootstrapping dotfiles\033[0m"

	if [[ ! -d ${DOTPATH} ]]; then
		if [[ -z ${BRANCH} ]]; then
			echo -e "📑 \033[1;33mCloning dotfiles...\033[0m"
			git clone https://github.com/martimlobao/dotfiles.git "${DOTPATH}"
			echo -e "✅ \033[1;32mCloned Dotfiles to ${DOTPATH}\033[0m"
		else
			echo -e "📑 \033[1;33mCloning dotfiles on branch ${BRANCH}...\033[0m"
			git clone https://github.com/martimlobao/dotfiles.git --branch "${BRANCH}" "${DOTPATH}"
			echo -e "✅ \033[1;32mCloned Dotfiles to ${DOTPATH} on branch ${BRANCH}\033[0m"
		fi
	else
		if [[ -z ${BRANCH} ]]; then
			echo -e "✅ \033[1;34mDotfiles already downloaded to ${DOTPATH}\033[0m"
		else
			echo -e "✅ \033[1;34mDotfiles already downloaded to ${DOTPATH}, checking out branch ${BRANCH}\033[0m"
			cd "${DOTPATH}"
			git stash
			git checkout "${BRANCH}"
			git pull origin "${BRANCH}"
		fi
	fi

	cd "${DOTPATH}"

	if [[ ${YES} == true ]]; then
		./run.sh -y
	else
		./run.sh
	fi

} # Prevent script from running if partially downloaded