package game_test

import (
	"iter"
	"math/rand/v2"
	"slices"
	"testing"

	"github.com/google/uuid"
	"github.com/iliadenisov/galaxy/internal/controller"
	e "github.com/iliadenisov/galaxy/internal/error"
	"github.com/iliadenisov/galaxy/internal/model/game"
	"github.com/stretchr/testify/assert"
)

func TestCargoCapacity(t *testing.T) {
	test := func(cargoSize float64, expectCapacity float64) {
		ship := game.ShipType{
			ShipTypeReport: game.ShipTypeReport{
				Drive:    1,
				Armament: 1,
				Weapons:  1,
				Shields:  1,
				Cargo:    cargoSize,
			},
		}
		sg := game.ShipGroup{
			Number:  1,
			State:   "In_Orbit",
			Drive:   1.5,
			Weapons: 1.1,
			Shields: 2.0,
			Cargo:   1.0,
		}
		assert.Equal(t, expectCapacity, sg.CargoCapacity(&ship))
	}
	test(1, 1.05)
	test(5, 6.25)
	test(10, 15)
	test(50, 175)
	test(100, 600)
}

func TestCarryingAndFullMass(t *testing.T) {
	Freighter := &game.ShipType{
		ShipTypeReport: game.ShipTypeReport{
			Name:     "Freighter",
			Drive:    8,
			Armament: 0,
			Weapons:  0,
			Shields:  2,
			Cargo:    10,
		},
	}
	sg := &game.ShipGroup{
		Number:  1,
		State:   "In_Orbit",
		Drive:   1.0,
		Weapons: 1.0,
		Shields: 1.0,
		Cargo:   1.0,
		Load:    0.0,
	}
	em := Freighter.EmptyMass()
	assert.Equal(t, 0.0, sg.CarryingMass())
	assert.Equal(t, em, sg.FullMass(Freighter))

	sg.Load = 10.0
	assert.Equal(t, 10.0, sg.CarryingMass())
	assert.Equal(t, em+10.0, sg.FullMass(Freighter))

	sg.Cargo = 2.5
	assert.Equal(t, 4.0, sg.CarryingMass())
	assert.Equal(t, em+4.0, sg.FullMass(Freighter))
}

func TestSpeed(t *testing.T) {
	Freighter := &game.ShipType{
		ShipTypeReport: game.ShipTypeReport{
			Name:     "Freighter",
			Drive:    8,
			Armament: 0,
			Weapons:  0,
			Shields:  2,
			Cargo:    10,
		},
	}
	sg := &game.ShipGroup{
		Number:  1,
		State:   "In_Orbit",
		Drive:   1.0,
		Weapons: 1.0,
		Shields: 1.0,
		Cargo:   1.0,
		Load:    0.0,
	}
	assert.Equal(t, 8.0, sg.Speed(Freighter))
	sg.Load = 5.0
	assert.Equal(t, 6.4, sg.Speed(Freighter))
	sg.Drive = 1.5
	assert.Equal(t, 9.6, sg.Speed(Freighter))
	sg.Load = 10
	sg.Cargo = 1.5
	assert.Equal(t, 9.0, sg.Speed(Freighter))
}

func TestBombingPower(t *testing.T) {
	Gunship := game.ShipType{
		ShipTypeReport: game.ShipTypeReport{
			Drive:    60.0,
			Armament: 3,
			Weapons:  30.0,
			Shields:  100.0,
			Cargo:    0.0,
		},
	}
	sg := game.ShipGroup{
		Number:  1,
		State:   "In_Orbit",
		Drive:   1.0,
		Weapons: 1.0,
		Shields: 1.0,
		Cargo:   1.0,
	}
	expectedBombingPower := 139.295
	result := sg.BombingPower(&Gunship)
	assert.Equal(t, expectedBombingPower, result)
}

func TestUpgradeCost(t *testing.T) {
	Cruiser := game.ShipType{
		ShipTypeReport: game.ShipTypeReport{
			Name:     "Cruiser",
			Drive:    15,
			Armament: 1,
			Weapons:  15,
			Shields:  15,
			Cargo:    0,
		},
	}

	sg := game.ShipGroup{
		Number:  1,
		State:   "In_Orbit",
		Drive:   1.0,
		Weapons: 1.0,
		Shields: 1.0,
		Cargo:   1.0,
	}
	upgradeCost := sg.UpgradeDriveCost(&Cruiser, 2.0) +
		sg.UpgradeWeaponsCost(&Cruiser, 2.0) +
		sg.UpgradeShieldsCost(&Cruiser, 2.0) +
		sg.UpgradeCargoCost(&Cruiser, 2.0)
	assert.Equal(t, 225., upgradeCost)
}

func TestDriveEffective(t *testing.T) {
	tc := []struct {
		driveShipType        float64
		driveTech            float64
		expectDriveEffective float64
	}{
		{1, 1, 1},
		{1, 2, 2},
		{2, 1, 2},
		{0, 1, 0},
		{0, 1.5, 0},
		{0, 10, 0},
		{1.5, 1.5, 2.25},
	}
	for i := range tc {
		someShip := game.ShipType{
			ShipTypeReport: game.ShipTypeReport{
				Drive:    tc[i].driveShipType,
				Armament: rand.UintN(30) + 1,
				Weapons:  rand.Float64()*30 + 1,
				Shields:  rand.Float64()*100 + 1,
				Cargo:    rand.Float64()*20 + 1,
			},
		}
		sg := game.ShipGroup{
			Number:  rand.UintN(4) + 1,
			State:   "In_Orbit",
			Drive:   tc[i].driveTech,
			Weapons: rand.Float64()*5 + 1,
			Shields: rand.Float64()*5 + 1,
			Cargo:   rand.Float64()*5 + 1,
		}
		assert.Equal(t, tc[i].expectDriveEffective, sg.DriveEffective(&someShip))
	}
}

func TestShipGroupEqual(t *testing.T) {
	fleetId := uuid.New()
	someUUID := uuid.New()
	mat := game.CargoMaterial
	cap := game.CargoCapital
	left := &game.ShipGroup{
		Index:  1,
		Number: 1,

		OwnerID:   uuid.New(),
		TypeID:    uuid.New(),
		FleetID:   &fleetId,
		State:     "In_Orbit",
		CargoType: &mat,
		Load:      123.45,
		Drive:     1.0,
		Weapons:   1.0,
		Shields:   1.0,
		Cargo:     1.0,
	}

	// essential properties
	right := *left
	assert.True(t, left.Equal(right))

	left.OwnerID = someUUID
	assert.False(t, left.Equal(right))

	right = *left
	left.TypeID = someUUID
	assert.False(t, left.Equal(right))

	right = *left
	left.FleetID = &someUUID
	assert.False(t, left.Equal(right))

	right = *left
	left.FleetID = nil
	assert.False(t, left.Equal(right))

	right = *left
	left.State = "In_Space"
	assert.False(t, left.Equal(right))

	right = *left
	left.CargoType = &cap
	assert.False(t, left.Equal(right))

	right = *left
	left.CargoType = nil
	assert.False(t, left.Equal(right))

	right = *left
	left.Load = 45.123
	assert.False(t, left.Equal(right))

	right = *left
	left.Drive = 1.1
	assert.False(t, left.Equal(right))

	right = *left
	left.Weapons = 1.1
	assert.False(t, left.Equal(right))

	right = *left
	left.Shields = 1.1
	assert.False(t, left.Equal(right))

	right = *left
	left.Cargo = 1.1
	assert.False(t, left.Equal(right))

	// non-essential properties
	right = *left

	left.Index = 2
	assert.True(t, left.Equal(right))
	left.Number = 5
	assert.True(t, left.Equal(right))
}

func TestJoinEqualGroups(t *testing.T) {
	g := &game.Game{
		Race: make([]game.Race, 2),
	}
	raceIdx := 0
	g.Race[raceIdx] = game.Race{
		ID:      uuid.New(),
		Name:    "Race_0",
		Drive:   1.1,
		Weapons: 1.2,
		Shields: 1.3,
		Cargo:   1.4,
	}
	g.Race[1] = game.Race{
		ID:      uuid.New(),
		Name:    "Race_1",
		Drive:   2.1,
		Weapons: 2.2,
		Shields: 2.3,
		Cargo:   2.4,
	}
	g.Map = game.Map{
		Width:  10,
		Height: 10,
		Planet: make([]game.Planet, 3),
	}
	g.Map.Planet[0] = controller.NewPlanet(0, "Planet_0", g.Race[0].ID, 0, 0, 100, 0, 0, 0, game.ProductionNone.AsType(uuid.Nil))
	g.Map.Planet[1] = controller.NewPlanet(1, "Planet_1", g.Race[1].ID, 1, 1, 100, 0, 0, 0, game.ProductionNone.AsType(uuid.Nil))
	g.Map.Planet[2] = controller.NewPlanet(1, "Planet_2", g.Race[0].ID, 2, 2, 100, 0, 0, 0, game.ProductionNone.AsType(uuid.Nil))

	err := g.CreateShipType("Race_0", "R0_Gunship", 60, 30, 100, 0, 3)
	assert.NoError(t, err)
	err = g.CreateShipType("Race_0", "R0_Freighter", 8, 0, 2, 10, 0)
	assert.NoError(t, err)

	err = g.CreateShipType("Race_1", "R1_Gunship", 60, 30, 100, 0, 3)
	assert.NoError(t, err)
	err = g.CreateShipType("Race_1", "R1_Freighter", 8, 0, 2, 10, 0)
	assert.NoError(t, err)

	err = g.CreateShips(raceIdx, "Freighter", 0, 2)
	assert.ErrorContains(t, err, e.GenericErrorText(e.ErrInputEntityNotExists))
	err = g.CreateShips(raceIdx, "R0_Gunship", 1, 2)
	assert.ErrorContains(t, err, e.GenericErrorText(e.ErrInputEntityNotOwned))

	err = g.CreateShips(raceIdx, "R0_Freighter", 0, 1) // 1 -> 2
	assert.NoError(t, err)
	assert.Len(t, collectGroups(g.ListShipGroups(raceIdx)), 1)

	err = g.CreateShips(1, "R1_Freighter", 1, 1)
	assert.NoError(t, err)
	assert.Len(t, collectGroups(g.ListShipGroups(1)), 1)

	err = g.CreateShips(raceIdx, "R0_Freighter", 0, 6) // (2)
	assert.NoError(t, err)
	assert.Len(t, collectGroups(g.ListShipGroups(raceIdx)), 2)

	err = g.CreateShips(raceIdx, "R0_Gunship", 0, 2) // (3)
	assert.NoError(t, err)
	assert.Len(t, collectGroups(g.ListShipGroups(raceIdx)), 3)

	err = g.CreateShips(1, "R1_Gunship", 1, 1)
	assert.NoError(t, err)
	assert.Len(t, collectGroups(g.ListShipGroups(1)), 2)

	g.Race[raceIdx].Drive = 1.5
	err = g.CreateShips(raceIdx, "R0_Gunship", 0, 9) // 4 -> 6
	assert.NoError(t, err)
	assert.Len(t, collectGroups(g.ListShipGroups(raceIdx)), 4)
	err = g.CreateShips(raceIdx, "R0_Freighter", 0, 7) // 5 -> 7
	assert.NoError(t, err)
	assert.Len(t, collectGroups(g.ListShipGroups(raceIdx)), 5)
	err = g.CreateShips(raceIdx, "R0_Gunship", 0, 4) // (6)
	assert.NoError(t, err)
	assert.Len(t, collectGroups(g.ListShipGroups(raceIdx)), 6)
	err = g.CreateShips(raceIdx, "R0_Freighter", 0, 4) // (7)
	assert.NoError(t, err)
	assert.Len(t, collectGroups(g.ListShipGroups(raceIdx)), 7)

	g.Race[1].Shields = 2.0
	err = g.CreateShips(1, "R1_Freighter", 1, 1)
	assert.NoError(t, err)
	assert.Len(t, collectGroups(g.ListShipGroups(1)), 3)

	err = g.JoinEqualGroups("Race_0")
	assert.NoError(t, err)

	assert.Len(t, collectGroups(g.ListShipGroups(1)), 3)
	assert.Len(t, collectGroups(g.ListShipGroups(raceIdx)), 4)

	shipTypeID := func(ri int, name string) uuid.UUID {
		st := slices.IndexFunc(g.Race[ri].ShipTypes, func(v game.ShipType) bool { return v.Name == name })
		if st < 0 {
			t.Fatalf("ShipType not found: %s", name)
			return uuid.Nil
		}
		return g.Race[ri].ShipTypes[st].ID
	}

	for _, sg := range g.ListShipGroups(raceIdx) {
		switch {
		case sg.TypeID == shipTypeID(raceIdx, "R0_Freighter") && sg.Drive == 1.1:
			assert.Equal(t, uint(7), sg.Number)
			assert.Equal(t, uint(2), sg.Index)
		case sg.TypeID == shipTypeID(raceIdx, "R0_Freighter") && sg.Drive == 1.5:
			assert.Equal(t, uint(11), sg.Number)
			assert.Equal(t, uint(7), sg.Index)
		case sg.TypeID == shipTypeID(raceIdx, "R0_Gunship") && sg.Drive == 1.1:
			assert.Equal(t, uint(2), sg.Number)
			assert.Equal(t, uint(3), sg.Index)
		case sg.TypeID == shipTypeID(raceIdx, "R0_Gunship") && sg.Drive == 1.5:
			assert.Equal(t, uint(13), sg.Number)
			assert.Equal(t, uint(6), sg.Index)
		default:
			t.Error("not all ship groups covered")
		}
	}
}

func collectGroups(i iter.Seq2[int, game.ShipGroup]) []game.ShipGroup {
	result := make([]game.ShipGroup, 0)
	for _, sg := range i {
		result = append(result, sg)
	}
	return result
}