package main

import (
	"os"
	"path/filepath"
	"strings"
	"testing"

	"github.com/stretchr/testify/require"
)

func TestFreeLeafHoldsAShortMessageWordForWord(t *testing.T) {
	message := Message{
		ID:   7,
		Kind: "user",
		Text: strings.Repeat("a", nodeBytes-len("user: ")),
	}

	node, free := freeLeaf(message)
	require.True(t, free)
	require.Equal(t, Node{
		Level: 0,
		Index: 7,
		Text:  "user: " + message.Text,
		Size:  nodeBytes,
	}, node)

	message.Text += "a"
	_, free = freeLeaf(message)
	require.False(t, free)
}

func TestFreeMergeJoinsShortSiblings(t *testing.T) {
	left := testNode(1, 2, "left")
	right := testNode(1, 3, "right")

	node, free := freeMerge(left, right)
	require.True(t, free)
	require.Equal(t, testNode(2, 1, "left\nright"), node)

	right = testNode(1, 3, strings.Repeat("r", nodeBytes-len("left")))
	_, free = freeMerge(left, right)
	require.False(t, free)
}

func TestTreeBuildsEachNodeOnce(t *testing.T) {
	tree, err := openTree(t.Context(), t.TempDir(), 2, newTestClock().Now)
	require.NoError(t, err)

	defer func() {
		require.NoError(t, tree.Close())
	}()

	require.NoError(t, tree.Add(testNode(0, 0, "user: a")))
	require.ErrorContains(t, tree.Add(testNode(0, 0, "user: b")), "already built")
	require.ErrorContains(t, tree.Add(testNode(1, 0, "a b")), "child that is not built")

	require.NoError(t, tree.Add(testNode(0, 1, "unii: b")))
	require.NoError(t, tree.Add(testNode(1, 0, "a b")))

	node, built := tree.Node(0, 0)
	require.True(t, built)
	require.Equal(t, "user: a", node.Text)
}

func TestTreeReopenLoadsTheSameNodes(t *testing.T) {
	root := t.TempDir()
	clock := newTestClock()
	tree, err := openTree(t.Context(), root, 2, clock.Now)
	require.NoError(t, err)

	nodes := []Node{
		testNode(0, 1, "unii: b"),
		testNode(0, 0, "user: a"),
		testNode(1, 0, "user: a\nunii: b"),
	}
	for _, node := range nodes {
		require.NoError(t, tree.Add(node))
	}

	require.NoError(t, tree.Close())

	data, err := os.ReadFile(filepath.Join(root, "tree", "2026-01-01.jsonl"))
	require.NoError(t, err)
	require.Equal(t, `{"l":0,"i":1,"text":"unii: b","size":7}`+"\n", strings.SplitAfter(string(data), "\n")[0])

	tree, err = openTree(t.Context(), root, 2, clock.Now)
	require.NoError(t, err)

	defer func() {
		require.NoError(t, tree.Close())
	}()

	require.Len(t, tree.nodes, len(nodes))
	for _, node := range nodes {
		loaded, built := tree.Node(node.Level, node.Index)
		require.True(t, built)
		require.Equal(t, node, loaded)
	}

	_, err = openTree(t.Context(), root, 1, clock.Now)
	require.ErrorContains(t, err, "covers messages after the log")
}

func testNode(level, index int, text string) Node {
	return Node{
		Level: level,
		Index: index,
		Text:  text,
		Size:  len(text),
	}
}