From af3e40c173eadc0050d341dfee2f698cf495e217 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 2 Apr 2024 12:48:00 -0300
Subject: [PATCH] Update poetry installation script to include pipx update
 check

---
 scripts/setup/update_poetry.sh | 95 +++++++++++++++++++++++++++++++---
 1 file changed, 88 insertions(+), 7 deletions(-)

diff --git a/scripts/setup/update_poetry.sh b/scripts/setup/update_poetry.sh
index b6ff9f841..6df158b99 100644
--- a/scripts/setup/update_poetry.sh
+++ b/scripts/setup/update_poetry.sh
@@ -7,26 +7,107 @@ then
     exit 1
 fi
 
-echo "Checking Pipx installation..."
+# Utility function to display an error message and exit
+exit_with_message() {
+    echo "$1" >&2
+    exit 1
+}
 
-# Check if pipx is installed
-if ! command -v pipx &> /dev/null
-then
+# Check if version argument is provided
+if [ -z "$1" ]; then
+    exit_with_message "No argument supplied. Please provide the Poetry version to check."
+fi
+
+# Detect Operating System
+OS="$(uname -s)"
+case "$OS" in
+    Darwin)
+        OS="macOS"
+        ;;
+    Linux)
+        OS="Linux"
+        ;;
+    *)
+        exit_with_message "Unsupported operating system. This script supports macOS and Linux."
+        ;;
+esac
+
+echo "Detected Operating System: $OS"
+
+# Installation of pipx based on the detected OS
+install_pipx() {
+    case $1 in
+        macOS)
+            # macOS installation using Homebrew
+            command -v brew >/dev/null 2>&1 || exit_with_message "Homebrew is not installed. Please install Homebrew first."
+            echo "Installing pipx using Homebrew..."
+            brew install pipx
+            pipx ensurepath
+            ;;
+        Linux)
+            # Linux installation. Further checks are needed to distinguish between distributions
+            if grep -qEi "(ubuntu|debian)" /etc/*release; then
+                echo "Installing pipx on Ubuntu/Debian..."
+                sudo apt update
+                sudo apt install pipx -y
+            elif grep -qEi "fedora" /etc/*release; then
+                echo "Installing pipx on Fedora..."
+                sudo dnf install pipx -y
+            else
+                echo "Installing pipx using pip (other Linux distributions)..."
+                python3 -m pip install --user pipx
+            fi
+            pipx ensurepath
+            ;;
+        *)
+            exit_with_message "Unsupported operating system for pipx installation."
+            ;;
+    esac
+}
+
+# Function to fetch the latest version of pipx from GitHub and compare with the installed version
+check_for_pipx_update() {
+    echo "Checking for updates to pipx..."
+    # Fetch the latest version of pipx, ensuring only to capture the numeric version without 'v' prefix.
+    local latest_version=$(curl -s https://api.github.com/repos/pypa/pipx/releases/latest | grep '"tag_name":' | sed -E 's/.*"tag_name": "v?([^"]+)".*/\1/')
+    # Extract the current installed version of pipx.
+    local current_version=$(pipx --version | grep -oE '[0-9]+\.[0-9]+\.[0-9]+')
+
+    if [[ "$latest_version" == "$current_version" ]]; then
+        echo "You have the latest version of pipx ($current_version)."
+    else
+        echo "A newer version of pipx ($latest_version) is available. You have $current_version. Do you want to update? (yes/no)"
+        read -r user_input
+        if [[ "$user_input" == "yes" ]]; then
+            echo "Updating pipx..."
+            python3 -m pip install --user --upgrade pipx
+            pipx ensurepath
+            echo "pipx updated to version $latest_version"
+        else
+            echo "Not updating pipx at this time."
+        fi
+    fi
+}
+
+# Now, modify the existing check to call check_for_pipx_update even if pipx is installed
+if ! command -v pipx &> /dev/null; then
     echo "Pipx is not installed. Installing..."
-    python3 -m pip install --user pipx
-    python3 -m pipx ensurepath
+    install_pipx "$OS"
     echo "Pipx installed successfully."
 else
     echo "Pipx is already installed."
+    check_for_pipx_update
 fi
 
+
 echo "Checking Poetry installation..."
 
 # Check if Poetry is installed
 if ! command -v poetry &> /dev/null
 then
     echo "Poetry is not installed. Installing..."
-    pipx install poetry
+    # Also install python 3.10 and use
+    pipx install poetry --python python3.10  --fetch-missing-python
     echo "Poetry installed successfully."
 else
     echo "Poetry is already installed."