#!/bin/bash

RED='\033[0;31m'
GREEN='\033[0;32m'
NC='\033[0m'

ROOT="${1:-.}"

echo "Scanning for Python virtual environments under: $(realpath "$ROOT")"
echo ""

find "$ROOT" -name "pyvenv.cfg" -type f 2>/dev/null | while read cfg; do
    venv_dir="$(dirname "$cfg")"
    pip="$venv_dir/bin/pip"
    [ ! -x "$pip" ] && pip="$venv_dir/Scripts/pip" # Windows-style
    [ ! -x "$pip" ] && continue

    version=$("$pip" show litellm 2>/dev/null | grep -i '^Version:' | awk '{print $2}')

    echo "Venv: $venv_dir"
    if [ -z "$version" ]; then
        echo "  LiteLLM not found"
    elif [ "$version" = "1.82.7" ] || [ "$version" = "1.82.8" ]; then
        echo -e "  LiteLLM version: $version ${RED}***COMPROMISED***${NC}"
    else
        echo -e "  LiteLLM version: $version ${GREEN}OK${NC}"
    fi
    echo ""
done