InventoryTransactionPacket.php

<?php
 
/*
 * This file is part of BedrockProtocol.
 * Copyright (C) 2014-2022 PocketMine Team <https://github.com/pmmp/BedrockProtocol>
 *
 * BedrockProtocol 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.
 */
 
declare(strict_types=1);
 
namespace pocketmine\network\mcpe\protocol;
 
use pmmp\encoding\ByteBufferReader;
use pmmp\encoding\ByteBufferWriter;
use pmmp\encoding\VarInt;
use pocketmine\network\mcpe\protocol\serializer\CommonTypes;
use pocketmine\network\mcpe\protocol\types\inventory\InventoryTransactionChangedSlotsHack;
use pocketmine\network\mcpe\protocol\types\inventory\MismatchTransactionData;
use pocketmine\network\mcpe\protocol\types\inventory\NormalTransactionData;
use pocketmine\network\mcpe\protocol\types\inventory\ReleaseItemTransactionData;
use pocketmine\network\mcpe\protocol\types\inventory\TransactionData;
use pocketmine\network\mcpe\protocol\types\inventory\UseItemOnEntityTransactionData;
use pocketmine\network\mcpe\protocol\types\inventory\UseItemTransactionData;
use function count;
 
class InventoryTransactionPacket extends DataPacket implements ClientboundPacket, ServerboundPacket{
    public const NETWORK_ID = ProtocolInfo::INVENTORY_TRANSACTION_PACKET;
 
    public const TYPE_NORMAL = 0;
    public const TYPE_MISMATCH = 1;
    public const TYPE_USE_ITEM = 2;
    public const TYPE_USE_ITEM_ON_ENTITY = 3;
    public const TYPE_RELEASE_ITEM = 4;
 
    public int $requestId;
    public array $requestChangedSlots;
    public TransactionData $trData;
 
    public static function create(int $requestId, array $requestChangedSlots, TransactionData $trData) : self{
        $result = new self;
        $result->requestId = $requestId;
        $result->requestChangedSlots = $requestChangedSlots;
        $result->trData = $trData;
        return $result;
    }
 
    protected function decodePayload(ByteBufferReader $in) : void{
        $this->requestId = CommonTypes::readLegacyItemStackRequestId($in);
        $this->requestChangedSlots = [];
        if($this->requestId !== 0){
            for($i = 0, $len = VarInt::readUnsignedInt($in); $i < $len; ++$i){
                $this->requestChangedSlots[] = InventoryTransactionChangedSlotsHack::read($in);
            }
        }
 
        $transactionType = VarInt::readUnsignedInt($in);
 
        $this->trData = match($transactionType){
            NormalTransactionData::ID => new NormalTransactionData(),
            MismatchTransactionData::ID => new MismatchTransactionData(),
            UseItemTransactionData::ID => new UseItemTransactionData(),
            UseItemOnEntityTransactionData::ID => new UseItemOnEntityTransactionData(),
            ReleaseItemTransactionData::ID => new ReleaseItemTransactionData(),
            default => throw new PacketDecodeException("Unknown transaction type $transactionType"),
        };
 
        $this->trData->decode($in);
    }
 
    protected function encodePayload(ByteBufferWriter $out) : void{
        CommonTypes::writeLegacyItemStackRequestId($out, $this->requestId);
        if($this->requestId !== 0){
            VarInt::writeUnsignedInt($out, count($this->requestChangedSlots));
            foreach($this->requestChangedSlots as $changedSlots){
                $changedSlots->write($out);
            }
        }
 
        VarInt::writeUnsignedInt($out, $this->trData->getTypeId());
 
        $this->trData->encode($out);
    }
 
    public function handle(PacketHandlerInterface $handler) : bool{
        return $handler->handleInventoryTransaction($this);
    }
}