#!/bin/bash
# Script pour désactiver le nettoyage prématuré des screenshots
# POC/MVP : Les screenshots doivent rester jusqu'après l'apprentissage

set -e

echo "🔧 Désactivation du Nettoyage Prématuré des Screenshots"
echo "========================================================"
echo ""

if [ "$EUID" -ne 0 ]; then
    echo "❌ Ce script doit être exécuté avec sudo"
    echo "Usage: sudo bash disable_premature_cleanup.sh"
    exit 1
fi

FILE="/opt/rpa_vision_v3/server/processing_pipeline.py"

# Créer une sauvegarde
BACKUP="${FILE}.backup_$(date +%Y%m%d_%H%M%S)"
echo "💾 Création sauvegarde: $BACKUP"
cp "$FILE" "$BACKUP"
echo "✓ Sauvegarde créée"
echo ""

# Commenter la ligne de nettoyage (ligne 164)
echo "🔧 Modification du fichier..."
sed -i '164s/^            self._cleanup_raw_files/            # DISABLED for POC\/MVP: self._cleanup_raw_files/' "$FILE"

# Vérifier la modification
if grep -q "# DISABLED for POC/MVP: self._cleanup_raw_files" "$FILE"; then
    echo "✅ Ligne de nettoyage désactivée avec succès"
else
    echo "❌ Erreur lors de la modification"
    echo "Restauration de la sauvegarde..."
    cp "$BACKUP" "$FILE"
    exit 1
fi
echo ""

# Afficher le contexte
echo "📋 Contexte modifié:"
echo "-------------------"
grep -A 2 -B 2 "DISABLED for POC" "$FILE" || echo "Erreur affichage"
echo ""

# Redémarrer le worker
echo "🔄 Redémarrage du worker..."
systemctl restart rpa-vision-v3-worker.service
sleep 3

# Vérifier le statut
STATUS=$(systemctl is-active rpa-vision-v3-worker.service)
if [ "$STATUS" = "active" ]; then
    echo "✅ Worker redémarré avec succès"
else
    echo "❌ Worker en état: $STATUS"
    echo "Voir les logs: sudo journalctl -u rpa-vision-v3-worker -n 50"
fi
echo ""

echo "✅ MODIFICATION TERMINÉE"
echo ""
echo "📝 Résumé:"
echo "  - Nettoyage prématuré DÉSACTIVÉ"
echo "  - Screenshots seront conservés après traitement"
echo "  - Sauvegarde disponible: $BACKUP"
echo ""
echo "🎯 Prochaines étapes:"
echo "  1. Tester l'agent: cd agent_v0 && ./run.sh"
echo "  2. Vérifier les screenshots: ls -lhR /opt/rpa_vision_v3/data/training/sessions/sess_*/shots/"
echo ""
echo "💡 TODO Après POC:"
echo "  - Implémenter nettoyage APRÈS apprentissage complet"
echo "  - Conserver uniquement: embeddings, screen_states, workflows"
echo ""
echo "💾 Pour restaurer:"
echo "   sudo cp $BACKUP $FILE"
echo "   sudo systemctl restart rpa-vision-v3-worker.service"