Optimize GUIColorDrawer

Author: vanifatovvlad

Editor.Extras/Drawers/GUIColorDrawer.cs

@@ -1,4 +1,5 @@
-using TriInspector;
+using JetBrains.Annotations;
+using TriInspector;
 using TriInspector.Drawers;
 using TriInspector.Resolvers;
 using UnityEngine;
@@ -9,27 +10,33 @@ namespace TriInspector.Drawers
 {
     public class GUIColorDrawer : TriAttributeDrawer<GUIColorAttribute>
     {
-        public override void OnGUI(Rect position, TriProperty property, TriElement next)
+        [CanBeNull] private ValueResolver<Color> _colorResolver;
+
+        public override TriExtensionInitializationResult Initialize(TriPropertyDefinition propertyDefinition)
         {
-            var oldColor = GUI.color;
-            var newColor = Color.white;
-            
-            if (string.IsNullOrEmpty(Attribute.GetColor))
+            if (!string.IsNullOrEmpty(Attribute.GetColor))
             {
-                newColor = Attribute.Color;
+                _colorResolver = ValueResolver.Resolve<Color>(propertyDefinition, Attribute.GetColor);
             }
-            else
+
+            if (_colorResolver != null && _colorResolver.TryGetErrorString(out var error))
             {
-                var colorResolver = ValueResolver.Resolve<Color>(property.Definition, Attribute.GetColor ?? "");
-                
-                newColor = colorResolver.GetValue(property, Color.white);
+                return error;
             }
-            
+
+            return TriExtensionInitializationResult.Ok;
+        }
+
+        public override void OnGUI(Rect position, TriProperty property, TriElement next)
+        {
+            var oldColor = GUI.color;
+            var newColor = _colorResolver?.GetValue(property, Color.white) ?? Attribute.Color;
+
             GUI.color = newColor;
             GUI.contentColor = newColor;
-            
+
             next.OnGUI(position);
-            
+
             GUI.color = oldColor;
             GUI.contentColor = oldColor;
         }