agentremote/reaction_helpers.go at main · beeper/agentremote · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
package agentremote

import (
	"time"

	"github.com/rs/zerolog"
	"go.mau.fi/util/variationselector"
	"maunium.net/go/mautrix/bridgev2"
	"maunium.net/go/mautrix/bridgev2/database"
	"maunium.net/go/mautrix/bridgev2/networkid"
	"maunium.net/go/mautrix/bridgev2/simplevent"
)

func reactionEventMeta(
	eventType bridgev2.RemoteEventType,
	portal networkid.PortalKey,
	sender bridgev2.EventSender,
	targetMessage networkid.MessageID,
	logKey string,
	timing EventTiming,
) simplevent.EventMeta {
	return simplevent.EventMeta{
		Type:        eventType,
		PortalKey:   portal,
		Sender:      sender,
		Timestamp:   timing.Timestamp,
		StreamOrder: timing.StreamOrder,
		LogContext: func(c zerolog.Context) zerolog.Context {
			return c.Str(logKey, string(targetMessage))
		},
	}
}

// BuildReactionEvent creates a reaction add event with normalized emoji data.
func BuildReactionEvent(
	portal networkid.PortalKey,
	sender bridgev2.EventSender,
	targetMessage networkid.MessageID,
	emoji string,
	emojiID networkid.EmojiID,
	timestamp time.Time,
	streamOrder int64,
	logKey string,
	dbMeta *database.Reaction,
	extraContent map[string]any,
) *simplevent.Reaction {
	normalized := variationselector.Remove(emoji)
	if normalized == "" {
		normalized = variationselector.Remove(string(emojiID))
	}
	if emojiID == "" {
		emojiID = networkid.EmojiID(normalized)
	}
	timing := ResolveEventTiming(timestamp, streamOrder)
	return &simplevent.Reaction{
		EventMeta:      reactionEventMeta(bridgev2.RemoteEventReaction, portal, sender, targetMessage, logKey, timing),
		TargetMessage:  targetMessage,
		Emoji:          normalized,
		EmojiID:        emojiID,
		ReactionDBMeta: dbMeta,
		ExtraContent:   extraContent,
	}
}

// BuildReactionRemoveEvent creates a reaction removal event with explicit timing.
func BuildReactionRemoveEvent(
	portal networkid.PortalKey,
	sender bridgev2.EventSender,
	targetMessage networkid.MessageID,
	emojiID networkid.EmojiID,
	timestamp time.Time,
	streamOrder int64,
	logKey string,
) *simplevent.Reaction {
	timing := ResolveEventTiming(timestamp, streamOrder)
	return &simplevent.Reaction{
		EventMeta:     reactionEventMeta(bridgev2.RemoteEventReactionRemove, portal, sender, targetMessage, logKey, timing),
		TargetMessage: targetMessage,
		EmojiID:       emojiID,
	}
}