1 files changed, 148 insertions, 0 deletions
diff --git a/vendor/go.mau.fi/whatsmeow/receipt.go b/vendor/go.mau.fi/whatsmeow/receipt.go
new file mode 100644
index 00000000..dec03c5d
--- /dev/null
+++ b/vendor/go.mau.fi/whatsmeow/receipt.go
@@ -0,0 +1,148 @@
+// Copyright (c) 2021 Tulir Asokan
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+package whatsmeow
+
+import (
+	"fmt"
+	"time"
+
+	waBinary "go.mau.fi/whatsmeow/binary"
+	"go.mau.fi/whatsmeow/types"
+	"go.mau.fi/whatsmeow/types/events"
+)
+
+func (cli *Client) handleReceipt(node *waBinary.Node) {
+	receipt, err := cli.parseReceipt(node)
+	if err != nil {
+		cli.Log.Warnf("Failed to parse receipt: %v", err)
+	} else {
+		if receipt.Type == events.ReceiptTypeRetry {
+			go func() {
+				err := cli.handleRetryReceipt(receipt, node)
+				if err != nil {
+					cli.Log.Errorf("Failed to handle retry receipt for %s/%s from %s: %v", receipt.Chat, receipt.MessageIDs[0], receipt.Sender, err)
+				}
+			}()
+		}
+		go cli.dispatchEvent(receipt)
+	}
+	go cli.sendAck(node)
+}
+
+func (cli *Client) parseReceipt(node *waBinary.Node) (*events.Receipt, error) {
+	ag := node.AttrGetter()
+	source, err := cli.parseMessageSource(node)
+	if err != nil {
+		return nil, err
+	}
+	receipt := events.Receipt{
+		MessageSource: source,
+		Timestamp:     time.Unix(ag.Int64("t"), 0),
+		Type:          events.ReceiptType(ag.OptionalString("type")),
+	}
+	mainMessageID := ag.String("id")
+	if !ag.OK() {
+		return nil, fmt.Errorf("failed to parse read receipt attrs: %+v", ag.Errors)
+	}
+
+	receiptChildren := node.GetChildren()
+	if len(receiptChildren) == 1 && receiptChildren[0].Tag == "list" {
+		listChildren := receiptChildren[0].GetChildren()
+		receipt.MessageIDs = make([]string, 1, len(listChildren)+1)
+		receipt.MessageIDs[0] = mainMessageID
+		for _, item := range listChildren {
+			if id, ok := item.Attrs["id"].(string); ok && item.Tag == "item" {
+				receipt.MessageIDs = append(receipt.MessageIDs, id)
+			}
+		}
+	} else {
+		receipt.MessageIDs = []types.MessageID{mainMessageID}
+	}
+	return &receipt, nil
+}
+
+func (cli *Client) sendAck(node *waBinary.Node) {
+	attrs := waBinary.Attrs{
+		"class": node.Tag,
+		"id":    node.Attrs["id"],
+	}
+	attrs["to"] = node.Attrs["from"]
+	if participant, ok := node.Attrs["participant"]; ok {
+		attrs["participant"] = participant
+	}
+	if recipient, ok := node.Attrs["recipient"]; ok {
+		attrs["recipient"] = recipient
+	}
+	if receiptType, ok := node.Attrs["type"]; node.Tag != "message" && ok {
+		attrs["type"] = receiptType
+	}
+	err := cli.sendNode(waBinary.Node{
+		Tag:   "ack",
+		Attrs: attrs,
+	})
+	if err != nil {
+		cli.Log.Warnf("Failed to send acknowledgement for %s %s: %v", node.Tag, node.Attrs["id"], err)
+	}
+}
+
+// MarkRead sends a read receipt for the given message IDs including the given timestamp as the read at time.
+//
+// The first JID parameter (chat) must always be set to the chat ID (user ID in DMs and group ID in group chats).
+// The second JID parameter (sender) must be set in group chats and must be the user ID who sent the message.
+func (cli *Client) MarkRead(ids []types.MessageID, timestamp time.Time, chat, sender types.JID) error {
+	node := waBinary.Node{
+		Tag: "receipt",
+		Attrs: waBinary.Attrs{
+			"id":   ids[0],
+			"type": "read",
+			"to":   chat,
+			"t":    timestamp.Unix(),
+		},
+	}
+	if cli.GetPrivacySettings().ReadReceipts == types.PrivacySettingNone {
+		node.Attrs["type"] = "read-self"
+	}
+	if !sender.IsEmpty() && chat.Server != types.DefaultUserServer {
+		node.Attrs["participant"] = sender.ToNonAD()
+	}
+	if len(ids) > 1 {
+		children := make([]waBinary.Node, len(ids)-1)
+		for i := 1; i < len(ids); i++ {
+			children[i-1].Tag = "item"
+			children[i-1].Attrs = waBinary.Attrs{"id": ids[i]}
+		}
+		node.Content = []waBinary.Node{{
+			Tag:     "list",
+			Content: children,
+		}}
+	}
+	return cli.sendNode(node)
+}
+
+func (cli *Client) sendMessageReceipt(info *types.MessageInfo) {
+	attrs := waBinary.Attrs{
+		"id": info.ID,
+	}
+	if info.IsFromMe {
+		attrs["type"] = "sender"
+	} else {
+		attrs["type"] = "inactive"
+	}
+	attrs["to"] = info.Chat
+	if info.IsGroup {
+		attrs["participant"] = info.Sender
+	} else if info.IsFromMe {
+		attrs["recipient"] = info.Sender
+	}
+	err := cli.sendNode(waBinary.Node{
+		Tag:   "receipt",
+		Attrs: attrs,
+	})
+	if err != nil {
+		cli.Log.Warnf("Failed to send receipt for %s: %v", info.ID, err)
+	}
+}