#!/bin/bash
# Script de démarrage de tous les services Effi
# À exécuter sur folzo-Corp après un boot ou WOL

echo "================================"
echo "Démarrage des services Effi"
echo "================================"
echo ""

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

# Fonction de démarrage
start_service() {
    local name=$1
    local dir=$2
    local compose_file=${3:-docker-compose.yml}
    
    echo -n "Démarrage $name... "
    cd "$dir" || { echo -e "${RED}✗${NC} Répertoire non trouvé"; return 1; }
    
    if sudo docker compose -f "$compose_file" up -d 2>/dev/null; then
        echo -e "${GREEN}✓${NC} OK"
        return 0
    else
        echo -e "${RED}✗${NC} Erreur"
        return 1
    fi
}

# 1. Vérifier Docker
if ! systemctl is-active --quiet docker; then
    echo "Démarrage de Docker..."
    sudo systemctl start docker
    sleep 3
fi

# 2. Démarrer les services
echo "Démarrage des conteneurs :"
echo ""

start_service "Ollama" "$HOME" "docker-compose.yml"
start_service "ComfyUI" "$HOME/Documents/OpenCode/Projets/Effi/comfyui"
start_service "Effi-STT" "$HOME/Documents/OpenCode/Projets/Effi/effi-stt"
start_service "openedai-speech" "$HOME/openedai-speech"

echo ""
echo "================================"
echo "Vérification des services"
echo "================================"
echo ""

sleep 5

# Vérifier que tout répond
services_ok=0
services_total=4

echo -n "Ollama (11434)... "
if curl -s http://localhost:11434 > /dev/null 2>&1; then
    echo -e "${GREEN}✓${NC}"
    ((services_ok++))
else
    echo -e "${RED}✗${NC}"
fi

echo -n "ComfyUI (8188)... "
if curl -s http://localhost:8188 > /dev/null 2>&1; then
    echo -e "${GREEN}✓${NC}"
    ((services_ok++))
else
    echo -e "${RED}✗${NC}"
fi

echo -n "Effi-STT (8200)... "
if curl -s http://localhost:8200/health > /dev/null 2>&1; then
    echo -e "${GREEN}✓${NC}"
    ((services_ok++))
else
    echo -e "${RED}✗${NC}"
fi

echo -n "openedai-speech (8100)... "
if curl -s http://localhost:8100/health > /dev/null 2>&1; then
    echo -e "${GREEN}✓${NC}"
    ((services_ok++))
else
    echo -e "${RED}✗${NC}"
fi

echo ""
echo "================================"
echo "Résultat : $services_ok/$services_total services OK"
echo "================================"

if [ $services_ok -eq $services_total ]; then
    echo -e "${GREEN}Tous les services sont démarrés !${NC}"
    echo ""
    echo "Vous pouvez utiliser Effi normalement."
    exit 0
else
    echo -e "${RED}Certains services n'ont pas démarré.${NC}"
    echo ""
    echo "Pour voir les logs :"
    echo "  sudo docker logs ollama"
    echo "  sudo docker logs comfyui"
    echo "  sudo docker logs effi-stt"
    echo "  sudo docker logs openedai-speech"
    exit 1
fi