.zshrc

# Overrides 'pip' based on the virtual environment.
# Uses 'uv pip' if the environment was created by 'uv', otherwise uses 'pip'.

function pip() {
    RED='\033[0;31m'
    GREEN='\033[0;32m'
    BOLD='\033[1m'
    NC='\033[0m'

    if [[ -n "$VIRTUAL_ENV" ]]; then
        if [[ -f "$VIRTUAL_ENV/pyvenv.cfg" ]]; then
            if grep -q "uv" "$VIRTUAL_ENV/pyvenv.cfg"; then
                if command -v uv > /dev/null 2>&1; then
                    echo -e "${GREEN}Running ${BOLD}'uv pip $@'${NC}"
                    uv pip "$@"
                else
                    echo -e "${RED}'uv' command not found. Running ${BOLD}'pip $@'${NC}"
                    command pip "$@"
                fi
            else
                echo -e "${GREEN}Running ${BOLD}'pip $@'${NC}"
                command pip "$@"
            fi
        else
            echo -e "${RED}pyvenv.cfg file does not exist. Running ${BOLD}'pip $@'${NC}"
            command pip "$@"
        fi
    else
        echo -e "${RED}No virtual environment is activated. Running ${BOLD}'pip $@'${NC}"
        command pip "$@"
    fi
}