cmd: send fleet

This commit is contained in:
Ilia Denisov
2026-01-05 19:46:33 +02:00
parent a6093a1c29
commit 5f3a416abd
7 changed files with 271 additions and 38 deletions
+85 -22
View File
@@ -1,6 +1,7 @@
package game
import (
"fmt"
"iter"
"math"
"slices"
@@ -13,10 +14,50 @@ type Fleet struct {
ID uuid.UUID `json:"id"`
OwnerID uuid.UUID `json:"ownerId"`
Name string `json:"name"`
}
Destination uint `json:"destination"`
Origin *uint `json:"origin,omitempty"`
Range *float64 `json:"range,omitempty"`
func FleetState(g *Game, fleetID uuid.UUID) (ShipGroupState, *uint, *InSpace) {
fi := slices.IndexFunc(g.Fleets, func(f Fleet) bool { return f.ID == fleetID })
if fi < 0 {
panic("FleetState: fleet id not found: " + fleetID.String())
}
ri := slices.IndexFunc(g.Race, func(r Race) bool { return r.ID == g.Fleets[fi].OwnerID })
if ri < 0 {
panic("FleetState: race id not found: " + g.Fleets[fi].OwnerID.String())
}
var state *ShipGroupState
var onPlanet *uint
var is *InSpace
for sg := range FleetGroups(g, ri, fi) {
if state == nil {
s := sg.State()
state = &s
if planet, ok := sg.OnPlanet(); ok {
onPlanet = &planet
}
is = sg.StateInSpace
continue
}
if *state != sg.State() {
panic(fmt.Sprintf("FleetState: one or more ships in race's %q fleet %q has different states", g.Race[ri].Name, g.Fleets[fi].Name))
}
if planet, ok := sg.OnPlanet(); ok && onPlanet != nil && *onPlanet != planet {
for sg := range FleetGroups(g, ri, fi) {
fmt.Println("group", sg.Index, "fleet", sg.FleetID, g.Fleets[fi].Name, "state", sg.State(), "on", sg.Destination)
}
panic(fmt.Sprintf("FleetState: one or more ships in race's %q fleet %q are on different planets: %d <> %d", g.Race[ri].Name, g.Fleets[fi].Name, *onPlanet, planet))
}
if (is == nil && sg.StateInSpace != nil) || (is != nil && sg.StateInSpace == nil) {
panic(fmt.Sprintf("FleetState: one or more ships in race's %q fleet %q on_planet and in_space at the same time", g.Race[ri].Name, g.Fleets[fi].Name))
}
if is != nil && sg.StateInSpace != nil && !is.Equal(*sg.StateInSpace) {
panic(fmt.Sprintf("FleetState: one or more ships in race's %q fleet %q has different is_space states", g.Race[ri].Name, g.Fleets[fi].Name))
}
}
if state == nil {
panic(fmt.Sprintf("FleetState: race's %q fleet %q has no ships", g.Race[ri].Name, g.Fleets[fi].Name))
}
return *state, onPlanet, is
}
// TODO: Hello! Wanna know fleet's speed? Good. Implement & test this func first.
@@ -51,7 +92,7 @@ func (g *Game) JoinFleets(raceName, fleetSourceName, fleetTargetName string) err
return g.joinFleetsInternal(ri, fleetSourceName, fleetTargetName)
}
func (g *Game) joinShipGroupToFleetInternal(ri int, fleetName string, group, count uint) (err error) {
func (g *Game) joinShipGroupToFleetInternal(ri int, fleetName string, groupIndex, quantity uint) (err error) {
name, ok := validateTypeName(fleetName)
if !ok {
return e.NewEntityTypeNameValidationError("%q", name)
@@ -59,7 +100,7 @@ func (g *Game) joinShipGroupToFleetInternal(ri int, fleetName string, group, cou
sgi := -1
var maxIndex uint
for i, sg := range g.listIndexShipGroups(ri) {
if sgi < 0 && sg.Index == group {
if sgi < 0 && sg.Index == groupIndex {
sgi = i
}
if sg.Index > maxIndex {
@@ -67,34 +108,40 @@ func (g *Game) joinShipGroupToFleetInternal(ri int, fleetName string, group, cou
}
}
if sgi < 0 {
return e.NewEntityNotExistsError("group #%d", group)
return e.NewEntityNotExistsError("group #%d", groupIndex)
}
if g.ShipGroups[sgi].State() != StateInOrbit {
return e.NewShipsBusyError()
}
if g.ShipGroups[sgi].Number < count {
return e.NewJoinFleetGroupNumberNotEnoughError("%d<%d", g.ShipGroups[sgi].Number, count)
if g.ShipGroups[sgi].Number < quantity {
return e.NewJoinFleetGroupNumberNotEnoughError("%d<%d", g.ShipGroups[sgi].Number, quantity)
}
fi := g.fleetIndex(ri, name)
if fi < 0 {
fi, err = g.createFleet(ri, name, g.ShipGroups[sgi].Destination)
fi, err = g.createFleet(ri, name)
if err != nil {
return err
}
} else {
if g.Fleets[fi].Destination != g.ShipGroups[sgi].Destination || g.Fleets[fi].Origin != nil || g.Fleets[fi].Range != nil {
state, onPlanet, _ := FleetState(g, g.Fleets[fi].ID)
if state != StateInOrbit || *onPlanet != g.ShipGroups[sgi].Destination {
return e.NewShipsNotOnSamePlanetError("fleet: %s", fleetName)
}
}
// FIXME: if g.ShipGroups[sgi].FleetID != nil { // delete old fleet if empty, ALSO mind breaking group }
if count > 0 && g.ShipGroups[sgi].Number != count {
if quantity > 0 && quantity < g.ShipGroups[sgi].Number {
// nsgi, err := g.breakGroupSafe(ri, groupIndex, quantity)
// if err != nil {
// return err
// }
// sgi = nsgi
newGroup := g.ShipGroups[sgi]
newGroup.Number -= count
g.ShipGroups[sgi].Number = count
newGroup.Number -= quantity
g.ShipGroups[sgi].Number = quantity
newGroup.Index = maxIndex + 1
g.ShipGroups = append(g.ShipGroups, newGroup)
}
@@ -103,7 +150,7 @@ func (g *Game) joinShipGroupToFleetInternal(ri int, fleetName string, group, cou
return nil
}
func (g *Game) createFleet(ri int, name string, planetNumber uint) (int, error) {
func (g *Game) createFleet(ri int, name string) (int, error) {
n, ok := validateTypeName(name)
if !ok {
return 0, e.NewEntityTypeNameValidationError("%q", n)
@@ -111,12 +158,13 @@ func (g *Game) createFleet(ri int, name string, planetNumber uint) (int, error)
if fl := g.fleetIndex(ri, n); fl >= 0 {
return 0, e.NewEntityTypeNameDuplicateError("fleet %w", g.Fleets[fl].Name)
}
g.Fleets = append(g.Fleets, Fleet{
ID: uuid.New(),
OwnerID: g.Race[ri].ID,
Name: n,
Destination: planetNumber,
fleets := slices.Clone(g.Fleets)
fleets = append(fleets, Fleet{
ID: uuid.New(),
OwnerID: g.Race[ri].ID,
Name: n,
})
g.Fleets = fleets
return len(g.Fleets) - 1, nil
}
@@ -129,9 +177,9 @@ func (g *Game) joinFleetsInternal(ri int, fleetSourceName, fleetTargetName strin
if fiTarget < 0 {
return e.NewEntityNotExistsError("target fleet %s", fleetTargetName)
}
if g.Fleets[fiSource].Destination != g.Fleets[fiTarget].Destination ||
g.Fleets[fiSource].Origin != nil || g.Fleets[fiTarget].Origin != nil ||
g.Fleets[fiSource].Range != nil || g.Fleets[fiTarget].Range != nil {
srcState, planet1, _ := FleetState(g, g.Fleets[fiSource].ID)
tgtState, planet2, _ := FleetState(g, g.Fleets[fiTarget].ID)
if srcState != StateInOrbit || srcState != tgtState || *planet1 != *planet2 {
return e.NewShipsNotOnSamePlanetError()
}
for sgi, sg := range g.listIndexShipGroups(ri) {
@@ -181,3 +229,18 @@ func (g Game) listIndexFleets(ri int) iter.Seq2[int, Fleet] {
}
}
}
func FleetGroups(g *Game, ri, fi int) iter.Seq[ShipGroup] {
if len(g.Fleets) < fi+1 {
panic(fmt.Sprintf("FleetGroups: game fleets index %d invalid: len=%d", fi, len(g.Fleets)))
}
return func(yield func(ShipGroup) bool) {
for sg := range g.listShipGroups(ri) {
if sg.FleetID != nil && *sg.FleetID == g.Fleets[fi].ID {
if !yield(sg) {
break
}
}
}
}
}