using System.Collections.Generic;
using UnityEngine;
using VRC.SDK3.Avatars.Components;
using VRC.SDK3.Dynamics.Constraint.Components;
using TMPro;
using VRC.Dynamics;


#if UNITY_EDITOR
using UnityEditor;
#endif

namespace gay.lilyy.AviLabels
{
  
  [AddComponentMenu("LillithRosePup/Avatar Label/Driver")]
  public class AviLabelParent : MonoBehaviour, VRC.SDKBase.IEditorOnly
  {
    public Color defaultColor = Color.white;
    public TMP_FontAsset defaultFont = null;
    public void UpdateLabels()
    {
      List<Transform> childrenToDestroy = new List<Transform>();
      foreach (Transform child in transform)
      {
        childrenToDestroy.Add(child);
      }
      foreach (Transform child in childrenToDestroy)
      {
        DestroyImmediate(child.gameObject);
      }
      var avatarDescriptors = GameObject.FindObjectsOfType<VRCAvatarDescriptor>();
      foreach (var avatarDescriptor in avatarDescriptors)
      {
        LabelConfig labelConfig = LabelConfig.GetDefault();
        if (avatarDescriptor.GetComponent<LabelConfig>() != null)
        {
          labelConfig = avatarDescriptor.GetComponent<LabelConfig>();
        }
        if (labelConfig.font == null)
        {
          labelConfig.font = defaultFont;
        }
        if (labelConfig.color == Color.white)
        {
          labelConfig.color = defaultColor;
        }
        var child = new GameObject(avatarDescriptor.name);
        child.transform.SetParent(transform);
        child.transform.localPosition = Vector3.zero;
        child.transform.localRotation = Quaternion.identity;
        child.transform.localScale = Vector3.one;


        var parentConstraint = child.AddComponent<VRCParentConstraint>();
        parentConstraint.AffectsPositionY = false;
        parentConstraint.AffectsRotationY = false;
        parentConstraint.Sources.Add(new VRCConstraintSource {
          SourceTransform = avatarDescriptor.transform,
          Weight = 1
        });
        parentConstraint.ActivateConstraint();
        parentConstraint.ZeroConstraint();


        var textMeshPro = child.AddComponent<TextMeshPro>();
        if (labelConfig.font != null)
        {
          textMeshPro.font = labelConfig.font;
        }
        textMeshPro.color = labelConfig.color;
        textMeshPro.text = !string.IsNullOrEmpty(labelConfig.overrideName) ? labelConfig.overrideName : avatarDescriptor.name;
        textMeshPro.alignment = TextAlignmentOptions.Center;
        textMeshPro.fontSize = 1;
      }
    }
  }

  #if UNITY_EDITOR
  [CustomEditor(typeof(AviLabelParent))]
  public class AviLabelParentInspector : Editor
  {
    public override void OnInspectorGUI()
    {
      base.OnInspectorGUI();
      if (GUILayout.Button("Update Labels"))
      {
        ((AviLabelParent)target).UpdateLabels();
      }
    }
  }
  #endif
}