Tri-Inspector/Editor/Utilities/TriGuiHelper.cs

using System;
using System.Collections.Generic;
using UnityEditor;
using UnityEngine;
using Object = UnityEngine.Object;

namespace TriInspector.Utilities
{
    public static class TriGuiHelper
    {
        private static readonly Stack<Object> TargetObjects = new Stack<Object>();

        internal static bool IsAnyEditorPushed()
        {
            return TargetObjects.Count > 0;
        }

        internal static bool IsEditorTargetPushed(Object obj)
        {
            foreach (var targetObject in TargetObjects)
            {
                if (targetObject == obj)
                {
                    return true;
                }
            }

            return false;
        }

        internal static EditorScope PushEditorTarget(Object obj)
        {
            return new EditorScope(obj);
        }

        public static LabelWidthScope PushLabelWidth(float labelWidth)
        {
            return new LabelWidthScope(labelWidth);
        }

        public static IndentLevelScope PushIndentLevel(int indent = 1)
        {
            return new IndentLevelScope(indent);
        }

        public static GuiColorScope PushColor(Color color)
        {
            return new GuiColorScope(color);
        }

        public readonly struct EditorScope : IDisposable
        {
            public EditorScope(Object obj)
            {
                TargetObjects.Push(obj);
            }

            public void Dispose()
            {
                TargetObjects.Pop();
            }
        }

        public readonly struct LabelWidthScope : IDisposable
        {
            private readonly float _oldLabelWidth;

            public LabelWidthScope(float labelWidth)
            {
                _oldLabelWidth = EditorGUIUtility.labelWidth;

                if (labelWidth > 0)
                {
                    EditorGUIUtility.labelWidth = labelWidth;
                }
            }

            public void Dispose()
            {
                EditorGUIUtility.labelWidth = _oldLabelWidth;
            }
        }

        public readonly struct IndentLevelScope : IDisposable
        {
            private readonly int _oldIndentLevel;

            public IndentLevelScope(int indentLevel)
            {
                _oldIndentLevel = EditorGUI.indentLevel;

                EditorGUI.indentLevel += indentLevel;
            }

            public void Dispose()
            {
                EditorGUI.indentLevel = _oldIndentLevel;
            }
        }

        public readonly struct GuiColorScope : IDisposable
        {
            private readonly Color _oldColor;

            public GuiColorScope(Color color)
            {
                _oldColor = GUI.color;

                GUI.color = color;
            }

            public void Dispose()
            {
                GUI.color = _oldColor;
            }
        }
    }
}