install_python.sh

#!/bin/bash
# Exit immediately if a command fails
set -e

# Remove any existing Conda installations first
if [ -d "${HOME}/conda" ] || [ -d "${HOME}/miniconda3" ] || [ -d "${HOME}/anaconda3" ] || [ -d "opt/conda" ]; then
    echo "Removing existing Conda installations..."
    rm -rf "${HOME}/conda" "${HOME}/miniconda3" "${HOME}/anaconda3" "/opt/conda"
fi

# Remove pyenv if present
if [ -d "${HOME}/.pyenv" ]; then
    echo "Removing existing pyenv installation..."
    rm -rf "${HOME}/.pyenv"
fi

# Install uv
echo "Installing uv..."
# Install to ~/.local/bin to ensure it works across different Ubuntu versions
curl -LsSf https://astral.sh/uv/install.sh | DEST=$HOME/.local/bin sh

# Create a persistent PATH update script that we can source immediately
echo "Setting up path for uv..."
UV_PATH_SCRIPT="${HOME}/.local/bin/setup_uv_path.sh"

cat << EOT > "$UV_PATH_SCRIPT"
#!/bin/bash
# Add uv installation directories to PATH
export PATH="\$HOME/.local/bin:\$PATH"
EOT
chmod +x "$UV_PATH_SCRIPT"

# Source the path script for the current session
source "$UV_PATH_SCRIPT"

# Explicitly add to PATH for the current script session
export PATH="$HOME/.local/bin:$PATH"

# Verify PATH contains our bin directories
echo "PATH now includes: (grepping for local/bin)"
echo "$PATH" | tr ':' '\n' | grep -E 'local/bin' || echo "⚠️ PATH update may not have succeeded"

# Verify uv is available
if ! command -v uv > /dev/null 2>&1; then
    echo "ERROR: uv installation failed or not in PATH. Trying alternative installation..."
    
    # Check if uv exists but isn't in PATH
    for uv_path in "$HOME/.local/bin/uv" "/usr/local/bin/uv" "/usr/bin/uv"; do
        if [ -f "$uv_path" ]; then
            echo "Found uv at $uv_path, adding to PATH and making executable"
            chmod +x "$uv_path"
            export PATH="$(dirname "$uv_path"):$PATH"
            break
        fi
    done
    
    
    # Final fallback: direct download of x86_64 binary
    if ! command -v uv > /dev/null 2>&1; then
        echo "Attempting direct binary download for x86_64..."
        mkdir -p "$HOME/.local/bin"
        curl -L "https://github.com/astral-sh/uv/releases/latest/download/uv-x86_64-unknown-linux-gnu.tar.gz" | tar -xz -C "$HOME/.local/bin"
        chmod +x "$HOME/.local/bin/uv"
        export PATH="$HOME/.local/bin:$PATH"
    fi
    
    # Final check
    if ! command -v uv > /dev/null 2>&1; then
        echo "ERROR: Could not install uv. Some functionality will be limited."
        echo "Please install uv manually after installation completes."
        echo "Visit https://github.com/astral-sh/uv for installation instructions."
    else
        echo "uv successfully installed via alternative method: $(uv --version)"
        echo "uv path: $(which uv)"
    fi
else
    echo "uv successfully installed: $(uv --version)"
    echo "uv path: $(which uv)"
fi

# Install Python using uv if available
if command -v uv > /dev/null 2>&1; then
    echo "Installing Python 3.11.4 using uv..."
    uv python install --force 3.11.4 || echo "Python installation failed, continuing anyway"
    
    # Install tools with error handling
    echo "Installing Python tools with uv..."
    
    # Install core Python tools with retries if needed
    for tool in pipx pipenv; do
        echo "Installing $tool..."
        uv tool install --force $tool || {
            echo "First attempt to install $tool failed, retrying with less aggressive options"
            uv tool install $tool --no-binary || {
                echo "⚠️ Failed to install $tool. Continuing with installation."
            }
        }
    done
    
    # Install additional utilities with error handling
    echo "Installing additional utilities with uv..."
    for pkg in cruft dive-bin hadolint-bin lazydocker-bin; do
        echo "Installing $pkg..."
        uv tool install --force $pkg || {
            echo "⚠️ Failed to install $pkg. Continuing with installation."
        }
    done
    
    # Verify the tools were installed
    echo "Verifying tool installation..."
    for tool in pipx pipenv cruft dive hadolint lazydocker; do
        if command -v $tool > /dev/null 2>&1; then
            echo "✅ $tool successfully installed: $(which $tool)"
        else
            echo "⚠️ $tool installation may have failed"
            # Check if binary exists but isn't executable
            for path in "$HOME/.local/bin/$tool" "$HOME/.cargo/bin/$tool"; do
                if [ -f "$path" ] && [ ! -x "$path" ]; then
                    echo "   Found non-executable binary at $path, fixing permissions"
                    chmod +x "$path"
                fi
            done
        fi
    done
else
    echo "Skipping uv-based Python installations."
fi

# Process shell configuration files to add uv paths and remove pyenv
for rc_file in "${HOME}/.zshrc" "${HOME}/.bashrc" "${HOME}/.profile" "${HOME}/.bash_profile"; do
    if [ -f "$rc_file" ]; then
        # Create backup of the rc file
        cp "$rc_file" "${rc_file}.bak.$(date +%s)"
        
        # Remove pyenv configuration if present
        if grep -q "PYENV_ROOT" "$rc_file"; then
            echo "Removing pyenv configuration from $rc_file..."
            sed -i '/# pyenv configuration/,/fi/d' "$rc_file" || true
            sed -i '/PYENV_ROOT/d' "$rc_file" || true
            sed -i '/pyenv init/d' "$rc_file" || true
        fi
        
        
        # Add ~/.local/bin to PATH if it doesn't exist
        if ! grep -q ".local/bin" "$rc_file"; then
            echo 'export PATH="$HOME/.local/bin:$PATH"  # For uv tools and pipx' >> "$rc_file"
        fi
    fi
done

# Ensure ~/.local/bin is in PATH for all users
if ! grep -q ".local/bin" "${HOME}/.profile" 2>/dev/null; then
    echo 'export PATH="$HOME/.local/bin:$PATH"' >> "${HOME}/.profile"
    echo "Added ~/.local/bin to PATH in .profile"
fi

echo "Python environment setup completed"