#!/bin/bash

echo "🔍 Running local validation..."

FAILED=0

# ============================================================
# 📂 FILE CHECK
# ============================================================

REQUIRED_FILES=(
    "pydebug_optimizer/env.py"
    "pydebug_optimizer/models.py"
    "pydebug_optimizer/grader.py"
    "pydebug_optimizer/tasks.py"
    "inference.py"
    "openenv.yaml"
)

for file in "${REQUIRED_FILES[@]}"; do
    if [ ! -f "$file" ]; then
        echo "❌ Missing file: $file"
        FAILED=1
    fi
done

# ============================================================
# 🧪 RUN TESTS
# ============================================================

echo "🧪 Running pytest..."
pytest tests/
if [ $? -ne 0 ]; then
    echo "❌ Tests failed"
    FAILED=1
fi

# ============================================================
# 🚀 ENTRYPOINT CHECK
# ============================================================

echo "🚀 Checking inference entrypoint..."
python inference.py > /dev/null 2>&1
if [ $? -ne 0 ]; then
    echo "❌ inference.py failed"
    FAILED=1
fi

# ============================================================
# ✅ FINAL RESULT
# ============================================================

if [ $FAILED -eq 0 ]; then
    echo -e "\033[0;32m✅ VALIDATION SUCCESSFUL\033[0m"
    exit 0
else
    echo -e "\033[0;31m❌ VALIDATION FAILED\033[0m"
    exit 1
fi