match some ai::Query

1 files changed, 11 insertions, 2 deletions

diff --git a/src/Game/AI/Query/queryIsSuccessEndLastDemoAction.cpp b/src/Game/AI/Query/queryIsSuccessEndLastDemoAction.cpp
index a34df9e3..37d33fc2 100644
--- a/src/Game/AI/Query/queryIsSuccessEndLastDemoAction.cpp
+++ b/src/Game/AI/Query/queryIsSuccessEndLastDemoAction.cpp
@@ -1,5 +1,7 @@
 #include "Game/AI/Query/queryIsSuccessEndLastDemoAction.h"
 #include <evfl/Query.h>
+#include "KingSystem/ActorSystem/actActor.h"
+#include "KingSystem/ActorSystem/actAiRoot.h"
 
 namespace uking::query {
 
@@ -8,9 +10,16 @@ IsSuccessEndLastDemoAction::IsSuccessEndLastDemoAction(const InitArg& arg)
 
 IsSuccessEndLastDemoAction::~IsSuccessEndLastDemoAction() = default;
 
-// FIXME: implement
 int IsSuccessEndLastDemoAction::doQuery() {
-    return -1;
+    if (mActor != nullptr && mActor->getRootAi() != nullptr) {
+        auto* action = mActor->getRootAi()->getCurrentAction();
+        if (action == nullptr)
+            return 0;
+
+        if (action->isFinished())
+            return 1;
+    }
+    return 0;
 }
 
 void IsSuccessEndLastDemoAction::loadParams(const evfl::QueryArg& arg) {}