From ef901a62f6694314a91f6cbd1a0a0c83c223bf87 Mon Sep 17 00:00:00 2001
From: xLuxy <67131061+xLuxy@users.noreply.github.com>
Date: Fri, 1 Sep 2023 23:58:41 +0200
Subject: [PATCH] server: Add vehicle passengers getter

---
 server/src/CNodeResourceImpl.cpp | 51 ++++++++++++++++++++++++++++++++
 server/src/CNodeResourceImpl.h   |  1 +
 server/src/bindings/Vehicle.cpp  | 22 ++++++++++++++
 shared/IRuntimeEventHandler.h    |  4 +++
 shared/V8ResourceImpl.h          |  4 +++
 5 files changed, 82 insertions(+)

diff --git a/server/src/CNodeResourceImpl.cpp b/server/src/CNodeResourceImpl.cpp
index 0c70c51a..09e32741 100644
--- a/server/src/CNodeResourceImpl.cpp
+++ b/server/src/CNodeResourceImpl.cpp
@@ -140,6 +140,8 @@ void CNodeResourceImpl::OnEvent(const alt::CEvent* e)
     v8::Context::Scope scope(GetContext());
     // env->PushAsyncCallbackScope();
 
+    HandleVehiclePassengerSeatEvents(e);
+
     V8Helpers::EventHandler* handler = V8Helpers::EventHandler::Get(e);
     if(!handler) return;
 
@@ -186,6 +188,55 @@ void CNodeResourceImpl::OnEvent(const alt::CEvent* e)
     return;
 }
 
+void CNodeResourceImpl::HandleVehiclePassengerSeatEvents(const alt::CEvent* ev)
+{
+    auto evType = ev->GetType();
+
+    if (evType == alt::CEvent::Type::PLAYER_ENTER_VEHICLE)
+    {
+        auto event = static_cast<const alt::CPlayerEnterVehicleEvent*>(ev);
+        auto vehicle = event->GetTarget();
+        auto player = event->GetPlayer();
+        auto seat = event->GetSeat();
+
+        if (!vehiclePassengers.contains(vehicle))
+            vehiclePassengers[vehicle] = {};
+
+        vehiclePassengers[vehicle][seat] = player;
+    }
+    else if (evType == alt::CEvent::Type::PLAYER_LEAVE_VEHICLE)
+    {
+        auto event = static_cast<const alt::CPlayerLeaveVehicleEvent*>(ev);
+        auto vehicle = event->GetTarget();
+        auto player = event->GetPlayer();
+
+        if (vehiclePassengers.contains(vehicle))
+        {
+            for ( auto it = vehiclePassengers[vehicle].begin(); it != vehiclePassengers[vehicle].end(); )
+            {
+                if (it->second == player)
+                    it = vehiclePassengers[vehicle].erase(it);
+                else
+                    ++it;
+            }
+
+            if (vehiclePassengers[vehicle].empty())
+                vehiclePassengers.erase(vehicle);
+        }
+    }
+    else if (evType == alt::CEvent::Type::PLAYER_CHANGE_VEHICLE_SEAT)
+    {
+        auto event = static_cast<const alt::CPlayerChangeVehicleSeatEvent*>(ev);
+        auto vehicle = event->GetTarget();
+
+        if (!vehiclePassengers.contains(vehicle))
+            vehiclePassengers[vehicle] = {};
+
+        vehiclePassengers[vehicle].erase(event->GetOldSeat());
+        vehiclePassengers[vehicle][event->GetNewSeat()] = event->GetPlayer();
+    }
+}
+
 void CNodeResourceImpl::OnTick()
 {
     v8::Locker locker(isolate);
diff --git a/server/src/CNodeResourceImpl.h b/server/src/CNodeResourceImpl.h
index d17242f4..3581c63a 100644
--- a/server/src/CNodeResourceImpl.h
+++ b/server/src/CNodeResourceImpl.h
@@ -25,6 +25,7 @@ class CNodeResourceImpl : public V8ResourceImpl
     bool Stop() override;
 
     void OnEvent(const alt::CEvent* ev) override;
+    void HandleVehiclePassengerSeatEvents(const alt::CEvent* ev);
     void OnTick() override;
 
     bool MakeClient(alt::IResource::CreationInfo* info, std::vector<std::string>) override;
diff --git a/server/src/bindings/Vehicle.cpp b/server/src/bindings/Vehicle.cpp
index dd841a35..48f79231 100644
--- a/server/src/bindings/Vehicle.cpp
+++ b/server/src/bindings/Vehicle.cpp
@@ -229,6 +229,27 @@ static void RoofClosedGetter(v8::Local<v8::String> name, const v8::PropertyCallb
     V8_RETURN_BOOLEAN(_this->GetRoofState() == 1);
 }
 
+static void GetPassengers(v8::Local<v8::String>, const v8::PropertyCallbackInfo<v8::Value>& info)
+{
+    V8_GET_ISOLATE_CONTEXT_RESOURCE();
+    V8_GET_THIS_BASE_OBJECT(_this, IVehicle);
+
+    auto obj = v8::Object::New(isolate);
+    auto passengers = V8ResourceImpl::vehiclePassengers[_this];
+
+    if (!passengers.empty())
+    {
+        for (auto& [seat, player] : passengers)
+        {
+            auto entity = resource->GetBaseObjectOrNull(player);
+            if (!entity->IsNull())
+                obj->Set(ctx, seat, entity);
+        }
+    }
+
+    V8_RETURN(obj);
+}
+
 extern V8Class v8Entity;
 extern V8Class v8Vehicle("Vehicle",
                          v8Entity,
@@ -244,6 +265,7 @@ extern V8Class v8Vehicle("Vehicle",
                              // Common getter/setters
                              V8Helpers::SetAccessor<IVehicle, bool, &IVehicle::IsDestroyed>(isolate, tpl, "destroyed");
                              V8Helpers::SetAccessor<IVehicle, IPlayer*, &IVehicle::GetDriver>(isolate, tpl, "driver");
+                             V8Helpers::SetAccessor(isolate, tpl, "passengers", &GetPassengers);
                              V8Helpers::SetAccessor<IVehicle, Vector3f, &IVehicle::GetVelocity>(isolate, tpl, "velocity");
                              V8Helpers::SetAccessor<IVehicle, Quaternion, &IVehicle::GetQuaternion, &IVehicle::SetQuaternion>(isolate, tpl, "quaternion");
 
diff --git a/shared/IRuntimeEventHandler.h b/shared/IRuntimeEventHandler.h
index e18cc9e2..f5507202 100644
--- a/shared/IRuntimeEventHandler.h
+++ b/shared/IRuntimeEventHandler.h
@@ -15,6 +15,10 @@ class IRuntimeEventHandler
     // All events the module uses, which need to be enabled and never disabled
     static constexpr std::array internalEvents = {
         EventType::CONNECTION_COMPLETE, EventType::DISCONNECT_EVENT, EventType::GAME_ENTITY_CREATE, EventType::GAME_ENTITY_DESTROY, EventType::RESOURCE_STOP
+#ifdef ALT_SERVER_API
+        // required for vehicle seat stuff
+        , EventType::PLAYER_ENTER_VEHICLE, EventType::PLAYER_LEAVE_VEHICLE, EventType::PLAYER_CHANGE_VEHICLE_SEAT
+#endif
     };
 
     // Keeps track of the event handlers registered for all events
diff --git a/shared/V8ResourceImpl.h b/shared/V8ResourceImpl.h
index a382e9f4..8e7d69d3 100644
--- a/shared/V8ResourceImpl.h
+++ b/shared/V8ResourceImpl.h
@@ -313,6 +313,10 @@ class V8ResourceImpl : public alt::IResource::Impl
         return static_cast<alt::IResource*>(ctx->GetAlignedPointerFromEmbedderData(1));
     }
 
+#ifdef ALT_SERVER_API
+    static inline std::unordered_map<alt::IVehicle*, std::unordered_map<uint8_t, alt::IPlayer*>> vehiclePassengers{};
+#endif
+
 protected:
     v8::Isolate* isolate;
     alt::IResource* resource;