bootstrap

#!/bin/bash
set -e
export LC_ALL=C
if [ $# -eq 0 ]; then
  install=false
else
  if [ "$1" = "install" ]; then
      install=true
  else
      echo "Invalid command, supported commands are: 'install'"
      exit 1
  fi
fi

PYTHON=${PYTHON:-"python3"}
VENV=${VENV:-"virtualenv"}

case "$(uname -s)" in
Linux)
    case "$(lsb_release --id --short)" in
    Ubuntu|Debian|LinuxMint)
        # Ensure setuptools is installed
        if [[ "$PYTHON" =~ "python2" ]]; then
            deps=(qemu-utils python-dev libssl-dev python-pip python-virtualenv libev-dev libvirt-dev libffi-dev libyaml-dev)
        else
            deps=(qemu-utils python3-dev libssl-dev python3-pip python3-virtualenv libev-dev libvirt-dev libffi-dev libyaml-dev)
        fi
        for package in ${deps[@]}; do
            if [ "$(dpkg --status -- $package|sed -n 's/^Status: //p')" != "install ok installed" ]; then
                # add a space after old values
                missing="${missing:+$missing }$package"
            fi
        done
        if [ -n "$missing" ]; then
            echo "$0: missing required packages:" 1>&2
            echo "$missing"
            if [ "$install" = true ]; then
                echo "Installing missing packages..."
                sudo apt-get -y install $missing
            else
                echo "Please install missing packages or run './bootstrap install' if you have sudo"
                echo "sudo apt-get -y install $missing"
                exit 1
            fi
        fi
        ;;
    RedHatEnterpriseWorkstation|RedHatEnterpriseServer|RedHatEnterprise|CentOS)
        if [[ "$PYTHON" =~ "python2" ]]; then
            deps=(python-pip python-devel python-virtualenv mariadb-devel libev-devel libvirt-devel libffi-devel)
        else
            deps=(python3-pip python3-devel python3-virtualenv mariadb-devel libev-devel libvirt-devel libffi-devel)
        fi
        for package in ${deps[@]}; do
          if [ "$(rpm -q $package)" == "package $package is not installed" ]; then
              missing="${missing:+$missing }$package"
          fi
        done
        if [ -n "$missing" ]; then
            echo "$0: missing required packages:" 1>&2
            echo "$missing"
            if [ "$install" = true ]; then
                echo "Installing missing packages..."
                sudo yum -y install $missing
            else
                echo "Please install missing packages or run './bootstrap install' if you have sudo"
                echo "sudo yum -y install $missing"
                exit 1
            fi
        fi
        ;;
    Fedora)
        if [[ "$PYTHON" =~ "python2" ]]; then
            deps=(python-pip python-devel python-virtualenv libev-devel libvirt-devel libffi-devel)
        else
            deps=(python3-pip python3-devel python3-virtualenv libev-devel libvirt-devel libffi-devel)
        fi
        for package in ${deps[@]}; do
          if [ "$(rpm -q $package)" == "package $package is not installed" ]; then
              missing="${missing:+$missing }$package"
          fi
        done
        fedora_release=$(lsb_release -rs)
        package_manager=dnf
        if [ $fedora_release -lt 23 ]; then
            package_manager=yum
        fi
        if [ -n "$missing" ]; then
            echo "$0: missing required packages:" 1>&2
            echo "$missing"
            if [ "$install" = true ]; then
                echo "Installing missing packages..."
                sudo $package_manager -y install $missing
            else
                echo "Please install missing packages or run './bootstrap install' if you have sudo"
                echo "sudo $package_manager -y install $missing"
                exit 1
            fi
        fi
        ;;
    "openSUSE project"|"SUSE LINUX"|"openSUSE")

	if [[ "$PYTHON" =~ "python2" ]]; then
	    deps=(python-pip python-devel python-virtualenv libev-devel libvirt-devel libffi-devel)
	else
	    deps=(python3-pip python3-devel python3-virtualenv libev-devel libvirt-devel libffi-devel)
	fi
	for package in ${deps[@]}; do
            if [ "$(rpm -q $package)" == "package $package is not installed" ]; then
                if [ "$(rpm -q --whatprovides $package)" == "no package provides $package" ]; then
                    missing="${missing:+$missing }$package"
                fi
            fi
        done
        if [ -n "$missing" ]; then
            echo "$0: missing required packages, please install them:" 1>&2
            echo "sudo zypper install $missing"
            exit 1
        fi
        ;;
    *)
        echo "This script does not support your Linux distribution yet. Patches encouraged!"
        exit 1
        ;;
    esac
    ;;
Darwin)
    if ! which brew > /dev/null; then
        echo "You need Homebrew: http://brew.sh/"
        exit 1
    fi
    for keg in python libvirt libev libffi; do
        if brew list $keg >/dev/null 2>&1; then
            echo "Found $keg"
        else
            brew install $keg
        fi
    done
    ;;
*)
    echo "This script does not support your OS yet. Patches encouraged!"
    exit 1
    ;;
esac

# Forcibly remove old virtualenvs which used system site-packages
if [ -e ./$VENV ]  && [ ! -e ./$VENV/lib/python*/no-global-site-packages.txt ]; then
    echo "Removing old virtualenv because it uses system site-packages"
    rm -rf ./$VENV
fi

if [ -z "$NO_CLOBBER" ] || [ ! -e ./$VENV ]; then
    if ! which virtualenv > /dev/null; then
        pip install virtualenv
    fi
    virtualenv --python=$PYTHON $VENV
fi


# be compatible with pip shipped by distro older v20.2
if ./$VENV/bin/pip --use-feature=2020-resolver >/dev/null 2>&1 ; then
    PIP_INSTALL="./$VENV/bin/pip install --use-feature=2020-resolver"
else
    PIP_INSTALL="./$VENV/bin/pip install"
fi
# Upgrade pip first - then we have a new pip version with the --use-feature flag
$PIP_INSTALL --upgrade pip

# Ensure setuptools is installed
./$VENV/bin/pip install --use-feature=2020-resolver setuptools --upgrade

# Install all requirements
./$VENV/bin/pip install --use-feature=2020-resolver --upgrade -r requirements.txt

# Check to make sure requirements are met
./$VENV/bin/pip check

# Remove leftover .pyc files
find teuthology -name '*.pyc' -exec rm {} \;

# Install teuthology
./$VENV/bin/python setup.py develop