match more ai::Query

1 files changed, 14 insertions, 2 deletions

diff --git a/src/Game/AI/Query/queryCheckDyeShopSelect.cpp b/src/Game/AI/Query/queryCheckDyeShopSelect.cpp
index d0cea451..02f6a7ed 100644
--- a/src/Game/AI/Query/queryCheckDyeShopSelect.cpp
+++ b/src/Game/AI/Query/queryCheckDyeShopSelect.cpp
@@ -1,5 +1,6 @@
 #include "Game/AI/Query/queryCheckDyeShopSelect.h"
 #include <evfl/Query.h>
+#include "KingSystem/GameData/gdtManager.h"
 
 namespace uking::query {
 
@@ -7,9 +8,20 @@ CheckDyeShopSelect::CheckDyeShopSelect(const InitArg& arg) : ksys::act::ai::Quer
 
 CheckDyeShopSelect::~CheckDyeShopSelect() = default;
 
-// FIXME: implement
 int CheckDyeShopSelect::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") && (screen_type != 0)) {
+            if (screen_type == 9) {
+                return 0;
+            }
+            if (screen_type == 10) {
+                return 1;
+            }
+        }
+    }
+    return 2;
 }
 
 void CheckDyeShopSelect::loadParams(const evfl::QueryArg& arg) {}