quantum-space-buddies/QSB/Tools/QSBFlashlight.cs

using OWML.ModHelper.Events;
using UnityEngine;

namespace QSB.Tools
{
    public class QSBFlashlight : MonoBehaviour
    {
        private OWLight2[] _lights;
        private OWLight2 _illuminationCheckLight;
        private Transform _root;
        private Transform _basePivot;
        private Transform _wobblePivot;

        private bool _flashlightOn;
        private Vector3 _baseForward;
        private Quaternion _baseRotation;

        private void Start()
        {
            _baseForward = _basePivot.forward;
            _baseRotation = _basePivot.rotation;
        }

        public void Init(Flashlight oldComponent)
        {
            _lights = oldComponent.GetValue<OWLight2[]>("_lights");
            _illuminationCheckLight = oldComponent.GetValue<OWLight2>("_illuminationCheckLight");
            _root = oldComponent.GetValue<Transform>("_root");
            _basePivot = oldComponent.GetValue<Transform>("_basePivot");
            _wobblePivot = oldComponent.GetValue<Transform>("_wobblePivot");
        }

        public void TurnOn()
        {
            if (_flashlightOn)
            {
                return;
            }
            foreach (var light in _lights)
            {
                light.GetLight().enabled = true;
            }
            _flashlightOn = true;
            var rotation = _root.rotation;
            _basePivot.rotation = rotation;
            _baseRotation = rotation;
            _baseForward = _basePivot.forward;
        }

        public void TurnOff()
        {
            if (!_flashlightOn)
            {
                return;
            }
            foreach (var light in _lights)
            {
                light.GetLight().enabled = false;
            }
            _flashlightOn = false;
        }

        private void FixedUpdate()
        {
            var lhs = Quaternion.FromToRotation(_basePivot.up, _root.up) * Quaternion.FromToRotation(_baseForward, _root.forward);
            var b = lhs * _baseRotation;
            _baseRotation = Quaternion.Slerp(_baseRotation, b, 6f * Time.deltaTime);
            _basePivot.rotation = _baseRotation;
            _baseForward = _basePivot.forward;
            _wobblePivot.localRotation = OWUtilities.GetWobbleRotation(0.3f, 0.15f) * Quaternion.identity;
        }
    }
}