package daemon

import (
	"testing"

	"github.com/stretchr/testify/require"

	renderer "github.com/bnema/vev-vt"
	"github.com/bnema/vev/internal/domain"
	scopy "github.com/bnema/vev/internal/usecase/copy"
	promptui "github.com/bnema/vev/internal/usecase/prompt"
	"github.com/bnema/vev/internal/usecase/visualsearch"
)

func TestOverlayRuntimeActive(t *testing.T) {
	ac := &attachedClient{}
	ac.initOverlays()
	require.False(t, ac.overlays.Active())

	ac.overlays.promptMu.Lock()
	ac.overlays.prompt = promptui.New(" Test ", "")
	ac.overlays.promptMu.Unlock()
	require.True(t, ac.overlays.Active())
}

func TestOverlayRuntimeHandleInputPrecedence(t *testing.T) {
	cases := []struct {
		name  string
		setup func(t *testing.T, d *Daemon, sess *session, ac *attachedClient)
		input []byte
		check func(t *testing.T, ac *attachedClient)
	}{
		{
			name: "prompt before palette copy",
			setup: func(t *testing.T, d *Daemon, sess *session, ac *attachedClient) {
				t.Helper()
				d.enterCopyMode(sess, ac)
				d.enterPalette(sess, ac)
				d.enterPrompt(sess, ac, "Rename", "", func(string) error { return nil })
			},
			input: []byte("x"),
			check: func(t *testing.T, ac *attachedClient) {
				t.Helper()
				require.Equal(t, "x", ac.overlays.prompt.Value())
				require.Equal(t, "", ac.overlays.palette.Query())
				require.True(t, ac.overlays.copyActive())
			},
		},
		{
			name: "palette before copy",
			setup: func(t *testing.T, d *Daemon, sess *session, ac *attachedClient) {
				t.Helper()
				d.enterCopyMode(sess, ac)
				d.enterPalette(sess, ac)
			},
			input: []byte("x"),
			check: func(t *testing.T, ac *attachedClient) {
				t.Helper()
				require.Equal(t, "x", ac.overlays.palette.Query())
				require.True(t, ac.overlays.copyActive())
			},
		},
		{
			name: "copy",
			setup: func(t *testing.T, d *Daemon, sess *session, ac *attachedClient) {
				t.Helper()
				d.enterCopyMode(sess, ac)
			},
			input: []byte("q"),
			check: func(t *testing.T, ac *attachedClient) {
				t.Helper()
				require.False(t, ac.overlays.copyActive())
			},
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			p1, releasePTY1 := newBlockingPTY(t)
			p2, releasePTY2 := newBlockingPTY(t)
			d, sess, ac, _ := newManualSessionWithPTYs(t, p1, p2)
			defer releasePTY1()
			defer releasePTY2()
			tc.setup(t, d, sess, ac)

			require.True(t, ac.overlays.HandleInput(d, tc.input))

			tc.check(t, ac)
		})
	}
}

func TestOverlayRuntimeSnapshotDoesNotAliasCopySearchState(t *testing.T) {
	ac := &attachedClient{}
	ac.initOverlays()
	snap := scopy.NewSnapshotFromRows([][]renderer.Cell{testRow("alpha"), testRow("beta alpha")}, 16, 2)
	mode := scopy.NewMode(scopy.NewDocument(snap, domain.DefaultWordSeparators))
	require.True(t, mode.Search("alpha"))
	search := visualsearch.New(snap)
	for _, r := range "alpha" {
		search.Insert(r)
	}

	ac.overlays.copyMu.Lock()
	ac.overlays.copyMode = mode
	ac.overlays.copySearch = search
	ac.overlays.copyMu.Unlock()

	renderSnap := ac.overlays.SnapshotForRender()
	defer renderSnap.Unlock()

	ac.overlays.copyMu.Lock()
	ac.overlays.copyMode.Searches[0].Row = 99
	ac.overlays.copySearch.Insert('z')
	ac.overlays.copyMu.Unlock()

	require.Equal(t, 0, renderSnap.copyMode.Searches[0].Row)
	require.Equal(t, "alpha", renderSnap.copySearchModel.Query())
}

func TestOverlayRuntimeHandleInputInactive(t *testing.T) {
	ac := &attachedClient{}
	ac.initOverlays()

	require.False(t, ac.overlays.HandleInput(&Daemon{}, []byte("x")))
}

func TestOverlayRuntimeSnapshotCarriesCopyModeDocumentAndClearReleasesIt(t *testing.T) {
	ac := &attachedClient{}
	ac.initOverlays()
	p := newPane("floating", nil, domain.Size{Cols: 4, Rows: 2})
	document := scopy.NewDocument(scopy.NewSnapshotFromRows([][]renderer.Cell{testRow("row")}, 4, 2), domain.DefaultWordSeparators)

	ac.overlays.copyMu.Lock()
	ac.overlays.copyMode = scopy.NewMode(document)
	ac.overlays.copyPane = p
	ac.overlays.copyDocument = document
	ac.overlays.copyMu.Unlock()

	renderSnap := ac.overlays.SnapshotForRender()
	require.Same(t, p, renderSnap.copyPane)
	require.NotNil(t, renderSnap.copyMode)
	require.Same(t, document, renderSnap.copyMode.Document())
	renderSnap.Unlock()

	ac.overlays.copyMu.Lock()
	ac.overlays.clearCopyModeLocked()
	require.Nil(t, ac.overlays.copyPane)
	require.Nil(t, ac.overlays.copyMode)
	require.Nil(t, ac.overlays.copyDocument)
	ac.overlays.copyMu.Unlock()
}

func TestOverlayRuntimePointerEpochInvalidatesTransferAndPaneClose(t *testing.T) {
	ac := &attachedClient{}
	ac.initOverlays()
	p := newPane("pane", nil, domain.Size{Cols: 4, Rows: 1})
	document := scopy.NewDocument(scopy.NewSnapshotFromRows([][]renderer.Cell{testRow("row")}, 4, 1), domain.DefaultWordSeparators)

	ac.overlays.copyMu.Lock()
	ac.overlays.beginCopyPointerLocked(copyPointerState{pane: p, document: document, press: scopy.Pos{}})
	staleEpoch := ac.overlays.copyPointer.epoch
	ac.overlays.clearCopyPointerForTransferLocked()
	ac.overlays.invalidateCopyPointerLocked(true) // release/close while publish revalidates
	require.NotEqual(t, staleEpoch, ac.overlays.copyPointerEpoch)
	require.False(t, ac.overlays.copyPointer.valid)
	ac.overlays.copyCandidate = scopy.NewMode(document)
	ac.overlays.copyDocument = document
	ac.overlays.copyPane = p
	ac.overlays.copyMu.Unlock()

	require.True(t, ac.overlays.clearCopyModeForPane(p))
	ac.overlays.copyMu.Lock()
	require.False(t, ac.overlays.copyPointer.valid)
	ac.overlays.copyMu.Unlock()
}

func TestOverlayRuntimeCopyModeDocumentSurvivesRingOverwrite(t *testing.T) {
	ac := &attachedClient{}
	ac.initOverlays()
	history := newTestHistory(1)
	appendHistoryRow(t, history, testRow("before"))
	document := scopy.NewDocument(scopy.NewSnapshot(history, renderer.NewFrame(6, 1), nil, nil), domain.DefaultWordSeparators)

	ac.overlays.copyMu.Lock()
	ac.overlays.copyMode = scopy.NewMode(document)
	ac.overlays.copyDocument = document
	ac.overlays.copyMu.Unlock()

	appendHistoryRow(t, history, testRow("after"))
	renderSnap := ac.overlays.SnapshotForRender()
	defer renderSnap.Unlock()
	require.NotNil(t, renderSnap.copyMode)
	require.Same(t, document, renderSnap.copyMode.Document())
	require.Equal(t, "before", rowText(renderSnap.copyMode.Document().Snapshot().Row(0)))
}