From cf9fa71646747754cd445b94ba45b58bb33cadd3 Mon Sep 17 00:00:00 2001
From: wdvxdr <wdvxdr1123@gmail.com>
Date: Fri, 11 Feb 2022 14:17:54 +0800
Subject: [PATCH] coolq: fix private reply id

Fixes #1368
---
 coolq/cqcode.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/coolq/cqcode.go b/coolq/cqcode.go
index 7f95b1f..c30de37 100644
--- 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 {