From 32565dd18898ee5d2cd8405d769881664af956e1 Mon Sep 17 00:00:00 2001
From: skiffer-git <44203734@qq.com>
Date: Thu, 9 Jun 2022 13:57:10 +0800
Subject: [PATCH] config for msg read receipt

---
 internal/rpc/msg/send_msg.go    | 24 +++++++++++++++++++++-
 pkg/common/config/config.go     | 15 +++++++-------
 pkg/common/constant/constant.go | 35 +++++++++++++++++----------------
 pkg/common/constant/error.go    | 13 ++++++------
 4 files changed, 56 insertions(+), 31 deletions(-)

diff --git a/internal/rpc/msg/send_msg.go b/internal/rpc/msg/send_msg.go
index 1000c6771..be6a52f66 100644
--- a/internal/rpc/msg/send_msg.go
+++ b/internal/rpc/msg/send_msg.go
@@ -49,6 +49,24 @@ type MsgCallBackResp struct {
 	}
 }
 
+func isMessageHasReadEnabled(pb *pbChat.SendMsgReq) (bool, int32, string) {
+	switch pb.MsgData.ContentType {
+	case constant.HasReadReceipt:
+		if config.Config.SingleMessageHasReadReceiptEnable {
+			return true, 0, ""
+		} else {
+			return false, constant.ErrMessageHasReadDisable.ErrCode, constant.ErrMessageHasReadDisable.ErrMsg
+		}
+	case constant.GroupHasReadReceipt:
+		if config.Config.GroupMessageHasReadReceiptEnable {
+			return true, 0, ""
+		} else {
+			return false, constant.ErrMessageHasReadDisable.ErrCode, constant.ErrMessageHasReadDisable.ErrMsg
+		}
+	}
+	return true, 0, ""
+}
+
 func userRelationshipVerification(data *pbChat.SendMsgReq) (bool, int32, string) {
 	if data.MsgData.SessionType == constant.GroupChatType {
 		return true, 0, ""
@@ -144,7 +162,11 @@ func (rpc *rpcChat) SendMsg(_ context.Context, pb *pbChat.SendMsgReq) (*pbChat.S
 	replay := pbChat.SendMsgResp{}
 	newTime := db.GetCurrentTimestampByMill()
 	log.NewWarn(pb.OperationID, "rpc sendMsg come here", pb.String(), pb.MsgData.ClientMsgID)
-	flag, errCode, errMsg := userRelationshipVerification(pb)
+	flag, errCode, errMsg := isMessageHasReadEnabled(pb)
+	if !flag {
+		return returnMsg(&replay, pb, errCode, errMsg, "", 0)
+	}
+	flag, errCode, errMsg = userRelationshipVerification(pb)
 	if !flag {
 		return returnMsg(&replay, pb, errCode, errMsg, "", 0)
 	}
diff --git a/pkg/common/config/config.go b/pkg/common/config/config.go
index 2b9a4d891..9da3fdf29 100644
--- a/pkg/common/config/config.go
+++ b/pkg/common/config/config.go
@@ -214,13 +214,14 @@ type config struct {
 			MsgToPush         string `yaml:"msgToPush"`
 		}
 	}
-	Secret               string `yaml:"secret"`
-	MultiLoginPolicy     int    `yaml:"multiloginpolicy"`
-	ChatPersistenceMysql bool   `yaml:"chatpersistencemysql"`
-	ReliableStorage      bool   `yaml:"reliablestorage"`
-	MsgCacheTimeout      int    `yaml:"msgCacheTimeout"`
-
-	TokenPolicy struct {
+	Secret                            string `yaml:"secret"`
+	MultiLoginPolicy                  int    `yaml:"multiloginpolicy"`
+	ChatPersistenceMysql              bool   `yaml:"chatpersistencemysql"`
+	ReliableStorage                   bool   `yaml:"reliablestorage"`
+	MsgCacheTimeout                   int    `yaml:"msgCacheTimeout"`
+	GroupMessageHasReadReceiptEnable  bool   `yaml:"groupMessageHasReadReceiptEnable"`
+	SingleMessageHasReadReceiptEnable bool   `yaml:"singleMessageHasReadReceiptEnable"`
+	TokenPolicy                       struct {
 		AccessSecret string `yaml:"accessSecret"`
 		AccessExpire int64  `yaml:"accessExpire"`
 	}
diff --git a/pkg/common/constant/constant.go b/pkg/common/constant/constant.go
index 53a4cce38..a726111b3 100644
--- a/pkg/common/constant/constant.go
+++ b/pkg/common/constant/constant.go
@@ -28,23 +28,24 @@ const (
 
 	///ContentType
 	//UserRelated
-	Text           = 101
-	Picture        = 102
-	Voice          = 103
-	Video          = 104
-	File           = 105
-	AtText         = 106
-	Merger         = 107
-	Card           = 108
-	Location       = 109
-	Custom         = 110
-	Revoke         = 111
-	HasReadReceipt = 112
-	Typing         = 113
-	Quote          = 114
-	Common         = 200
-	GroupMsg       = 201
-	SignalMsg      = 202
+	Text                = 101
+	Picture             = 102
+	Voice               = 103
+	Video               = 104
+	File                = 105
+	AtText              = 106
+	Merger              = 107
+	Card                = 108
+	Location            = 109
+	Custom              = 110
+	Revoke              = 111
+	HasReadReceipt      = 112
+	Typing              = 113
+	Quote               = 114
+	GroupHasReadReceipt = 116
+	Common              = 200
+	GroupMsg            = 201
+	SignalMsg           = 202
 
 	//SysRelated
 	NotificationBegin                     = 1000
diff --git a/pkg/common/constant/error.go b/pkg/common/constant/error.go
index b0c28725e..52c1253e1 100644
--- a/pkg/common/constant/error.go
+++ b/pkg/common/constant/error.go
@@ -49,12 +49,13 @@ var (
 	ErrTokenUnknown     = ErrInfo{705, TokenUnknownMsg.Error()}
 	ErrTokenKicked      = ErrInfo{706, TokenUserKickedMsg.Error()}
 
-	ErrAccess    = ErrInfo{ErrCode: 801, ErrMsg: AccessMsg.Error()}
-	ErrDB        = ErrInfo{ErrCode: 802, ErrMsg: DBMsg.Error()}
-	ErrArgs      = ErrInfo{ErrCode: 803, ErrMsg: ArgsMsg.Error()}
-	ErrStatus    = ErrInfo{ErrCode: 804, ErrMsg: StatusMsg.Error()}
-	ErrCallback  = ErrInfo{ErrCode: 809, ErrMsg: CallBackMsg.Error()}
-	ErrSendLimit = ErrInfo{ErrCode: 810, ErrMsg: "send msg limit, to many request, try again later"}
+	ErrAccess                = ErrInfo{ErrCode: 801, ErrMsg: AccessMsg.Error()}
+	ErrDB                    = ErrInfo{ErrCode: 802, ErrMsg: DBMsg.Error()}
+	ErrArgs                  = ErrInfo{ErrCode: 803, ErrMsg: ArgsMsg.Error()}
+	ErrStatus                = ErrInfo{ErrCode: 804, ErrMsg: StatusMsg.Error()}
+	ErrCallback              = ErrInfo{ErrCode: 809, ErrMsg: CallBackMsg.Error()}
+	ErrSendLimit             = ErrInfo{ErrCode: 810, ErrMsg: "send msg limit, to many request, try again later"}
+	ErrMessageHasReadDisable = ErrInfo{ErrCode: 811, ErrMsg: "message has read disable"}
 )
 
 var (