SERVER-49015 Automatically install db-contrib-tool

1 files changed, 76 insertions, 0 deletions

diff --git a/buildscripts/auto_install_db_contrib_tool.sh b/buildscripts/auto_install_db_contrib_tool.sh
new file mode 100755
index 00000000000..1ba77c12460
--- /dev/null
+++ b/buildscripts/auto_install_db_contrib_tool.sh
@@ -0,0 +1,76 @@
+#!/bin/bash
+
+set -o errexit
+
+echo "+-----------------------------------------------------------------------------------------------------------+"
+echo "|   Running a script to automatically install db-contrib-tool (https://github.com/10gen/db-contrib-tool).   |"
+echo "+-----------------------------------------------------------------------------------------------------------+"
+echo
+
+if [[ -d "/opt/mongodbtoolchain/v3/bin" ]]; then
+  export PATH="/opt/mongodbtoolchain/v3/bin:$PATH"
+fi
+
+rc_file=""
+if [[ -f "$HOME/.bashrc" ]]; then
+  rc_file="$HOME/.bashrc"
+fi
+
+if [[ -f "$HOME/.zshrc" ]]; then
+  rc_file="$HOME/.zshrc"
+fi
+
+if ! command -v db-contrib-tool &> /dev/null; then
+  if ! python3 -c "import sys; sys.exit(sys.version_info < (3, 7))" &> /dev/null; then
+    actual_version=$(python3 -c 'import sys; print(sys.version)')
+    echo "You must have python3.7+ installed. Detected version $actual_version."
+    echo "To avoid unexpected issues, python3.7+ will not be automatically installed."
+    echo "Please, do it yourself."
+    echo
+    echo "On macOS you can run:"
+    echo
+    echo "    brew install python3"
+    echo
+    exit 1
+  fi
+
+  if command -v pipx &> /dev/null; then
+    echo "Found pipx: $(command -v pipx)."
+    echo "Using it to install 'db-contrib-tool'."
+    echo
+
+    pipx ensurepath &> /dev/null
+    if [[ -f "$rc_file" ]]; then
+      source "$rc_file"
+    fi
+
+    pipx install db-contrib-tool --python $(command -v python3) --force
+    echo
+  else
+    if ! python3 -m pipx --version &> /dev/null; then
+      echo "Couldn't find pipx. Installing it as python3 module:"
+      echo "    $(command -v python3) -m pip install pipx"
+      echo
+      python3 -m pip install pipx
+      echo
+    else
+      echo "Found pipx installed as python3 module:"
+      echo "    $(command -v python3) -m pipx --version"
+      echo "Using it to install 'db-contrib-tool'."
+      echo
+    fi
+
+    python3 -m pipx ensurepath &> /dev/null
+    if [[ -f "$rc_file" ]]; then
+      source "$rc_file"
+    fi
+
+    python3 -m pipx install db-contrib-tool --force
+    echo
+  fi
+fi
+
+echo "Please, open a new shell or run:"
+echo
+echo "    source $rc_file"
+echo