match some ai::Query

1 files changed, 10 insertions, 2 deletions

diff --git a/src/Game/AI/Query/queryHasEnoughPayment.cpp b/src/Game/AI/Query/queryHasEnoughPayment.cpp
index 5ef119da..52b86453 100644
--- a/src/Game/AI/Query/queryHasEnoughPayment.cpp
+++ b/src/Game/AI/Query/queryHasEnoughPayment.cpp
@@ -1,5 +1,6 @@
 #include "Game/AI/Query/queryHasEnoughPayment.h"
 #include <evfl/Query.h>
+#include "KingSystem/GameData/gdtManager.h"
 
 namespace uking::query {
 
@@ -7,9 +8,16 @@ HasEnoughPayment::HasEnoughPayment(const InitArg& arg) : ksys::act::ai::Query(ar
 
 HasEnoughPayment::~HasEnoughPayment() = default;
 
-// FIXME: implement
 int HasEnoughPayment::doQuery() {
-    return -1;
+    auto* gdm = ksys::gdt::Manager::instance();
+    if (gdm != nullptr) {
+        s32 state = -1;
+        if (!gdm->getParam().get().getS32(&state, "Shop_CurrentItemState"))
+            return 1;
+        if (state == 2 || state == 3)
+            return 0;
+    }
+    return 1;
 }
 
 void HasEnoughPayment::loadParams(const evfl::QueryArg& arg) {}