From a4334ecb02e9d340c3f094e46bd4d84659559755 Mon Sep 17 00:00:00 2001
From: KiritoDev <36680385+KiritoDv@users.noreply.github.com>
Date: Mon, 8 Aug 2022 18:53:43 -0500
Subject: [PATCH] Changed input disable to only be forced on switch (#941)

* Changed input disable to only be forced on switch

* Inverted if

* Finally fixed validation

* Removed enhancement if its a switch build

* Input now only gets blocked when the device is not a keyboard or its on switch

* Input gets blocked when the controller view is opened

* gControlNav is enabled by default on switch

* Fixed compilation issues
---
 libultraship/libultraship/ControlDeck.cpp | 38 ++++++++++++++++-------
 libultraship/libultraship/ImGuiImpl.cpp   | 13 ++++++--
 libultraship/libultraship/Window.cpp      |  7 ++---
 3 files changed, 39 insertions(+), 19 deletions(-)

diff --git a/libultraship/libultraship/ControlDeck.cpp b/libultraship/libultraship/ControlDeck.cpp
index 31eb90067..7437afeee 100644
--- a/libultraship/libultraship/ControlDeck.cpp
+++ b/libultraship/libultraship/ControlDeck.cpp
@@ -6,6 +6,7 @@
 #include "KeyboardController.h"
 #include "SDLController.h"
 #include <Utils/StringHelper.h>
+#include "Cvar.h"
 
 namespace Ship {
     uint8_t* controllerBits;
@@ -51,18 +52,31 @@ namespace Ship {
         *controllerBits |= (backend->Connected()) << slot;
     }
 
-    void ControlDeck::WriteToPad(OSContPad* pad) const {
-        for (size_t i = 0; i < virtualDevices.size(); i++) {
-            const std::shared_ptr<Controller> backend = physicalDevices[virtualDevices[i]];
-            if (backend->GetGuid() == "Auto") {
-                for (const auto& device : physicalDevices) {
-                    device->Read(&pad[i], i);
-                }
-                continue;
-            }
-            backend->Read(&pad[i], i);
-        }
-    }
+	void ControlDeck::WriteToPad(OSContPad* pad) const {
+
+	#ifdef __SWITCH__
+		bool shouldBlockGameInput = CVar_GetS32("gOpenMenuBar", 0);
+	#else
+		bool shouldBlockGameInput = CVar_GetS32("gOpenMenuBar", 0) && CVar_GetS32("gControlNav", 0);
+	#endif
+
+		for (size_t i = 0; i < virtualDevices.size(); i++) {
+			const std::shared_ptr<Controller> backend = physicalDevices[virtualDevices[i]];
+			if (backend->GetGuid() == "Auto") {
+				for (const auto& device : physicalDevices) {
+					if(shouldBlockGameInput && device->GetGuid() != "Keyboard") {
+						continue;
+					}
+					device->Read(&pad[i], i);
+				}
+				continue;
+			}
+			if(shouldBlockGameInput && backend->GetGuid() != "Keyboard") {
+				continue;
+			}
+			backend->Read(&pad[i], i);
+		}
+	}
 
 #define NESTED(key, ...) StringHelper::Sprintf("Controllers.%s.Slot_%d." key, device->GetGuid().c_str(), slot, __VA_ARGS__)
 
diff --git a/libultraship/libultraship/ImGuiImpl.cpp b/libultraship/libultraship/ImGuiImpl.cpp
index b73b65886..0d00ec32c 100644
--- a/libultraship/libultraship/ImGuiImpl.cpp
+++ b/libultraship/libultraship/ImGuiImpl.cpp
@@ -822,7 +822,12 @@ namespace SohImGui {
             GlobalCtx2::GetInstance()->GetWindow()->SetMenuBar(menu_bar);
             ShowCursor(menu_bar, Dialogues::dMenubar);
             GlobalCtx2::GetInstance()->GetWindow()->GetControlDeck()->SaveControllerSettings();
-            if (CVar_GetS32("gControlNav", 0)) {
+        #ifdef __SWITCH__
+            bool enableControllerNavigation = true;
+        #else
+            bool enableControllerNavigation = CVar_GetS32("gControlNav", 0);
+        #endif
+            if (enableControllerNavigation) {
                 if (CVar_GetS32("gOpenMenuBar", 0)) {
                     io->ConfigFlags |=ImGuiConfigFlags_NavEnableGamepad | ImGuiConfigFlags_NavEnableKeyboard;
                 } else {
@@ -892,9 +897,11 @@ namespace SohImGui {
 
             if (ImGui::BeginMenu("Controller"))
             {
+
+            #ifndef __SWITCH__
                 EnhancementCheckbox("Use Controller Navigation", "gControlNav");
                 Tooltip("Allows controller navigation of the menu bar\nD-pad to move between items, A to select, and X to grab focus on the menu bar");
-
+            #endif
                 EnhancementCheckbox("Controller Configuration", "gControllerConfigurationEnabled");
                 controller->Opened = CVar_GetS32("gControllerConfigurationEnabled", 0);
 
@@ -1365,7 +1372,7 @@ namespace SohImGui {
                         {
                             val = 20;
                         }
-                        
+
                         CVar_SetS32(fps_cvar, val);
                         needs_save = true;
                     }
diff --git a/libultraship/libultraship/Window.cpp b/libultraship/libultraship/Window.cpp
index 0db33792e..f4f047428 100644
--- a/libultraship/libultraship/Window.cpp
+++ b/libultraship/libultraship/Window.cpp
@@ -19,7 +19,7 @@
 #include <string>
 #include <chrono>
 #include "Console.h"
-#include "Cvar.h"
+#include "ImGuiImpl.h"
 
 #include <iostream>
 
@@ -67,10 +67,9 @@ extern "C" {
         pad->gyro_x = 0;
         pad->gyro_y = 0;
 
-        if (!CVar_GetS32("gOpenMenuBar", 0)) {
-            Ship::GlobalCtx2::GetInstance()->GetWindow()->GetControlDeck()->WriteToPad(pad);
-        }
+	    if (SohImGui::controller->Opened) return;
 
+        Ship::GlobalCtx2::GetInstance()->GetWindow()->GetControlDeck()->WriteToPad(pad);
         Ship::ExecuteHooks<Ship::ControllerRead>(pad);
     }