From 6e2dbea24525c34e1343c8c8e383efc5fa680028 Mon Sep 17 00:00:00 2001
From: Lex Lim <hyperzlink@outlook.com>
Date: Fri, 14 Jan 2022 21:12:25 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0Hook:=20IsekaiAIReviewResult?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 includes/AIReviewJob.php | 16 ++++++++++------
 includes/Hooks.php       |  5 +++--
 2 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/includes/AIReviewJob.php b/includes/AIReviewJob.php
index 65b31d4..2d1de07 100644
--- a/includes/AIReviewJob.php
+++ b/includes/AIReviewJob.php
@@ -24,15 +24,15 @@ class AIReviewJob extends Job {
 
         $robotUser = User::newFromId($wgAIReviewRobotUID);
 
-        $modid = $this->params['mod_id'];
-        $modUser = $dbr->selectField('moderation', 'mod_user', ['mod_id' => $modid], __METHOD__);
+        $mod_id = $this->params['mod_id'];
+        $modUser = $dbr->selectField('moderation', 'mod_user', ['mod_id' => $mod_id], __METHOD__);
 
         $services = MediaWikiServices::getInstance();
         $entryFactory = $services->getService('Moderation.EntryFactory');
         $consequenceManager = $services->getService('Moderation.ConsequenceManager');
 
         /** @var ModerationViewableEntry $contentEntry  */
-        $contentEntry = $entryFactory->findViewableEntry($modid);
+        $contentEntry = $entryFactory->findViewableEntry($mod_id);
         $title = $contentEntry->getTitle();
 
         $context = RequestContext::getMain();
@@ -50,7 +50,9 @@ class AIReviewJob extends Job {
                     'isekai-aireview',
                     'Reject revision on: ' . $title->getText() . ', reason: ' . Utils::getReadableReason($result['reason'])
                 );
-                Utils::addAIReviewLog('reject', $robotUser, $modUser, $title, $modid, $result['reason']);
+                Utils::addAIReviewLog('reject', $robotUser, $modUser, $title, $mod_id, $result['reason']);
+                $services->getHookContainer()->run("IsekaiAIReviewResult",
+                    [ false, $title, $mod_id, $modUser, $result['reason'] ]);
                 return true;
             }
         }
@@ -60,9 +62,11 @@ class AIReviewJob extends Job {
             'isekai-aireview',
             'Approve revision on: ' . $title->getText()
         );
-        Utils::addAIReviewLog('approve', $robotUser, $modUser, $title, $modid);
-        $approveEntry = $entryFactory->findApprovableEntry($modid);
+        $approveEntry = $entryFactory->findApprovableEntry($mod_id);
         $approveEntry->approve($robotUser);
+        Utils::addAIReviewLog('approve', $robotUser, $modUser, $title, $mod_id);
+        $services->getHookContainer()->run("IsekaiAIReviewResult",
+            [ true, $title, $mod_id, $modUser, '' ]);
         return true;
     }
 }
\ No newline at end of file
diff --git a/includes/Hooks.php b/includes/Hooks.php
index e23f1c4..91215c3 100644
--- a/includes/Hooks.php
+++ b/includes/Hooks.php
@@ -5,9 +5,10 @@ use JobQueueGroup;
 use Title;
 
 class Hooks {
-    public static function onModerationPending($fields, $modid){
+    public static function onModerationPending($fields, $mod_id){
         //加入审核队列
-        $job = new AIReviewJob(Title::newFromText($fields['mod_title']), ['mod_id' => $modid]);
+        $title = Title::newFromText($fields['mod_title']);
+        $job = new AIReviewJob($title, ['mod_id' => $mod_id]);
         JobQueueGroup::singleton()->push($job);
     }
 }
\ No newline at end of file