[OneBot] fix check sequence

LagrangeDev · Nov 26, 2024 · a0c881a · a0c881a
1 parent 5e7072c
commit a0c881a
Show file tree

Hide file tree

Showing 6 changed files with 17 additions and 3 deletions.
diff --git a/Lagrange.OneBot/Core/Operation/Message/SendGroupAiRecordOperation.cs b/Lagrange.OneBot/Core/Operation/Message/SendGroupAiRecordOperation.cs
@@ -26,9 +26,12 @@ public async Task<OneBotResult> HandleOperation(BotContext context, JsonNode? pa
             );
             if (code != 0 || recordEntity == null) return new OneBotResult(null, code, "failed");
 
-
             var chain = MessageBuilder.Group(message.GroupId).Add(recordEntity).Build();
             var result = await context.SendMessage(chain);
+
+            if (result.Result != 0) return new OneBotResult(null, (int)result.Result, "failed");
+            if (result.Sequence == null || result.Sequence == 0) return new OneBotResult(null, 9000, "failed");
+
             int hash = MessageRecord.CalcMessageHash(chain.MessageId, result.Sequence ?? 0);
 
             return new OneBotResult(new OneBotMessageResponse(hash), (int)result.Result, "ok");

diff --git a/Lagrange.OneBot/Core/Operation/Message/SendGroupForwardOperation.cs b/Lagrange.OneBot/Core/Operation/Message/SendGroupForwardOperation.cs
@@ -23,6 +23,10 @@ public async Task<OneBotResult> HandleOperation(BotContext context, JsonNode? pa
             var multi = new MultiMsgEntity(chains);
             var chain = MessageBuilder.Group(forward.GroupId).Add(multi).Build();
             var ret = await context.SendMessage(chain);
+
+            if (ret.Result != 0) return new OneBotResult(null, (int)ret.Result, "failed");
+            if (ret.Sequence == null || ret.Sequence == 0) return new OneBotResult(null, 9000, "failed");
+
             int hash = MessageRecord.CalcMessageHash(chain.MessageId, ret.Sequence ?? 0);
 
             return new OneBotResult(new OneBotForwardResponse(hash, multi.ResId ?? ""), (int)ret.Result, "ok");

diff --git a/Lagrange.OneBot/Core/Operation/Message/SendGroupMessageOperation.cs b/Lagrange.OneBot/Core/Operation/Message/SendGroupMessageOperation.cs
@@ -21,10 +21,14 @@ public async Task<OneBotResult> HandleOperation(BotContext context, JsonNode? pa
             OneBotGroupMessageText messageText => common.ParseChain(messageText).Build(),
             _ => throw new Exception()
         };
-        
+
         var result = await context.SendMessage(chain);
+
+        if (result.Result != 0) return new OneBotResult(null, (int)result.Result, "failed");
+        if (result.Sequence == null || result.Sequence == 0) return new OneBotResult(null, 9000, "failed");
+
         int hash = MessageRecord.CalcMessageHash(chain.MessageId, result.Sequence ?? 0);
-        
+
         return new OneBotResult(new OneBotMessageResponse(hash), (int)result.Result, "ok");
     }
 }
diff --git a/Lagrange.OneBot/Core/Operation/Message/SendMessageOperation.cs b/Lagrange.OneBot/Core/Operation/Message/SendMessageOperation.cs
@@ -26,6 +26,7 @@ public async Task<OneBotResult> HandleOperation(BotContext context, JsonNode? pa
         var result = await context.SendMessage(chain);
 
         if (result.Result != 0) return new OneBotResult(null, (int)result.Result, "failed");
+        if (result.Sequence == null || result.Sequence == 0) return new OneBotResult(null, 9000, "failed");
 
         int hash = MessageRecord.CalcMessageHash(result.MessageId, result.Sequence ?? 0);
 

diff --git a/Lagrange.OneBot/Core/Operation/Message/SendPrivateForwardOperation.cs b/Lagrange.OneBot/Core/Operation/Message/SendPrivateForwardOperation.cs
@@ -27,6 +27,7 @@ public async Task<OneBotResult> HandleOperation(BotContext context, JsonNode? pa
             var result = await context.SendMessage(chain);
 
             if (result.Result != 0) return new OneBotResult(null, (int)result.Result, "failed");
+            if (result.Sequence == null || result.Sequence == 0) return new OneBotResult(null, 9000, "failed");
 
             int hash = MessageRecord.CalcMessageHash(result.MessageId, result.Sequence ?? 0);
 

diff --git a/Lagrange.OneBot/Core/Operation/Message/SendPrivateMessageOperation.cs b/Lagrange.OneBot/Core/Operation/Message/SendPrivateMessageOperation.cs
@@ -26,6 +26,7 @@ public async Task<OneBotResult> HandleOperation(BotContext context, JsonNode? pa
         var result = await context.SendMessage(chain);
 
         if (result.Result != 0) return new OneBotResult(null, (int)result.Result, "failed");
+        if (result.Sequence == null || result.Sequence == 0) return new OneBotResult(null, 9000, "failed");
 
         int hash = MessageRecord.CalcMessageHash(result.MessageId, result.Sequence ?? 0);