package repository

import (
	"fmt"
	"testing"
	"time"

	"git.mchus.pro/mchus/quoteforge/internal/models"
	"github.com/glebarez/sqlite"
	"gorm.io/gorm"
)

func TestGenerateVersion_FirstOfDay(t *testing.T) {
	repo := newTestPricelistRepository(t)

	version, err := repo.GenerateVersionBySource(string(models.PricelistSourceEstimate))
	if err != nil {
		t.Fatalf("GenerateVersionBySource returned error: %v", err)
	}

	today := time.Now().Format("2006-01-02")
	want := fmt.Sprintf("E-%s-001", today)
	if version != want {
		t.Fatalf("expected %s, got %s", want, version)
	}
}

func TestGenerateVersion_UsesMaxSuffixNotCount(t *testing.T) {
	repo := newTestPricelistRepository(t)
	today := time.Now().Format("2006-01-02")

	seed := []models.Pricelist{
		{Source: string(models.PricelistSourceEstimate), Version: fmt.Sprintf("E-%s-001", today), CreatedBy: "test", IsActive: true},
		{Source: string(models.PricelistSourceEstimate), Version: fmt.Sprintf("E-%s-003", today), CreatedBy: "test", IsActive: true},
	}
	for _, pl := range seed {
		if err := repo.Create(&pl); err != nil {
			t.Fatalf("seed insert failed: %v", err)
		}
	}

	version, err := repo.GenerateVersionBySource(string(models.PricelistSourceEstimate))
	if err != nil {
		t.Fatalf("GenerateVersionBySource returned error: %v", err)
	}

	want := fmt.Sprintf("E-%s-004", today)
	if version != want {
		t.Fatalf("expected %s, got %s", want, version)
	}
}

func TestGenerateVersion_IsolatedBySource(t *testing.T) {
	repo := newTestPricelistRepository(t)
	today := time.Now().Format("2006-01-02")

	seed := []models.Pricelist{
		{Source: string(models.PricelistSourceEstimate), Version: fmt.Sprintf("E-%s-009", today), CreatedBy: "test", IsActive: true},
		{Source: string(models.PricelistSourceWarehouse), Version: fmt.Sprintf("S-%s-002", today), CreatedBy: "test", IsActive: true},
	}
	for _, pl := range seed {
		if err := repo.Create(&pl); err != nil {
			t.Fatalf("seed insert failed: %v", err)
		}
	}

	version, err := repo.GenerateVersionBySource(string(models.PricelistSourceWarehouse))
	if err != nil {
		t.Fatalf("GenerateVersionBySource returned error: %v", err)
	}

	want := fmt.Sprintf("S-%s-003", today)
	if version != want {
		t.Fatalf("expected %s, got %s", want, version)
	}
}

func TestGetItems_WarehouseAvailableQtyUsesPrefixResolver(t *testing.T) {
	repo := newTestPricelistRepository(t)
	db := repo.db

	warehouse := models.Pricelist{
		Source:    string(models.PricelistSourceWarehouse),
		Version:   "S-2026-02-07-001",
		CreatedBy: "test",
		IsActive:  true,
	}
	if err := db.Create(&warehouse).Error; err != nil {
		t.Fatalf("create pricelist: %v", err)
	}
	if err := db.Create(&models.PricelistItem{
		PricelistID: warehouse.ID,
		LotName:     "SSD_NVME_03.2T",
		Price:       100,
	}).Error; err != nil {
		t.Fatalf("create pricelist item: %v", err)
	}
	if err := db.Create(&models.Lot{LotName: "SSD_NVME_03.2T"}).Error; err != nil {
		t.Fatalf("create lot: %v", err)
	}
	qty := 5.0
	if err := db.Create(&models.StockLog{
		Partnumber: "SSD_NVME_03.2T_GEN3_P4610",
		Date:       time.Now(),
		Price:      200,
		Qty:        &qty,
	}).Error; err != nil {
		t.Fatalf("create stock log: %v", err)
	}

	items, total, err := repo.GetItems(warehouse.ID, 0, 20, "")
	if err != nil {
		t.Fatalf("GetItems: %v", err)
	}
	if total != 1 {
		t.Fatalf("expected total=1, got %d", total)
	}
	if len(items) != 1 {
		t.Fatalf("expected 1 item, got %d", len(items))
	}
	if items[0].AvailableQty == nil {
		t.Fatalf("expected available qty to be set")
	}
	if *items[0].AvailableQty != 5 {
		t.Fatalf("expected available qty=5, got %v", *items[0].AvailableQty)
	}
}

func newTestPricelistRepository(t *testing.T) *PricelistRepository {
	t.Helper()

	db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
	if err != nil {
		t.Fatalf("open sqlite: %v", err)
	}
	if err := db.AutoMigrate(&models.Pricelist{}, &models.PricelistItem{}, &models.Lot{}, &models.LotPartnumber{}, &models.StockLog{}); err != nil {
		t.Fatalf("migrate: %v", err)
	}
	return NewPricelistRepository(db)
}