package usage

import (
	"context"
	"encoding/json"
	"testing"
	"time"

	"github.com/carlosframework/platform/internal/release"
)

func TestKeys(t *testing.T) {
	if got := CensusDayKey("2026-08-11"); got != "usage/census/2026-08-11.json" {
		t.Fatalf("census key = %q", got)
	}
	if got := StorageDayKey("2026-08-11"); got != "usage/storage/2026-08-11.json" {
		t.Fatalf("storage key = %q", got)
	}
	if got := StatementKey("bab", "2026-08"); got != "usage/accounts/bab/2026-08.json" {
		t.Fatalf("statement key = %q", got)
	}
}

func TestAppStorageStoreProvisionedWireShapeIsAdditive(t *testing.T) {
	raw, err := json.Marshal(AppStorage{BucketBytes: 12, StoreProvisioned: true})
	if err != nil {
		t.Fatal(err)
	}
	if got, want := string(raw), `{"bucket_bytes":12,"store_provisioned":true}`; got != want {
		t.Fatalf("provisioned app storage = %s, want %s", got, want)
	}

	// Objects written before StoreProvisioned existed remain valid and read as
	// unprovisioned. New objects always carry the field so a consumer inspecting
	// the raw shape can distinguish a legacy snapshot from a known zero count.
	var legacy AppStorage
	if err := json.Unmarshal([]byte(`{"bucket_bytes":7,"store_bytes":3}`), &legacy); err != nil {
		t.Fatal(err)
	}
	if legacy.StoreProvisioned {
		t.Fatal("legacy app storage read as provisioned")
	}
	raw, err = json.Marshal(AppStorage{BucketBytes: 12})
	if err != nil {
		t.Fatal(err)
	}
	if got, want := string(raw), `{"bucket_bytes":12,"store_provisioned":false}`; got != want {
		t.Fatalf("unprovisioned app storage = %s, want %s", got, want)
	}
}

func TestCreateDayObjectIsCreateOnly(t *testing.T) {
	s := release.NewDirStore(t.TempDir())
	ctx := context.Background()
	day := CensusDay{V: 1, Date: "2026-08-11", Accounts: map[string]CensusAccount{"bab": {Apps: 2, Routes: 3}}}
	created, err := CreateDayObject(ctx, s, CensusDayKey(day.Date), day)
	if err != nil || !created {
		t.Fatalf("first create = %v %v", created, err)
	}
	// Second console racing the same day: no error, no overwrite.
	loser := CensusDay{V: 1, Date: "2026-08-11", Accounts: map[string]CensusAccount{"bab": {Apps: 99}}}
	created, err = CreateDayObject(ctx, s, CensusDayKey(day.Date), loser)
	if err != nil || created {
		t.Fatalf("second create = %v %v", created, err)
	}
	raw, _, _ := s.Get(ctx, CensusDayKey(day.Date))
	var got CensusDay
	if err := json.Unmarshal(raw, &got); err != nil {
		t.Fatal(err)
	}
	if got.Accounts["bab"].Apps != 2 {
		t.Fatalf("winner overwritten: %+v", got)
	}
}

func TestUpsertStatementReplaces(t *testing.T) {
	s := release.NewDirStore(t.TempDir())
	ctx := context.Background()
	st := Statement{V: 1, Account: "bab", Month: "2026-08",
		Meters: map[string]float64{MeterEgressGB: 1.5}, CoveredDays: 10, DaysInMonth: 31,
		UpdatedAt: time.Date(2026, 8, 11, 0, 0, 0, 0, time.UTC)}
	if err := UpsertStatement(ctx, s, st); err != nil {
		t.Fatal(err)
	}
	st.Meters[MeterEgressGB] = 2.5
	st.CoveredDays = 11
	if err := UpsertStatement(ctx, s, st); err != nil {
		t.Fatal(err)
	}
	raw, _, _ := s.Get(ctx, StatementKey("bab", "2026-08"))
	var got Statement
	if err := json.Unmarshal(raw, &got); err != nil {
		t.Fatal(err)
	}
	if got.Meters[MeterEgressGB] != 2.5 || got.CoveredDays != 11 {
		t.Fatalf("statement = %+v", got)
	}
}