Spawn In Bed

namespace YangBed { public sealed class YangBedModSystem : ModSystem { private ICoreServerAPI sapi; private readonly Dictionary<string, Action> mountListenersByPlayerUid = new Dictionary<string, Action>();

public override bool ShouldLoad(EnumAppSide forSide) { return forSide == EnumAppSide.Server; }

public override void StartServerSide(ICoreServerAPI api) { sapi = api;

sapi.Event.PlayerNowPlaying += OnPlayerNowPlaying; sapi.Event.PlayerDisconnect += OnPlayerDisconnect; }

private void OnPlayerNowPlaying(IServerPlayer player) { if (player == null || player.Entity == null) return;

UnregisterPlayer(player); string playerUid = player.PlayerUID;

// Beds set tiredness.isSleeping via an enqueued task during mount, one-shot callback. No need to check anything each tick or something stupid like that. Action listener = () => { sapi.Event.RegisterCallback(_ => TrySetSpawnFromSleep(playerUid), 50); };

mountListenersByPlayerUid[playerUid] = listener; player.Entity.WatchedAttributes.RegisterModifiedListener("mountedOn", listener); }

private void OnPlayerDisconnect(IServerPlayer player) { if (player == null) return; UnregisterPlayer(player); }

private void UnregisterPlayer(IServerPlayer player) { if (!mountListenersByPlayerUid.TryGetValue(player.PlayerUID, out Action listener)) return;

player.Entity?.WatchedAttributes.UnregisterListener(listener); mountListenersByPlayerUid.Remove(player.PlayerUID); }

private void TrySetSpawnFromSleep(string playerUid) { IServerPlayer player = sapi.World.PlayerByUid(playerUid) as IServerPlayer; if (player == null || player.Entity == null) return;

EntityPlayer entity = player.Entity; if (entity.MountedOn == null || !IsSleeping(entity)) return;

EntityPos seatPos = entity.MountedOn.SeatPosition ?? entity.Pos; player.SetSpawnPosition(new PlayerSpawnPos { x = (int)Math.Floor(seatPos.X), y = (int)Math.Ceiling(seatPos.InternalY), z = (int)Math.Floor(seatPos.Z), yaw = seatPos.Yaw, pitch = 0f, roll = 0f, RemainingUses = -1 });

player.SendIngameError("yangbed-confirm", "Re-spawn Point Set..."); }

private static bool IsSleeping(EntityPlayer entity) { ITreeAttribute tiredness = entity.WatchedAttributes.GetTreeAttribute("tiredness"); return tiredness != null && tiredness.GetInt("isSleeping") > 0; }

public override void Dispose() { if (sapi != null) { sapi.Event.PlayerNowPlaying -= OnPlayerNowPlaying; sapi.Event.PlayerDisconnect -= OnPlayerDisconnect; foreach (IPlayer player in sapi.World.AllOnlinePlayers) { if (player is IServerPlayer serverPlayer) { UnregisterPlayer(serverPlayer); } } }