cmd: create/delete science

pkg/model/game/ship.go

@@ -4,6 +4,7 @@ import (
 	"math"
 
 	"github.com/google/uuid"
+	e "github.com/iliadenisov/galaxy/pkg/error"
 	"github.com/iliadenisov/galaxy/pkg/number"
 )
 
@@ -128,3 +129,126 @@ func (fl Fleet) Speed() float64 {
 	}
 	return result
 }
+
+func (g Game) ShipTypes(raceName string) ([]ShipType, error) {
+	raceID, err := g.hostRaceID(raceName)
+	if err != nil {
+		return nil, err
+	}
+	return g.shipTypesInternal(raceID)
+}
+
+func (g Game) shipTypesInternal(race uuid.UUID) ([]ShipType, error) {
+	for r := range g.Race {
+		if g.Race[r].ID == race {
+			return g.Race[r].ShipTypes, nil
+		}
+	}
+	return nil, e.NewGameStateError("ShipTypes: race %v not found", race)
+}
+
+func (g Game) DeleteShipType(raceName, typeName string) error {
+	raceID, err := g.hostRaceID(raceName)
+	if err != nil {
+		return err
+	}
+	return g.deleteShipTypeInternal(raceID, typeName)
+}
+
+func (g Game) deleteShipTypeInternal(race uuid.UUID, name string) error {
+	for r := range g.Race {
+		if g.Race[r].ID == race {
+			for st := range g.Race[r].ShipTypes {
+				if g.Race[r].ShipTypes[st].Name == name {
+					for sg := range g.Race[r].ShipGroups {
+						if g.Race[r].ShipGroups[sg].TypeID == g.Race[r].ShipTypes[st].ID {
+							return e.NewDeleteShipTypeExistingGroupError(g.Race[r].ShipGroups[sg].Number)
+						}
+					}
+					for pl := range g.Map.Planet {
+						if g.Map.Planet[pl].Owner == race &&
+							g.Map.Planet[pl].Production.Production == ProductionShip &&
+							g.Map.Planet[pl].Production.SubjectID != nil &&
+							g.Race[r].ShipTypes[st].ID == *g.Map.Planet[pl].Production.SubjectID {
+							return e.NewDeleteShipTypePlanetProductionError(g.Map.Planet[pl].Name)
+						}
+					}
+					g.Race[r].ShipTypes = append(g.Race[r].ShipTypes[:st], g.Race[r].ShipTypes[st+1:]...)
+					return nil
+				}
+			}
+			return e.NewEntityTypeNameNotExistsError("ship type %w", name)
+		}
+	}
+	return e.NewGameStateError("DeleteShipType: race %v not found", race)
+}
+
+func (g Game) CreateShipType(raceName, typeName string, d, w, s, c float64, a int) error {
+	raceID, err := g.hostRaceID(raceName)
+	if err != nil {
+		return err
+	}
+	return g.createShipTypeInternal(raceID, typeName, d, w, s, c, a)
+}
+
+func (g Game) createShipTypeInternal(race uuid.UUID, n string, d, w, s, c float64, a int) error {
+	if err := checkShipTypeValues(d, w, s, c, a); err != nil {
+		return err
+	}
+	name, ok := validateTypeName(n)
+	if !ok {
+		return e.NewEntityTypeNameValidationError("%q", name)
+	}
+	for r := range g.Race {
+		if g.Race[r].ID == race {
+			for st := range g.Race[r].ShipTypes {
+				if g.Race[r].ShipTypes[st].Name == name {
+					return e.NewEntityTypeNameDuplicateError("ship type %w", g.Race[r].ShipTypes[st].Name)
+				}
+			}
+			id := uuid.New()
+			g.Race[r].ShipTypes = append(g.Race[r].ShipTypes, ShipType{
+				ID: id,
+				ShipTypeReport: ShipTypeReport{
+					Name:     name,
+					Drive:    d,
+					Weapons:  w,
+					Shields:  s,
+					Cargo:    c,
+					Armament: uint(a),
+				},
+			})
+			return nil
+		}
+	}
+	return e.NewGameStateError("CreateShipType: race %v not found", race)
+}
+
+func checkShipTypeValues(d, w, s, c float64, a int) error {
+	if !checkShipTypeValueDWSC(d) {
+		return e.NewDriveValueError(d)
+	}
+	if !checkShipTypeValueDWSC(w) {
+		return e.NewWeaponsValueError(w)
+	}
+	if !checkShipTypeValueDWSC(s) {
+		return e.NewShieldsValueError(s)
+	}
+	if !checkShipTypeValueDWSC(c) {
+		return e.NewCargoValueError(s)
+	}
+	if a < 0 {
+		return e.NewShipTypeArmamentValueError(a)
+	}
+	if (w == 0 && a > 0) || (a == 0 && w > 0) {
+		return e.NewShipTypeArmamentAndWeaponsValueError("A=%d W=%.0f", a, w)
+	}
+	if d == 0 && w == 0 && s == 0 && c == 0 && a == 0 {
+		return e.NewShipTypeShipTypeZeroValuesError()
+	}
+	return nil
+}
+
+func checkShipTypeValueDWSC(v float64) bool {
+	return v == 0 || v >= 1
+}