package history

import (
	"context"
	"database/sql"
	"strings"
	"testing"

	"lingniu-vehicle-ingest/go/vehicle-gateway/internal/envelope"
)

func TestSchemaStatementsCreateCoreStables(t *testing.T) {
	statements := strings.Join(SchemaStatements("test_ts"), "\n")
	for _, want := range []string{"CREATE DATABASE IF NOT EXISTS test_ts", "raw_frames", "vehicle_locations", "vehicle_mileage_points"} {
		if !strings.Contains(statements, want) {
			t.Fatalf("schema missing %q:\n%s", want, statements)
		}
	}
}

func TestWriterAppendsRawLocationAndMileage(t *testing.T) {
	exec := &recordingExec{}
	writer := NewWriter(exec)
	env := sampleEnvelope()

	if err := writer.AppendAll(context.Background(), env); err != nil {
		t.Fatalf("AppendAll() error = %v", err)
	}

	if got := countSQL(exec.calls, "USING raw_frames"); got != 1 {
		t.Fatalf("raw child create count = %d", got)
	}
	if got := countSQL(exec.calls, "USING vehicle_locations"); got != 1 {
		t.Fatalf("location child create count = %d", got)
	}
	if got := countSQL(exec.calls, "USING vehicle_mileage_points"); got != 1 {
		t.Fatalf("mileage child create count = %d", got)
	}
	if got := countSQL(exec.calls, "INSERT INTO raw_"); got != 1 {
		t.Fatalf("raw insert count = %d", got)
	}
	if got := countSQL(exec.calls, "INSERT INTO loc_"); got != 1 {
		t.Fatalf("location insert count = %d", got)
	}
	if got := countSQL(exec.calls, "INSERT INTO mil_"); got != 1 {
		t.Fatalf("mileage insert count = %d", got)
	}
}

func TestWriterSkipsSparseDerivedRows(t *testing.T) {
	exec := &recordingExec{}
	writer := NewWriter(exec)
	env := sampleEnvelope()
	env.Fields = map[string]any{}

	if err := writer.AppendAll(context.Background(), env); err != nil {
		t.Fatalf("AppendAll() error = %v", err)
	}

	if got := countSQL(exec.calls, "INSERT INTO raw_"); got != 1 {
		t.Fatalf("raw insert count = %d", got)
	}
	if got := countSQL(exec.calls, "INSERT INTO loc_"); got != 0 {
		t.Fatalf("location insert count = %d", got)
	}
	if got := countSQL(exec.calls, "INSERT INTO mil_"); got != 0 {
		t.Fatalf("mileage insert count = %d", got)
	}
}

func sampleEnvelope() envelope.FrameEnvelope {
	return envelope.FrameEnvelope{
		Protocol:       envelope.ProtocolJT808,
		MessageID:      "0x0200",
		Sequence:       1,
		VIN:            "LNBVIN00000000001",
		Phone:          "013307795425",
		SourceEndpoint: "127.0.0.1:18080",
		EventTimeMS:    1782745114000,
		ReceivedAtMS:   1782745114999,
		RawHex:         "7E0200",
		Parsed:         map[string]any{"message": "location"},
		Fields: map[string]any{
			envelope.FieldLongitude:      121.069881,
			envelope.FieldLatitude:       30.590151,
			envelope.FieldSpeedKMH:       23.0,
			envelope.FieldTotalMileageKM: 10241.2,
			"direction_deg":              uint16(79),
			"alarm_flag":                 uint32(0),
			"status_flag":                uint32(72),
		},
		ParseStatus: envelope.ParseOK,
	}
}

func countSQL(calls []execCall, pattern string) int {
	var count int
	for _, call := range calls {
		if strings.Contains(call.query, pattern) {
			count++
		}
	}
	return count
}

type execCall struct {
	query string
	args  []any
}

type recordingExec struct {
	calls []execCall
}

func (e *recordingExec) ExecContext(_ context.Context, query string, args ...any) (sql.Result, error) {
	e.calls = append(e.calls, execCall{query: query, args: args})
	return nil, nil
}