ItemCreatorSystem.cs 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. using Content.Server.Ninja.Events;
  2. using Content.Server.Power.EntitySystems;
  3. using Content.Shared.Hands.EntitySystems;
  4. using Content.Shared.Ninja.Components;
  5. using Content.Shared.Ninja.Systems;
  6. using Content.Shared.Popups;
  7. namespace Content.Server.Ninja.Systems;
  8. public sealed class ItemCreatorSystem : SharedItemCreatorSystem
  9. {
  10. [Dependency] private readonly BatterySystem _battery = default!;
  11. [Dependency] private readonly SharedHandsSystem _hands = default!;
  12. [Dependency] private readonly SharedPopupSystem _popup = default!;
  13. public override void Initialize()
  14. {
  15. base.Initialize();
  16. SubscribeLocalEvent<ItemCreatorComponent, CreateItemEvent>(OnCreateItem);
  17. SubscribeLocalEvent<ItemCreatorComponent, NinjaBatteryChangedEvent>(OnBatteryChanged);
  18. }
  19. private void OnCreateItem(Entity<ItemCreatorComponent> ent, ref CreateItemEvent args)
  20. {
  21. var (uid, comp) = ent;
  22. if (comp.Battery is not {} battery)
  23. return;
  24. args.Handled = true;
  25. var user = args.Performer;
  26. if (!_battery.TryUseCharge(battery, comp.Charge))
  27. {
  28. _popup.PopupEntity(Loc.GetString(comp.NoPowerPopup), user, user);
  29. return;
  30. }
  31. var ev = new CreateItemAttemptEvent(user);
  32. RaiseLocalEvent(uid, ref ev);
  33. if (ev.Cancelled)
  34. return;
  35. // try to put throwing star in hand, otherwise it goes on the ground
  36. var star = Spawn(comp.SpawnedPrototype, Transform(user).Coordinates);
  37. _hands.TryPickupAnyHand(user, star);
  38. }
  39. private void OnBatteryChanged(Entity<ItemCreatorComponent> ent, ref NinjaBatteryChangedEvent args)
  40. {
  41. if (ent.Comp.Battery == args.Battery)
  42. return;
  43. ent.Comp.Battery = args.Battery;
  44. Dirty(ent, ent.Comp);
  45. }
  46. }