From 1cd8e35c255a516a1ff31eca61368aeaf34e4314 Mon Sep 17 00:00:00 2001
From: Ed <96445749+TheShuEd@users.noreply.github.com>
Date: Sat, 30 Mar 2024 01:54:31 +0300
Subject: [PATCH] Roundstart planetmap (#2)

* planetmap component

* refactor code
---
 .../SpawnMapBiome/StationBiomeComponent.cs    | 19 ++++++++++
 .../SpawnMapBiome/StationBiomeSystem.cs       | 35 +++++++++++++++++++
 Resources/Prototypes/Maps/debug.yml           |  1 +
 3 files changed, 55 insertions(+)
 create mode 100644 Content.Server/CrystallPunk/SpawnMapBiome/StationBiomeComponent.cs
 create mode 100644 Content.Server/CrystallPunk/SpawnMapBiome/StationBiomeSystem.cs

diff --git a/Content.Server/CrystallPunk/SpawnMapBiome/StationBiomeComponent.cs b/Content.Server/CrystallPunk/SpawnMapBiome/StationBiomeComponent.cs
new file mode 100644
index 0000000000..a61f30b370
--- /dev/null
+++ b/Content.Server/CrystallPunk/SpawnMapBiome/StationBiomeComponent.cs
@@ -0,0 +1,19 @@
+using Content.Shared.Parallax.Biomes;
+using Robust.Shared.Prototypes;
+
+namespace Content.Server.Corvax.CrystallPunk.SpawnMapBiome;
+
+/// <summary>
+/// allows you to initialize a planet on a specific map at initialization time.
+/// </summary>
+
+[RegisterComponent, Access(typeof(StationBiomeSystem))]
+public sealed partial class StationBiomeComponent : Component
+{
+    [DataField]
+    public ProtoId<BiomeTemplatePrototype> Biome = "Grasslands";
+
+    // If null, its random
+    [DataField]
+    public int? Seed = null;
+}
diff --git a/Content.Server/CrystallPunk/SpawnMapBiome/StationBiomeSystem.cs b/Content.Server/CrystallPunk/SpawnMapBiome/StationBiomeSystem.cs
new file mode 100644
index 0000000000..c794474e95
--- /dev/null
+++ b/Content.Server/CrystallPunk/SpawnMapBiome/StationBiomeSystem.cs
@@ -0,0 +1,35 @@
+using Content.Server.Parallax;
+using Content.Server.Station.Components;
+using Content.Server.Station.Events;
+using Content.Server.Station.Systems;
+using Robust.Shared.Map;
+using Robust.Shared.Prototypes;
+
+namespace Content.Server.Corvax.CrystallPunk.SpawnMapBiome;
+public sealed partial class StationBiomeSystem : EntitySystem
+{
+    [Dependency] private readonly BiomeSystem _biome = default!;
+    [Dependency] private readonly IMapManager _mapManager = default!;
+    [Dependency] private readonly IPrototypeManager _proto = default!;
+    [Dependency] private readonly StationSystem _station = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+        SubscribeLocalEvent<StationBiomeComponent, StationPostInitEvent>(OnStationPostInit);
+    }
+
+    private void OnStationPostInit(Entity<StationBiomeComponent> map, ref StationPostInitEvent args)
+    {
+        if (!TryComp(map, out StationDataComponent? dataComp))
+            return;
+
+        var station = _station.GetLargestGrid(dataComp);
+        if (station == null) return;
+
+        var mapId = Transform(station.Value).MapID;
+        var mapUid = _mapManager.GetMapEntityId(mapId);
+
+        _biome.EnsurePlanet(mapUid, _proto.Index(map.Comp.Biome), map.Comp.Seed);
+    }
+}
diff --git a/Resources/Prototypes/Maps/debug.yml b/Resources/Prototypes/Maps/debug.yml
index 74cbf4ecc6..d3b414a73b 100644
--- a/Resources/Prototypes/Maps/debug.yml
+++ b/Resources/Prototypes/Maps/debug.yml
@@ -24,6 +24,7 @@
     Dev:
       stationProto: StandardNanotrasenStation
       components:
+        - type: StationBiome
         - type: StationNameSetup
           mapNameTemplate: "Dev"
         - type: StationJobs