Added NbtElement.GetEnumerator

Author: TechPizzaDev

MCServerSharp.NBT/NbtElement.ContainerEnumerator.cs

@@ -20,7 +20,7 @@ public NbtElement Current
             {
                 get
                 {
-                    if (_currentIndex < 0)
+                    if (_currentIndex < 0 || _currentIndex >= _targetEndIndex)
                         return default;
                     return new NbtElement(_container._parent, _currentIndex);
                 }

MCServerSharp.NBT/NbtElement.cs

@@ -1,5 +1,6 @@
 using System;
 using System.Buffers;
+using System.Collections;
 using System.Collections.Generic;
 using System.Diagnostics;
 using System.IO;
@@ -9,7 +10,7 @@
 namespace MCServerSharp.NBT
 {
     [DebuggerDisplay("{ToString(), nq}")]
-    public readonly partial struct NbtElement
+    public readonly partial struct NbtElement : IEnumerable<NbtElement>
     {
         private readonly NbtDocument _parent;
         private readonly int _index;
@@ -26,18 +27,16 @@ public readonly partial struct NbtElement
 
         public NbtElement this[ReadOnlySpan<char> name] => GetCompoundElement(name);
 
-        public NbtElement this[string name] => this[name.AsSpan()];
-
         public NbtOptions Options => _parent.Options;
 
         public bool IsValid => _parent != null && _parent.Bytes.Length != 0;
 
-        public NbtType Type => _parent != null 
+        public NbtType Type => _parent != null
             ? _parent.GetTagType(_index)
             : NbtType.Undefined;
 
-        public NbtFlags Flags => _parent != null 
-            ? _parent.GetFlags(_index) 
+        public NbtFlags Flags => _parent != null
+            ? _parent.GetFlags(_index)
             : NbtFlags.None;
 
         public Utf8Memory Name => _parent != null
@@ -287,6 +286,26 @@ public ContainerEnumerator EnumerateContainer()
             return new ContainerEnumerator(this);
         }
 
+        public ContainerEnumerator GetEnumerator()
+        {
+            AssertValidInstance();
+
+            if (!Type.IsContainer())
+                return default;
+
+            return new ContainerEnumerator(this);
+        }
+
+        IEnumerator<NbtElement> IEnumerable<NbtElement>.GetEnumerator()
+        {
+            return GetEnumerator();
+        }
+
+        IEnumerator IEnumerable.GetEnumerator()
+        {
+            return GetEnumerator();
+        }
+
         public ArrayEnumerator<T> EnumerateArray<T>()
             where T : unmanaged
         {