setup-env

#!/usr/bin/env bash

set -o nounset
set -o errexit
set -o pipefail

USAGE=$(
  cat << 'END_OF_LINE'
Configure a development environment for this repository.

It does the following:
  - Allows the user to specify the Python version to use for the virtual environment.
  - Allows the user to specify a name for the virtual environment.
  - Verifies pyenv and pyenv-virtualenv are installed.
  - Creates the Python virtual environment.
  - Configures the activation of the virtual enviroment for the repo directory.
  - Installs the requirements needed for development (including mypy type stubs).
  - Installs git pre-commit hooks.
  - Configures git remotes for upstream "lineage" repositories.

Usage:
  setup-env [--venv-name venv_name] [--python-version python_version]
  setup-env (-h | --help)

Options:
  -f | --force             Delete virtual enviroment if it already exists.
  -h | --help              Show this message.
  -i | --install-hooks     Install hook environments for all environments in the
                           pre-commit config file.
  -l | --list-versions     List available Python versions and select one interactively.
  -v | --venv-name         Specify the name of the virtual environment.
  -p | --python-version    Specify the Python version for the virtual environment.

END_OF_LINE
)

# Display pyenv's installed Python versions
python_versions() {
  pyenv versions --bare --skip-aliases --skip-envs
}

# Flag to force deletion and creation of virtual environment
FORCE=0

# Initialize the other flags
INSTALL_HOOKS=0
LIST_VERSIONS=0
PYTHON_VERSION=""
VENV_NAME=""

# Define long options
LONGOPTS="force,help,install-hooks,list-versions,python-version:,venv-name:"

# Define short options for getopt
SHORTOPTS="fhilp:v:"

# Check for GNU getopt by matching a specific pattern ("getopt from util-linux")
# in its version output. This approach presumes the output format remains stable.
# Be aware that format changes could invalidate this check.
if [[ $(getopt --version 2> /dev/null) != *"getopt from util-linux"* ]]; then
  cat << 'END_OF_LINE'

  Please note, this script requires GNU getopt due to its enhanced
  functionality and compatibility with certain script features that
  are not supported by the POSIX getopt found in some systems, particularly
  those with a non-GNU version of getopt. This distinction is crucial
  as a system might have a non-GNU version of getopt installed by default,
  which could lead to unexpected behavior.

  On macOS, we recommend installing brew (https://brew.sh/).  Then installation
  is as simple as `brew install gnu-getopt` and adding this to your
  profile:

  export PATH="$(brew --prefix)/opt/gnu-getopt/bin:$PATH"

  GNU getopt must be explicitly added to the PATH since it
  is keg-only (https://docs.brew.sh/FAQ#what-does-keg-only-mean).

END_OF_LINE
  exit 1
fi

# Check to see if pyenv is installed
if [ -z "$(command -v pyenv)" ] || { [ -z "$(command -v pyenv-virtualenv)" ] && [ ! -f "$(pyenv root)/plugins/pyenv-virtualenv/bin/pyenv-virtualenv" ]; }; then
  echo "pyenv and pyenv-virtualenv are required."
  if [[ "$OSTYPE" == "darwin"* ]]; then
    cat << 'END_OF_LINE'

  On macOS, we recommend installing brew, https://brew.sh/.  Then installation
  is as simple as `brew install pyenv pyenv-virtualenv` and adding this to your
  profile:

  eval "$(pyenv init -)"
  eval "$(pyenv virtualenv-init -)"

END_OF_LINE

  fi
  cat << 'END_OF_LINE'
  For Linux, Windows Subsystem for Linux (WSL), or macOS (if you don't want
  to use "brew") you can use https://github.com/pyenv/pyenv-installer to install
  the necessary tools. Before running this ensure that you have installed the
  prerequisites for your platform according to the pyenv wiki page,
  https://github.com/pyenv/pyenv/wiki/common-build-problems.

  On WSL you should treat your platform as whatever Linux distribution you've
  chosen to install.

  Once you have installed "pyenv" you will need to add the following lines to
  your ".bashrc":

  export PATH="$PATH:$HOME/.pyenv/bin"
  eval "$(pyenv init -)"
  eval "$(pyenv virtualenv-init -)"
END_OF_LINE
  exit 1
fi

# Use GNU getopt to parse options
if ! PARSED=$(getopt --options $SHORTOPTS --longoptions $LONGOPTS --name "$0" -- "$@"); then
  echo "Error parsing options"
  exit 1
fi
eval set -- "$PARSED"

while true; do
  case "$1" in
    -f | --force)
      FORCE=1
      shift
      ;;
    -h | --help)
      echo "$USAGE"
      exit 0
      ;;
    -i | --install-hooks)
      INSTALL_HOOKS=1
      shift
      ;;
    -l | --list-versions)
      LIST_VERSIONS=1
      shift
      ;;
    -p | --python-version)
      PYTHON_VERSION="$2"
      shift 2
      # Check the Python versions being passed in.
      if [ -n "${PYTHON_VERSION+x}" ]; then
        if python_versions | grep -E "^${PYTHON_VERSION}$" > /dev/null; then
          echo Using Python version "$PYTHON_VERSION"
        else
          echo Error: Python version "$PYTHON_VERSION" is not installed.
          echo Installed Python versions are:
          python_versions
          exit 1
        fi
      fi
      ;;
    -v | --venv-name)
      VENV_NAME="$2"
      shift 2
      ;;
    --)
      shift
      break
      ;;
    *)
      # Unreachable due to GNU getopt handling all options
      echo "Programming error"
      exit 64
      ;;
  esac
done

# Determine the virtual environment name
if [ -n "$VENV_NAME" ]; then
  # Use the user-provided environment name
  env_name="$VENV_NAME"
else
  # Set the environment name to the last part of the working directory.
  env_name=${PWD##*/}
fi

# List Python versions and select one interactively.
if [ $LIST_VERSIONS -ne 0 ]; then
  echo Available Python versions:
  python_versions
  # Read the user's desired Python version.
  # -r: treat backslashes as literal, -p: display prompt before input.
  read -r -p "Enter the desired Python version: " PYTHON_VERSION
  # Check the Python versions being passed in.
  if [ -n "${PYTHON_VERSION+x}" ]; then
    if python_versions | grep -E "^${PYTHON_VERSION}$" > /dev/null; then
      echo Using Python version "$PYTHON_VERSION"
    else
      echo Error: Python version "$PYTHON_VERSION" is not installed.
      exit 1
    fi
  fi
fi

# Remove any lingering local configuration.
if [ $FORCE -ne 0 ]; then
  rm -f .python-version
  pyenv virtualenv-delete --force "${env_name}" || true
elif [[ -f .python-version ]]; then
  cat << 'END_OF_LINE'
  An existing .python-version file was found.  Either remove this file yourself
  or re-run with the --force option to have it deleted along with the associated
  virtual environment.

  rm .python-version

END_OF_LINE
  exit 1
fi

# Create a new virtual environment for this project
#
# If $PYTHON_VERSION is undefined then the current pyenv Python version will be used.
#
# We can't quote ${PYTHON_VERSION:=} below since if the variable is
# undefined then we want nothing to appear; this is the reason for the
# "shellcheck disable" line below.
#
# shellcheck disable=SC2086
if ! pyenv virtualenv ${PYTHON_VERSION:=} "${env_name}"; then
  cat << END_OF_LINE
  An existing virtual environment named $env_name was found.  Either delete this
  environment yourself or re-run with the --force option to have it deleted.

  pyenv virtualenv-delete ${env_name}

END_OF_LINE
  exit 1
fi

# Set the local application-specific Python version(s) by writing the
# version name to a file named `.python-version'.
pyenv local "${env_name}"

# Upgrade pip and friends
python3 -m pip install --upgrade pip setuptools wheel

# Find a requirements file (if possible) and install
for req_file in "requirements-dev.txt" "requirements-test.txt" "requirements.txt"; do
  if [[ -f $req_file ]]; then
    pip install --requirement $req_file
    break
  fi
done

# Install all necessary mypy type stubs
mypy --install-types src/

# Install git pre-commit hooks now or later.
pre-commit install ${INSTALL_HOOKS:+"--install-hooks"}

# Setup git remotes from lineage configuration
# This could fail if the remotes are already setup, but that is ok.
set +o errexit

eval "$(
  python3 << 'END_OF_LINE'
from pathlib import Path
import yaml
import sys

LINEAGE_CONFIG = Path(".github/lineage.yml")

if not LINEAGE_CONFIG.exists():
    print("No lineage configuration found.", file=sys.stderr)
    sys.exit(0)

with LINEAGE_CONFIG.open("r") as f:
    lineage = yaml.safe_load(stream=f)

if lineage["version"] == "1":
    for parent_name, v in lineage["lineage"].items():
        remote_url = v["remote-url"]
        print(f"git remote add {parent_name} {remote_url};")
        print(f"git remote set-url --push {parent_name} no_push;")
else:
    print(f'Unsupported lineage version: {lineage["version"]}', file=sys.stderr)
END_OF_LINE
)"

# Qapla'
echo "Success!"