snnrb

#!/bin/bash

# export PATH="$PWD:$PATH"

## declare the list of github repositories in an array.
declare -a arr=("snnadaptation" "snnalgorithms" "snnbackends" "snncompare" "snnradiation") #"simsnn")
# Specify global variables.
# DIR_WITH_REPOS="/home/$(echo $(whoami))/git/snn/"
DIR_WITH_REPOS="$( cd -- "$(dirname "$0")" >/dev/null 2>&1 ; pwd -P )"
DIR_WITH_REPOS="$DIR_WITH_REPOS/../"
CONDA_ENVIRONMENT_NAME="snncompare"


while [[ "$#" -gt 0 ]]; do
    case $1 in
        -b|--branch)
		    CHECKOUT_BRANCH=1;
			BRANCH_NAME="$2"
      		shift # past argument
      		shift # past value
      		;;
		-nb|--new-branch)
			CHECKOUT_NEW_BRANCH=1;
			BRANCH_NAME="$2"
      		shift # past argument
      		shift # past value
      		;;
		-r|--rebuild) rebuild=1; ;;
	    -p|--pull) pull=1 ;;
		-pc|--precommit) precommit=1 ;;
		-pu|--precommit-update) precommit_update=1 ;;
		-pi|--precommit-install) precommit_install=1 ;;
		-pip|--publish-to-pip) publish_to_pip=1 ;;
		-c|--commitpush)
			commitpush=1
			COMMIT_MESSAGE="$2"
      		shift # past argument
      		shift # past value
      		;;
        *) echo "Unknown parameter passed: $1"; exit 1 ;;
    esac
    shift
done

#######################################
# Verifies a directory exists, throws error otherwise.
# Local variables:
#  dirpath
# Globals:
#  None.
# Arguments:
#  Relative folderpath of folder whose existence is verified.
# Returns:
#  0 If folder was found.
#  31 If the folder was not found.
# Outputs:
#  Nothing
#######################################
manual_assert_dir_exists() {
	local dirpath="$1"
	if [ ! -d "$dirpath" ]; then
		echo "The dir: $dirpath does not exist, even though one would expect it does."
		exit 31
	fi
}


#######################################
#
# Local variables:
#
# Globals:
#
# Arguments:
#
# Returns:
#  0 If function was evaluated successful.
# Outputs:
#
# TODO(a-t-0):
#######################################
# Source: https://stackoverflow.com/questions/70597896/check-if-conda-env-exists-and-create-if-not-in-bash
find_in_conda_env(){
    conda env list | grep "${@}" >/dev/null 2>/dev/null
}


#######################################
#
# Local variables:
#
# Globals:
#
# Arguments:
#
# Returns:
#  0 If function was evaluated successful.
# Outputs:
#
# TODO(a-t-0):
#######################################
conda_env_exists() {
	local some_envirionment="$1"
	if find_in_conda_env ".*$some_envirionment.*" ; then
   		echo "FOUND"
	else
		echo "NOTFOUND"
	fi
}


run_precommit() {
    local git_path="$1"
    if [ "$(conda_env_exists $CONDA_ENVIRONMENT_NAME)" == "FOUND" ]; then
		eval "$(conda shell.bash hook)"
		cd "$git_path" && conda deactivate && conda activate snncompare && git add *
		cd "$git_path" && conda deactivate && conda activate snncompare && git add .* && pre-commit run --all-files
		cd "$git_path" && conda deactivate && conda activate snncompare && pre-commit install
		cd "$git_path" && conda deactivate && conda activate snncompare && pre-commit run --all-files
	else
		echo "Error, conda environment name:$CONDA_ENVIRONMENT_NAME not found."
        exit 5
	fi
}

run_precommit_install() {
    local git_path="$1"
    if [ "$(conda_env_exists $CONDA_ENVIRONMENT_NAME)" == "FOUND" ]; then
	eval "$(conda shell.bash hook)"
	cd "$git_path" && conda deactivate && conda activate snncompare && pre-commit install
    else
	echo "Error, conda environment name:$CONDA_ENVIRONMENT_NAME not found."
        exit 5
    fi
}

run_precommit_update() {
    local git_path="$1"
    if [ "$(conda_env_exists $CONDA_ENVIRONMENT_NAME)" == "FOUND" ]; then
		eval "$(conda shell.bash hook)"
		cd "$git_path" && conda deactivate && conda activate snncompare && pre-commit autoupdate
	else
		echo "Error, conda environment name:$CONDA_ENVIRONMENT_NAME not found."
        exit 5
	fi
}


git_pull() {
	local git_path="$1"
    if [ "$(conda_env_exists $CONDA_ENVIRONMENT_NAME)" == "FOUND" ]; then
		eval "$(conda shell.bash hook)"
		cd "$git_path" && conda deactivate && conda activate snncompare && git pull
	else
		echo "Error, conda environment name:$CONDA_ENVIRONMENT_NAME not found."
        exit 5
	fi
}

checkout_new_branch() {
    local git_path="$1"
	local branch_name="$2"
	local new_branch="$3"
    if [ "$(conda_env_exists $CONDA_ENVIRONMENT_NAME)" == "FOUND" ]; then
		eval "$(conda shell.bash hook)"
		if [ "$new_branch" == 1 ]; then
			cd "$git_path" && conda deactivate && conda activate snncompare && git checkout -b "$branch_name"
			cd "$git_path" && conda deactivate && conda activate snncompare && git add -A
			cd "$git_path" && conda deactivate && conda activate snncompare && git commit -m "$message"
			cd "$git_path" && conda deactivate && conda activate snncompare && git push --set-upstream origin "$branch_name"
		else
			cd "$git_path" && conda deactivate && conda activate snncompare && git checkout "$branch_name"
			cd "$git_path" && conda deactivate && conda activate snncompare && git pull
		fi
	else
		echo "Error, conda environment name:$CONDA_ENVIRONMENT_NAME not found."
        exit 5
	fi
}

commit_and_push() {
    local git_path="$1"
	local message="$2"
    if [ "$(conda_env_exists $CONDA_ENVIRONMENT_NAME)" == "FOUND" ]; then
		eval "$(conda shell.bash hook)"
		cd "$git_path" && conda deactivate && conda activate snncompare && git add *
		cd "$git_path" && conda deactivate && conda activate snncompare && git add -A
		cd "$git_path" && conda deactivate && conda activate snncompare && git commit -m "$message"
		cd "$git_path" && conda deactivate && conda activate snncompare && git push
	else
		echo "Error, conda environment name:$CONDA_ENVIRONMENT_NAME not found."
        exit 5
	fi
}

build_pip_package() {
    local git_path="$1"
    if [ "$(conda_env_exists $CONDA_ENVIRONMENT_NAME)" == "FOUND" ]; then
		eval "$(conda shell.bash hook)"
		cd "$git_path" && conda deactivate && conda activate snncompare && python3 setup.py sdist bdist_wheel
	else
		echo "Error, conda environment name:$CONDA_ENVIRONMENT_NAME not found."
        exit 5
	fi
}

publish_to_pip() {
    local git_path="$1"
    if [ "$(conda_env_exists $CONDA_ENVIRONMENT_NAME)" == "FOUND" ]; then
		eval "$(conda shell.bash hook)"
		cd "$git_path" && conda deactivate && conda activate snncompare && python3 setup.py sdist bdist_wheel
		cd "$git_path" && conda deactivate && conda activate snncompare && python -m twine upload dist/\*
	else
		echo "Error, conda environment name:$CONDA_ENVIRONMENT_NAME not found."
        exit 5
	fi
}

install_pip_package() {
    local git_path="$1"
    if [ "$(conda_env_exists $CONDA_ENVIRONMENT_NAME)" == "FOUND" ]; then
		eval "$(conda shell.bash hook)"
		cd "$git_path" && conda deactivate && conda activate snncompare && pip install -e .
	else
		echo "Error, conda environment name:$CONDA_ENVIRONMENT_NAME not found."
        exit 5
	fi
}

for reponame in "${arr[@]}"
do
	echo ""
	echo ""
	echo "$reponame"

	# Assert the required folders exist.
	manual_assert_dir_exists "$DIR_WITH_REPOS$reponame"

	if [ "$rebuild" == 1 ]; then
		# Build the pip package
   		build_pip_package "$DIR_WITH_REPOS$reponame"

   		# Install package locally
   		install_pip_package "$DIR_WITH_REPOS$reponame"
	fi

	if [ "$CHECKOUT_BRANCH" == 1 ] || [ "$CHECKOUT_NEW_BRANCH" == 1 ]; then
		checkout_new_branch "$DIR_WITH_REPOS$reponame" "$BRANCH_NAME" "$CHECKOUT_NEW_BRANCH"
	fi

	if [ "$precommit" == 1 ]; then
		run_precommit "$DIR_WITH_REPOS$reponame"
	fi

	if [ "$precommit_install" == 1 ]; then
		run_precommit_install "$DIR_WITH_REPOS$reponame"
	fi

	if [ "$precommit_update" == 1 ]; then
		run_precommit_update "$DIR_WITH_REPOS$reponame"
		if [ "$reponame" == "snncompare" ]; then
			if [ "$(conda_env_exists $CONDA_ENVIRONMENT_NAME)" == "FOUND" ]; then
				cd "$DIR_WITH_REPOS$reponame" && conda deactivate
				conda env update --file environment.yml
				cd "$DIR_WITH_REPOS$reponame" && conda deactivate && conda activate snncompare
			fi
		fi
	fi

	if [ "$publish_to_pip" == 1 ]; then
		publish_to_pip "$DIR_WITH_REPOS$reponame"
	fi

	if [ "$pull" == 1 ]; then
		git_pull "$DIR_WITH_REPOS$reponame"
	fi

	if [ "$commitpush" == 1 ]; then
		commit_and_push "$DIR_WITH_REPOS$reponame" "$COMMIT_MESSAGE"
	fi

   	# TODO: if CLI arg is passed, upload the pip packages with upgraded version.
done