This repository has been archived by the owner on Nov 1, 2024. It is now read-only.
forked from new-frontiers-14/frontier-station-14
-
Notifications
You must be signed in to change notification settings - Fork 38
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
7 changed files
with
130 additions
and
12 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
73 changes: 73 additions & 0 deletions
73
Content.Server/Mech/Equipment/EntitySystems/CorvaxMechGunSystem.cs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,73 @@ | ||
/* | ||
This code under CC-BY-SA 3.0, and taken from https://github.com/space-wizards/space-station-14/pull/19263/files#diff-b878680d623afff2ab173822240fcb66e3c989d11e9786490e6b91947383c198. | ||
*/ | ||
|
||
using Content.Server.Mech.Systems; | ||
using Content.Server.Power.Components; | ||
using Content.Server.Power.EntitySystems; | ||
using Content.Shared.Mech.Components; | ||
using Content.Shared.Mech.Equipment.Components; | ||
using Content.Shared.Throwing; | ||
using Content.Shared.Weapons.Ranged.Systems; | ||
using Robust.Shared.Random; | ||
|
||
namespace Content.Server.Mech.Equipment.EntitySystems; | ||
public sealed class CorvaxMechGunSystem : EntitySystem | ||
{ | ||
[Dependency] private readonly IRobustRandom _random = default!; | ||
[Dependency] private readonly ThrowingSystem _throwing = default!; | ||
[Dependency] private readonly MechSystem _mech = default!; | ||
[Dependency] private readonly BatterySystem _battery = default!; | ||
|
||
public override void Initialize() | ||
{ | ||
base.Initialize(); | ||
SubscribeLocalEvent<MechEquipmentComponent, GunShotEvent>(MechGunShot); | ||
} | ||
|
||
private void MechGunShot(EntityUid uid, MechEquipmentComponent component, ref GunShotEvent args) | ||
{ | ||
if (!component.EquipmentOwner.HasValue) | ||
return; | ||
|
||
if (!TryComp<MechComponent>(component.EquipmentOwner.Value, out var mech)) | ||
return; | ||
|
||
if (TryComp<BatteryComponent>(uid, out var battery)) | ||
{ | ||
ChargeGunBattery(uid, battery); | ||
return; | ||
} | ||
|
||
foreach (var (ent, _) in args.Ammo) | ||
{ | ||
if (ent.HasValue && mech.EquipmentContainer.Contains(ent.Value)) | ||
{ | ||
mech.EquipmentContainer.Remove(ent.Value); | ||
_throwing.TryThrow(ent.Value, _random.NextVector2(), _random.Next(5)); | ||
} | ||
} | ||
} | ||
|
||
private void ChargeGunBattery(EntityUid uid, BatteryComponent component) | ||
{ | ||
if (!TryComp<MechEquipmentComponent>(uid, out var mechEquipment) || !mechEquipment.EquipmentOwner.HasValue) | ||
return; | ||
|
||
if (!TryComp<MechComponent>(mechEquipment.EquipmentOwner.Value, out var mech)) | ||
return; | ||
|
||
var maxCharge = component.MaxCharge; | ||
var currentCharge = component.CurrentCharge; | ||
|
||
var chargeDelta = maxCharge - currentCharge; | ||
|
||
if (chargeDelta <= 0 || mech.Energy - chargeDelta < 0) | ||
return; | ||
|
||
if (!_mech.TryChangeEnergy(mechEquipment.EquipmentOwner.Value, -chargeDelta, mech)) | ||
return; | ||
|
||
_battery.SetCharge(uid, component.MaxCharge, component); | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters