Nodes - Reorganized code

src/Artemis.VisualScripting/Nodes/Operators/Screens/EnumEqualsNodeCustomView.axaml

@@ -0,0 +1,15 @@
+<UserControl xmlns="https://github.com/avaloniaui"
+             xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"
+             xmlns:d="http://schemas.microsoft.com/expression/blend/2008"
+             xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006"
+             xmlns:screens="clr-namespace:Artemis.VisualScripting.Nodes.Operators.Screens"
+             mc:Ignorable="d" d:DesignWidth="800" d:DesignHeight="450"
+             x:Class="Artemis.VisualScripting.Nodes.Operators.Screens.EnumEqualsNodeCustomView"
+             x:DataType="screens:EnumEqualsNodeCustomViewModel">
+    <ComboBox IsEnabled="{CompiledBinding EnumValues.Count}"
+              Items="{CompiledBinding EnumValues}"
+              SelectedItem="{CompiledBinding CurrentValue}"
+              PlaceholderText="Connect a pin..."
+              Classes="condensed"
+              VerticalAlignment="Center" />
+</UserControl>

src/Artemis.VisualScripting/Nodes/Operators/Screens/EnumEqualsNodeCustomView.axaml.cs

@@ -0,0 +1,18 @@
+using Avalonia.Markup.Xaml;
+using Avalonia.ReactiveUI;
+
+namespace Artemis.VisualScripting.Nodes.Operators.Screens
+{
+    public partial class EnumEqualsNodeCustomView : ReactiveUserControl<EnumEqualsNodeCustomViewModel>
+    {
+        public EnumEqualsNodeCustomView()
+        {
+            InitializeComponent();
+        }
+
+        private void InitializeComponent()
+        {
+            AvaloniaXamlLoader.Load(this);
+        }
+    }
+}

src/Artemis.VisualScripting/Nodes/Operators/Screens/EnumEqualsNodeCustomViewModel.cs

@@ -0,0 +1,59 @@
+using System.Collections.ObjectModel;
+using System.Reactive.Linq;
+using Artemis.Core;
+using Artemis.Core.Events;
+using Artemis.UI.Shared.Services.NodeEditor;
+using Artemis.UI.Shared.Services.NodeEditor.Commands;
+using Artemis.UI.Shared.VisualScripting;
+using Avalonia.Controls.Mixins;
+using DynamicData;
+using ReactiveUI;
+
+namespace Artemis.VisualScripting.Nodes.Operators.Screens;
+
+public class EnumEqualsNodeCustomViewModel : CustomNodeViewModel
+{
+    private readonly EnumEqualsNode _node;
+    private readonly INodeEditorService _nodeEditorService;
+
+    public EnumEqualsNodeCustomViewModel(EnumEqualsNode node, INodeScript script, INodeEditorService nodeEditorService) : base(node, script)
+    {
+        _node = node;
+        _nodeEditorService = nodeEditorService;
+
+        NodeModified += (_, _) => this.RaisePropertyChanged(nameof(CurrentValue));
+        this.WhenActivated(d =>
+        {
+            if (_node.InputPin.ConnectedTo.Any())
+            {
+                EnumValues.Clear();
+                EnumValues.AddRange(Enum.GetValues(_node.InputPin.ConnectedTo.First().Type).Cast<Enum>());
+
+                this.RaisePropertyChanged(nameof(CurrentValue));
+            }
+
+            Observable.FromEventPattern<SingleValueEventArgs<IPin>>(x => _node.InputPin.PinConnected += x, x => _node.InputPin.PinConnected -= x)
+                .Subscribe(p =>
+                {
+                    EnumValues.AddRange(Enum.GetValues(p.EventArgs.Value.Type).Cast<Enum>());
+                    this.RaisePropertyChanged(nameof(CurrentValue));
+                })
+                .DisposeWith(d);
+            Observable.FromEventPattern<SingleValueEventArgs<IPin>>(x => _node.InputPin.PinDisconnected += x, x => _node.InputPin.PinDisconnected -= x)
+                .Subscribe(_ => EnumValues.Clear())
+                .DisposeWith(d);
+        });
+    }
+
+    public ObservableCollection<Enum> EnumValues { get; } = new();
+
+    public Enum? CurrentValue
+    {
+        get => _node.Storage;
+        set
+        {
+            if (value != null && !Equals(_node.Storage, value))
+                _nodeEditorService.ExecuteCommand(Script, new UpdateStorage<Enum>(_node, value));
+        }
+    }
+}