cache player controllers on handle new player (ServersHub#8)

iterate all players to find the player by eos ID when when it wasn't cached yet

AsaApi/Core/Private/Ark/ApiUtils.cpp

@@ -85,6 +85,26 @@ namespace AsaApi
 		{
 			found_player = iter->second;
 		}
+
+		if (found_player == nullptr)
+		{
+			const auto& player_controllers = GetWorld()->PlayerControllerListField();
+			for (TWeakObjectPtr<APlayerController> player_controller : player_controllers)
+			{
+				AShooterPlayerController* shooter_pc = static_cast<AShooterPlayerController*>(player_controller.Get());
+
+				if (shooter_pc)
+				{
+					const FString& iter_eos_id = *API::game_api->GetApiUtils()->GetEOSIDFromController(shooter_pc);
+					if (iter_eos_id == eos_id)
+					{
+						found_player = shooter_pc;
+						break;
+					}
+				}
+			}
+		}
+
 		return found_player;
 	}
 

AsaApi/Core/Private/Ark/HooksImpl.cpp

@@ -37,6 +37,7 @@ namespace AsaApi
 	DECLARE_HOOK(URCONServer_Init, bool, URCONServer*, FString*, unsigned int, UShooterCheatManager*);
 	DECLARE_HOOK(AShooterPlayerController_OnPossess, void, AShooterPlayerController*, APawn*);
 	DECLARE_HOOK(AShooterGameMode_Logout, void, AShooterGameMode*, AController*);
+	DECLARE_HOOK(AShooterGameMode_HandleNewPlayer_Implementation, bool, AShooterGameMode*, AShooterPlayerController*, UPrimalPlayerData*, AShooterCharacter*, bool);
 
 	void InitHooks()
 	{
@@ -52,6 +53,7 @@ namespace AsaApi
 		hooks->SetHook("URCONServer.Init(FString,int,UShooterCheatManager*)", &Hook_URCONServer_Init, &URCONServer_Init_original);
 		hooks->SetHook("AShooterPlayerController.OnPossess(APawn*)", &Hook_AShooterPlayerController_OnPossess, &AShooterPlayerController_OnPossess_original);
 		hooks->SetHook("AShooterGameMode.Logout(AController*)", &Hook_AShooterGameMode_Logout, &AShooterGameMode_Logout_original);
+		hooks->SetHook("AShooterGameMode.HandleNewPlayer_Implementation(AShooterPlayerController*,UPrimalPlayerData*,AShooterCharacter*,bool)", &Hook_AShooterGameMode_HandleNewPlayer_Implementation, &AShooterGameMode_HandleNewPlayer_Implementation_original);
 
 		Log::GetLog()->info("Initialized hooks\n");
 	}
@@ -186,4 +188,11 @@ namespace AsaApi
 
 		AShooterGameMode_Logout_original(_this, Exiting);
 	}
+
+	bool Hook_AShooterGameMode_HandleNewPlayer_Implementation(AShooterGameMode* _this, AShooterPlayerController* NewPlayer, UPrimalPlayerData* PlayerData, AShooterCharacter* PlayerCharacter, bool bIsFromLogin)
+	{
+		dynamic_cast<ApiUtils&>(*API::game_api->GetApiUtils()).SetPlayerController(NewPlayer);
+
+		return AShooterGameMode_HandleNewPlayer_Implementation_original(_this, NewPlayer, PlayerData, PlayerCharacter, bIsFromLogin);
+	}
 } // namespace AsaApi