From 1b7ef9a4cdcce27a164a3d3579ce6db4103b5c38 Mon Sep 17 00:00:00 2001
From: Tofaa <82680183+Tofaa2@users.noreply.github.com>
Date: Tue, 2 Jan 2024 02:27:34 +0300
Subject: [PATCH] finish integrating ObjectData

---
 .../tofaa/entitylib/entity/WrapperEntity.java | 35 +++++++++++++++++--
 1 file changed, 33 insertions(+), 2 deletions(-)

diff --git a/src/main/java/me/tofaa/entitylib/entity/WrapperEntity.java b/src/main/java/me/tofaa/entitylib/entity/WrapperEntity.java
index 6518a3a..e03b9d2 100644
--- a/src/main/java/me/tofaa/entitylib/entity/WrapperEntity.java
+++ b/src/main/java/me/tofaa/entitylib/entity/WrapperEntity.java
@@ -3,6 +3,8 @@ package me.tofaa.entitylib.entity;
 import com.github.retrooper.packetevents.protocol.entity.type.EntityType;
 import com.github.retrooper.packetevents.protocol.player.User;
 import com.github.retrooper.packetevents.protocol.world.Location;
+import com.github.retrooper.packetevents.util.Vector3d;
+import com.github.retrooper.packetevents.util.Vector3f;
 import com.github.retrooper.packetevents.wrapper.PacketWrapper;
 import com.github.retrooper.packetevents.wrapper.play.server.*;
 import me.tofaa.entitylib.EntityLib;
@@ -24,6 +26,7 @@ public class WrapperEntity {
     private Location location;
     private boolean onGround;
     private boolean spawned;
+    private Vector3d velocity = Vector3d.zero();
 
     public WrapperEntity(@NotNull UUID uuid, EntityType entityType, EntityMeta meta) {
         this.uuid = Optional.of(uuid);
@@ -43,8 +46,23 @@ public class WrapperEntity {
         this.spawned = true;
 
         int data = 0;
+        Optional<Vector3d> velocity;
+        double veloX = 0, veloY = 0, veloZ = 0;
         if (meta instanceof ObjectData) {
-            data = ((ObjectData) meta).getObjectData();
+            ObjectData od = (ObjectData) meta;
+
+            data = od.getObjectData();
+            if (od.requiresVelocityPacketAtSpawn()) {
+                final WrapperPlayServerEntityVelocity veloPacket = getVelocityPacket();
+                veloX = veloPacket.getVelocity().getX();
+                veloY = veloPacket.getVelocity().getY();
+                veloZ = veloPacket.getVelocity().getZ();
+            }
+        }
+        if (veloX == 0 && veloY == 0 && veloZ == 0) {
+            velocity = Optional.empty();
+        } else {
+            velocity = Optional.of(new Vector3d(veloX, veloY, veloZ));
         }
         sendPacketToViewers(
                 new WrapperPlayServerSpawnEntity(
@@ -56,7 +74,7 @@ public class WrapperEntity {
                         location.getYaw(),
                         location.getYaw(),
                         data,
-                        Optional.empty()
+                        velocity
                 )
         );
         sendPacketToViewers(meta.createPacket());
@@ -161,4 +179,17 @@ public class WrapperEntity {
         return spawned;
     }
 
+    public Vector3d getVelocity() {
+        return velocity;
+    }
+
+    public void setVelocity(Vector3d velocity) {
+        this.velocity = velocity;
+    }
+
+    private WrapperPlayServerEntityVelocity getVelocityPacket() {
+        Vector3d velocity = this.velocity.multiply(8000.0f / 20.0f);
+        return new WrapperPlayServerEntityVelocity(entityId, velocity);
+    }
+
 }