package cereal

import (
	"capnproto.org/go/capnp/v3"
	"encoding/base64"
	"encoding/json"
	"os"
	"pfeifer.dev/mapd/cereal/custom"
	"pfeifer.dev/mapd/cereal/log"
	"testing"
)

// This same actual Go wire fixture is decoded by the host Python schema tests.
func TestMapdV1HostWireFixture(t *testing.T) {
	msg, seg, err := capnp.NewMessage(capnp.SingleSegment(nil))
	if err != nil {
		t.Fatal(err)
	}
	event, err := log.NewRootEvent(seg)
	if err != nil {
		t.Fatal(err)
	}
	event.SetLogMonoTime(456000000000)
	event.SetValid(true)
	out, err := event.NewMapdOut()
	if err != nil {
		t.Fatal(err)
	}
	out.SetRoadName("Synthetic provider road")
	out.SetSpeedLimit(13.4112)
	out.SetTileLoaded(true)
	out.SetRoadContext(custom.RoadContext_city)
	out.SetWaySelectionType(custom.WaySelectionType_possible)
	out.SetHighwayClass(custom.HighwayClass_residential)
	out.SetWayId(1234567890123)
	out.SetConditionalSpeedLimit("20 @ (Mo-Fr 08:00-09:00)")
	out.SetSampleVersion(1)
	out.SetRoadStatus(custom.MapdOut_SampleStatus_matchedLimit)
	out.SetGpsSource(custom.MapdOut_GpsSource_external)
	out.SetGpsMonoTime(455950000000)
	out.SetComputedMonoTime(455990000000)
	out.SetSourceGeneration(3)
	out.SetProducerSession(0x1234abcd5011)
	raw, err := msg.Marshal()
	if err != nil {
		t.Fatal(err)
	}

	fixtureBytes, err := os.ReadFile("testdata/mapd_v1_wire.json")
	if err != nil {
		t.Fatal(err)
	}
	var fixture struct {
		EventBase64 string `json:"event_base64"`
	}
	if err := json.Unmarshal(fixtureBytes, &fixture); err != nil {
		t.Fatal(err)
	}
	expected, err := base64.StdEncoding.DecodeString(fixture.EventBase64)
	if err != nil {
		t.Fatal(err)
	}
	actualMessage, err := capnp.Unmarshal(raw)
	if err != nil {
		t.Fatal(err)
	}
	oldMessage, err := capnp.Unmarshal(expected)
	if err != nil {
		t.Fatal(err)
	}
	actualRoot, err := actualMessage.Root()
	if err != nil {
		t.Fatal(err)
	}
	oldRoot, err := oldMessage.Root()
	if err != nil {
		t.Fatal(err)
	}
	// Appended default fields can enlarge storage without changing old values.
	equal, err := capnp.Equal(actualRoot, oldRoot)
	if err != nil || !equal {
		t.Fatal("provider wire differs from shared host fixture; review both schemas")
	}
}

func TestPreviewHostWireFixture(t *testing.T) {
	data, err := os.ReadFile("testdata/mapd_preview_wire.json")
	if err != nil {
		t.Fatal(err)
	}
	var fixture struct {
		EventBase64 string `json:"event_base64"`
	}
	if err := json.Unmarshal(data, &fixture); err != nil {
		t.Fatal(err)
	}
	raw, err := base64.StdEncoding.DecodeString(fixture.EventBase64)
	if err != nil {
		t.Fatal(err)
	}
	msg, err := capnp.Unmarshal(raw)
	if err != nil {
		t.Fatal(err)
	}
	event, err := log.ReadRootEvent(msg)
	if err != nil {
		t.Fatal(err)
	}
	out, err := event.MapdOut()
	if err != nil {
		t.Fatal(err)
	}
	boundary, err := out.PreviewBoundaryId()
	if err != nil || !event.Valid() || out.SampleVersion() != 2 || out.SpeedLimit() != 20 || out.PreviewSpeedLimit() != 15 || out.PreviewDistance() != 60 || len(boundary) != 64 || out.PreviewWayId() != 2 {
		t.Fatal("provider preview differs from shared host fixture")
	}
}