Refactor how jobs are handed out (#5422)
* Completely refactor how job spawning works * Remove remains of old system. * Squash the final bug, cleanup. * Attempt to fix tests * Adjusts packed's round-start crew roster, re-enables a bunch of old roles. Also adds the Central Command Official as a proper role. * pretty up ui * refactor StationSystem into the correct folder & namespace. * remove a log, make sure the lobby gets updated if a new map is spontaneously added. * re-add accidentally removed log * We do a little logging * we do a little resolving * we do a little documenting * Renamed OverflowJob to FallbackOverflowJob Allows stations to configure their own roundstart overflow job list. * narrator: it did not compile * oops * support having no overflow jobs * filescope for consistency * small fixes * Bumps a few role counts for Packed, namely engis * log moment * E * Update Resources/Prototypes/Entities/Objects/Misc/identification_cards.yml Co-authored-by: Leon Friedrich <60421075+ElectroJr@users.noreply.github.com> * Update Content.Server/Maps/GameMapPrototype.cs Co-authored-by: Leon Friedrich <60421075+ElectroJr@users.noreply.github.com> * factored job logic, cleanup. * e * Address reviews * Remove the concept of a "default" grid. It has no future in our new multi-station world * why was clickable using that in the first place * fix bad evil bug that almost slipped through also adds chemist * rms obsolete things from chemist * Adds a sanity fallback * address reviews * adds ability to set name * fuck * cleanup joingame
This commit is contained in:
Moony
GitHub
165
Content.Server/Station/StationSystem.cs
Normal file
165
Content.Server/Station/StationSystem.cs
Normal file
@@ -0,0 +1,165 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using Content.Server.GameTicking;
|
||||
using Content.Server.Maps;
|
||||
using Content.Shared.Station;
|
||||
using Robust.Shared.GameObjects;
|
||||
using Robust.Shared.IoC;
|
||||
using Robust.Shared.Log;
|
||||
|
||||
namespace Content.Server.Station;
|
||||
|
||||
/// <summary>
|
||||
/// System that manages the jobs available on a station, and maybe other things later.
|
||||
/// </summary>
|
||||
public class StationSystem : EntitySystem
|
||||
{
|
||||
[Dependency] private GameTicker _gameTicker = default!;
|
||||
private uint _idCounter = 1;
|
||||
|
||||
private Dictionary<StationId, StationInfoData> _stationInfo = new();
|
||||
/// <summary>
|
||||
/// List of stations for the current round.
|
||||
/// </summary>
|
||||
public IReadOnlyDictionary<StationId, StationInfoData> StationInfo => _stationInfo;
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
SubscribeLocalEvent<GameRunLevelChangedEvent>(OnRoundEnd);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Cleans up station info.
|
||||
/// </summary>
|
||||
private void OnRoundEnd(GameRunLevelChangedEvent eventArgs)
|
||||
{
|
||||
if (eventArgs.New == GameRunLevel.PostRound)
|
||||
_stationInfo = new();
|
||||
}
|
||||
|
||||
public class StationInfoData
|
||||
{
|
||||
public readonly string Name;
|
||||
|
||||
/// <summary>
|
||||
/// Job list associated with the game map.
|
||||
/// </summary>
|
||||
public readonly GameMapPrototype MapPrototype;
|
||||
|
||||
/// <summary>
|
||||
/// The round job list.
|
||||
/// </summary>
|
||||
private readonly Dictionary<string, int> _jobList;
|
||||
|
||||
public IReadOnlyDictionary<string, int> JobList => _jobList;
|
||||
|
||||
public StationInfoData(string name, GameMapPrototype mapPrototype, Dictionary<string, int> jobList)
|
||||
{
|
||||
Name = name;
|
||||
MapPrototype = mapPrototype;
|
||||
_jobList = jobList;
|
||||
}
|
||||
|
||||
public bool TryAssignJob(string jobName)
|
||||
{
|
||||
if (_jobList.ContainsKey(jobName))
|
||||
{
|
||||
switch (_jobList[jobName])
|
||||
{
|
||||
case > 0:
|
||||
_jobList[jobName]--;
|
||||
return true;
|
||||
case -1:
|
||||
return true;
|
||||
default:
|
||||
return false;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Creates a new station and attaches it to the given grid.
|
||||
/// </summary>
|
||||
/// <param name="mapGrid">grid to attach to</param>
|
||||
/// <param name="mapPrototype">game map prototype of the station</param>
|
||||
/// <param name="stationName">name of the station to assign, if not the default</param>
|
||||
/// <param name="gridComponent">optional grid component of the grid.</param>
|
||||
/// <returns>The ID of the resulting station</returns>
|
||||
/// <exception cref="ArgumentException">Thrown when the given entity is not a grid.</exception>
|
||||
public StationId InitialSetupStationGrid(EntityUid mapGrid, GameMapPrototype mapPrototype, string? stationName = null, IMapGridComponent? gridComponent = null)
|
||||
{
|
||||
if (!Resolve(mapGrid, ref gridComponent))
|
||||
throw new ArgumentException("Tried to initialize a station on a non-grid entity!");
|
||||
|
||||
var jobListDict = mapPrototype.AvailableJobs.ToDictionary(x => x.Key, x => x.Value[1]);
|
||||
var id = AllocateStationInfo();
|
||||
|
||||
_stationInfo[id] = new StationInfoData(stationName ?? mapPrototype.MapName, mapPrototype, jobListDict);
|
||||
var station = EntityManager.AddComponent<StationComponent>(mapGrid);
|
||||
station.Station = id;
|
||||
|
||||
_gameTicker.UpdateJobsAvailable(); // new station means new jobs, tell any lobby-goers.
|
||||
|
||||
Logger.InfoS("stations",
|
||||
$"Setting up new {mapPrototype.ID} called {mapPrototype.MapName} on grid {mapGrid}:{gridComponent.GridIndex}");
|
||||
|
||||
return id;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Adds the given grid to the given station.
|
||||
/// </summary>
|
||||
/// <param name="mapGrid">grid to attach</param>
|
||||
/// <param name="station">station to attach the grid to</param>
|
||||
/// <param name="gridComponent">optional grid component of the grid.</param>
|
||||
/// <exception cref="ArgumentException">Thrown when the given entity is not a grid.</exception>
|
||||
public void AddGridToStation(EntityUid mapGrid, StationId station, IMapGridComponent? gridComponent = null)
|
||||
{
|
||||
if (!Resolve(mapGrid, ref gridComponent))
|
||||
throw new ArgumentException("Tried to initialize a station on a non-grid entity!");
|
||||
var stationComponent = EntityManager.AddComponent<StationComponent>(mapGrid);
|
||||
stationComponent.Station = station;
|
||||
|
||||
Logger.InfoS("stations", $"Adding grid {mapGrid}:{gridComponent.GridIndex} to station {station} named {_stationInfo[station].Name}");
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Attempts to assign a job on the given station.
|
||||
/// </summary>
|
||||
/// <param name="stationId">station to assign to</param>
|
||||
/// <param name="jobName">name of the job</param>
|
||||
/// <returns>assignment success</returns>
|
||||
public bool TryAssignJobToStation(StationId stationId, string jobName)
|
||||
{
|
||||
if (stationId != StationId.Invalid)
|
||||
return _stationInfo[stationId].TryAssignJob(jobName);
|
||||
else
|
||||
return false;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Checks if the given job is available.
|
||||
/// </summary>
|
||||
/// <param name="stationId">station to check</param>
|
||||
/// <param name="jobName">name of the job</param>
|
||||
/// <returns>job availability</returns>
|
||||
public bool IsJobAvailableOnStation(StationId stationId, string jobName)
|
||||
{
|
||||
if (_stationInfo[stationId].JobList.TryGetValue(jobName, out var amount))
|
||||
return amount != 0;
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
private StationId AllocateStationInfo()
|
||||
{
|
||||
return new StationId(_idCounter++);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user