mirror of
https://github.com/pmmp/PocketMine-MP.git
synced 2025-07-10 03:47:48 +00:00
437 lines
10 KiB
PHP
437 lines
10 KiB
PHP
<?php
|
|
|
|
/*
|
|
*
|
|
* ____ _ _ __ __ _ __ __ ____
|
|
* | _ \ ___ ___| | _____| |_| \/ (_)_ __ ___ | \/ | _ \
|
|
* | |_) / _ \ / __| |/ / _ \ __| |\/| | | '_ \ / _ \_____| |\/| | |_) |
|
|
* | __/ (_) | (__| < __/ |_| | | | | | | | __/_____| | | | __/
|
|
* |_| \___/ \___|_|\_\___|\__|_| |_|_|_| |_|\___| |_| |_|_|
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Lesser General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* @author PocketMine Team
|
|
* @link http://www.pocketmine.net/
|
|
*
|
|
*
|
|
*/
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace pocketmine\inventory;
|
|
|
|
use pocketmine\entity\Entity;
|
|
use pocketmine\event\entity\EntityInventoryChangeEvent;
|
|
use pocketmine\event\inventory\InventoryOpenEvent;
|
|
use pocketmine\item\Item;
|
|
use pocketmine\network\mcpe\protocol\InventoryContentPacket;
|
|
use pocketmine\network\mcpe\protocol\InventorySlotPacket;
|
|
use pocketmine\Player;
|
|
use pocketmine\Server;
|
|
|
|
abstract class BaseInventory implements Inventory{
|
|
|
|
/** @var int */
|
|
protected $maxStackSize = Inventory::MAX_STACK;
|
|
/** @var string */
|
|
protected $name;
|
|
/** @var string */
|
|
protected $title;
|
|
/** @var \SplFixedArray<Item> */
|
|
protected $slots = [];
|
|
/** @var Player[] */
|
|
protected $viewers = [];
|
|
/** @var InventoryHolder */
|
|
protected $holder;
|
|
|
|
/**
|
|
* @param InventoryHolder $holder
|
|
* @param Item[] $items
|
|
* @param int $size
|
|
* @param string $title
|
|
*/
|
|
public function __construct(InventoryHolder $holder, array $items = [], int $size = null, string $title = null){
|
|
$this->holder = $holder;
|
|
|
|
$this->slots = new \SplFixedArray($size ?? $this->getDefaultSize());
|
|
$this->title = $title ?? $this->getName();
|
|
|
|
$this->setContents($items);
|
|
}
|
|
|
|
abstract public function getName() : string;
|
|
|
|
public function getTitle() : string{
|
|
return $this->title;
|
|
}
|
|
|
|
/**
|
|
* Returns the Minecraft PE inventory type used to show the inventory window to clients.
|
|
* @return int
|
|
*/
|
|
abstract public function getNetworkType() : int;
|
|
|
|
/**
|
|
* Returns the size of the inventory.
|
|
* @return int
|
|
*/
|
|
public function getSize() : int{
|
|
return $this->slots->getSize();
|
|
}
|
|
|
|
/**
|
|
* Sets the new size of the inventory.
|
|
* WARNING: If the size is smaller, any items past the new size will be lost.
|
|
*
|
|
* @param int $size
|
|
*/
|
|
public function setSize(int $size){
|
|
$this->slots->setSize($size);
|
|
}
|
|
|
|
abstract public function getDefaultSize() : int;
|
|
|
|
public function getMaxStackSize() : int{
|
|
return $this->maxStackSize;
|
|
}
|
|
|
|
public function getItem(int $index) : Item{
|
|
return $this->slots[$index] !== null ? clone $this->slots[$index] : Item::get(Item::AIR, 0, 0);
|
|
}
|
|
|
|
public function getContents() : array{
|
|
return array_filter($this->slots->toArray(), function(Item $item = null){ return $item !== null; });
|
|
}
|
|
|
|
/**
|
|
* @param Item[] $items
|
|
*/
|
|
public function setContents(array $items){
|
|
if(count($items) > $this->getSize()){
|
|
$items = array_slice($items, 0, $this->getSize(), true);
|
|
}
|
|
|
|
for($i = 0, $size = $this->getSize(); $i < $size; ++$i){
|
|
if(!isset($items[$i])){
|
|
if($this->slots[$i] !== null){
|
|
$this->clear($i, false);
|
|
}
|
|
}else{
|
|
if(!$this->setItem($i, $items[$i], false)){
|
|
$this->clear($i, false);
|
|
}
|
|
}
|
|
}
|
|
|
|
$this->sendContents($this->getViewers());
|
|
}
|
|
|
|
public function setItem(int $index, Item $item, bool $send = true) : bool{
|
|
if($item->isNull()){
|
|
$item = Item::get(Item::AIR, 0, 0);
|
|
}else{
|
|
$item = clone $item;
|
|
}
|
|
|
|
$holder = $this->getHolder();
|
|
if($holder instanceof Entity){
|
|
Server::getInstance()->getPluginManager()->callEvent($ev = new EntityInventoryChangeEvent($holder, $this->getItem($index), $item, $index));
|
|
if($ev->isCancelled()){
|
|
$this->sendSlot($index, $this->getViewers());
|
|
return false;
|
|
}
|
|
$item = $ev->getNewItem();
|
|
}
|
|
|
|
$old = $this->getItem($index);
|
|
$this->slots[$index] = $item->isNull() ? null : $item;
|
|
$this->onSlotChange($index, $old, $send);
|
|
|
|
return true;
|
|
}
|
|
|
|
public function contains(Item $item) : bool{
|
|
$count = max(1, $item->getCount());
|
|
$checkDamage = !$item->hasAnyDamageValue();
|
|
$checkTags = $item->hasCompoundTag();
|
|
foreach($this->getContents() as $i){
|
|
if($item->equals($i, $checkDamage, $checkTags)){
|
|
$count -= $i->getCount();
|
|
if($count <= 0){
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
public function all(Item $item) : array{
|
|
$slots = [];
|
|
$checkDamage = !$item->hasAnyDamageValue();
|
|
$checkTags = $item->hasCompoundTag();
|
|
foreach($this->getContents() as $index => $i){
|
|
if($item->equals($i, $checkDamage, $checkTags)){
|
|
$slots[$index] = $i;
|
|
}
|
|
}
|
|
|
|
return $slots;
|
|
}
|
|
|
|
public function remove(Item $item){
|
|
$checkDamage = !$item->hasAnyDamageValue();
|
|
$checkTags = $item->hasCompoundTag();
|
|
|
|
foreach($this->getContents() as $index => $i){
|
|
if($item->equals($i, $checkDamage, $checkTags)){
|
|
$this->clear($index);
|
|
}
|
|
}
|
|
}
|
|
|
|
public function first(Item $item) : int{
|
|
$count = max(1, $item->getCount());
|
|
$checkDamage = !$item->hasAnyDamageValue();
|
|
$checkTags = $item->hasCompoundTag();
|
|
|
|
foreach($this->getContents() as $index => $i){
|
|
if($item->equals($i, $checkDamage, $checkTags) and $i->getCount() >= $count){
|
|
return $index;
|
|
}
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
public function firstEmpty() : int{
|
|
foreach($this->slots as $i => $slot){
|
|
if($slot === null or $slot->isNull()){
|
|
return $i;
|
|
}
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
public function canAddItem(Item $item) : bool{
|
|
$item = clone $item;
|
|
$checkDamage = !$item->hasAnyDamageValue();
|
|
$checkTags = $item->hasCompoundTag();
|
|
for($i = 0; $i < $this->getSize(); ++$i){
|
|
$slot = $this->getItem($i);
|
|
if($item->equals($slot, $checkDamage, $checkTags)){
|
|
if(($diff = $slot->getMaxStackSize() - $slot->getCount()) > 0){
|
|
$item->setCount($item->getCount() - $diff);
|
|
}
|
|
}elseif($slot->getId() === Item::AIR){
|
|
$item->setCount($item->getCount() - $this->getMaxStackSize());
|
|
}
|
|
|
|
if($item->getCount() <= 0){
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
public function addItem(Item ...$slots) : array{
|
|
/** @var Item[] $itemSlots */
|
|
/** @var Item[] $slots */
|
|
$itemSlots = [];
|
|
foreach($slots as $slot){
|
|
if($slot->getId() !== 0 and $slot->getCount() > 0){
|
|
$itemSlots[] = clone $slot;
|
|
}
|
|
}
|
|
|
|
$emptySlots = [];
|
|
|
|
for($i = 0; $i < $this->getSize(); ++$i){
|
|
$item = $this->getItem($i);
|
|
if($item->getId() === Item::AIR or $item->getCount() <= 0){
|
|
$emptySlots[] = $i;
|
|
}
|
|
|
|
foreach($itemSlots as $index => $slot){
|
|
if($slot->equals($item) and $item->getCount() < $item->getMaxStackSize()){
|
|
$amount = min($item->getMaxStackSize() - $item->getCount(), $slot->getCount(), $this->getMaxStackSize());
|
|
if($amount > 0){
|
|
$slot->setCount($slot->getCount() - $amount);
|
|
$item->setCount($item->getCount() + $amount);
|
|
$this->setItem($i, $item);
|
|
if($slot->getCount() <= 0){
|
|
unset($itemSlots[$index]);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if(count($itemSlots) === 0){
|
|
break;
|
|
}
|
|
}
|
|
|
|
if(count($itemSlots) > 0 and count($emptySlots) > 0){
|
|
foreach($emptySlots as $slotIndex){
|
|
//This loop only gets the first item, then goes to the next empty slot
|
|
foreach($itemSlots as $index => $slot){
|
|
$amount = min($slot->getMaxStackSize(), $slot->getCount(), $this->getMaxStackSize());
|
|
$slot->setCount($slot->getCount() - $amount);
|
|
$item = clone $slot;
|
|
$item->setCount($amount);
|
|
$this->setItem($slotIndex, $item);
|
|
if($slot->getCount() <= 0){
|
|
unset($itemSlots[$index]);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
return $itemSlots;
|
|
}
|
|
|
|
public function removeItem(Item ...$slots) : array{
|
|
/** @var Item[] $itemSlots */
|
|
/** @var Item[] $slots */
|
|
$itemSlots = [];
|
|
foreach($slots as $slot){
|
|
if($slot->getId() !== 0 and $slot->getCount() > 0){
|
|
$itemSlots[] = clone $slot;
|
|
}
|
|
}
|
|
|
|
for($i = 0; $i < $this->getSize(); ++$i){
|
|
$item = $this->getItem($i);
|
|
if($item->getId() === Item::AIR or $item->getCount() <= 0){
|
|
continue;
|
|
}
|
|
|
|
foreach($itemSlots as $index => $slot){
|
|
if($slot->equals($item, !$slot->hasAnyDamageValue(), $slot->hasCompoundTag())){
|
|
$amount = min($item->getCount(), $slot->getCount());
|
|
$slot->setCount($slot->getCount() - $amount);
|
|
$item->setCount($item->getCount() - $amount);
|
|
$this->setItem($i, $item);
|
|
if($slot->getCount() <= 0){
|
|
unset($itemSlots[$index]);
|
|
}
|
|
}
|
|
}
|
|
|
|
if(count($itemSlots) === 0){
|
|
break;
|
|
}
|
|
}
|
|
|
|
return $itemSlots;
|
|
}
|
|
|
|
public function clear(int $index, bool $send = true) : bool{
|
|
return $this->setItem($index, Item::get(Item::AIR, 0, 0), $send);
|
|
}
|
|
|
|
public function clearAll(){
|
|
for($i = 0, $size = $this->getSize(); $i < $size; ++$i){
|
|
$this->clear($i, false);
|
|
}
|
|
|
|
$this->sendContents($this->getViewers());
|
|
}
|
|
|
|
/**
|
|
* @return Player[]
|
|
*/
|
|
public function getViewers() : array{
|
|
return $this->viewers;
|
|
}
|
|
|
|
public function getHolder(){
|
|
return $this->holder;
|
|
}
|
|
|
|
public function setMaxStackSize(int $size){
|
|
$this->maxStackSize = $size;
|
|
}
|
|
|
|
public function open(Player $who) : bool{
|
|
$who->getServer()->getPluginManager()->callEvent($ev = new InventoryOpenEvent($this, $who));
|
|
if($ev->isCancelled()){
|
|
return false;
|
|
}
|
|
$this->onOpen($who);
|
|
|
|
return true;
|
|
}
|
|
|
|
public function close(Player $who){
|
|
$this->onClose($who);
|
|
}
|
|
|
|
public function onOpen(Player $who){
|
|
$this->viewers[spl_object_hash($who)] = $who;
|
|
}
|
|
|
|
public function onClose(Player $who){
|
|
unset($this->viewers[spl_object_hash($who)]);
|
|
}
|
|
|
|
public function onSlotChange(int $index, Item $before, bool $send){
|
|
if($send){
|
|
$this->sendSlot($index, $this->getViewers());
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Player|Player[] $target
|
|
*/
|
|
public function sendContents($target){
|
|
if($target instanceof Player){
|
|
$target = [$target];
|
|
}
|
|
|
|
$pk = new InventoryContentPacket();
|
|
for($i = 0; $i < $this->getSize(); ++$i){
|
|
$pk->items[$i] = $this->getItem($i);
|
|
}
|
|
|
|
foreach($target as $player){
|
|
if(($id = $player->getWindowId($this)) === -1 or $player->spawned !== true){
|
|
$this->close($player);
|
|
continue;
|
|
}
|
|
$pk->windowId = $id;
|
|
$player->dataPacket($pk);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param int $index
|
|
* @param Player|Player[] $target
|
|
*/
|
|
public function sendSlot(int $index, $target){
|
|
if($target instanceof Player){
|
|
$target = [$target];
|
|
}
|
|
|
|
$pk = new InventorySlotPacket();
|
|
$pk->inventorySlot = $index;
|
|
$pk->item = clone $this->getItem($index);
|
|
|
|
foreach($target as $player){
|
|
if(($id = $player->getWindowId($this)) === -1){
|
|
$this->close($player);
|
|
continue;
|
|
}
|
|
$pk->windowId = $id;
|
|
$player->dataPacket($pk);
|
|
}
|
|
}
|
|
}
|