quantum-space-buddies/QSB/PlayerInfo.cs

using QSB.Animation;
using QSB.Tools;
using QSB.TransformSync;
using QSB.Utility;
using System.Linq;
using UnityEngine;

namespace QSB
{
    public class PlayerInfo
    {
        public uint PlayerId { get; }
        public GameObject Camera { get; set; }
        public GameObject Body { get; set; }

        // Tools
        public GameObject ProbeBody { get; set; }
        public QSBProbe Probe { get; set; }
        public QSBFlashlight FlashLight => Camera?.GetComponentInChildren<QSBFlashlight>();
        public QSBTool Signalscope => GetToolByType(ToolType.Signalscope);
        public QSBTool Translator => GetToolByType(ToolType.Translator);
        public QSBTool ProbeLauncher => GetToolByType(ToolType.ProbeLauncher);

        public PlayerHUDMarker HudMarker { get; set; }
        public string Name { get; set; }
        public bool IsReady { get; set; }
        public int CurrentDialogueID { get; set; }
        public GameObject CurrentDialogueBox { get; set; }
        public State State { get; set; }

        public PlayerInfo(uint id)
        {
            DebugLog.DebugWrite($"Creating PlayerInfo with id {id}");
            PlayerId = id;
            CurrentDialogueID = -1;
        }

        public void UpdateState(State state, bool value)
        {
            var states = State;
            if (value)
            {
                FlagsHelper.Set(ref states, state);
            }
            else
            {
                FlagsHelper.Unset(ref states, state);
            }
            State = states;
        }

        public void UpdateStateObjects()
        {
            if (OWInput.GetInputMode() == InputMode.None)
            {
                return;
            }
            FlashLight?.UpdateState(FlagsHelper.IsSet(State, State.Flashlight));
            Translator?.ChangeEquipState(FlagsHelper.IsSet(State, State.Translator));
            ProbeLauncher?.ChangeEquipState(FlagsHelper.IsSet(State, State.ProbeLauncher));
            Signalscope?.ChangeEquipState(FlagsHelper.IsSet(State, State.Signalscope));
            QSB.Helper.Events.Unity.RunWhen(() => PlayerRegistry.GetSyncObject<AnimationSync>(PlayerId) != null,
                () => PlayerRegistry.GetSyncObject<AnimationSync>(PlayerId).SetSuitState(FlagsHelper.IsSet(State, State.Suit)));
        }

        public bool GetState(State state)
        {
            return FlagsHelper.IsSet(State, state);
        }

        private QSBTool GetToolByType(ToolType type)
        {
            return Camera?.GetComponentsInChildren<QSBTool>()
                .FirstOrDefault(x => x.Type == type);
        }
    }
}