coolq: fix private reply id

Fixes #1368
Mrs4s · Feb 11, 2022 · cf9fa71 · cf9fa71
1 parent 115a5f1
commit cf9fa71
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/coolq/cqcode.go b/coolq/cqcode.go
@@ -114,7 +114,7 @@ func ToArrayMessage(e []message.IMessageElement, source MessageSource) (r []glob
 	if reply != nil && source.SourceType&(MessageSourceGroup|MessageSourcePrivate) != 0 {
 		replyElem := reply.(*message.ReplyElement)
 		rid := int64(source.PrimaryID)
-		if rid == 0 {
+		if source.SourceType == MessageSourcePrivate {
 			rid = replyElem.Sender
 		}
 		if replyElem.GroupID != 0 {
@@ -280,7 +280,7 @@ func ToStringMessage(e []message.IMessageElement, source MessageSource, isRaw ..
 	if reply != nil && source.SourceType&(MessageSourceGroup|MessageSourcePrivate) != 0 {
 		replyElem := reply.(*message.ReplyElement)
 		rid := int64(source.PrimaryID)
-		if rid == 0 {
+		if source.SourceType == MessageSourcePrivate {
 			rid = replyElem.Sender
 		}
 		if replyElem.GroupID != 0 {