Replicated the weapon flying tick setup using resources

addons/forge/editor/statescript/resolvers/AttributeResolverEditor.cs

@@ -0,0 +1,197 @@
+// Copyright © Gamesmiths Guild.
+
+#if TOOLS
+using System;
+using Gamesmiths.Forge.Godot.Resources.Statescript;
+using Gamesmiths.Forge.Godot.Resources.Statescript.Resolvers;
+using Gamesmiths.Forge.Statescript;
+using Godot;
+
+namespace Gamesmiths.Forge.Godot.Editor.Statescript.Resolvers;
+
+/// <summary>
+/// Resolver editor that reads a value from a Forge entity attribute. Shows attribute set and attribute dropdowns.
+/// </summary>
+[Tool]
+internal sealed partial class AttributeResolverEditor : NodeEditorProperty
+{
+	private OptionButton? _setDropdown;
+	private OptionButton? _attributeDropdown;
+	private string _selectedSetClass = string.Empty;
+	private string _selectedAttribute = string.Empty;
+	private Action? _onChanged;
+
+	/// <inheritdoc/>
+	public override string DisplayName => "Attribute";
+
+	/// <inheritdoc/>
+	public override string ResolverTypeId => "Attribute";
+
+	/// <inheritdoc/>
+	public override bool IsCompatibleWith(Type expectedType)
+	{
+		return expectedType == typeof(int) || expectedType == typeof(Variant128);
+	}
+
+	/// <inheritdoc/>
+	public override void Setup(
+		StatescriptGraph graph,
+		StatescriptNodeProperty? property,
+		Type expectedType,
+		Action onChanged,
+		bool isArray)
+	{
+		_onChanged = onChanged;
+
+		SizeFlagsHorizontal = SizeFlags.ExpandFill;
+		var vBox = new VBoxContainer { SizeFlagsHorizontal = SizeFlags.ExpandFill };
+		AddChild(vBox);
+
+		if (property?.Resolver is AttributeResolverResource attrRes)
+		{
+			_selectedSetClass = attrRes.AttributeSetClass;
+			_selectedAttribute = attrRes.AttributeName;
+		}
+
+		var setRow = new HBoxContainer { SizeFlagsHorizontal = SizeFlags.ExpandFill };
+		vBox.AddChild(setRow);
+
+		setRow.AddChild(new Label
+		{
+			Text = "Set:",
+			CustomMinimumSize = new Vector2(45, 0),
+			HorizontalAlignment = HorizontalAlignment.Right,
+		});
+
+		_setDropdown = new OptionButton { SizeFlagsHorizontal = SizeFlags.ExpandFill };
+		PopulateSetDropdown();
+		setRow.AddChild(_setDropdown);
+
+		var attrRow = new HBoxContainer { SizeFlagsHorizontal = SizeFlags.ExpandFill };
+		vBox.AddChild(attrRow);
+
+		attrRow.AddChild(new Label
+		{
+			Text = "Attr:",
+			CustomMinimumSize = new Vector2(45, 0),
+			HorizontalAlignment = HorizontalAlignment.Right,
+		});
+
+		_attributeDropdown = new OptionButton { SizeFlagsHorizontal = SizeFlags.ExpandFill };
+		PopulateAttributeDropdown();
+		attrRow.AddChild(_attributeDropdown);
+
+		_setDropdown.ItemSelected += OnSetDropdownItemSelected;
+		_attributeDropdown.ItemSelected += OnAttributeDropdownItemSelected;
+	}
+
+	/// <inheritdoc/>
+	public override void SaveTo(StatescriptNodeProperty property)
+	{
+		property.Resolver = new AttributeResolverResource
+		{
+			AttributeSetClass = _selectedSetClass,
+			AttributeName = _selectedAttribute,
+		};
+	}
+
+	/// <inheritdoc/>
+	public override void ClearCallbacks()
+	{
+		base.ClearCallbacks();
+		_onChanged = null;
+	}
+
+	private void OnSetDropdownItemSelected(long index)
+	{
+		if (_setDropdown is null)
+		{
+			return;
+		}
+
+		_selectedSetClass = _setDropdown.GetItemText(_setDropdown.Selected);
+		_selectedAttribute = string.Empty;
+		PopulateAttributeDropdown();
+		_onChanged?.Invoke();
+	}
+
+	private void OnAttributeDropdownItemSelected(long index)
+	{
+		if (_attributeDropdown is null)
+		{
+			return;
+		}
+
+		_selectedAttribute = _attributeDropdown.GetItemText(_attributeDropdown.Selected);
+		_onChanged?.Invoke();
+	}
+
+	private void PopulateSetDropdown()
+	{
+		if (_setDropdown is null)
+		{
+			return;
+		}
+
+		_setDropdown.Clear();
+
+		foreach (var option in EditorUtils.GetAttributeSetOptions())
+		{
+			_setDropdown.AddItem(option);
+		}
+
+		// Restore selection.
+		if (!string.IsNullOrEmpty(_selectedSetClass))
+		{
+			for (var i = 0; i < _setDropdown.GetItemCount(); i++)
+			{
+				if (_setDropdown.GetItemText(i) == _selectedSetClass)
+				{
+					_setDropdown.Selected = i;
+					return;
+				}
+			}
+		}
+
+		// Default to first if available.
+		if (_setDropdown.GetItemCount() > 0)
+		{
+			_setDropdown.Selected = 0;
+			_selectedSetClass = _setDropdown.GetItemText(0);
+		}
+	}
+
+	private void PopulateAttributeDropdown()
+	{
+		if (_attributeDropdown is null)
+		{
+			return;
+		}
+
+		_attributeDropdown.Clear();
+
+		foreach (var option in EditorUtils.GetAttributeOptions(_selectedSetClass))
+		{
+			_attributeDropdown.AddItem(option);
+		}
+
+		if (!string.IsNullOrEmpty(_selectedAttribute))
+		{
+			for (var i = 0; i < _attributeDropdown.GetItemCount(); i++)
+			{
+				if (_attributeDropdown.GetItemText(i) == _selectedAttribute)
+				{
+					_attributeDropdown.Selected = i;
+					return;
+				}
+			}
+		}
+
+		if (_attributeDropdown.GetItemCount() > 0)
+		{
+			_attributeDropdown.Selected = 0;
+			_selectedAttribute = _attributeDropdown.GetItemText(0);
+		}
+	}
+}
+#endif