package service

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

	"github.com/mhsanaei/3x-ui/v3/internal/database"
	"github.com/mhsanaei/3x-ui/v3/internal/database/model"
	"github.com/mhsanaei/3x-ui/v3/internal/xray"
)

// pinPanelZone fixes the panel time zone so the assertions below can talk about
// calendar days without the test machine's own zone shifting them.
func pinPanelZone(t *testing.T, name string) *time.Location {
	t.Helper()
	loc, err := time.LoadLocation(name)
	if err != nil {
		t.Skipf("zone database unavailable: %v", err)
	}
	if err := database.GetDB().Create(&model.Setting{Key: "timeLocation", Value: name}).Error; err != nil {
		t.Fatalf("pin panel zone: %v", err)
	}
	return loc
}

// Calendar mode renews on the same day each month. The interval mode drifts —
// 30 days from 31 January is 2 March — which is the whole reason for the mode.
func TestAutoRenewClients_CalendarModeLandsOnTheBillingDay(t *testing.T) {
	setupBulkDB(t)
	svc := &InboundService{}
	db := database.GetDB()
	zone := pinPanelZone(t, "UTC")

	// Expired two calendar months ago, billed on the 15th.
	past := time.Date(2026, time.April, 15, 0, 0, 0, 0, time.UTC)
	clients := []model.Client{
		{Email: "cal@x", ID: "11111111-1111-1111-1111-111111111111", Enable: false, ResetDay: 15, ExpiryTime: past.UnixMilli()},
	}
	ib := mkInbound(t, 30201, model.VLESS, clientsSettings(t, clients))
	if err := svc.clientService.SyncInbound(nil, ib.Id, clients); err != nil {
		t.Fatalf("SyncInbound: %v", err)
	}
	if err := db.Create(&xray.ClientTraffic{
		InboundId: ib.Id, Email: "cal@x", Enable: false, Up: 5, Down: 6,
		ResetDay: 15, ExpiryTime: past.UnixMilli(),
	}).Error; err != nil {
		t.Fatalf("seed client_traffics: %v", err)
	}

	if _, count, err := svc.autoRenewClients(db, newTrafficMutationBatch()); err != nil {
		t.Fatalf("autoRenewClients: %v", err)
	} else if count != 1 {
		t.Fatalf("renewed count = %d, want 1", count)
	}

	var row xray.ClientTraffic
	if err := db.Where("email = ?", "cal@x").First(&row).Error; err != nil {
		t.Fatal(err)
	}
	got := time.UnixMilli(row.ExpiryTime).In(zone)
	if got.Day() != 15 {
		t.Fatalf("renewed to %s, want the 15th: calendar mode must not drift", got.Format(time.RFC3339))
	}
	if !got.After(time.Now()) {
		t.Fatalf("renewed to %s, which is not in the future", got.Format(time.RFC3339))
	}
	if h, m, s := got.Clock(); h != 0 || m != 0 || s != 0 {
		t.Fatalf("renewed to %02d:%02d:%02d, want midnight", h, m, s)
	}
	if row.Up != 0 || row.Down != 0 {
		t.Fatalf("counters not reset: up=%d down=%d", row.Up, row.Down)
	}
	if !row.Enable {
		t.Fatal("a renewed client must be re-enabled")
	}
}

// A client billed on the 31st keeps that day, borrowing the last day only in
// months that are too short for it.
func TestAutoRenewClients_CalendarModeClampsShortMonths(t *testing.T) {
	setupBulkDB(t)
	svc := &InboundService{}
	db := database.GetDB()
	zone := pinPanelZone(t, "UTC")

	past := time.Date(2026, time.January, 31, 0, 0, 0, 0, time.UTC)
	clients := []model.Client{
		{Email: "eom@x", ID: "22222222-2222-2222-2222-222222222222", Enable: false, ResetDay: 31, ExpiryTime: past.UnixMilli()},
	}
	ib := mkInbound(t, 30202, model.VLESS, clientsSettings(t, clients))
	if err := svc.clientService.SyncInbound(nil, ib.Id, clients); err != nil {
		t.Fatalf("SyncInbound: %v", err)
	}
	if err := db.Create(&xray.ClientTraffic{
		InboundId: ib.Id, Email: "eom@x", Enable: false, ResetDay: 31, ExpiryTime: past.UnixMilli(),
	}).Error; err != nil {
		t.Fatalf("seed client_traffics: %v", err)
	}

	if _, _, err := svc.autoRenewClients(db, newTrafficMutationBatch()); err != nil {
		t.Fatalf("autoRenewClients: %v", err)
	}

	var row xray.ClientTraffic
	if err := db.Where("email = ?", "eom@x").First(&row).Error; err != nil {
		t.Fatal(err)
	}
	got := time.UnixMilli(row.ExpiryTime).In(zone)
	want := firstBillingMidnightAfter(t, time.Now().In(zone), 31, zone)
	if !got.Equal(want) {
		t.Fatalf("renewed to %s, want %s", got.Format(time.RFC3339), want.Format(time.RFC3339))
	}
}

// Deliberately not built on nextCalendarRenewal: it walks a day at a time and
// derives month length from time.Date's own zero-day trick, so it can disagree.
func firstBillingMidnightAfter(t *testing.T, from time.Time, day int, loc *time.Location) time.Time {
	t.Helper()
	cur := time.Date(from.Year(), from.Month(), from.Day(), 0, 0, 0, 0, loc)
	for i := 0; i < 400; i++ {
		cur = cur.AddDate(0, 0, 1)
		want := day
		if last := time.Date(cur.Year(), cur.Month()+1, 0, 0, 0, 0, 0, loc).Day(); want > last {
			want = last
		}
		if cur.Day() == want {
			return cur
		}
	}
	t.Fatalf("no billing midnight for day %d within a year of %s", day, from)
	return time.Time{}
}

// Interval clients must be untouched by the new field.
func TestAutoRenewClients_IntervalModeUnchanged(t *testing.T) {
	setupBulkDB(t)
	svc := &InboundService{}
	db := database.GetDB()

	past := time.Now().Add(-48 * time.Hour).UnixMilli()
	clients := []model.Client{
		{Email: "days@x", ID: "33333333-3333-3333-3333-333333333333", Enable: false, Reset: 30, ExpiryTime: past},
	}
	ib := mkInbound(t, 30203, model.VLESS, clientsSettings(t, clients))
	if err := svc.clientService.SyncInbound(nil, ib.Id, clients); err != nil {
		t.Fatalf("SyncInbound: %v", err)
	}
	if err := db.Create(&xray.ClientTraffic{
		InboundId: ib.Id, Email: "days@x", Enable: false, Reset: 30, ExpiryTime: past,
	}).Error; err != nil {
		t.Fatalf("seed client_traffics: %v", err)
	}

	if _, count, err := svc.autoRenewClients(db, newTrafficMutationBatch()); err != nil {
		t.Fatalf("autoRenewClients: %v", err)
	} else if count != 1 {
		t.Fatalf("renewed count = %d, want 1", count)
	}

	var row xray.ClientTraffic
	if err := db.Where("email = ?", "days@x").First(&row).Error; err != nil {
		t.Fatal(err)
	}
	if want := past + 30*86400000; row.ExpiryTime != want {
		t.Fatalf("interval renewal moved to %d, want the old fixed step %d", row.ExpiryTime, want)
	}
}

// The selection filter is what keeps a row with neither mode configured out of
// the renewal loop. That matters more than it looks: the interval step is
// reset*24h, so a zero interval reaching that loop would spin forever on the
// single traffic writer. The guard in the loop is a second line of defence and
// is deliberately unreachable while this filter holds.
func TestAutoRenewClients_RowWithNoModeIsNotSelected(t *testing.T) {
	setupBulkDB(t)
	svc := &InboundService{}
	db := database.GetDB()

	past := time.Now().Add(-48 * time.Hour).UnixMilli()
	clients := []model.Client{
		{Email: "none@x", ID: "44444444-4444-4444-4444-444444444444", Enable: false, ExpiryTime: past},
	}
	ib := mkInbound(t, 30204, model.VLESS, clientsSettings(t, clients))
	if err := svc.clientService.SyncInbound(nil, ib.Id, clients); err != nil {
		t.Fatalf("SyncInbound: %v", err)
	}
	// Seeded straight into the table with both modes off, the shape the
	// selection filter is supposed to exclude.
	if err := db.Create(&xray.ClientTraffic{
		InboundId: ib.Id, Email: "none@x", Enable: false, Reset: 0, ResetDay: 0, ExpiryTime: past,
	}).Error; err != nil {
		t.Fatalf("seed client_traffics: %v", err)
	}

	// Asserted against the query rather than by running the loop: the failure
	// this guards is a hang, and a stuck goroutine outlives the test's DB.
	var selected int64
	if err := db.Model(&xray.ClientTraffic{}).
		Where("(reset > 0 or reset_day > 0) and expiry_time > 0 and expiry_time <= ?", time.Now().UnixMilli()).
		Where("email = ?", "none@x").
		Count(&selected).Error; err != nil {
		t.Fatal(err)
	}
	if selected != 0 {
		t.Fatal("a row with no renewal mode was selected for renewal: it would reach the interval loop and spin forever")
	}

	if _, count, err := svc.autoRenewClients(db, newTrafficMutationBatch()); err != nil {
		t.Fatalf("autoRenewClients: %v", err)
	} else if count != 0 {
		t.Fatalf("renewed count = %d, want 0", count)
	}

	var row xray.ClientTraffic
	if err := db.Where("email = ?", "none@x").First(&row).Error; err != nil {
		t.Fatal(err)
	}
	if row.ExpiryTime != past {
		t.Fatalf("a client with no renewal mode was renewed to %d", row.ExpiryTime)
	}
}

// The billing day has to survive the clients table, not just the settings JSON:
// an ordinary edit rebuilds the client from the record and writes it back (#6106).
func TestClientEditKeepsTheBillingDay(t *testing.T) {
	setupBulkDB(t)
	svc := &InboundService{}
	db := database.GetDB()

	clients := []model.Client{
		{Email: "keep@x", ID: "55555555-5555-5555-5555-555555555555", Enable: true, ResetDay: 20, ExpiryTime: time.Now().Add(24 * time.Hour).UnixMilli()},
	}
	ib := mkInbound(t, 30205, model.VLESS, clientsSettings(t, clients))
	if err := svc.clientService.SyncInbound(nil, ib.Id, clients); err != nil {
		t.Fatalf("SyncInbound: %v", err)
	}
	mkTraffic(t, ib.Id, "keep@x", 10, 20, 0, 0, true)

	rec, err := svc.clientService.GetRecordByEmail(nil, "keep@x")
	if err != nil {
		t.Fatalf("GetRecordByEmail: %v", err)
	}
	if rec.ResetDay != 20 {
		t.Fatalf("clients.reset_day = %d, want the 20 the client was created with", rec.ResetDay)
	}

	// What the edit dialog does: hydrate the record, change something else, save.
	edited := rec.ToClient()
	edited.Comment = "renamed"
	if _, err := svc.clientService.Update(svc, rec.Id, *edited, rec.LimitHwid); err != nil {
		t.Fatalf("Update: %v", err)
	}

	// The inbound JSON is what xray and the edit dialog read back, and it is
	// rebuilt from the record, so it is where a dropped converter field shows.
	var stored model.Inbound
	if err := db.Where("id = ?", ib.Id).First(&stored).Error; err != nil {
		t.Fatal(err)
	}
	var settings struct {
		Clients []model.Client `json:"clients"`
	}
	if err := json.Unmarshal([]byte(stored.Settings), &settings); err != nil {
		t.Fatalf("parse inbound settings: %v", err)
	}
	if len(settings.Clients) != 1 {
		t.Fatalf("inbound holds %d clients, want 1", len(settings.Clients))
	}
	if settings.Clients[0].ResetDay != 20 {
		t.Fatalf("inbound settings resetDay = %d after an unrelated edit, want 20: calendar mode was silently turned off", settings.Clients[0].ResetDay)
	}

	rec, err = svc.clientService.GetRecordByEmail(nil, "keep@x")
	if err != nil {
		t.Fatalf("GetRecordByEmail after edit: %v", err)
	}
	if rec.ResetDay != 20 {
		t.Fatalf("clients.reset_day = %d after an unrelated edit, want 20", rec.ResetDay)
	}

	var row xray.ClientTraffic
	if err := db.Where("email = ?", "keep@x").First(&row).Error; err != nil {
		t.Fatal(err)
	}
	if row.ResetDay != 20 {
		t.Fatalf("client_traffics.reset_day = %d after an unrelated edit, want 20", row.ResetDay)
	}
}

// The billing day is useless if it can only be chosen once. The test above
// passes even without the record write, because nothing overwrites the value
// it checks; this one fails without it.
func TestClientEditChangesTheBillingDay(t *testing.T) {
	setupBulkDB(t)
	svc := &InboundService{}

	clients := []model.Client{
		{
			Email: "chg@x", ID: "77777777-7777-7777-7777-777777777777", Enable: true, ResetDay: 20,
			ExpiryTime: time.Now().Add(24 * time.Hour).UnixMilli(),
		},
	}
	ib := mkInbound(t, 30206, model.VLESS, clientsSettings(t, clients))
	if err := svc.clientService.SyncInbound(nil, ib.Id, clients); err != nil {
		t.Fatalf("SyncInbound: %v", err)
	}
	mkTraffic(t, ib.Id, "chg@x", 0, 0, 0, 0, true)

	rec, err := svc.clientService.GetRecordByEmail(nil, "chg@x")
	if err != nil {
		t.Fatalf("GetRecordByEmail: %v", err)
	}
	edited := rec.ToClient()
	edited.ResetDay = 5
	if _, err := svc.clientService.Update(svc, rec.Id, *edited, rec.LimitHwid); err != nil {
		t.Fatalf("Update: %v", err)
	}

	rec, err = svc.clientService.GetRecordByEmail(nil, "chg@x")
	if err != nil {
		t.Fatalf("GetRecordByEmail after edit: %v", err)
	}
	if rec.ResetDay != 5 {
		t.Fatalf("clients.reset_day = %d after the operator moved the billing day to the 5th", rec.ResetDay)
	}

	// Turning calendar mode off has to work too.
	edited = rec.ToClient()
	edited.ResetDay = 0
	edited.Reset = 30
	if _, err := svc.clientService.Update(svc, rec.Id, *edited, rec.LimitHwid); err != nil {
		t.Fatalf("Update back to interval mode: %v", err)
	}
	rec, err = svc.clientService.GetRecordByEmail(nil, "chg@x")
	if err != nil {
		t.Fatalf("GetRecordByEmail after switching mode: %v", err)
	}
	if rec.ResetDay != 0 {
		t.Fatalf("clients.reset_day = %d after the operator switched back to interval mode", rec.ResetDay)
	}
}

// The two renewal features meet here: a calendar client is capped like an
// interval one, spending one allowance per month rather than per tick.
func TestAutoRenewClients_CalendarModeSpendsOneAllowancePerMonth(t *testing.T) {
	setupBulkDB(t)
	svc := &InboundService{}
	db := database.GetDB()
	zone := pinPanelZone(t, "UTC")

	// Three calendar months behind with one allowance left: a single month step
	// cannot reach the present, so the client stays expired on its billing day.
	past := time.Now().In(zone).AddDate(0, -3, 0)
	past = time.Date(past.Year(), past.Month(), 10, 0, 0, 0, 0, zone)
	clients := []model.Client{
		{Email: "calcap@x", ID: "22222222-2222-2222-2222-222222222222", Enable: false, ResetDay: 10, ResetMax: 3, ExpiryTime: past.UnixMilli()},
	}
	ib := mkInbound(t, 30205, model.VLESS, clientsSettings(t, clients))
	if err := svc.clientService.SyncInbound(nil, ib.Id, clients); err != nil {
		t.Fatalf("SyncInbound: %v", err)
	}
	if err := db.Create(&xray.ClientTraffic{
		InboundId: ib.Id, Email: "calcap@x", Enable: false, ResetDay: 10, ResetMax: 3, ResetCount: 2,
		Up: 111, Down: 222, ExpiryTime: past.UnixMilli(),
	}).Error; err != nil {
		t.Fatalf("seed client_traffics: %v", err)
	}

	if _, _, err := svc.autoRenewClients(db, newTrafficMutationBatch()); err != nil {
		t.Fatalf("autoRenewClients: %v", err)
	}

	var row xray.ClientTraffic
	if err := db.Where("email = ?", "calcap@x").First(&row).Error; err != nil {
		t.Fatal(err)
	}
	got := time.UnixMilli(row.ExpiryTime).In(zone)
	if want := past.AddDate(0, 1, 0); !got.Equal(want) {
		t.Fatalf("renewed to %s, want exactly one month on to %s", got.Format(time.RFC3339), want.Format(time.RFC3339))
	}
	if row.ResetCount != 3 {
		t.Fatalf("resetCount = %d, want 3: one allowance per month stepped", row.ResetCount)
	}
	if row.Enable {
		t.Fatal("a client still expired after a truncated catch-up was enabled")
	}
	if row.Up != 111 || row.Down != 222 {
		t.Fatalf("counters zeroed for a month the client can never use: up=%d down=%d", row.Up, row.Down)
	}
}