EntityUidConverter.cs 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. using System.Text.Json;
  2. using Robust.Shared.Player;
  3. namespace Content.Server.Administration.Logs.Converters;
  4. [AdminLogConverter]
  5. public sealed class EntityUidConverter : AdminLogConverter<EntityUid>
  6. {
  7. // System.Text.Json actually keeps hold of your JsonSerializerOption instances in a cache on .NET 7.
  8. // Use a weak reference to avoid holding server instances live too long in integration tests.
  9. private WeakReference<IEntityManager> _entityManager = default!;
  10. public override void Init(IDependencyCollection dependencies)
  11. {
  12. _entityManager = new WeakReference<IEntityManager>(dependencies.Resolve<IEntityManager>());
  13. }
  14. public static void Write(Utf8JsonWriter writer, EntityUid value, JsonSerializerOptions options, IEntityManager entities)
  15. {
  16. writer.WriteStartObject();
  17. writer.WriteNumber("id", (int) value);
  18. if (entities.TryGetComponent(value, out MetaDataComponent? metaData))
  19. {
  20. writer.WriteString("name", metaData.EntityName);
  21. }
  22. if (entities.TryGetComponent(value, out ActorComponent? actor))
  23. {
  24. writer.WriteString("player", actor.PlayerSession.UserId.UserId);
  25. }
  26. writer.WriteEndObject();
  27. }
  28. public override void Write(Utf8JsonWriter writer, EntityUid value, JsonSerializerOptions options)
  29. {
  30. if (!_entityManager.TryGetTarget(out var entityManager))
  31. throw new InvalidOperationException("EntityManager got garbage collected!");
  32. Write(writer, value, options, entityManager);
  33. }
  34. }