From c1aecf21d7599112ae2d039eb5acab3e5db84a3e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=9C=A7=E9=9B=A8=E7=83=A8?= <Web_13234@outlook.com>
Date: Thu, 13 Jun 2024 10:43:54 +0800
Subject: [PATCH] Implements IEnumerable<ItemSlotNode> for IItemContainer.

---
 scripts/inventory/IItemContainer.cs         | 16 ++++++++++++----
 scripts/inventory/UniversalItemContainer.cs | 16 ++++++++++++++++
 2 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/scripts/inventory/IItemContainer.cs b/scripts/inventory/IItemContainer.cs
index e10b9bd..5adcedb 100644
--- a/scripts/inventory/IItemContainer.cs
+++ b/scripts/inventory/IItemContainer.cs
@@ -16,7 +16,7 @@ namespace ColdMint.scripts.inventory;
 ///<para>Item containers can store items. Things like backpacks and Hotbars are containers with visual pages.</para>
 ///<para>物品容器可以储存物品。像背包和hotbar是具有可视化页面的容器。</para>
 /// </remarks>
-public interface IItemContainer
+public interface IItemContainer : IEnumerable<ItemSlotNode>
 {
     /// <summary>
     /// <para>Can the specified item be added to the container?</para>
@@ -58,7 +58,7 @@ public interface IItemContainer
     /// </summary>
     /// <returns></returns>
     ItemSlotNode? GetSelectItemSlotNode();
-    
+
     /// <summary>
     /// <para>If present, remove an item from the slot at the currently selected location and return it.</para>
     /// <para>如果存在，移除当前选中位置的槽位中的一个物品并将其返回</para>
@@ -75,7 +75,7 @@ public interface IItemContainer
     /// <para>要取出的数量，小于0的输入代表全部物品</para>
     /// </param>
     /// <seealso cref="PickItemsFromItemSlot"/><seealso cref="PickItemFromItemSlotBySelectIndex"/>
-    IItemStack? PickItemsFromItemSlotBySelectIndex( int value);
+    IItemStack? PickItemsFromItemSlotBySelectIndex(int value);
 
     /// <summary>
     /// <para>Removes an item from the inventory at the currently selected location</para>
@@ -110,6 +110,14 @@ public interface IItemContainer
     /// <returns></returns>
     ItemSlotNode? GetItemSlotNode(int index);
 
+    /// <summary>
+    /// <para>Gets the item slot for the specified location, equals to <see cref="GetItemSlotNode"/></para>
+    /// <para>获取指定位置的物品槽，等同于<see cref="GetItemSlotNode"/></para>
+    /// </summary>
+    /// <param name="index"></param>
+    /// <returns></returns>
+    ItemSlotNode? this[int index] => GetItemSlotNode(index);
+
     /// <summary>
     /// <para>If present, remove an item from the slot in the specified location and return it.</para>
     /// <para>如果存在，移除指定位置的槽位中的一个物品并将其返回</para>
@@ -128,7 +136,7 @@ public interface IItemContainer
     /// </param>
     /// <seealso cref="PickItemFromItemSlot"/>
     IItemStack? PickItemsFromItemSlot(int itemSlotIndex, int value);
-    
+
     /// <summary>
     /// <para>Removes an item from the item slot in the specified location</para>
     /// <para>在指定位置的物品槽内移除物品</para>
diff --git a/scripts/inventory/UniversalItemContainer.cs b/scripts/inventory/UniversalItemContainer.cs
index 6532bac..998cda1 100644
--- a/scripts/inventory/UniversalItemContainer.cs
+++ b/scripts/inventory/UniversalItemContainer.cs
@@ -1,4 +1,5 @@
 using System;
+using System.Collections;
 using System.Collections.Generic;
 using System.Linq;
 
@@ -9,6 +10,8 @@ using ColdMint.scripts.utils;
 
 using Godot;
 
+using JetBrains.Annotations;
+
 namespace ColdMint.scripts.inventory;
 
 /// <summary>
@@ -348,4 +351,17 @@ public class UniversalItemContainer : IItemContainer
 
         _selectIndex = newSelectIndex;
     }
+
+
+    [MustDisposeResource]
+    public IEnumerator<ItemSlotNode> GetEnumerator()
+    {
+        return _itemSlotNodes?.GetEnumerator() ?? Enumerable.Empty<ItemSlotNode>().GetEnumerator();
+    }
+
+    [MustDisposeResource]
+    IEnumerator IEnumerable.GetEnumerator()
+    {
+        return GetEnumerator();
+    }
 }
\ No newline at end of file