69 lines
2.4 KiB
C#
69 lines
2.4 KiB
C#
using Content.Shared.Procedural;
|
|
using Content.Shared.Salvage.Expeditions;
|
|
|
|
namespace Content.Server.Salvage;
|
|
|
|
public sealed partial class SalvageSystem
|
|
{
|
|
private void OnSalvageClaimMessage(EntityUid uid, SalvageExpeditionConsoleComponent component, ClaimSalvageMessage args)
|
|
{
|
|
var station = _station.GetOwningStation(uid);
|
|
|
|
if (!TryComp<SalvageExpeditionDataComponent>(station, out var data) || data.Claimed)
|
|
return;
|
|
|
|
if (!data.Missions.TryGetValue(args.Index, out var missionparams))
|
|
return;
|
|
|
|
SpawnMission(missionparams, station.Value);
|
|
|
|
data.ActiveMission = args.Index;
|
|
var mission = GetMission(_prototypeManager.Index<SalvageDifficultyPrototype>(missionparams.Difficulty), missionparams.Seed);
|
|
data.NextOffer = _timing.CurTime + mission.Duration + TimeSpan.FromSeconds(1);
|
|
UpdateConsoles((station.Value, data));
|
|
}
|
|
|
|
private void OnSalvageConsoleInit(Entity<SalvageExpeditionConsoleComponent> console, ref ComponentInit args)
|
|
{
|
|
UpdateConsole(console);
|
|
}
|
|
|
|
private void OnSalvageConsoleParent(Entity<SalvageExpeditionConsoleComponent> console, ref EntParentChangedMessage args)
|
|
{
|
|
UpdateConsole(console);
|
|
}
|
|
|
|
private void UpdateConsoles(Entity<SalvageExpeditionDataComponent> component)
|
|
{
|
|
var state = GetState(component);
|
|
|
|
var query = AllEntityQuery<SalvageExpeditionConsoleComponent, UserInterfaceComponent, TransformComponent>();
|
|
while (query.MoveNext(out var uid, out _, out var uiComp, out var xform))
|
|
{
|
|
var station = _station.GetOwningStation(uid, xform);
|
|
|
|
if (station != component.Owner)
|
|
continue;
|
|
|
|
_ui.TrySetUiState(uid, SalvageConsoleUiKey.Expedition, state, ui: uiComp);
|
|
}
|
|
}
|
|
|
|
private void UpdateConsole(Entity<SalvageExpeditionConsoleComponent> component)
|
|
{
|
|
var station = _station.GetOwningStation(component);
|
|
SalvageExpeditionConsoleState state;
|
|
|
|
if (TryComp<SalvageExpeditionDataComponent>(station, out var dataComponent))
|
|
{
|
|
state = GetState(dataComponent);
|
|
}
|
|
else
|
|
{
|
|
state = new SalvageExpeditionConsoleState(TimeSpan.Zero, false, true, 0, new List<SalvageMissionParams>());
|
|
}
|
|
|
|
_ui.TrySetUiState(component, SalvageConsoleUiKey.Expedition, state);
|
|
}
|
|
}
|