PlayerArmorDamagePacket.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 pocketmine\network\mcpe\protocol\serializer\PacketSerializer;
 
class PlayerArmorDamagePacket extends DataPacket implements ClientboundPacket{
    public const NETWORK_ID = ProtocolInfo::PLAYER_ARMOR_DAMAGE_PACKET;
 
    private const FLAG_HEAD = 0;
    private const FLAG_CHEST = 1;
    private const FLAG_LEGS = 2;
    private const FLAG_FEET = 3;
    private const FLAG_BODY = 4;
 
    private ?int $headSlotDamage;
    private ?int $chestSlotDamage;
    private ?int $legsSlotDamage;
    private ?int $feetSlotDamage;
    private ?int $bodySlotDamage;
 
    public static function create(?int $headSlotDamage, ?int $chestSlotDamage, ?int $legsSlotDamage, ?int $feetSlotDamage, ?int $bodySlotDamage) : self{
        $result = new self;
        $result->headSlotDamage = $headSlotDamage;
        $result->chestSlotDamage = $chestSlotDamage;
        $result->legsSlotDamage = $legsSlotDamage;
        $result->feetSlotDamage = $feetSlotDamage;
        $result->bodySlotDamage = $bodySlotDamage;
        return $result;
    }
 
    public function getHeadSlotDamage() : ?int{ return $this->headSlotDamage; }
 
    public function getChestSlotDamage() : ?int{ return $this->chestSlotDamage; }
 
    public function getLegsSlotDamage() : ?int{ return $this->legsSlotDamage; }
 
    public function getFeetSlotDamage() : ?int{ return $this->feetSlotDamage; }
 
    public function getBodySlotDamage() : ?int{ return $this->bodySlotDamage; }
 
    private function maybeReadDamage(int $flags, int $flag, PacketSerializer $in) : ?int{
        if(($flags & (1 << $flag)) !== 0){
            return $in->getVarInt();
        }
        return null;
    }
 
    protected function decodePayload(PacketSerializer $in) : void{
        $flags = $in->getByte();
 
        $this->headSlotDamage = $this->maybeReadDamage($flags, self::FLAG_HEAD, $in);
        $this->chestSlotDamage = $this->maybeReadDamage($flags, self::FLAG_CHEST, $in);
        $this->legsSlotDamage = $this->maybeReadDamage($flags, self::FLAG_LEGS, $in);
        $this->feetSlotDamage = $this->maybeReadDamage($flags, self::FLAG_FEET, $in);
        $this->bodySlotDamage = $this->maybeReadDamage($flags, self::FLAG_BODY, $in);
    }
 
    private function composeFlag(?int $field, int $flag) : int{
        return $field !== null ? (1 << $flag) : 0;
    }
 
    private function maybeWriteDamage(?int $field, PacketSerializer $out) : void{
        if($field !== null){
            $out->putVarInt($field);
        }
    }
 
    protected function encodePayload(PacketSerializer $out) : void{
        $out->putByte(
            $this->composeFlag($this->headSlotDamage, self::FLAG_HEAD) |
            $this->composeFlag($this->chestSlotDamage, self::FLAG_CHEST) |
            $this->composeFlag($this->legsSlotDamage, self::FLAG_LEGS) |
            $this->composeFlag($this->feetSlotDamage, self::FLAG_FEET) |
            $this->composeFlag($this->bodySlotDamage, self::FLAG_BODY)
        );
 
        $this->maybeWriteDamage($this->headSlotDamage, $out);
        $this->maybeWriteDamage($this->chestSlotDamage, $out);
        $this->maybeWriteDamage($this->legsSlotDamage, $out);
        $this->maybeWriteDamage($this->feetSlotDamage, $out);
        $this->maybeWriteDamage($this->bodySlotDamage, $out);
    }
 
    public function handle(PacketHandlerInterface $handler) : bool{
        return $handler->handlePlayerArmorDamage($this);
    }
}