From 4b7cb88eef49ac4a86923f1de43a547d7af18421 Mon Sep 17 00:00:00 2001
From: zimzaza4 <3625282098@qq.com>
Date: Thu, 1 Aug 2024 21:44:45 +0800
Subject: [PATCH] add a check for bedrock

---
 .../re/imc/geysermodelengine/model/BedrockMountControl.java | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/main/java/re/imc/geysermodelengine/model/BedrockMountControl.java b/src/main/java/re/imc/geysermodelengine/model/BedrockMountControl.java
index 0f0938b..c640e21 100644
--- a/src/main/java/re/imc/geysermodelengine/model/BedrockMountControl.java
+++ b/src/main/java/re/imc/geysermodelengine/model/BedrockMountControl.java
@@ -8,6 +8,7 @@ import com.ticxo.modelengine.api.mount.controller.MountController;
 import org.bukkit.Bukkit;
 import org.bukkit.entity.Player;
 import org.bukkit.scheduler.BukkitRunnable;
+import org.geysermc.floodgate.api.FloodgateApi;
 import re.imc.geysermodelengine.GeyserModelEngine;
 
 public class BedrockMountControl {
@@ -17,6 +18,9 @@ public class BedrockMountControl {
             @Override
             public void run() {
                 for (Player player : Bukkit.getOnlinePlayers()) {
+                    if (!FloodgateApi.getInstance().isFloodgatePlayer(player.getUniqueId())) {
+                        continue;
+                    }
                     float pitch = player.getPitch();
                     Pair<ActiveModel, Mount> seat = GeyserModelEngine.getInstance().getDrivers().get(player);
                     if (seat != null) {
@@ -31,7 +35,7 @@ public class BedrockMountControl {
                                 }
                             }
                         }
-                        if (pitch > 50) {
+                        if (pitch > 80) {
                             MountController controller = ModelEngineAPI.getMountPairManager()
                                     .getController(player.getUniqueId());
                             if (controller != null) {