package store

import "testing"

// Retention prune keeps newest per (ticker, source) and deletes old rows.
func TestPruneSnapshots(t *testing.T) {
	db, err := Open(t.TempDir() + "/prune.db")
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	_ = db.SaveSnapshot("OLD", "2020-01-01", "daily", `[{"close":1}]`)
	_ = db.SaveSnapshot("OLD", "2026-09-11", "daily", `[{"close":2}]`)
	_ = db.SaveSnapshot("NEW", "2026-09-11", "daily", `[{"close":3}]`)
	n, err := db.PruneSnapshots("2026-01-01")
	if err != nil {
		t.Fatal(err)
	}
	if n != 1 {
		t.Fatalf("pruned = %d, want 1", n)
	}
	if _, _, err := db.LatestSnapshot("OLD", "daily"); err != nil {
		t.Fatal("newest OLD row must survive")
	}
}

// DeleteRoutine + FilingsSince + ListReports parity.
func TestStoreParity(t *testing.T) {
	db, err := Open(t.TempDir() + "/parity.db")
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	id, _ := db.CreateRoutine("demo", "weekend-review", "0 9 * * 6", nil)
	ok, err := db.DeleteRoutine(id, "demo")
	if err != nil || !ok {
		t.Fatalf("delete = %v, %v", ok, err)
	}
	if ok2, _ := db.DeleteRoutine(999999, "demo"); ok2 {
		t.Fatal("delete of missing id must report false")
	}
	rows, _ := db.ListRoutines("demo")
	for _, r := range rows {
		if r.ID == id {
			t.Fatal("routine not deleted")
		}
	}
	_ = db.InsertFiling("BBCA", "2026-09-10", "insider", "buy", 500000, 9850)
	fils, err := db.FilingsSince("BBCA", "2026-09-01", 10)
	if err != nil || len(fils) != 1 {
		t.Fatalf("filings = %v, %v", fils, err)
	}
	_, _ = db.SaveReport("BBCA", `{"a":1}`, `[]`, "tester")
	reps, err := db.ListReports("BBCA", 10)
	if err != nil || len(reps) != 1 {
		t.Fatalf("reports = %v, %v", reps, err)
	}
}

// Destination CRUD: masks nothing at store layer, scopes by owner, kind immutable via API.
func TestDestinationCRUD(t *testing.T) {
	db, err := Open(t.TempDir() + "/dest.db")
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	id, err := db.CreateDestination(Destination{UserKey: "u1", Kind: DestTelegram, Label: "bot", BotToken: "tok", ChatID: "123", Enabled: true})
	if err != nil || id == 0 {
		t.Fatalf("create = %d, %v", id, err)
	}
	rows, _ := db.ListDestinations("u1")
	if len(rows) != 1 || rows[0].BotToken != "tok" {
		t.Fatalf("list = %+v", rows)
	}
	if other, _ := db.ListDestinations("u2"); len(other) != 0 {
		t.Fatal("cross-user leak")
	}
	en, _ := db.ListEnabledDestinations("u1")
	if len(en) != 1 {
		t.Fatal("enabled list must include it")
	}
	off := false
	ok, err := db.UpdateDestination(id, "u1", nil, &off, nil, nil, nil)
	if err != nil || !ok {
		t.Fatalf("disable = %v, %v", ok, err)
	}
	if en, _ := db.ListEnabledDestinations("u1"); len(en) != 0 {
		t.Fatal("disabled row must drop from enabled list")
	}
	// Omitted secrets keep stored values.
	ok, err = db.UpdateDestination(id, "u1", nil, nil, nil, nil, nil)
	if err != nil || !ok {
		t.Fatalf("noop patch = %v, %v", ok, err)
	}
	got, _ := db.GetDestination(id, "u1")
	if got == nil || got.BotToken != "tok" {
		t.Fatalf("secrets must persist on empty patch: %+v", got)
	}
	if ok, _ := db.UpdateDestination(id, "u2", nil, nil, nil, nil, nil); ok {
		t.Fatal("cross-user patch must report false")
	}
	ok, err = db.DeleteDestination(id, "u1")
	if err != nil || !ok {
		t.Fatalf("delete = %v, %v", ok, err)
	}
	if ok, _ := db.DeleteDestination(id, "u1"); ok {
		t.Fatal("double delete must report false")
	}
}

// Universe save + local ticker search + company-name lookup.
func TestUniverseSearch(t *testing.T) {
	db, err := Open(t.TempDir() + "/univ.db")
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	uni := []UniverseRow{
		{Symbol: "BBCA.JK", Close: 100, Date: "2026-09-16", CompanyName: "PT Bank Central Asia Tbk"},
		{Symbol: "TLKM.JK", Close: 90, Date: "2026-09-16"},
		{Symbol: "BBRI.JK", Close: 80, Date: "2026-09-16", CompanyName: "PT Bank Rakyat Indonesia"},
	}
	if err := db.SaveUniverse(uni); err != nil {
		t.Fatal(err)
	}
	// Search by ticker prefix (case-insensitive).
	hits, err := db.SearchTickers("bca", 10)
	if err != nil {
		t.Fatal(err)
	}
	if len(hits) != 1 || hits[0].Symbol != "BBCA" {
		t.Fatalf("bca hits = %+v, want BBCA", hits)
	}
	// Search by company name.
	hits, err = db.SearchTickers("bank", 10)
	if err != nil {
		t.Fatal(err)
	}
	if len(hits) != 2 {
		t.Fatalf("bank hits = %d, want 2 (BBCA+BBRI by name)", len(hits))
	}
	// .JK stripped on save, order by close desc.
	if hits[0].Symbol != "BBCA" || hits[1].Symbol != "BBRI" {
		t.Fatalf("bank order = %s,%s, want BBCA,BBRI", hits[0].Symbol, hits[1].Symbol)
	}
	// CompanyNames returns only tickers with stored names.
	names, err := db.CompanyNames([]string{"BBCA", "TLKM", "BBRI"})
	if err != nil {
		t.Fatal(err)
	}
	if names["BBCA"] != "PT Bank Central Asia Tbk" || names["BBRI"] != "PT Bank Rakyat Indonesia" {
		t.Fatalf("names = %+v, want BBCA+BBRI only", names)
	}
	if _, ok := names["TLKM"]; ok {
		t.Fatal("TLKM has no name, must be omitted")
	}
}

// SetCompanyNames fills blank names; never overwrites existing ones; skips
// blanks and ticker-equal names.
func TestSetCompanyNames(t *testing.T) {
	db, err := Open(t.TempDir() + "/names.db")
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	uni := []UniverseRow{
		{Symbol: "BBCA.JK", Close: 100, Date: "2026-09-16", CompanyName: "PT Bank Central Asia Tbk"},
		{Symbol: "TLKM.JK", Close: 90, Date: "2026-09-16"},
		{Symbol: "BBRI.JK", Close: 80, Date: "2026-09-16"},
	}
	if err := db.SaveUniverse(uni); err != nil {
		t.Fatal(err)
	}
	// Fill TLKM + BBRI; BBCA already has a name (must NOT be clobbered);
	// "UNVR" would be a new row (INSERT OR IGNORE means it's skipped).
	err = db.SetCompanyNames(map[string]string{
		"TLKM": "PT Telkom Indonesia (Persero) Tbk",
		"BBRI": "PT Bank Rakyat Indonesia (Persero) Tbk",
		"BBCA": "SHOULD NOT OVERWRITE",
		"UNVR": "PT Unilever Indonesia Tbk",
		"EMTK": "PT Elang Mahkota Teknologi Tbk", // no universe row -> no-op
		"BLAH": "BLAH",                           // ticker-equal -> skip
	})
	if err != nil {
		t.Fatal(err)
	}
	names, err := db.CompanyNames([]string{"BBCA", "TLKM", "BBRI", "UNVR"})
	if err != nil {
		t.Fatal(err)
	}
	if names["BBCA"] != "PT Bank Central Asia Tbk" {
		t.Fatalf("BBCA clobbered: %q", names["BBCA"])
	}
	if names["TLKM"] != "PT Telkom Indonesia (Persero) Tbk" || names["BBRI"] != "PT Bank Rakyat Indonesia (Persero) Tbk" {
		t.Fatalf("names = %+v, want TLKM+BBRI filled", names)
	}
	if _, ok := names["UNVR"]; ok {
		t.Fatal("UNVR not in universe must be absent")
	}
}