space-game/apps/backend/Universe/Scenario/WorldBuilder.cs

using static SpaceGame.Api.Universe.Scenario.LoaderSupport;

namespace SpaceGame.Api.Universe.Scenario;

internal sealed class WorldBuilder(
  WorldGenerationOptions worldGeneration,
  DataCatalogLoader dataLoader,
  SystemGenerationService generationService,
  SpatialBuilder spatialBuilder,
  WorldSeedingService seedingService)
{
  internal SimulationWorld Build()
  {
    var catalog = dataLoader.LoadCatalog();
    var systems = generationService.ExpandSystems(
      generationService.InjectSpecialSystems(catalog.AuthoredSystems),
      worldGeneration.TargetSystemCount);
    var scenario = dataLoader.NormalizeScenarioToAvailableSystems(
      catalog.Scenario,
      systems.Select(system => system.Id).ToList());

    var systemRuntimes = systems
      .Select(definition => new SystemRuntime
      {
        Definition = definition,
        Position = ToVector(definition.Position),
      })
      .ToList();
    var systemsById = systemRuntimes.ToDictionary(system => system.Definition.Id, StringComparer.Ordinal);
    var spatialLayout = spatialBuilder.BuildLayout(systemRuntimes, catalog.Balance);

    var stations = CreateStations(
      scenario,
      systemsById,
      spatialLayout.SystemGraphs,
      spatialLayout.Celestials,
      catalog.ModuleDefinitions,
      catalog.ItemDefinitions);

    seedingService.InitializeStationStockpiles(stations);
    var refinery = seedingService.SelectRefineryStation(stations, scenario);
    var patrolRoutes = BuildPatrolRoutes(scenario, systemsById);
    var ships = CreateShips(scenario, systemsById, spatialLayout.Celestials, catalog.Balance, catalog.ShipDefinitions, patrolRoutes, stations, refinery);

    var factions = seedingService.CreateFactions(stations, ships);
    seedingService.BootstrapFactionEconomy(factions, stations);
    var policies = seedingService.CreatePolicies(factions);
    var commanders = seedingService.CreateCommanders(factions, stations, ships);
    var nowUtc = DateTimeOffset.UtcNow;
    var claims = seedingService.CreateClaims(stations, spatialLayout.Celestials, nowUtc);
    var bootstrapWorld = new SimulationWorld
    {
      Label = "Split Viewer / Bootstrap World",
      Seed = WorldSeed,
      Balance = catalog.Balance,
      Systems = systemRuntimes,
      Celestials = spatialLayout.Celestials,
      Nodes = spatialLayout.Nodes,
      Stations = stations,
      Ships = ships,
      Factions = factions,
      Commanders = commanders,
      Claims = claims,
      ConstructionSites = [],
      MarketOrders = [],
      Policies = policies,
      ShipDefinitions = new Dictionary<string, ShipDefinition>(catalog.ShipDefinitions, StringComparer.Ordinal),
      ItemDefinitions = new Dictionary<string, ItemDefinition>(catalog.ItemDefinitions, StringComparer.Ordinal),
      ModuleDefinitions = new Dictionary<string, ModuleDefinition>(catalog.ModuleDefinitions, StringComparer.Ordinal),
      ModuleRecipes = new Dictionary<string, ModuleRecipeDefinition>(catalog.ModuleRecipes, StringComparer.Ordinal),
      Recipes = new Dictionary<string, RecipeDefinition>(catalog.Recipes, StringComparer.Ordinal),
      ProductionGraph = catalog.ProductionGraph,
      OrbitalTimeSeconds = WorldSeed * 97d,
      GeneratedAtUtc = nowUtc,
    };
    var (constructionSites, marketOrders) = seedingService.CreateConstructionSites(bootstrapWorld);

    return new SimulationWorld
    {
      Label = "Split Viewer / Simulation World",
      Seed = WorldSeed,
      Balance = catalog.Balance,
      Systems = systemRuntimes,
      Celestials = spatialLayout.Celestials,
      Nodes = spatialLayout.Nodes,
      Stations = stations,
      Ships = ships,
      Factions = factions,
      Commanders = commanders,
      Claims = claims,
      ConstructionSites = constructionSites,
      MarketOrders = marketOrders,
      Policies = policies,
      ShipDefinitions = new Dictionary<string, ShipDefinition>(catalog.ShipDefinitions, StringComparer.Ordinal),
      ItemDefinitions = new Dictionary<string, ItemDefinition>(catalog.ItemDefinitions, StringComparer.Ordinal),
      ModuleDefinitions = new Dictionary<string, ModuleDefinition>(catalog.ModuleDefinitions, StringComparer.Ordinal),
      ModuleRecipes = new Dictionary<string, ModuleRecipeDefinition>(catalog.ModuleRecipes, StringComparer.Ordinal),
      Recipes = new Dictionary<string, RecipeDefinition>(catalog.Recipes, StringComparer.Ordinal),
      ProductionGraph = catalog.ProductionGraph,
      OrbitalTimeSeconds = WorldSeed * 97d,
      GeneratedAtUtc = DateTimeOffset.UtcNow,
    };
  }

  private static List<StationRuntime> CreateStations(
    ScenarioDefinition scenario,
    IReadOnlyDictionary<string, SystemRuntime> systemsById,
    IReadOnlyDictionary<string, SystemSpatialGraph> systemGraphs,
    IReadOnlyCollection<CelestialRuntime> celestials,
    IReadOnlyDictionary<string, ModuleDefinition> moduleDefinitions,
    IReadOnlyDictionary<string, ItemDefinition> itemDefinitions)
  {
    var stations = new List<StationRuntime>();
    var stationIdCounter = 0;

    foreach (var plan in scenario.InitialStations)
    {
      if (!systemsById.TryGetValue(plan.SystemId, out var system))
      {
        continue;
      }

      var placement = SpatialBuilder.ResolveStationPlacement(plan, system, systemGraphs[system.Definition.Id], celestials);
      var station = new StationRuntime
      {
        Id = $"station-{++stationIdCounter}",
        SystemId = system.Definition.Id,
        Label = plan.Label,
        Color = plan.Color,
        Objective = StationSimulationService.NormalizeStationObjective(plan.Objective),
        Position = placement.Position,
        FactionId = plan.FactionId ?? DefaultFactionId,
        CelestialId = placement.AnchorCelestial.Id,
        Health = 600f,
        MaxHealth = 600f,
      };

      stations.Add(station);
      placement.AnchorCelestial.OccupyingStructureId = station.Id;

      var startingModules = BuildStartingModules(plan, moduleDefinitions, itemDefinitions);

      foreach (var moduleId in startingModules)
      {
        AddStationModule(station, moduleDefinitions, moduleId);
      }
    }

    return stations;
  }

  private static IReadOnlyList<string> BuildStartingModules(
    InitialStationDefinition plan,
    IReadOnlyDictionary<string, ModuleDefinition> moduleDefinitions,
    IReadOnlyDictionary<string, ItemDefinition> itemDefinitions)
  {
    var startingModules = new List<string>(plan.StartingModules.Count > 0
      ? plan.StartingModules
      : ["module_arg_dock_m_01_lowtech", "module_gen_prod_energycells_01", "module_arg_stor_container_m_01"]);

    EnsureStartingModule(startingModules, "module_arg_dock_m_01_lowtech");

    var objectiveModuleId = GetObjectiveStartingModuleId(plan.Objective);
    if (!string.IsNullOrWhiteSpace(objectiveModuleId))
    {
      EnsureStartingModule(startingModules, objectiveModuleId);

      if (!string.Equals(objectiveModuleId, "module_gen_prod_energycells_01", StringComparison.Ordinal))
      {
        EnsureStartingModule(startingModules, "module_gen_prod_energycells_01");
      }

      foreach (var storageModuleId in GetRequiredStartingStorageModules(objectiveModuleId, moduleDefinitions, itemDefinitions))
      {
        EnsureStartingModule(startingModules, storageModuleId);
      }
    }

    return startingModules;
  }

  private static string? GetObjectiveStartingModuleId(string? objective) =>
    StationSimulationService.NormalizeStationObjective(objective) switch
    {
      "power" => "module_gen_prod_energycells_01",
      "refinery" => "module_gen_ref_ore_01",
      "graphene" => "module_gen_prod_graphene_01",
      "siliconwafers" => "module_gen_prod_siliconwafers_01",
      "hullparts" => "module_gen_prod_hullparts_01",
      "claytronics" => "module_gen_prod_claytronics_01",
      "quantumtubes" => "module_gen_prod_quantumtubes_01",
      "antimattercells" => "module_gen_prod_antimattercells_01",
      "superfluidcoolant" => "module_gen_prod_superfluidcoolant_01",
      "water" => "module_gen_prod_water_01",
      _ => null,
    };

  private static IEnumerable<string> GetRequiredStartingStorageModules(
    string moduleId,
    IReadOnlyDictionary<string, ModuleDefinition> moduleDefinitions,
    IReadOnlyDictionary<string, ItemDefinition> itemDefinitions)
  {
    if (!moduleDefinitions.TryGetValue(moduleId, out var moduleDefinition))
    {
      yield break;
    }

    foreach (var wareId in moduleDefinition.Production
      .SelectMany(production => production.Wares.Select(ware => ware.ItemId))
      .Concat(moduleDefinition.Products)
      .Distinct(StringComparer.Ordinal))
    {
      if (!itemDefinitions.TryGetValue(wareId, out var itemDefinition))
      {
        continue;
      }

      var storageModuleId = itemDefinition.CargoKind switch
      {
        "solid" => "module_arg_stor_solid_m_01",
        "liquid" => "module_arg_stor_liquid_m_01",
        _ => "module_arg_stor_container_m_01",
      };

      yield return storageModuleId;
    }
  }

  private static void EnsureStartingModule(List<string> modules, string moduleId)
  {
    if (!modules.Contains(moduleId, StringComparer.Ordinal))
    {
      modules.Add(moduleId);
    }
  }

  private static Dictionary<string, List<Vector3>> BuildPatrolRoutes(
    ScenarioDefinition scenario,
    IReadOnlyDictionary<string, SystemRuntime> systemsById)
  {
    return scenario.PatrolRoutes
      .GroupBy(route => route.SystemId, StringComparer.Ordinal)
      .ToDictionary(
        group => group.Key,
        group => group
          .SelectMany(route => route.Points)
          .Select(point => NormalizeScenarioPoint(systemsById[group.Key], point))
          .ToList(),
        StringComparer.Ordinal);
  }

  private static List<ShipRuntime> CreateShips(
    ScenarioDefinition scenario,
    IReadOnlyDictionary<string, SystemRuntime> systemsById,
    IReadOnlyCollection<CelestialRuntime> celestials,
    BalanceDefinition balance,
    IReadOnlyDictionary<string, ShipDefinition> shipDefinitions,
    IReadOnlyDictionary<string, List<Vector3>> patrolRoutes,
    IReadOnlyCollection<StationRuntime> stations,
    StationRuntime? refinery)
  {
    var ships = new List<ShipRuntime>();
    var shipIdCounter = 0;

    foreach (var formation in scenario.ShipFormations)
    {
      if (!shipDefinitions.TryGetValue(formation.ShipId, out var definition))
      {
        continue;
      }

      for (var index = 0; index < formation.Count; index += 1)
      {
        var offset = new Vector3((index % 3) * 18f, balance.YPlane, (index / 3) * 18f);
        var position = Add(NormalizeScenarioPoint(systemsById[formation.SystemId], formation.Center), offset);

        ships.Add(new ShipRuntime
        {
          Id = $"ship-{++shipIdCounter}",
          SystemId = formation.SystemId,
          Definition = definition,
          FactionId = formation.FactionId ?? DefaultFactionId,
          Position = position,
          TargetPosition = position,
          SpatialState = SpatialBuilder.CreateInitialShipSpatialState(formation.SystemId, position, celestials),
          DefaultBehavior = WorldSeedingService.CreateBehavior(
            definition,
            formation.SystemId,
            formation.FactionId ?? DefaultFactionId,
            scenario,
            patrolRoutes,
            stations,
            refinery),
          ControllerTask = new ControllerTaskRuntime { Kind = ControllerTaskKind.Idle, Threshold = balance.ArrivalThreshold, Status = WorkStatus.Pending },
          Health = definition.MaxHealth,
        });

        foreach (var (itemId, amount) in formation.StartingInventory)
        {
          if (amount > 0f)
          {
            ships[^1].Inventory[itemId] = amount;
          }
        }
      }
    }

    return ships;
  }
}