match more ai::Query

1 files changed, 14 insertions, 2 deletions

diff --git a/src/Game/AI/Query/queryCheckAppShopSelect.cpp b/src/Game/AI/Query/queryCheckAppShopSelect.cpp
index 3aab27d1..62dddb08 100644
--- a/src/Game/AI/Query/queryCheckAppShopSelect.cpp
+++ b/src/Game/AI/Query/queryCheckAppShopSelect.cpp
@@ -1,5 +1,6 @@
 #include "Game/AI/Query/queryCheckAppShopSelect.h"
 #include <evfl/Query.h>
+#include "KingSystem/GameData/gdtManager.h"
 
 namespace uking::query {
 
@@ -7,9 +8,20 @@ CheckAppShopSelect::CheckAppShopSelect(const InitArg& arg) : ksys::act::ai::Quer
 
 CheckAppShopSelect::~CheckAppShopSelect() = default;
 
-// FIXME: implement
 int CheckAppShopSelect::doQuery() {
-    return -1;
+    auto* gdm = ksys::gdt::Manager::instance();
+    if (gdm != nullptr) {
+        int screen_type = -1;
+        if (gdm->getParam().get().getS32(&screen_type, "Shop_ScreenType")) {
+            if (screen_type == 15) {
+                return 0;
+            }
+            if (screen_type == 0) {
+                return 1;
+            }
+        }
+    }
+    return 1;
 }
 
 void CheckAppShopSelect::loadParams(const evfl::QueryArg& arg) {}