package controller_test

import (
	"slices"
	"testing"

	e "github.com/iliadenisov/galaxy/internal/error"

	"github.com/iliadenisov/galaxy/internal/model/game"
	"github.com/stretchr/testify/assert"
)

func TestSendGroup(t *testing.T) {
	c, g := newCache()
	// group #1 - in_orbit, free to upgrade
	assert.NoError(t, c.CreateShips(Race_0_idx, ShipType_Cruiser, R0_Planet_0_num, 10))
	// group #2 - in_space
	assert.NoError(t, c.CreateShips(Race_0_idx, ShipType_Cruiser, R0_Planet_0_num, 1))
	// g.ShipGroups[1].StateInSpace = &game.InSpace{Origin: 2, Range: 1.23}
	c.ShipGroup(1).StateInSpace = &game.InSpace{Origin: 2, Range: 1.23}
	// group #3 - in_orbit, unmovable
	g.CreateShipType(Race_0.Name, "Fortress", 0, 50, 30, 100, 0)
	assert.NoError(t, c.CreateShips(Race_0_idx, "Fortress", R0_Planet_0_num, 1))

	assert.ErrorContains(t,
		g.SendGroup("UnknownRace", 1, 2, 0),
		e.GenericErrorText(e.ErrInputUnknownRace))
	assert.ErrorContains(t,
		g.SendGroup(Race_0.Name, 555, 2, 0),
		e.GenericErrorText(e.ErrInputEntityNotExists))
	assert.ErrorContains(t,
		g.SendGroup(Race_0.Name, 1, 222, 0),
		e.GenericErrorText(e.ErrInputEntityNotExists))
	assert.ErrorContains(t,
		g.SendGroup(Race_0.Name, 2, 1, 0),
		e.GenericErrorText(e.ErrShipsBusy))
	assert.ErrorContains(t,
		g.SendGroup(Race_0.Name, 3, 2, 0),
		e.GenericErrorText(e.ErrSendShipHasNoDrives))
	assert.ErrorContains(t,
		g.SendGroup(Race_0.Name, 1, 2, 100),
		e.GenericErrorText(e.ErrBeakGroupNumberNotEnough))
	assert.ErrorContains(t,
		g.SendGroup(Race_0.Name, 1, 3, 0),
		e.GenericErrorText(e.ErrSendUnreachableDestination))

	assert.NoError(t, g.SendGroup(Race_0.Name, 1, R0_Planet_2_num, 3)) // send 3 of 10
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 4)
	assert.Equal(t, uint(7), c.ShipGroup(0).Number)
	assert.Equal(t, game.StateInOrbit, c.ShipGroup(0).State())
	assert.Equal(t, uint(3), c.ShipGroup(3).Number)
	assert.Equal(t, game.StateLaunched, c.ShipGroup(3).State())
	assert.NotNil(t, c.ShipGroup(3).StateInSpace)
	assert.Equal(t, c.MustPlanet(R0_Planet_0_num).X.F(), c.ShipGroup(3).StateInSpace.X)
	assert.Equal(t, c.MustPlanet(R0_Planet_0_num).Y.F(), c.ShipGroup(3).StateInSpace.Y)

	assert.NoError(t, g.SendGroup(Race_0.Name, 4, R0_Planet_0_num, 2)) // un-send 2 of 3
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 4)
	assert.Equal(t, uint(9), c.MustShipGroup(Race_0_idx, 1).Number)
	assert.Equal(t, game.StateInOrbit, c.MustShipGroup(Race_0_idx, 1).State())
	assert.Equal(t, uint(1), c.MustShipGroup(Race_0_idx, 4).Number)
	assert.Equal(t, game.StateLaunched, c.MustShipGroup(Race_0_idx, 4).State())
	assert.NotNil(t, c.MustShipGroup(Race_0_idx, 4).StateInSpace)
	assert.Equal(t, c.MustPlanet(R0_Planet_0_num).X.F(), c.MustShipGroup(Race_0_idx, 4).StateInSpace.X)
	assert.Equal(t, c.MustPlanet(R0_Planet_0_num).Y.F(), c.MustShipGroup(Race_0_idx, 4).StateInSpace.Y)

	assert.NoError(t, g.SendGroup(Race_0.Name, 4, R0_Planet_0_num, 0)) // un-send the rest 1
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 3)
	assert.Equal(t, uint(10), c.MustShipGroup(Race_0_idx, 1).Number)
	assert.Equal(t, game.StateInOrbit, c.MustShipGroup(Race_0_idx, 1).State())

	assert.NoError(t, g.SendGroup(Race_0.Name, 1, R0_Planet_2_num, 0))
	assert.Len(t, slices.Collect(c.RaceShipGroups(Race_0_idx)), 3)
	assert.Equal(t, uint(10), c.MustShipGroup(Race_0_idx, 1).Number)
	assert.Equal(t, game.StateLaunched, c.MustShipGroup(Race_0_idx, 1).State())
}