From ce4e2700f8fd992231c3a5eee5ca38074d976e74 Mon Sep 17 00:00:00 2001
From: MakStashkevich <makstashkevich@gmail.com>
Date: Sat, 21 Dec 2019 19:55:20 +0300
Subject: [PATCH] add new functions

---
 client/Bot.php                  | 243 +++++++++++++++++++++++++++++++-
 client/PocketEditionClient.php  |  24 +++-
 client/entity/EntityHelpers.php | 143 +++++++++++++++++++
 console/Console.php             |   4 -
 4 files changed, 401 insertions(+), 13 deletions(-)
 create mode 100644 client/entity/EntityHelpers.php

diff --git a/client/Bot.php b/client/Bot.php
index 49fbca5..e28f360 100644
--- a/client/Bot.php
+++ b/client/Bot.php
@@ -2,6 +2,7 @@
 
 namespace client;
 
+use client\entity\EntityHelpers;
 use client\level\Level;
 use pocketmine\entity\Attribute;
 use pocketmine\math\Vector3;
@@ -85,6 +86,15 @@ class Bot
 	public $flags = 0;
 	public $userPermission = 0;
 
+	/** @var array */
+	public $metadata = [];
+
+	/** @var int */
+	public $gamemode = 0;
+
+	/** @var array */
+	public $playersOnline = [];
+
 	/**
 	 * Bot constructor.
 	 * @param string $username
@@ -101,6 +111,46 @@ function __construct(string $username = null, string $password = null, Address $
 		$this->level = new Level();
 	}
 
+	/**
+	 * @param array $players
+	 */
+	function addPlayersOnline(array $players = [])
+	{
+		array_merge($this->playersOnline, $players);
+	}
+
+	/**
+	 * @param array $players
+	 */
+	function removePlayersOnline(array $players = [])
+	{
+		//todo
+	}
+
+	/**
+	 * @return array
+	 */
+	function getPlayersOnline(): array
+	{
+		return $this->playersOnline;
+	}
+
+	/**
+	 * @param int $mode
+	 */
+	function setGamemode(int $mode)
+	{
+		$this->gamemode = $mode;
+	}
+
+	/**
+	 * @return int
+	 */
+	function getGamemode(): int
+	{
+		return $this->gamemode;
+	}
+
 	/**
 	 * @param float $x
 	 * @param float $y
@@ -517,10 +567,199 @@ function setUserPermission(int $permission = 0)
 	}
 
 	/**
-	 * @return bool
+	 * @return int
 	 */
-	function getUserPermission(): bool
+	function getUserPermission(): int
 	{
 		return $this->userPermission;
 	}
+
+	/**
+	 * @return bool
+	 */
+	function isOp(): bool
+	{
+		return $this->getUserPermission() === 1;
+	}
+
+	/**
+	 * @param array $metadata
+	 */
+	function addMetadata(array $metadata = [])
+	{
+		if ($this->metadata === []) $this->setMetadata($metadata);
+		else $this->metadata = array_merge($this->metadata, $metadata);
+	}
+
+	/**
+	 * @param int $id
+	 */
+	function removeMetadata(int $id)
+	{
+		unset($this->metadata[$id]);
+	}
+
+	/**
+	 * @param array $metadata
+	 */
+	function setMetadata(array $metadata = [])
+	{
+		$this->metadata = $metadata;
+	}
+
+	/**
+	 * @return array
+	 */
+	function getMetadata(): array
+	{
+		return $this->metadata;
+	}
+
+	/**
+	 * @param int $id
+	 * @return bool
+	 */
+	function hasMetadata(int $id): bool
+	{
+		return isset($this->metadata[$id]);
+	}
+
+	/**
+	 * @return array
+	 */
+	function getMetadataFlags(): array
+	{
+		return $this->metadata[EntityHelpers::DATA_FLAGS] ?? [];
+	}
+
+	/**
+	 * @param $id
+	 * @return mixed|null
+	 */
+	function getDataProperty(int $id)
+	{
+		$data = $this->getMetadataFlags();
+		return isset($data[$id]) ? $data[$id][1] : null;
+	}
+
+	/**
+	 * @param $id
+	 * @return mixed|null
+	 */
+	function getDataPropertyType(int $id)
+	{
+		$data = $this->getMetadataFlags();
+		return isset($data[$id]) ? $data[$id][0] : null;
+	}
+
+	/**
+	 * @param $propertyId
+	 * @param $id
+	 * @return bool
+	 */
+	function getDataFlag(int $propertyId, int $id): bool
+	{
+		return (((int)$this->getDataProperty($propertyId)) & (1 << $id)) > 0;
+	}
+
+	/**
+	 * @return bool
+	 */
+	function isImmobile(): bool
+	{
+		return $this->getDataFlag(EntityHelpers::DATA_FLAGS, EntityHelpers::DATA_FLAG_IMMOBILE);
+	}
+
+	/**
+	 * @return bool
+	 */
+	function isGliding(): bool
+	{
+		return $this->getDataFlag(EntityHelpers::DATA_FLAGS, EntityHelpers::DATA_FLAG_GLIDING);
+	}
+
+	/**
+	 * @return bool
+	 */
+	function isSprinting(): bool
+	{
+		return $this->getDataFlag(EntityHelpers::DATA_FLAGS, EntityHelpers::DATA_FLAG_SPRINTING);
+	}
+
+	/**
+	 * @return bool
+	 */
+	function isSneaking(): bool
+	{
+		return $this->getDataFlag(EntityHelpers::DATA_FLAGS, EntityHelpers::DATA_FLAG_SNEAKING);
+	}
+
+	/**
+	 * @return float
+	 */
+	function getScale(): float
+	{
+		return (float)$this->getDataProperty(EntityHelpers::DATA_SCALE);
+	}
+
+	/**
+	 * @return string
+	 */
+	function getNameTag(): string
+	{
+		return $this->getDataProperty(EntityHelpers::DATA_NAMETAG);
+	}
+
+	/**
+	 * @return bool
+	 */
+	function isNameTagVisible(): bool
+	{
+		return $this->getDataFlag(EntityHelpers::DATA_FLAGS, EntityHelpers::DATA_FLAG_CAN_SHOW_NAMETAG);
+	}
+
+	/**
+	 * @return bool
+	 */
+	function isNameTagAlwaysVisible(): bool
+	{
+		return $this->getDataFlag(EntityHelpers::DATA_FLAGS, EntityHelpers::DATA_FLAG_ALWAYS_SHOW_NAMETAG);
+	}
+
+	/**
+	 * Returns whether the entity is able to climb blocks such as ladders or vines.
+	 * @return bool
+	 */
+	function canClimb(): bool
+	{
+		return $this->getDataFlag(EntityHelpers::DATA_FLAGS, EntityHelpers::DATA_FLAG_CAN_CLIMB);
+	}
+
+	/**
+	 * Returns whether this entity is climbing a block. By default this is only true if the entity is climbing a ladder or vine or similar block.
+	 *
+	 * @return bool
+	 */
+	function canClimbWalls(): bool
+	{
+		return $this->getDataFlag(EntityHelpers::DATA_FLAGS, EntityHelpers::DATA_FLAG_WALLCLIMBING);
+	}
+
+	/**
+	 * Returns the entity ID of the owning entity, or null if the entity doesn't have an owner.
+	 * @return int|string|null
+	 */
+	function getOwningEntityId()
+	{
+		return $this->getDataProperty(EntityHelpers::DATA_OWNER_EID);
+	}
+
+	/**
+	 * Returns the entity ID of the entity's target, or null if it doesn't have a target.
+	 * @return int|string|null
+	 */
+	function getTargetEntityId()
+	{
+		return $this->getDataProperty(EntityHelpers::DATA_TARGET_EID);
+	}
 }
\ No newline at end of file
diff --git a/client/PocketEditionClient.php b/client/PocketEditionClient.php
index e5f0c6a..e521a87 100644
--- a/client/PocketEditionClient.php
+++ b/client/PocketEditionClient.php
@@ -787,7 +787,11 @@ protected function handleDataPacket(DataPacket $packet): void
 			$pk = new RequestChunkRadiusPacket();
 			$pk->radius = 8;
 			$this->sendDataPacket($pk);
-			$this->player->setLocation($packet->x, $packet->y, $packet->z, $packet->yaw, $packet->pitch);
+
+			$player = $this->getPlayer();
+			$player->setId($packet->entityRuntimeId);
+			$player->setGamemode($packet->playerGamemode);
+			$player->setLocation($packet->x, $packet->y, $packet->z, $packet->yaw, $packet->pitch);
 		} elseif ($packet instanceof UpdateAttributesPacket) {
 			$player = $this->player;
 			if ($player->getId() === $packet->entityRuntimeId) {
@@ -812,12 +816,18 @@ protected function handleDataPacket(DataPacket $packet): void
 			$player->setFlags((int)$packet->flags);
 			$player->setUserPermission((int)$packet->userPermission);
 			return;
+		} elseif ($packet instanceof SetEntityDataPacket) {
+			$id = $packet->entityRuntimeId;
+			$player = $this->getPlayer();
+			if ($id === $player->getId()) {
+				$player->addMetadata((array)$packet->metadata);
+			}
+			return;
 		} elseif ($packet instanceof PlayerListPacket) {
-			foreach ($packet->entries as $id => $e) {
-				if (isset($e[2]) && $e[2] === $this->player->getName()) {
-					$this->player->setId((int)$e[1]);
-					return;
-				}
+			if ($packet->type === PlayerListPacket::TYPE_ADD) {
+				$this->getPlayer()->addPlayersOnline($packet->entries);
+			} else {
+				$this->getPlayer()->removePlayersOnline($packet->entries);
 			}
 			return;
 		} elseif ($packet instanceof FullChunkDataPacket) {
@@ -873,7 +883,7 @@ protected function handleDataPacket(DataPacket $packet): void
 			PlayerListPacket::NETWORK_ID, SetEntityDataPacket::NETWORK_ID, AddPlayerPacket::NETWORK_ID,
 			RemoveEntityPacket::NETWORK_ID, MovePlayerPacket::NETWORK_ID, MoveEntityPacket::NETWORK_ID,
 			LevelSoundEventPacket::NETWORK_ID, PlayerActionPacket::NETWORK_ID, InventoryActionPacket::NETWORK_ID,
-			EntityEventPacket::NETWORK_ID, SetEntityDataPacket::NETWORK_ID, AnimatePacket::NETWORK_ID,
+			EntityEventPacket::NETWORK_ID, AnimatePacket::NETWORK_ID,
 			SetEntityMotionPacket::NETWORK_ID, LevelEventPacket::NETWORK_ID, UpdateBlockPacket::NETWORK_ID,
 			MobArmorEquipmentPacket::NETWORK_ID, AddItemEntityPacket::NETWORK_ID, BlockEventPacket::NETWORK_ID,
 			SetEntityLinkPacket::NETWORK_ID, MobEquipmentPacket::NETWORK_ID, AddEntityPacket::NETWORK_ID,
diff --git a/client/entity/EntityHelpers.php b/client/entity/EntityHelpers.php
new file mode 100644
index 0000000..cdd6869
--- /dev/null
+++ b/client/entity/EntityHelpers.php
@@ -0,0 +1,143 @@
+<?php
+
+namespace client\entity;
+
+class EntityHelpers
+{
+	const NETWORK_ID = -1;
+
+	const DATA_TYPE_BYTE = 0;
+	const DATA_TYPE_SHORT = 1;
+	const DATA_TYPE_INT = 2;
+	const DATA_TYPE_FLOAT = 3;
+	const DATA_TYPE_STRING = 4;
+	const DATA_TYPE_SLOT = 5;
+	const DATA_TYPE_POS = 6;
+	const DATA_TYPE_LONG = 7;
+	const DATA_TYPE_VECTOR3F = 8;
+
+	const DATA_FLAGS = 0;
+	const DATA_HEALTH = 1; //int (minecart/boat)
+	const DATA_VARIANT = 2; //int
+	const DATA_COLOR = 3, DATA_COLOUR = 3; //byte
+	const DATA_NAMETAG = 4; //string
+	const DATA_OWNER_EID = 5; //long
+	const DATA_TARGET_EID = 6; //long
+	const DATA_AIR = 7; //short
+	const DATA_POTION_COLOR = 8; //int (ARGB!)
+	const DATA_POTION_AMBIENT = 9; //byte
+	/* 10 (byte) */
+	const DATA_HURT_TIME = 11; //int (minecart/boat)
+	const DATA_HURT_DIRECTION = 12; //int (minecart/boat)
+	const DATA_PADDLE_TIME_LEFT = 13; //float
+	const DATA_PADDLE_TIME_RIGHT = 14; //float
+	const DATA_EXPERIENCE_VALUE = 15; //int (xp orb)
+	const DATA_MINECART_DISPLAY_BLOCK = 16; //int (id | (data << 16))
+	const DATA_MINECART_DISPLAY_OFFSET = 17; //int
+	const DATA_MINECART_HAS_DISPLAY = 18; //byte (must be 1 for minecart to show block inside)
+
+	//TODO: add more properties
+
+	const DATA_ENDERMAN_HELD_ITEM_ID = 23; //short
+	const DATA_ENDERMAN_HELD_ITEM_DAMAGE = 24; //short
+	const DATA_ENTITY_AGE = 25; //short
+
+	/* 27 (byte) player-specific flags
+	 * 28 (int) player "index"?
+	 * 29 (block coords) bed position */
+	const DATA_FIREBALL_POWER_X = 30; //float
+	const DATA_FIREBALL_POWER_Y = 31;
+	const DATA_FIREBALL_POWER_Z = 32;
+	/* 33 (unknown)
+	 * 34 (float) fishing bobber
+	 * 35 (float) fishing bobber
+	 * 36 (float) fishing bobber */
+	const DATA_POTION_AUX_VALUE = 37; //short
+	const DATA_LEAD_HOLDER_EID = 38; //long
+	const DATA_SCALE = 39; //float
+	const DATA_INTERACTIVE_TAG = 40; //string (button text)
+	const DATA_NPC_SKIN_ID = 41; //string
+	const DATA_URL_TAG = 42; //string
+	const DATA_MAX_AIR = 43; //short
+	const DATA_MARK_VARIANT = 44; //int
+	/* 45 (byte) container stuff
+	 * 46 (int) container stuff
+	 * 47 (int) container stuff */
+	const DATA_BLOCK_TARGET = 48; //block coords (ender crystal)
+	const DATA_WITHER_INVULNERABLE_TICKS = 49; //int
+	const DATA_WITHER_TARGET_1 = 50; //long
+	const DATA_WITHER_TARGET_2 = 51; //long
+	const DATA_WITHER_TARGET_3 = 52; //long
+	/* 53 (short) */
+	const DATA_BOUNDING_BOX_WIDTH = 54; //float
+	const DATA_BOUNDING_BOX_HEIGHT = 55; //float
+	const DATA_FUSE_LENGTH = 56; //int
+	const DATA_RIDER_SEAT_POSITION = 57; //vector3f
+	const DATA_RIDER_ROTATION_LOCKED = 58; //byte
+	const DATA_RIDER_MAX_ROTATION = 59; //float
+	const DATA_RIDER_MIN_ROTATION = 60; //float
+	const DATA_AREA_EFFECT_CLOUD_RADIUS = 61; //float
+	const DATA_AREA_EFFECT_CLOUD_WAITING = 62; //int
+	const DATA_AREA_EFFECT_CLOUD_PARTICLE_ID = 63; //int
+	/* 64 (int) shulker-related */
+	const DATA_SHULKER_ATTACH_FACE = 65; //byte
+	/* 66 (short) shulker-related */
+	const DATA_SHULKER_ATTACH_POS = 67; //block coords
+	const DATA_TRADING_PLAYER_EID = 68; //long
+
+	/* 70 (byte) command-block */
+	const DATA_COMMAND_BLOCK_COMMAND = 71; //string
+	const DATA_COMMAND_BLOCK_LAST_OUTPUT = 72; //string
+	const DATA_COMMAND_BLOCK_TRACK_OUTPUT = 73; //byte
+	const DATA_CONTROLLING_RIDER_SEAT_NUMBER = 74; //byte
+	const DATA_STRENGTH = 75; //int
+	const DATA_MAX_STRENGTH = 76; //int
+	/* 77 (int)
+	 * 78 (int) */
+
+
+	const DATA_FLAG_ONFIRE = 0;
+	const DATA_FLAG_SNEAKING = 1;
+	const DATA_FLAG_RIDING = 2;
+	const DATA_FLAG_SPRINTING = 3;
+	const DATA_FLAG_ACTION = 4;
+	const DATA_FLAG_INVISIBLE = 5;
+	const DATA_FLAG_TEMPTED = 6;
+	const DATA_FLAG_INLOVE = 7;
+	const DATA_FLAG_SADDLED = 8;
+	const DATA_FLAG_POWERED = 9;
+	const DATA_FLAG_IGNITED = 10;
+	const DATA_FLAG_BABY = 11;
+	const DATA_FLAG_CONVERTING = 12;
+	const DATA_FLAG_CRITICAL = 13;
+	const DATA_FLAG_CAN_SHOW_NAMETAG = 14;
+	const DATA_FLAG_ALWAYS_SHOW_NAMETAG = 15;
+	const DATA_FLAG_IMMOBILE = 16, DATA_FLAG_NO_AI = 16;
+	const DATA_FLAG_SILENT = 17;
+	const DATA_FLAG_WALLCLIMBING = 18;
+	const DATA_FLAG_CAN_CLIMB = 19;
+	const DATA_FLAG_SWIMMER = 20;
+	const DATA_FLAG_CAN_FLY = 21;
+	const DATA_FLAG_RESTING = 22;
+	const DATA_FLAG_SITTING = 23;
+	const DATA_FLAG_ANGRY = 24;
+	const DATA_FLAG_INTERESTED = 25;
+	const DATA_FLAG_CHARGED = 26;
+	const DATA_FLAG_TAMED = 27;
+	const DATA_FLAG_LEASHED = 28;
+	const DATA_FLAG_SHEARED = 29;
+	const DATA_FLAG_GLIDING = 30;
+	const DATA_FLAG_ELDER = 31;
+	const DATA_FLAG_MOVING = 32;
+	const DATA_FLAG_BREATHING = 33;
+	const DATA_FLAG_CHESTED = 34;
+	const DATA_FLAG_STACKABLE = 35;
+	const DATA_FLAG_SHOWBASE = 36;
+	const DATA_FLAG_REARING = 37;
+	const DATA_FLAG_VIBRATING = 38;
+	const DATA_FLAG_IDLING = 39;
+	const DATA_FLAG_EVOKER_SPELL = 40;
+	const DATA_FLAG_CHARGE_ATTACK = 41;
+
+	const DATA_FLAG_LINGER = 45;
+}
\ No newline at end of file
diff --git a/console/Console.php b/console/Console.php
index 947c9a4..cb862d7 100644
--- a/console/Console.php
+++ b/console/Console.php
@@ -91,10 +91,6 @@ function command(string $command)
 				BotHelpers::moveTo($client, $client->getPlayer()->getPosition()->add(0, (float)$args[1]));
 				break;
 
-			case 'test':
-				var_dump($this->client->getPocketClient()->getPlayer()->getLevel()->getChests());
-				break;
-
 			default:
 				if ($this->chat) $this->client->chat($command);
 				break;