package web

import (
	"context"
	"io"
	"log/slog"
	"net/http"
	"net/http/httptest"
	"path/filepath"
	"strings"
	"testing"
	"time"

	"gitea.mynodes.xyz/ilker/llmgw/internal/store"
)

func TestAPersonSeesTheirUseAndWhatIsLeftOfTheirBudget(t *testing.T) {
	ctx := context.Background()
	st, err := store.OpenSQLite(ctx, filepath.Join(t.TempDir(), "s.db"))
	if err != nil {
		t.Fatal(err)
	}
	t.Cleanup(func() { _ = st.Close() })
	budget := int64(50e6)
	st.SetPersonalDefaults(&store.Limits{Budget: &budget, BudgetPeriod: store.PeriodMonth})
	// 45 of 50 units this month: past four-fifths, so the bar says so.
	if err := st.WriteUsage(ctx, []store.Usage{
		{At: time.Now(), KeyID: "session:ada", UserID: "ada", Model: "m", ServedModel: "m",
			PromptTokens: 1200, CompletionTokens: 300, CostMicros: 45e6, Status: 200},
		{At: time.Now(), KeyID: "session:bob", UserID: "bob", Model: "m", ServedModel: "m",
			PromptTokens: 9, CompletionTokens: 9, CostMicros: 1e6, Status: 200},
	}); err != nil {
		t.Fatal(err)
	}

	a := fakeAccounts{user: store.User{ID: "ada", Email: "ada@example.com"}}
	p := NewAccountPages(a, st, nil, "v0", slog.New(slog.NewTextHandler(io.Discard, nil)))
	mux := http.NewServeMux()
	p.Register(mux)
	req := httptest.NewRequest("GET", "/profile", nil)
	req.AddCookie(&http.Cookie{Name: SessionCookie, Value: "good"})
	rec := httptest.NewRecorder()
	mux.ServeHTTP(rec, req)
	body := rec.Body.String()

	for _, want := range []string{
		"Your use and limits",
		// Ada's 1.5k tokens, and none of Bob's.
		">1.5k</div>", "tokens · 30 days", "Tokens per day",
		"Your budget, across everything you use", "<b>45.00</b> of 50.00 this month",
		`class="budget near"`, "width: 90%",
	} {
		if !strings.Contains(body, want) {
			t.Errorf("missing %q", want)
		}
	}
}