@occultus/toolkit 0.22.1 → 0.22.2

package/README.md

@@ -28,4 +28,3 @@ The above copyright notice and this permission notice shall be included in all c
 
 THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 ```
-

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@occultus/toolkit",
-  "version": "0.22.1",
+  "version": "0.22.2",
   "repository": {
     "type": "git",
     "url": "https://codeberg.org/TeamOccultus/StarTenonAPI"

package/src/entity/entityUtils.ts

@@ -5,7 +5,6 @@ import {
   Entity,
   EntityComponentTypes,
   EntityEffectOptions,
-  EntityEquippableComponent,
   EquipmentSlot,
   ItemStack,
   Player

package/src/item/container.ts

@@ -1,4 +1,9 @@
-import { Container, ItemStack } from "@minecraft/server";
+import { Container, EquipmentSlot, ItemStack, Player } from "@minecraft/server";
+import {
+  getContainer,
+  getEquipmentItem,
+  setEquipmentItem
+} from "../entity/entityUtils";
 
 /**
  * 获取容器中指定含指定物品标签的物品的数量
@@ -70,30 +75,109 @@ export function getItemAmountInContainer(
   return amount;
 }
 
+/**
+ * 获取玩家身上的所有物品（包含装备栏与副手）
+ * @param player 玩家对象
+ * @param typeId 要获取的物品 ID
+ * @returns 容器中指定物品的数量
+ * @author RawDiamondMC
+ */
+
+export function getItemAmountOfPlayer(player: Player, typeId: string): number {
+  const container = player.getComponent("minecraft:inventory")?.container;
+  if (!container) return 0;
+  let amount = getItemAmountInContainer(container, typeId);
+  // 不包含主手，防止重复计算
+  const slots = [
+    EquipmentSlot.Head,
+    EquipmentSlot.Chest,
+    EquipmentSlot.Legs,
+    EquipmentSlot.Feet,
+    EquipmentSlot.Offhand
+  ];
+  for (const slot of slots) {
+    const itemStack = getEquipmentItem(player, slot);
+    if (itemStack?.typeId === typeId) {
+      amount += itemStack.amount;
+    }
+  }
+  return amount;
+}
+
 /**
  * 从容器中移除指定数量的物品
  * @param container 容器对象
  * @param typeId 物品 ID
  * @param amount 要移除物品的数量
+ * @returns 未能够移除完的物品数量
  * @author RawDiamondMC
  */
 export function removeItemInContainer(
   container: Container,
   typeId: string,
   amount: number
-): void {
+): number {
   for (let slot = 0; slot < container.size; slot++) {
+    // 物品已经移除完毕
+    if (amount <= 0) return 0;
     const itemStack: undefined | ItemStack = container.getItem(slot);
     if (itemStack?.typeId === typeId) {
+      // 这个槽位的物品数量足够移除
       if (itemStack.amount > amount) {
         itemStack.amount -= amount;
         container.setItem(slot, itemStack);
-        return;
+        // 直接返回，因为肯定移除完毕了
+        return 0;
       }
       container.setItem(slot);
       amount -= itemStack.amount;
     }
   }
+  return amount;
+}
+
+/**
+ * 从玩家身上移除一定量的物品
+ * @param player 玩家对象
+ * @param typeId 物品 ID
+ * @param amount 要移除物品的数量
+ * @returns 未能够移除完的物品数量
+ * @author RawDiamondMC
+ */
+export function removeItemOfPlayer(
+  player: Player,
+  typeId: string,
+  amount: number
+): number {
+  // 玩家肯定得有inventory组件
+  let remaining = removeItemInContainer(getContainer(player)!, typeId, amount);
+  // 已经移除完毕
+  if (remaining <= 0) return 0;
+  // 不包含主手，防止重复计算
+  const slots = [
+    EquipmentSlot.Head,
+    EquipmentSlot.Chest,
+    EquipmentSlot.Legs,
+    EquipmentSlot.Feet,
+    EquipmentSlot.Offhand
+  ];
+  for (const slot of slots) {
+    // 物品已经移除完毕
+    if (remaining <= 0) return 0;
+    const itemStack = getEquipmentItem(player, slot);
+    if (itemStack?.typeId === typeId) {
+      // 这个槽位的物品数量足够移除
+      if (itemStack.amount > remaining) {
+        itemStack.amount -= remaining;
+        setEquipmentItem(player, itemStack, slot);
+        // 直接返回，因为肯定移除完毕了
+        return 0;
+      }
+      setEquipmentItem(player, itemStack, slot);
+      remaining -= itemStack.amount;
+    }
+  }
+  return remaining;
 }
 
 /**