cmd: upgrade group

internal/model/game/game.go

@@ -2,6 +2,9 @@ package game
 
 import (
 	"encoding/json"
+	"fmt"
+	"iter"
+	"maps"
 	"slices"
 	"strings"
 
@@ -9,6 +12,22 @@ import (
 	e "github.com/iliadenisov/galaxy/internal/error"
 )
 
+type TechSet map[Tech]float64
+
+func (ts TechSet) Value(t Tech) float64 {
+	if v, ok := ts[t]; ok {
+		return v
+	} else {
+		panic(fmt.Sprintf("TechSet: Value: %s's value not set", t.String()))
+	}
+}
+
+func (ts TechSet) Set(t Tech, v float64) TechSet {
+	m := maps.Clone(ts)
+	m[t] = v
+	return m
+}
+
 type Game struct {
 	ID         uuid.UUID   `json:"id"`
 	Age        uint        `json:"turn"` // Game's turn number
@@ -29,6 +48,26 @@ func (g Game) Votes(raceID uuid.UUID) float64 {
 	return pop / 1000.
 }
 
+func (g Game) PlanetByNumber(number uint) (Planet, error) {
+	pi := slices.IndexFunc(g.Map.Planet, func(p Planet) bool { return p.Number == number })
+	if pi < 0 {
+		return Planet{}, e.NewGameStateError("PlanetByNumber: planet with number=%d not found", number)
+	}
+	return g.Map.Planet[pi], nil
+}
+
+func (g Game) ShipsInUpgrade(planetNumber uint) iter.Seq[ShipGroup] {
+	return func(yield func(ShipGroup) bool) {
+		for sg := range g.ShipGroups {
+			if g.ShipGroups[sg].Destination == planetNumber && g.ShipGroups[sg].State() == StateUpgrade {
+				if !yield(g.ShipGroups[sg]) {
+					break
+				}
+			}
+		}
+	}
+}
+
 func (g Game) raceIndex(name string) (int, error) {
 	i := slices.IndexFunc(g.Race, func(r Race) bool { return r.Name == name })
 	if i < 0 {