#!/bin/bash
set -euo pipefail

echo "Bootstrap de l'infrastructure notariat-pipeline..."

# Aller dans le répertoire infra
cd "$(dirname "$0")/../infra"

# Copier le fichier d'environnement s'il n'existe pas
cp -n .env.example .env || true
echo "Fichier .env créé. Veuillez le modifier selon vos besoins."

# Télécharger les images Docker
echo "Téléchargement des images Docker..."
docker compose pull

# Démarrer les services de base
echo "Démarrage des services de base..."
docker compose up -d postgres redis minio opensearch neo4j ollama anythingsqlite

# Attendre que les services soient prêts
echo "Attente du démarrage des services..."
sleep 15

# Configuration MinIO
echo "Configuration de MinIO..."
# Créer l'alias MinIO
mc alias set local http://127.0.0.1:9000 $MINIO_ROOT_USER $MINIO_ROOT_PASSWORD || true
# Créer le bucket
mc mb -p local/$MINIO_BUCKET || true

# Télécharger les modèles Ollama
echo "Téléchargement des modèles Ollama..."
curl -s http://127.0.0.1:11434/api/pull -d '{"name":"llama3:8b"}' || echo "Erreur lors du téléchargement de llama3:8b"
curl -s http://127.0.0.1:11434/api/pull -d '{"name":"mistral:7b"}' || echo "Erreur lors du téléchargement de mistral:7b"

# Démarrer les services applicatifs
echo "Démarrage des services applicatifs..."
docker compose up -d host-api worker grafana prometheus

echo "Bootstrap terminé !"
echo "Services disponibles :"
echo "- API: http://localhost:8000/api"
echo "- AnythingLLM: http://localhost:3001"
echo "- Grafana: http://localhost:3000"
echo "- MinIO Console: http://localhost:9001"
echo "- Neo4j Browser: http://localhost:7474"
echo "- OpenSearch: http://localhost:9200"