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\Byte;
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 = CommonTypes::readOptional($in, static function(ByteBufferReader $in) : array{
            $result = [];
            for($i = 0, $len = VarInt::readUnsignedInt($in); $i < $len; ++$i){
                $result[] = InventoryTransactionChangedSlotsHack::read($in);
            }
            return $result;
        });
 
        if(Byte::readUnsigned($in) !== 1){
            throw new PacketDecodeException("Dummy optional bool for transactionType should always be 1");
        }
        $transactionType = VarInt::readUnsignedInt($in);
        if(Byte::readUnsigned($in) !== 1){
            throw new PacketDecodeException("Dummy optional bool for trData should always be 1");
        }
        $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->decodeTransaction($in);
    }
 
    protected function encodePayload(ByteBufferWriter $out) : void{
        CommonTypes::writeLegacyItemStackRequestId($out, $this->requestId);
 
        CommonTypes::writeOptional($out, $this->requestChangedSlots, static function(ByteBufferWriter $out, array $value) : void{
            VarInt::writeUnsignedInt($out, count($value));
            foreach($value as $changedSlots){
                $changedSlots->write($out);
            }
        });
 
        Byte::writeUnsigned($out, 1);
        VarInt::writeUnsignedInt($out, $this->trData->getTypeId());
        Byte::writeUnsigned($out, 1);
        $this->trData->encodeTransaction($out);
    }
 
    public function handle(PacketHandlerInterface $handler) : bool{
        return $handler->handleInventoryTransaction($this);
    }
}