agentremote/status_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
package agentremote

import (
	"context"
	"errors"

	"maunium.net/go/mautrix/bridgev2"
	"maunium.net/go/mautrix/event"
)

func UnsupportedMessageStatus(err error) error {
	return bridgev2.WrapErrorInStatus(err).
		WithStatus(event.MessageStatusFail).
		WithErrorReason(event.MessageStatusUnsupported).
		WithIsCertain(true).
		WithSendNotice(true).
		WithErrorAsMessage()
}

func MessageSendStatusError(
	err error,
	message string,
	reason event.MessageStatusReason,
	statusForError func(error) event.MessageStatus,
	reasonForError func(error) event.MessageStatusReason,
) error {
	if err == nil {
		err = errors.New(coalesceStrings(message, "message send failed"))
	}
	st := bridgev2.WrapErrorInStatus(err).WithSendNotice(true)
	if statusForError != nil {
		st = st.WithStatus(statusForError(err))
	}
	if reason != "" {
		st = st.WithErrorReason(reason)
	} else if reasonForError != nil {
		st = st.WithErrorReason(reasonForError(err))
	}
	if message != "" {
		st = st.WithMessage(message)
	} else {
		st = st.WithErrorAsMessage()
	}
	return st
}

func SendMatrixMessageStatus(
	ctx context.Context,
	portal *bridgev2.Portal,
	evt *event.Event,
	status bridgev2.MessageStatus,
) {
	if portal == nil || portal.Bridge == nil {
		return
	}
	info := MatrixMessageStatusEventInfo(portal, evt)
	if info == nil {
		return
	}
	portal.Bridge.Matrix.SendMessageStatus(ctx, &status, info)
}

func MatrixMessageStatusEventInfo(
	portal *bridgev2.Portal,
	evt *event.Event,
) *bridgev2.MessageStatusEventInfo {
	if portal == nil || evt == nil {
		return nil
	}
	info := bridgev2.StatusEventInfoFromEvent(evt)
	if info == nil {
		return nil
	}
	if info.RoomID == "" && portal.MXID != "" {
		info.RoomID = portal.MXID
	}
	return info
}