feat: 检查消息类型时额外判断TemplateMessage

SNWCreations · Nov 12, 2024 · b2b5ce6 · b2b5ce6
1 parent 6bee341
commit b2b5ce6
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/main/java/snw/kookbc/impl/message/MessageImpl.java b/src/main/java/snw/kookbc/impl/message/MessageImpl.java
@@ -163,7 +163,13 @@ protected final void checkCompatibleComponentType(BaseComponent newIncoming) {
         if (this.component == null) {
             return; // we don't know, let HTTP API check
         }
-        final boolean compatible = isCompatibleComponentType(this.component, newIncoming);
+        boolean compatible;
+        if (newIncoming instanceof TemplateMessage) {
+            int oldType = (int) MessageBuilder.serialize(this.component)[0];
+            compatible = ((TemplateMessage) newIncoming).getType() == oldType;
+        } else {
+            compatible = isCompatibleComponentType(this.component, newIncoming);
+        }
         if (!compatible) {
             throw new IllegalArgumentException("Incompatible component type, tried updating from "
                     + this.component.getClass() + " to " + newIncoming.getClass());