#!/bin/bash

# Copyright 2021 Anthony Perkins
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

if [ x$1 == x ]; then
	install_path=$1
	if [ $(id -u) -eq 0 ]; then
		bin_path=/usr/local/bin
	else
		bin_path="$HOME/.local/bin"
	fi
elif [ $(id -u) -eq 0 ]; then
	install_path=/opt/ansible
	bin_path=/usr/local/bin
else
	install_path="$HOME/opt/ansible"
	bin_path="$HOME/.local/bin"
fi

pipinstall () {
	./bin/pip install --upgrade $1
}

create_link () {
	ln -srf "$install_path/bin/$1" "$bin_path/$1"
}

py3cmd="$(command -v python3)"
if [ ! -x "$py3cmd" ]; then
	echo "python3 not found"
	exit 1
else
	echo "python3 found: $py3cmd"
fi

if ! $($py3cmd -c "import venv" 2> /dev/null); then
	echo "Python3 venv module missing. Try installing python3-venv package."
	exit 1
fi

if ! $($py3cmd -c "import ensurepip" 2> /dev/null); then
	echo "Python3 ensurepip module missing. Try installing python3-venv package."
	exit 1
fi

if [ ! -d "$install_path" ]; then
	python3 -m venv "$install_path" || exit 1
fi

pushd "$install_path"
pipinstall wheel
pipinstall pip
pipinstall ansible
popd

if [ ! -d "$bin_path" ]; then
	mkdir "$bin_path"
fi

for command in \
	ansible \
	ansible-config \
	ansible-connection \
	ansible-console \
	ansible-doc \
	ansible-galaxy \
	ansible-inventory \
	ansible-playbook \
	ansible-pull \
	ansible-test \
	ansible-vault; do
	create_link $command
done