Revert 'Revert 'Solution Entities'' (#23168)
This commit is contained in:
@@ -1,34 +1,34 @@
|
||||
using System.Linq;
|
||||
using Content.Server.Chemistry.Containers.EntitySystems;
|
||||
using Content.Server.Weapons.Ranged.Components;
|
||||
using Content.Shared.Chemistry.Components;
|
||||
using Content.Shared.Chemistry.EntitySystems;
|
||||
using Content.Shared.Weapons.Ranged.Events;
|
||||
using System.Linq;
|
||||
|
||||
namespace Content.Server.Weapons.Ranged.Systems
|
||||
{
|
||||
public sealed class ChemicalAmmoSystem : EntitySystem
|
||||
{
|
||||
[Dependency] private readonly SolutionContainerSystem _solutionSystem = default!;
|
||||
[Dependency] private readonly SolutionContainerSystem _solutionContainerSystem = default!;
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
SubscribeLocalEvent<ChemicalAmmoComponent, AmmoShotEvent>(OnFire);
|
||||
}
|
||||
|
||||
private void OnFire(EntityUid uid, ChemicalAmmoComponent component, AmmoShotEvent args)
|
||||
private void OnFire(Entity<ChemicalAmmoComponent> entity, ref AmmoShotEvent args)
|
||||
{
|
||||
if (!_solutionSystem.TryGetSolution(uid, component.SolutionName, out var ammoSolution))
|
||||
if (!_solutionContainerSystem.TryGetSolution(entity.Owner, entity.Comp.SolutionName, out var ammoSoln, out var ammoSolution))
|
||||
return;
|
||||
|
||||
var projectiles = args.FiredProjectiles;
|
||||
|
||||
var projectileSolutionContainers = new List<(EntityUid, Solution)>();
|
||||
var projectileSolutionContainers = new List<(EntityUid, Entity<SolutionComponent>)>();
|
||||
foreach (var projectile in projectiles)
|
||||
{
|
||||
if (_solutionSystem
|
||||
.TryGetSolution(projectile, component.SolutionName, out var projectileSolutionContainer))
|
||||
if (_solutionContainerSystem
|
||||
.TryGetSolution(projectile, entity.Comp.SolutionName, out var projectileSoln, out _))
|
||||
{
|
||||
projectileSolutionContainers.Add((uid, projectileSolutionContainer));
|
||||
projectileSolutionContainers.Add((projectile, projectileSoln.Value));
|
||||
}
|
||||
}
|
||||
|
||||
@@ -37,13 +37,13 @@ namespace Content.Server.Weapons.Ranged.Systems
|
||||
|
||||
var solutionPerProjectile = ammoSolution.Volume * (1 / projectileSolutionContainers.Count);
|
||||
|
||||
foreach (var (projectileUid, projectileSolution) in projectileSolutionContainers)
|
||||
foreach (var (_, projectileSolution) in projectileSolutionContainers)
|
||||
{
|
||||
var solutionToTransfer = _solutionSystem.SplitSolution(uid, ammoSolution, solutionPerProjectile);
|
||||
_solutionSystem.TryAddSolution(projectileUid, projectileSolution, solutionToTransfer);
|
||||
var solutionToTransfer = _solutionContainerSystem.SplitSolution(ammoSoln.Value, solutionPerProjectile);
|
||||
_solutionContainerSystem.TryAddSolution(projectileSolution, solutionToTransfer);
|
||||
}
|
||||
|
||||
_solutionSystem.RemoveAllSolution(uid, ammoSolution);
|
||||
_solutionContainerSystem.RemoveAllSolution(ammoSoln.Value);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user