match some ai::Query

1 files changed, 9 insertions, 2 deletions

diff --git a/src/Game/AI/Query/queryCheckManufactResult.cpp b/src/Game/AI/Query/queryCheckManufactResult.cpp
index 25bdf663..d87c84de 100644
--- a/src/Game/AI/Query/queryCheckManufactResult.cpp
+++ b/src/Game/AI/Query/queryCheckManufactResult.cpp
@@ -1,5 +1,6 @@
 #include "Game/AI/Query/queryCheckManufactResult.h"
 #include <evfl/Query.h>
+#include "KingSystem/GameData/gdtCommonFlagsUtils.h"
 
 namespace uking::query {
 
@@ -7,9 +8,15 @@ CheckManufactResult::CheckManufactResult(const InitArg& arg) : ksys::act::ai::Qu
 
 CheckManufactResult::~CheckManufactResult() = default;
 
-// FIXME: implement
+static constexpr s32 sManufactResults[8] = {0, 6, 4, 6, 1, 2, 3, 5};
+
 int CheckManufactResult::doQuery() {
-    return -1;
+    if (ksys::gdt::getFlag_Shop_IsDecide()) {
+        s32 state = ksys::gdt::getFlag_Shop_CurrentItemState();
+        if (state < 8 && state >= 0)
+            return sManufactResults[state];
+    }
+    return 6;
 }
 
 void CheckManufactResult::loadParams(const evfl::QueryArg& arg) {}