Deepen faction economy and station planning

apps/backend/Universe/Scenario/WorldSeedingService.cs

@@ -37,7 +37,7 @@ internal sealed class WorldSeedingService
         .ToList();
 
       var refineries = ownedStations
-        .Where(station => HasInstalledModules(station, "module_gen_prod_refinedmetals_01", "module_gen_prod_energycells_01", "module_arg_stor_liquid_m_01"))
+        .Where(station => string.Equals(StationSimulationService.DetermineStationRole(station), "refinery", StringComparison.Ordinal))
         .ToList();
 
       if (refineries.Count > 0)
@@ -65,10 +65,32 @@ internal sealed class WorldSeedingService
     foreach (var station in stations)
     {
       InitializeStationPopulation(station);
-      station.Inventory["refinedmetals"] = 120f;
+      if (station.InstalledModules.Contains("module_gen_prod_energycells_01", StringComparer.Ordinal))
+      {
+        station.Inventory["energycells"] = MathF.Max(GetInventoryAmount(station.Inventory, "energycells"), 240f);
+      }
+
+      if (station.InstalledModules.Contains("module_gen_prod_refinedmetals_01", StringComparer.Ordinal))
+      {
+        station.Inventory["ore"] = MathF.Max(GetInventoryAmount(station.Inventory, "ore"), 220f);
+      }
+
+      if (station.InstalledModules.Contains("module_gen_prod_hullparts_01", StringComparer.Ordinal))
+      {
+        station.Inventory["refinedmetals"] = MathF.Max(GetInventoryAmount(station.Inventory, "refinedmetals"), 240f);
+        station.Inventory["graphene"] = MathF.Max(GetInventoryAmount(station.Inventory, "graphene"), 80f);
+      }
+
+      if (station.InstalledModules.Contains("module_gen_prod_claytronics_01", StringComparer.Ordinal))
+      {
+        station.Inventory["antimattercells"] = MathF.Max(GetInventoryAmount(station.Inventory, "antimattercells"), 90f);
+        station.Inventory["microchips"] = MathF.Max(GetInventoryAmount(station.Inventory, "microchips"), 120f);
+        station.Inventory["quantumtubes"] = MathF.Max(GetInventoryAmount(station.Inventory, "quantumtubes"), 90f);
+      }
+
       if (station.Population > 0f)
       {
-        station.Inventory["water"] = MathF.Max(80f, station.Population * 1.5f);
+        station.Inventory["water"] = MathF.Max(60f, station.Population * 1.5f);
       }
     }
   }
@@ -76,10 +98,10 @@ internal sealed class WorldSeedingService
   internal StationRuntime? SelectRefineryStation(IReadOnlyCollection<StationRuntime> stations, ScenarioDefinition scenario)
   {
     return stations.FirstOrDefault(station =>
-        HasInstalledModules(station, "module_gen_prod_energycells_01", "module_arg_stor_liquid_m_01") &&
+        string.Equals(StationSimulationService.DetermineStationRole(station), "refinery", StringComparison.Ordinal) &&
         station.SystemId == scenario.MiningDefaults.RefinerySystemId)
       ?? stations.FirstOrDefault(station =>
-        HasInstalledModules(station, "module_gen_prod_energycells_01", "module_arg_stor_liquid_m_01"));
+        string.Equals(StationSimulationService.DetermineStationRole(station), "refinery", StringComparison.Ordinal));
   }
 
   internal List<ClaimRuntime> CreateClaims(
@@ -116,23 +138,21 @@ internal sealed class WorldSeedingService
   }
 
   internal (List<ConstructionSiteRuntime> ConstructionSites, List<MarketOrderRuntime> MarketOrders) CreateConstructionSites(
-    IReadOnlyCollection<StationRuntime> stations,
-    IReadOnlyCollection<ClaimRuntime> claims,
-    IReadOnlyDictionary<string, ModuleRecipeDefinition> moduleRecipes)
+    SimulationWorld world)
   {
     var sites = new List<ConstructionSiteRuntime>();
     var orders = new List<MarketOrderRuntime>();
 
-    foreach (var station in stations)
+    foreach (var station in world.Stations)
     {
-      var moduleId = GetNextConstructionSiteModule(station, moduleRecipes);
+      var moduleId = InfrastructureSimulationService.GetNextStationModuleToBuild(station, world);
       if (moduleId is null || station.CelestialId is null)
       {
         continue;
       }
 
-      var claim = claims.FirstOrDefault(candidate => candidate.CelestialId == station.CelestialId);
-      if (claim is null || !moduleRecipes.TryGetValue(moduleId, out var recipe))
+      var claim = world.Claims.FirstOrDefault(candidate => candidate.CelestialId == station.CelestialId);
+      if (claim is null || !world.ModuleRecipes.TryGetValue(moduleId, out var recipe))
       {
         continue;
       }
@@ -294,23 +314,40 @@ internal sealed class WorldSeedingService
   internal static DefaultBehaviorRuntime CreateBehavior(
     ShipDefinition definition,
     string systemId,
+    string factionId,
     ScenarioDefinition scenario,
     IReadOnlyDictionary<string, List<Vector3>> patrolRoutes,
+    IReadOnlyCollection<StationRuntime> stations,
     StationRuntime? refinery)
   {
-    if (string.Equals(definition.Kind, "construction", StringComparison.Ordinal) && refinery is not null)
+    var homeStation = stations.FirstOrDefault(station =>
+        string.Equals(station.FactionId, factionId, StringComparison.Ordinal)
+        && string.Equals(station.SystemId, systemId, StringComparison.Ordinal))
+      ?? stations.FirstOrDefault(station => string.Equals(station.FactionId, factionId, StringComparison.Ordinal))
+      ?? refinery;
+
+    if (string.Equals(definition.Kind, "construction", StringComparison.Ordinal) && homeStation is not null)
     {
       return new DefaultBehaviorRuntime
       {
         Kind = "construct-station",
-        StationId = refinery.Id,
+        StationId = homeStation.Id,
         Phase = "travel-to-station",
       };
     }
 
-    if (HasCapabilities(definition, "mining") && refinery is not null)
+    if (HasCapabilities(definition, "mining") && homeStation is not null)
     {
-      return CreateResourceHarvestBehavior("auto-mine", scenario.MiningDefaults.NodeSystemId, refinery.Id);
+      return CreateResourceHarvestBehavior("auto-mine", scenario.MiningDefaults.NodeSystemId, homeStation.Id);
+    }
+
+    if (string.Equals(definition.Kind, "transport", StringComparison.Ordinal))
+    {
+      return new DefaultBehaviorRuntime
+      {
+        Kind = "trade-haul",
+        Phase = "travel-to-source",
+      };
     }
 
     if (string.Equals(definition.Kind, "military", StringComparison.Ordinal) && patrolRoutes.TryGetValue(systemId, out var route))
@@ -318,6 +355,7 @@ internal sealed class WorldSeedingService
       return new DefaultBehaviorRuntime
       {
         Kind = "patrol",
+        StationId = homeStation?.Id,
         PatrolPoints = route,
         PatrolIndex = 0,
       };
@@ -340,6 +378,13 @@ internal sealed class WorldSeedingService
         Color = "#7ed4ff",
         Credits = MinimumFactionCredits,
       },
+      "asterion-league" => new FactionRuntime
+      {
+        Id = factionId,
+        Label = "Asterion League",
+        Color = "#ff8f70",
+        Credits = MinimumFactionCredits,
+      },
       _ => new FactionRuntime
       {
         Id = factionId,
@@ -350,31 +395,6 @@ internal sealed class WorldSeedingService
     };
   }
 
-  private static string? GetNextConstructionSiteModule(
-    StationRuntime station,
-    IReadOnlyDictionary<string, ModuleRecipeDefinition> moduleRecipes)
-  {
-    foreach (var (moduleId, targetCount) in new (string ModuleId, int TargetCount)[]
-    {
-      ("module_gen_prod_refinedmetals_01", 1),
-      ("module_arg_stor_container_m_01", 1),
-      ("module_gen_prod_hullparts_01", 2),
-      ("module_gen_prod_advancedelectronics_01", 1),
-      ("module_gen_build_l_01", 1),
-      ("module_gen_prod_energycells_01", 2),
-      ("module_arg_dock_m_01_lowtech", 2),
-    })
-    {
-      if (CountModules(station.InstalledModules, moduleId) < targetCount
-        && moduleRecipes.ContainsKey(moduleId))
-      {
-        return moduleId;
-      }
-    }
-
-    return null;
-  }
-
   private static void InitializeStationPopulation(StationRuntime station)
   {
     var habitatModules = CountModules(station.InstalledModules, "module_arg_hab_m_01");
@@ -406,6 +426,8 @@ internal sealed class WorldSeedingService
   {
     Kind = behavior.Kind,
     AreaSystemId = behavior.AreaSystemId,
+    TargetEntityId = behavior.TargetEntityId,
+    ItemId = behavior.ItemId,
     ModuleId = behavior.ModuleId,
     NodeId = behavior.NodeId,
     Phase = behavior.Phase,