#!/bin/sh

venv() {
	if ! command -V "venv_$1" | grep -q " function "
	then
		cat <<EOF
Usage:
	venv on
	venv off

Manage the python venv for a given project or directory.

Commands:
	on	Enable the venv for \$PWD
	off	Deactivate the current venv
EOF

		return 1
	fi

	"venv_$1"
}

venv_on() {
	export PYTHON_VENV_DIR="$PWD/.venv"
	export VIRTUAL_ENV_DISABLE_PROMPT=1

	if [ ! -d "$PYTHON_VENV_DIR" ]
	then
		printf "Initializing venv directory at %s\n" "$PYTHON_VENV_DIR"
		python3 -m venv "$PYTHON_VENV_DIR"
	fi

	. "$PYTHON_VENV_DIR/bin/activate"
}

venv_off() {
	if [ -z "$PYTHON_VENV_DIR" ]
	then
		printf "No venv active?\n"
		return 1
	fi

	deactivate

	unset PYTHON_VENV_DIR
	unset VIRTUAL_ENV_DISABLE_PROMPT
}