SDM_LethalCompany_Mod/ScarletMansion/ScarletMansion/GamePatch/Items/ScarletFlashlight.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using UnityEngine;

namespace ScarletMansion.GamePatch.Items {

  // While I would prefer to use my own script,
  // It would cause too many problems I think
  // Due to mods that reserve flashlights and such
  public class ScarletFlashlight : FlashlightItem, IScarletItem {

    [Header("Colors")]
    public MeshRenderer[] crystalRenderers;
    public int colorIndex;
    private Color lightColor;
    private int fallbackFlashlightTypeID;

    public override void Start() {
      base.Start();

      fallbackFlashlightTypeID = flashlightTypeID;
      var flashlight = Assets.GetFlashlight(itemProperties);
      if (flashlight == null || flashlight.scarletHelmetIndex == -1){
        StartCoroutine(WaitForRealFlashlightID());
        return;
      }
      
      flashlightTypeID = flashlight.scarletHelmetIndex;
    }

    public override int GetItemDataToSave() {
      base.GetItemDataToSave();
      return colorIndex;
    }

    public override void LoadItemSaveData(int saveData) {
      base.LoadItemSaveData(saveData);
      UpdateSpecialProperties(saveData);
    }

    public override void EquipItem() {
      // test to prevent future BS
      if (playerHeldBy && flashlightTypeID >= playerHeldBy.allHelmetLights.Length){
        Plugin.logger.LogWarning($"FlashlightTypeID {flashlightTypeID} is invalid somehow. Reverting to ID {fallbackFlashlightTypeID}");
        flashlightTypeID = fallbackFlashlightTypeID;
      }

      base.EquipItem();

      // once we fall back, we can no longer call this
      if (flashlightTypeID != fallbackFlashlightTypeID)
        UpdateHelmetColor();
    }

    public void UpdateHelmetColor(){
      var light = playerHeldBy.allHelmetLights[flashlightTypeID];
      light.color = lightColor;
    }

    public void UpdateSpecialProperties(int colorIndex) {
      this.colorIndex = colorIndex;

      bulbLight = new Material(bulbLight);
      bulbDark = new Material(bulbDark);

      var color = FlandreCrystal.colorVariants[colorIndex];
      lightColor = flashlightBulb.color = Color.Lerp(flashlightBulb.color, color, 0.5f);
      flashlightBulbGlow.color = Color.Lerp(flashlightBulbGlow.color, color, 0.5f);
      bulbLight.color = Color.Lerp(bulbLight.color, color, 0.75f);
      bulbDark.color = Color.Lerp(bulbDark.color, color, 0.75f);

      SwitchFlashlight(isBeingUsed);

      foreach(var r in crystalRenderers){
        r.material.color = color;
      }
    }

    public IEnumerator WaitForRealFlashlightID(){
      Plugin.logger.LogDebug("Waiting for real helmet index for scarlet flashlight");
      var t = Time.realtimeSinceStartup + 8f;
      while(true) {
        yield return null;

        var flashlight = Assets.GetFlashlight(itemProperties);
        if (flashlight != null && flashlight.scarletHelmetIndex != -1) {
          Plugin.logger.LogDebug("Got real helmet index for scarlet flashlight");
          flashlightTypeID = flashlight.scarletHelmetIndex;
          yield break;
        }

        if (Time.realtimeSinceStartup > t) {
          Plugin.logger.LogWarning("Failed to get helmet index for scarlet flashlight");
          yield break;
        }
      }
    }

  }
}