package loki

import (
	"strings"
	"testing"
)

// Un résultat coupé est rangé à part, relu en entier, et part avec sa discussion.
func TestToolResultsSaveLoadDelete(t *testing.T) {
	testHome(t)
	_ = putStr(bkChat, ckActive, "c42")
	long := strings.Repeat("ligne de sortie\n", 500)
	ev := fillToolResult(&ToolUsedEvent{Name: "bash", Done: true}, long)
	if ev.ResultID == "" || !strings.HasPrefix(ev.ResultID, "c42.") {
		t.Fatalf("id attendu préfixé par la discussion, got %q", ev.ResultID)
	}
	if len([]rune(ev.Result)) != toolPreviewChars || ev.ResultChars != len([]rune(long)) {
		t.Fatalf("aperçu %d car., taille %d — attendu %d et %d", len([]rune(ev.Result)), ev.ResultChars, toolPreviewChars, len([]rune(long)))
	}
	if got, ok := loadToolResult(ev.ResultID); !ok || got != long {
		t.Fatal("le résultat complet doit être relisible")
	}
	short := fillToolResult(&ToolUsedEvent{}, "ok")
	if short.ResultID != "" || short.Result != "ok" {
		t.Fatal("un résultat court part entier, sans id")
	}
	deleteToolResultsFor("c42")
	if _, ok := loadToolResult(ev.ResultID); ok {
		t.Fatal("les résultats doivent partir avec leur discussion")
	}
}

// Une sortie de commande coupée le DIT, avec sa taille réelle.
func TestTailOutputAnnouncesCut(t *testing.T) {
	if tailOutput("court") != "court" {
		t.Fatal("une sortie courte reste intacte")
	}
	out := tailOutput(strings.Repeat("x", toolMaxOutput+10))
	if !strings.HasPrefix(out, "[sortie tronquée : ") || !strings.Contains(out, "au total") {
		t.Fatalf("mention de coupe absente : %q", out[:60])
	}
}

// Mémoire chiffrée : le résultat complet est chiffré en base et relu intact ;
// verrouillée, rien n'est écrit (le flux portera le résultat entier) et le
// nettoyage des orphelins ne tourne pas (l'index illisible ferait tout effacer).
func TestToolResultsChiffres(t *testing.T) {
	testHome(t)
	clearMemDEK()
	if _, err := EnableMemEncryption("motdepasse-fort"); err != nil {
		t.Fatalf("EnableMemEncryption: %v", err)
	}
	full := strings.Repeat("ligne de sortie\n", 500)
	id := saveToolResult(full)
	if id == "" {
		t.Fatal("résultat non enregistré")
	}
	persistQ.flush() // l'écriture est différée (chat_persist.go)
	if !looksEncrypted(getBytes(bkToolRes, id)) {
		t.Fatal("résultat en clair alors que la mémoire est chiffrée")
	}
	if got, ok := loadToolResult(id); !ok || got != full {
		t.Fatal("résultat chiffré non restitué à l'identique")
	}
	clearMemDEK()
	if saveToolResult("autre") != "" {
		t.Fatal("mémoire verrouillée : rien n'aurait dû être écrit")
	}
	pruneToolResults()
	if getBytes(bkToolRes, id) == nil {
		t.Fatal("le nettoyage a effacé un résultat pendant que la mémoire était verrouillée")
	}
}