Merge pull request #41 from notyourav/queries

match some ai::Query

1 files changed, 6 insertions, 2 deletions

diff --git a/src/Game/AI/Query/queryHasArmorAlreadyDyed.cpp b/src/Game/AI/Query/queryHasArmorAlreadyDyed.cpp
index 7ad39f2c..b042a651 100644
--- a/src/Game/AI/Query/queryHasArmorAlreadyDyed.cpp
+++ b/src/Game/AI/Query/queryHasArmorAlreadyDyed.cpp
@@ -1,5 +1,6 @@
 #include "Game/AI/Query/queryHasArmorAlreadyDyed.h"
 #include <evfl/Query.h>
+#include "Game/UI/uiPauseMenuDataMgr.h"
 
 namespace uking::query {
 
@@ -7,9 +8,12 @@ HasArmorAlreadyDyed::HasArmorAlreadyDyed(const InitArg& arg) : ksys::act::ai::Qu
 
 HasArmorAlreadyDyed::~HasArmorAlreadyDyed() = default;
 
-// FIXME: implement
 int HasArmorAlreadyDyed::doQuery() {
-    return -1;
+    auto* pm = ui::PauseMenuDataMgr::instance();
+    if (pm == nullptr)
+        return 0;
+
+    return pm->countAlreadyDyedArmor() > 0;
 }
 
 void HasArmorAlreadyDyed::loadParams(const evfl::QueryArg& arg) {}