package store

import (
	"context"
	"crypto/sha256"
	"path/filepath"
	"testing"
	"time"

	rvboxv1 "github.com/rvbox/rvbox/gen/go/rvbox/v1"
	"github.com/rvbox/rvbox/internal/domain"
)

func TestReadCommandEventsValidatesSegmentAndPreservesOrder_HP_EVENT_04(t *testing.T) {
	ctx := context.Background()
	opened, err := Open(ctx, Options{DataDir: filepath.Join(t.TempDir(), "state"), BusyTimeout: time.Second})
	if err != nil {
		t.Fatal(err)
	}
	defer opened.Close()
	if _, err := opened.RegisterClientSession(ctx, ClientRegistration{ClientID: "win-client", Platform: 3, Architecture: "amd64", DaemonVersion: "test", DaemonCWD: `C:\`, SupportedShells: []byte{1}, ClientInstanceID: [16]byte{1}, SessionID: [16]byte{2}, ConnectedAt: time.Now()}); err != nil {
		t.Fatal(err)
	}
	issue, _ := domain.ParseUUIDv7("019c46f1-1d02-7000-8000-0000000000b1")
	now := time.Now().UTC()
	if _, err := opened.QueueCommand(ctx, QueueCommandInput{IssueUUID: issue, ClientID: "win-client", IssueTime: now, ReceiptTime: now, ImmutableSHA256: sha256.Sum256([]byte("request")), ExecutionSpec: []byte("spec")}); err != nil {
		t.Fatal(err)
	}
	if _, err := opened.ClaimNextDispatch(ctx, "win-client", 1, now); err != nil {
		t.Fatal(err)
	}
	if _, err := opened.RecordCommandAcceptance(ctx, issue, "win-client", 1, 1, true, now); err != nil {
		t.Fatal(err)
	}
	for sequence, lifecycle := range []rvboxv1.CommandLifecycle{rvboxv1.CommandLifecycle_COMMAND_RUNNING, rvboxv1.CommandLifecycle_COMMAND_SUCCEEDED} {
		event := []byte{byte(sequence + 1), 'e', 'v', 't'}
		var identity *rvboxv1.WindowsExecutionIdentity
		if sequence == 0 {
			effective := rvboxv1.WindowsExecutionContext_WINDOWS_EXECUTION_CONTEXT_ACTIVE_USER
			identity = &rvboxv1.WindowsExecutionIdentity{EffectiveContext: &effective, AttemptedContexts: []rvboxv1.WindowsExecutionContext{effective}, SelectionDetail: "selected"}
		}
		if _, err := opened.AppendCommandEvent(ctx, EventAppend{IssueUUID: [16]byte(issue), ClientID: "win-client", SessionGeneration: 1, EventSeq: uint64(sequence + 1), ObservedUnixNano: now.UnixNano(), ReceiptUnixNano: now.Add(time.Duration(sequence) * time.Millisecond).UnixNano(), EventType: 4, Compression: 1, RawLength: uint64(len(event)), Payload: event, ImmutableSHA256: sha256.Sum256(event), Lifecycle: &lifecycle, LifecycleRevision: 1, WindowsIdentity: identity}); err != nil {
			t.Fatal(err)
		}
	}
	view, err := opened.GetCommandView(ctx, "win-client", issue)
	if err != nil {
		t.Fatal(err)
	}
	if len(view.WindowsIdentity) == 0 {
		t.Fatal("running lifecycle did not persist Windows identity")
	}
	events, err := opened.ReadCommandEvents(ctx, issue, 0, 10)
	if err != nil || len(events) != 2 || events[0].EventSeq != 1 || events[1].EventSeq != 2 || string(events[1].Payload) != "\x02evt" {
		t.Fatalf("read events = %#v, %v", events, err)
	}
	decoded, err := DecodeEventPayload(events[0])
	if err != nil || string(decoded) != "\x01evt" {
		t.Fatalf("decoded event payload = %q, %v", decoded, err)
	}
	if _, err := opened.ReadCommandEvents(ctx, issue, 0, 0); err == nil {
		t.Fatal("zero event limit accepted")
	}
}