reactions: support TG -> Matrix
Signed-off-by: Sumner Evans <sumner.evans@automattic.com>
This commit is contained in:
+7
-174
@@ -6,7 +6,7 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"sync"
|
||||||
|
|
||||||
"github.com/gotd/td/telegram"
|
"github.com/gotd/td/telegram"
|
||||||
"github.com/gotd/td/telegram/updates"
|
"github.com/gotd/td/telegram/updates"
|
||||||
@@ -30,6 +30,8 @@ type TelegramClient struct {
|
|||||||
client *telegram.Client
|
client *telegram.Client
|
||||||
clientCancel context.CancelFunc
|
clientCancel context.CancelFunc
|
||||||
msgConv *msgconv.MessageConverter
|
msgConv *msgconv.MessageConverter
|
||||||
|
|
||||||
|
reactionMessageLocks map[int]*sync.Mutex
|
||||||
}
|
}
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -97,6 +99,7 @@ func NewTelegramClient(ctx context.Context, tc *TelegramConnector, login *bridge
|
|||||||
dispatcher.OnNewChannelMessage(client.onUpdateNewChannelMessage)
|
dispatcher.OnNewChannelMessage(client.onUpdateNewChannelMessage)
|
||||||
dispatcher.OnUserName(client.onUserName)
|
dispatcher.OnUserName(client.onUserName)
|
||||||
dispatcher.OnDeleteMessages(client.onDeleteMessages)
|
dispatcher.OnDeleteMessages(client.onDeleteMessages)
|
||||||
|
dispatcher.OnEditMessage(client.onMessageEdit)
|
||||||
|
|
||||||
store := tc.store.GetScopedStore(loginID)
|
store := tc.store.GetScopedStore(loginID)
|
||||||
|
|
||||||
@@ -118,6 +121,7 @@ func NewTelegramClient(ctx context.Context, tc *TelegramConnector, login *bridge
|
|||||||
})
|
})
|
||||||
client.msgConv = msgconv.NewMessageConverter(client.client, tc.Bridge.Matrix, tc.useDirectMedia)
|
client.msgConv = msgconv.NewMessageConverter(client.client, tc.Bridge.Matrix, tc.useDirectMedia)
|
||||||
client.clientCancel, err = connectTelegramClient(ctx, client.client)
|
client.clientCancel, err = connectTelegramClient(ctx, client.client)
|
||||||
|
client.reactionMessageLocks = map[int]*sync.Mutex{}
|
||||||
go func() {
|
go func() {
|
||||||
err = updatesManager.Run(ctx, client.client.API(), loginID, updates.AuthOptions{})
|
err = updatesManager.Run(ctx, client.client.API(), loginID, updates.AuthOptions{})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -161,179 +165,8 @@ func connectTelegramClient(ctx context.Context, client *telegram.Client) (contex
|
|||||||
return cancel, nil
|
return cancel, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *TelegramClient) onUpdateNewMessage(ctx context.Context, e tg.Entities, update *tg.UpdateNewMessage) error {
|
|
||||||
log := zerolog.Ctx(ctx)
|
|
||||||
switch msg := update.GetMessage().(type) {
|
|
||||||
case *tg.Message:
|
|
||||||
sender := t.getEventSender(msg)
|
|
||||||
|
|
||||||
if media, ok := msg.GetMedia(); ok && media.TypeID() == tg.MessageMediaContactTypeID {
|
|
||||||
contact := media.(*tg.MessageMediaContact)
|
|
||||||
// TODO update the corresponding puppet
|
|
||||||
log.Info().Int64("user_id", contact.UserID).Msg("received contact")
|
|
||||||
}
|
|
||||||
|
|
||||||
t.main.Bridge.QueueRemoteEvent(t.userLogin, &bridgev2.SimpleRemoteEvent[*tg.Message]{
|
|
||||||
Type: bridgev2.RemoteEventMessage,
|
|
||||||
LogContext: func(c zerolog.Context) zerolog.Context {
|
|
||||||
return c.
|
|
||||||
Int("message_id", update.Message.GetID()).
|
|
||||||
Str("sender", string(sender.Sender)).
|
|
||||||
Str("sender_login", string(sender.SenderLogin)).
|
|
||||||
Bool("is_from_me", sender.IsFromMe)
|
|
||||||
},
|
|
||||||
ID: ids.MakeMessageID(msg.ID),
|
|
||||||
Sender: sender,
|
|
||||||
PortalKey: ids.MakePortalID(msg.PeerID),
|
|
||||||
Data: msg,
|
|
||||||
CreatePortal: true,
|
|
||||||
ConvertMessageFunc: t.msgConv.ToMatrix,
|
|
||||||
Timestamp: time.Unix(int64(msg.Date), 0),
|
|
||||||
})
|
|
||||||
case *tg.MessageService:
|
|
||||||
fmt.Printf("message service\n")
|
|
||||||
fmt.Printf("%v\n", msg)
|
|
||||||
|
|
||||||
// sender := t.getEventSender(msg)
|
|
||||||
// switch action := msg.Action.(type) {
|
|
||||||
// case *tg.MessageActionChatEditTitle:
|
|
||||||
// case *tg.MessageActionChatCreate:
|
|
||||||
// case *tg.MessageActionChatEditPhoto:
|
|
||||||
// case *tg.MessageActionChatDeletePhoto:
|
|
||||||
// case *tg.MessageActionChatAddUser:
|
|
||||||
// case *tg.MessageActionChatDeleteUser:
|
|
||||||
// case *tg.MessageActionChatJoinedByLink:
|
|
||||||
// case *tg.MessageActionChannelCreate:
|
|
||||||
// case *tg.MessageActionChatMigrateTo:
|
|
||||||
// case *tg.MessageActionChannelMigrateFrom:
|
|
||||||
// case *tg.MessageActionPinMessage:
|
|
||||||
// case *tg.MessageActionHistoryClear:
|
|
||||||
// case *tg.MessageActionGameScore:
|
|
||||||
// case *tg.MessageActionPaymentSentMe:
|
|
||||||
// case *tg.MessageActionPaymentSent:
|
|
||||||
// case *tg.MessageActionPhoneCall:
|
|
||||||
// case *tg.MessageActionScreenshotTaken:
|
|
||||||
// case *tg.MessageActionCustomAction:
|
|
||||||
// case *tg.MessageActionBotAllowed:
|
|
||||||
// case *tg.MessageActionSecureValuesSentMe:
|
|
||||||
// case *tg.MessageActionSecureValuesSent:
|
|
||||||
// case *tg.MessageActionContactSignUp:
|
|
||||||
// case *tg.MessageActionGeoProximityReached:
|
|
||||||
// case *tg.MessageActionGroupCall:
|
|
||||||
// case *tg.MessageActionInviteToGroupCall:
|
|
||||||
// case *tg.MessageActionSetMessagesTTL:
|
|
||||||
// case *tg.MessageActionGroupCallScheduled:
|
|
||||||
// case *tg.MessageActionSetChatTheme:
|
|
||||||
// case *tg.MessageActionChatJoinedByRequest:
|
|
||||||
// case *tg.MessageActionWebViewDataSentMe:
|
|
||||||
// case *tg.MessageActionWebViewDataSent:
|
|
||||||
// case *tg.MessageActionGiftPremium:
|
|
||||||
// case *tg.MessageActionTopicCreate:
|
|
||||||
// case *tg.MessageActionTopicEdit:
|
|
||||||
// case *tg.MessageActionSuggestProfilePhoto:
|
|
||||||
// case *tg.MessageActionRequestedPeer:
|
|
||||||
// case *tg.MessageActionSetChatWallPaper:
|
|
||||||
// case *tg.MessageActionGiftCode:
|
|
||||||
// case *tg.MessageActionGiveawayLaunch:
|
|
||||||
// case *tg.MessageActionGiveawayResults:
|
|
||||||
// case *tg.MessageActionBoostApply:
|
|
||||||
// case *tg.MessageActionRequestedPeerSentMe:
|
|
||||||
// default:
|
|
||||||
// return fmt.Errorf("unknown action type %T", action)
|
|
||||||
// }
|
|
||||||
|
|
||||||
default:
|
|
||||||
return fmt.Errorf("unknown message type %T", msg)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type messageWithSender interface {
|
|
||||||
GetOut() bool
|
|
||||||
GetFromID() (tg.PeerClass, bool)
|
|
||||||
GetPeerID() tg.PeerClass
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *TelegramClient) getEventSender(msg messageWithSender) (sender bridgev2.EventSender) {
|
|
||||||
if msg.GetOut() {
|
|
||||||
sender.IsFromMe = true
|
|
||||||
sender.SenderLogin = ids.MakeUserLoginID(t.loginID)
|
|
||||||
sender.Sender = ids.MakeUserID(t.loginID)
|
|
||||||
} else if f, ok := msg.GetFromID(); ok {
|
|
||||||
switch from := f.(type) {
|
|
||||||
case *tg.PeerUser:
|
|
||||||
sender.SenderLogin = ids.MakeUserLoginID(from.UserID)
|
|
||||||
sender.Sender = ids.MakeUserID(from.UserID)
|
|
||||||
default:
|
|
||||||
fmt.Printf("%+v\n", f)
|
|
||||||
fmt.Printf("%T\n", f)
|
|
||||||
panic("unimplemented FromID")
|
|
||||||
}
|
|
||||||
} else if peer, ok := msg.GetPeerID().(*tg.PeerUser); ok {
|
|
||||||
sender.SenderLogin = ids.MakeUserLoginID(peer.UserID)
|
|
||||||
sender.Sender = ids.MakeUserID(peer.UserID)
|
|
||||||
} else {
|
|
||||||
panic("not from anyone")
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *TelegramClient) onUpdateNewChannelMessage(ctx context.Context, e tg.Entities, update *tg.UpdateNewChannelMessage) error {
|
|
||||||
fmt.Printf("update new channel message %+v\n", update)
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *TelegramClient) onUserName(ctx context.Context, e tg.Entities, update *tg.UpdateUserName) error {
|
|
||||||
ghost, err := t.main.Bridge.GetGhostByID(ctx, ids.MakeUserID(update.UserID))
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
name := util.FormatFullName(update.FirstName, update.LastName)
|
|
||||||
|
|
||||||
// TODO update identifiers?
|
|
||||||
ghost.UpdateInfo(ctx, &bridgev2.UserInfo{Name: &name})
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *TelegramClient) onDeleteMessages(ctx context.Context, e tg.Entities, update *tg.UpdateDeleteMessages) error {
|
|
||||||
for _, messageID := range update.Messages {
|
|
||||||
parts, err := t.main.Bridge.DB.Message.GetAllPartsByID(ctx, ids.MakeUserLoginID(t.loginID), ids.MakeMessageID(messageID))
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if len(parts) == 0 {
|
|
||||||
return fmt.Errorf("no parts found for message %d", messageID)
|
|
||||||
}
|
|
||||||
t.main.Bridge.QueueRemoteEvent(t.userLogin, &bridgev2.SimpleRemoteEvent[any]{
|
|
||||||
Type: bridgev2.RemoteEventMessageRemove,
|
|
||||||
LogContext: func(c zerolog.Context) zerolog.Context {
|
|
||||||
return c.
|
|
||||||
Str("action", "delete message").
|
|
||||||
Int("message_id", messageID)
|
|
||||||
},
|
|
||||||
PortalKey: parts[0].Room,
|
|
||||||
CreatePortal: false,
|
|
||||||
TargetMessage: ids.MakeMessageID(messageID),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *TelegramClient) onEntityUpdate(ctx context.Context, e tg.Entities) error {
|
|
||||||
for userID, user := range e.Users {
|
|
||||||
ghost, err := t.main.Bridge.GetGhostByID(ctx, ids.MakeUserID(userID))
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
userInfo, err := t.getUserInfoFromTelegramUser(ctx, user)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
ghost.UpdateInfo(ctx, userInfo)
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t *TelegramClient) Connect(ctx context.Context) (err error) {
|
func (t *TelegramClient) Connect(ctx context.Context) (err error) {
|
||||||
t.clientCancel, err = connectTelegramClient(ctx, t.client)
|
t.clientCancel, err = connectTelegramClient(ctx, t.client)
|
||||||
@@ -452,11 +285,11 @@ func (t *TelegramClient) GetUserInfo(ctx context.Context, ghost *bridgev2.Ghost)
|
|||||||
if user, ok := users[0].(*tg.User); !ok {
|
if user, ok := users[0].(*tg.User); !ok {
|
||||||
return nil, fmt.Errorf("returned user is not *tg.User")
|
return nil, fmt.Errorf("returned user is not *tg.User")
|
||||||
} else {
|
} else {
|
||||||
return t.getUserInfoFromTelegramUser(ctx, user)
|
return t.getUserInfoFromTelegramUser(user)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *TelegramClient) getUserInfoFromTelegramUser(ctx context.Context, user *tg.User) (*bridgev2.UserInfo, error) {
|
func (t *TelegramClient) getUserInfoFromTelegramUser(user *tg.User) (*bridgev2.UserInfo, error) {
|
||||||
var identifiers []string
|
var identifiers []string
|
||||||
for _, username := range user.Usernames {
|
for _, username := range user.Usernames {
|
||||||
identifiers = append(identifiers, fmt.Sprintf("telegram:%s", username.Username))
|
identifiers = append(identifiers, fmt.Sprintf("telegram:%s", username.Username))
|
||||||
|
|||||||
@@ -21,6 +21,7 @@ func MakeUserLoginID(userID int64) networkid.UserLoginID {
|
|||||||
return networkid.UserLoginID(strconv.FormatInt(userID, 10))
|
return networkid.UserLoginID(strconv.FormatInt(userID, 10))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TODO: add space ID
|
||||||
func MakeMessageID(messageID int) networkid.MessageID {
|
func MakeMessageID(messageID int) networkid.MessageID {
|
||||||
return networkid.MessageID(strconv.Itoa(messageID))
|
return networkid.MessageID(strconv.Itoa(messageID))
|
||||||
}
|
}
|
||||||
@@ -67,7 +68,7 @@ func (pt PeerType) AsPortalKey(chatID int64) networkid.PortalKey {
|
|||||||
return networkid.PortalKey{ID: networkid.PortalID(fmt.Sprintf("%s:%d", pt, chatID))}
|
return networkid.PortalKey{ID: networkid.PortalID(fmt.Sprintf("%s:%d", pt, chatID))}
|
||||||
}
|
}
|
||||||
|
|
||||||
func MakePortalID(peer tg.PeerClass) networkid.PortalKey {
|
func MakePortalKey(peer tg.PeerClass) networkid.PortalKey {
|
||||||
switch v := peer.(type) {
|
switch v := peer.(type) {
|
||||||
case *tg.PeerUser:
|
case *tg.PeerUser:
|
||||||
return networkid.PortalKey{ID: networkid.PortalID(fmt.Sprintf("%s:%d", PeerTypeUser, v.UserID))}
|
return networkid.PortalKey{ID: networkid.PortalID(fmt.Sprintf("%s:%d", PeerTypeUser, v.UserID))}
|
||||||
@@ -104,6 +105,30 @@ func InputPeerForPortalID(portalID networkid.PortalID) (tg.InputPeerClass, error
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func InputPeerForPortalKey(portalKey networkid.PortalKey) (tg.InputPeerClass, error) {
|
||||||
|
return InputPeerForPortalID(portalKey.ID)
|
||||||
|
}
|
||||||
|
|
||||||
func MakeAvatarID(photoID int64) networkid.AvatarID {
|
func MakeAvatarID(photoID int64) networkid.AvatarID {
|
||||||
return networkid.AvatarID(strconv.FormatInt(photoID, 10))
|
return networkid.AvatarID(strconv.FormatInt(photoID, 10))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func MakeEmojiIDFromDocumentID(documentID int64) networkid.EmojiID {
|
||||||
|
return networkid.EmojiID(fmt.Sprintf("d%d", documentID))
|
||||||
|
}
|
||||||
|
|
||||||
|
func MakeEmojiIDFromEmoticon(emoticon string) networkid.EmojiID {
|
||||||
|
return networkid.EmojiID(fmt.Sprintf("e%s", emoticon))
|
||||||
|
}
|
||||||
|
|
||||||
|
func ParseEmojiID(emojiID networkid.EmojiID) (documentID int64, emoticon string, err error) {
|
||||||
|
switch emojiID[0] {
|
||||||
|
case 'd':
|
||||||
|
documentID, err = strconv.ParseInt(string(emojiID[1:]), 10, 64)
|
||||||
|
case 'e':
|
||||||
|
emoticon = string(emojiID[1:])
|
||||||
|
default:
|
||||||
|
err = fmt.Errorf("invalid emoji ID type %s", string(emojiID[0]))
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|||||||
@@ -0,0 +1,451 @@
|
|||||||
|
package connector
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"slices"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/gotd/td/tg"
|
||||||
|
"github.com/rs/zerolog"
|
||||||
|
"maunium.net/go/mautrix/bridgev2"
|
||||||
|
"maunium.net/go/mautrix/bridgev2/database"
|
||||||
|
"maunium.net/go/mautrix/bridgev2/networkid"
|
||||||
|
|
||||||
|
"go.mau.fi/mautrix-telegram/pkg/connector/ids"
|
||||||
|
"go.mau.fi/mautrix-telegram/pkg/connector/util"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (t *TelegramClient) onUpdateNewMessage(ctx context.Context, e tg.Entities, update *tg.UpdateNewMessage) error {
|
||||||
|
log := zerolog.Ctx(ctx)
|
||||||
|
switch msg := update.GetMessage().(type) {
|
||||||
|
case *tg.Message:
|
||||||
|
sender := t.getEventSender(msg)
|
||||||
|
|
||||||
|
if err := t.handleTelegramReactions(ctx, msg); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if media, ok := msg.GetMedia(); ok && media.TypeID() == tg.MessageMediaContactTypeID {
|
||||||
|
contact := media.(*tg.MessageMediaContact)
|
||||||
|
// TODO update the corresponding puppet
|
||||||
|
log.Info().Int64("user_id", contact.UserID).Msg("received contact")
|
||||||
|
}
|
||||||
|
|
||||||
|
t.main.Bridge.QueueRemoteEvent(t.userLogin, &bridgev2.SimpleRemoteEvent[*tg.Message]{
|
||||||
|
Type: bridgev2.RemoteEventMessage,
|
||||||
|
LogContext: func(c zerolog.Context) zerolog.Context {
|
||||||
|
return c.
|
||||||
|
Int("message_id", update.Message.GetID()).
|
||||||
|
Str("sender", string(sender.Sender)).
|
||||||
|
Str("sender_login", string(sender.SenderLogin)).
|
||||||
|
Bool("is_from_me", sender.IsFromMe)
|
||||||
|
},
|
||||||
|
ID: ids.MakeMessageID(msg.ID),
|
||||||
|
Sender: sender,
|
||||||
|
PortalKey: ids.MakePortalKey(msg.PeerID),
|
||||||
|
Data: msg,
|
||||||
|
CreatePortal: true,
|
||||||
|
ConvertMessageFunc: t.msgConv.ToMatrix,
|
||||||
|
Timestamp: time.Unix(int64(msg.Date), 0),
|
||||||
|
})
|
||||||
|
case *tg.MessageService:
|
||||||
|
fmt.Printf("message service\n")
|
||||||
|
fmt.Printf("%v\n", msg)
|
||||||
|
|
||||||
|
// sender := t.getEventSender(msg)
|
||||||
|
// switch action := msg.Action.(type) {
|
||||||
|
// case *tg.MessageActionChatEditTitle:
|
||||||
|
// case *tg.MessageActionChatCreate:
|
||||||
|
// case *tg.MessageActionChatEditPhoto:
|
||||||
|
// case *tg.MessageActionChatDeletePhoto:
|
||||||
|
// case *tg.MessageActionChatAddUser:
|
||||||
|
// case *tg.MessageActionChatDeleteUser:
|
||||||
|
// case *tg.MessageActionChatJoinedByLink:
|
||||||
|
// case *tg.MessageActionChannelCreate:
|
||||||
|
// case *tg.MessageActionChatMigrateTo:
|
||||||
|
// case *tg.MessageActionChannelMigrateFrom:
|
||||||
|
// case *tg.MessageActionPinMessage:
|
||||||
|
// case *tg.MessageActionHistoryClear:
|
||||||
|
// case *tg.MessageActionGameScore:
|
||||||
|
// case *tg.MessageActionPaymentSentMe:
|
||||||
|
// case *tg.MessageActionPaymentSent:
|
||||||
|
// case *tg.MessageActionPhoneCall:
|
||||||
|
// case *tg.MessageActionScreenshotTaken:
|
||||||
|
// case *tg.MessageActionCustomAction:
|
||||||
|
// case *tg.MessageActionBotAllowed:
|
||||||
|
// case *tg.MessageActionSecureValuesSentMe:
|
||||||
|
// case *tg.MessageActionSecureValuesSent:
|
||||||
|
// case *tg.MessageActionContactSignUp:
|
||||||
|
// case *tg.MessageActionGeoProximityReached:
|
||||||
|
// case *tg.MessageActionGroupCall:
|
||||||
|
// case *tg.MessageActionInviteToGroupCall:
|
||||||
|
// case *tg.MessageActionSetMessagesTTL:
|
||||||
|
// case *tg.MessageActionGroupCallScheduled:
|
||||||
|
// case *tg.MessageActionSetChatTheme:
|
||||||
|
// case *tg.MessageActionChatJoinedByRequest:
|
||||||
|
// case *tg.MessageActionWebViewDataSentMe:
|
||||||
|
// case *tg.MessageActionWebViewDataSent:
|
||||||
|
// case *tg.MessageActionGiftPremium:
|
||||||
|
// case *tg.MessageActionTopicCreate:
|
||||||
|
// case *tg.MessageActionTopicEdit:
|
||||||
|
// case *tg.MessageActionSuggestProfilePhoto:
|
||||||
|
// case *tg.MessageActionRequestedPeer:
|
||||||
|
// case *tg.MessageActionSetChatWallPaper:
|
||||||
|
// case *tg.MessageActionGiftCode:
|
||||||
|
// case *tg.MessageActionGiveawayLaunch:
|
||||||
|
// case *tg.MessageActionGiveawayResults:
|
||||||
|
// case *tg.MessageActionBoostApply:
|
||||||
|
// case *tg.MessageActionRequestedPeerSentMe:
|
||||||
|
// default:
|
||||||
|
// return fmt.Errorf("unknown action type %T", action)
|
||||||
|
// }
|
||||||
|
|
||||||
|
default:
|
||||||
|
return fmt.Errorf("unknown message type %T", msg)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type messageWithSender interface {
|
||||||
|
GetOut() bool
|
||||||
|
GetFromID() (tg.PeerClass, bool)
|
||||||
|
GetPeerID() tg.PeerClass
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TelegramClient) getEventSender(msg messageWithSender) (sender bridgev2.EventSender) {
|
||||||
|
if msg.GetOut() {
|
||||||
|
sender.IsFromMe = true
|
||||||
|
sender.SenderLogin = ids.MakeUserLoginID(t.loginID)
|
||||||
|
sender.Sender = ids.MakeUserID(t.loginID)
|
||||||
|
} else if f, ok := msg.GetFromID(); ok {
|
||||||
|
switch from := f.(type) {
|
||||||
|
case *tg.PeerUser:
|
||||||
|
sender.SenderLogin = ids.MakeUserLoginID(from.UserID)
|
||||||
|
sender.Sender = ids.MakeUserID(from.UserID)
|
||||||
|
default:
|
||||||
|
fmt.Printf("%+v\n", f)
|
||||||
|
fmt.Printf("%T\n", f)
|
||||||
|
panic("unimplemented FromID")
|
||||||
|
}
|
||||||
|
} else if peer, ok := msg.GetPeerID().(*tg.PeerUser); ok {
|
||||||
|
sender.SenderLogin = ids.MakeUserLoginID(peer.UserID)
|
||||||
|
sender.Sender = ids.MakeUserID(peer.UserID)
|
||||||
|
} else {
|
||||||
|
panic("not from anyone")
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TelegramClient) onUpdateNewChannelMessage(ctx context.Context, e tg.Entities, update *tg.UpdateNewChannelMessage) error {
|
||||||
|
fmt.Printf("update new channel message %+v\n", update)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TelegramClient) onUserName(ctx context.Context, e tg.Entities, update *tg.UpdateUserName) error {
|
||||||
|
ghost, err := t.main.Bridge.GetGhostByID(ctx, ids.MakeUserID(update.UserID))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
name := util.FormatFullName(update.FirstName, update.LastName)
|
||||||
|
|
||||||
|
// TODO update identifiers?
|
||||||
|
ghost.UpdateInfo(ctx, &bridgev2.UserInfo{Name: &name})
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TelegramClient) onDeleteMessages(ctx context.Context, e tg.Entities, update *tg.UpdateDeleteMessages) error {
|
||||||
|
for _, messageID := range update.Messages {
|
||||||
|
parts, err := t.main.Bridge.DB.Message.GetAllPartsByID(ctx, ids.MakeUserLoginID(t.loginID), ids.MakeMessageID(messageID))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if len(parts) == 0 {
|
||||||
|
return fmt.Errorf("no parts found for message %d", messageID)
|
||||||
|
}
|
||||||
|
t.main.Bridge.QueueRemoteEvent(t.userLogin, &bridgev2.SimpleRemoteEvent[any]{
|
||||||
|
Type: bridgev2.RemoteEventMessageRemove,
|
||||||
|
LogContext: func(c zerolog.Context) zerolog.Context {
|
||||||
|
return c.
|
||||||
|
Str("action", "delete message").
|
||||||
|
Int("message_id", messageID)
|
||||||
|
},
|
||||||
|
PortalKey: parts[0].Room,
|
||||||
|
CreatePortal: false,
|
||||||
|
TargetMessage: ids.MakeMessageID(messageID),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TelegramClient) onEntityUpdate(ctx context.Context, e tg.Entities) error {
|
||||||
|
for userID, user := range e.Users {
|
||||||
|
ghost, err := t.main.Bridge.GetGhostByID(ctx, ids.MakeUserID(userID))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
userInfo, err := t.getUserInfoFromTelegramUser(user)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
ghost.UpdateInfo(ctx, userInfo)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TelegramClient) onMessageEdit(ctx context.Context, e tg.Entities, update *tg.UpdateEditMessage) error {
|
||||||
|
fmt.Printf("message edit %+v\n", update)
|
||||||
|
msg, ok := update.Message.(*tg.Message)
|
||||||
|
if !ok {
|
||||||
|
return fmt.Errorf("edit message is not *tg.Message")
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := t.handleTelegramReactions(ctx, msg); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// t.main.Bridge.QueueRemoteEvent(t.userLogin, &bridgev2.SimpleRemoteEvent[*tg.Message]{
|
||||||
|
// Type: bridgev2.RemoteEventEdit,
|
||||||
|
// LogContext: func(c zerolog.Context) zerolog.Context {
|
||||||
|
// return c.
|
||||||
|
// Str("action", "edit message").
|
||||||
|
// Int("message_id", message.ID)
|
||||||
|
// },
|
||||||
|
// Sender: sender,
|
||||||
|
// PortalKey: ids.MakePortalID(message.PeerID),
|
||||||
|
// })
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TelegramClient) handleTelegramReactions(ctx context.Context, msg *tg.Message) error {
|
||||||
|
if _, set := msg.GetReactions(); !set {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
fmt.Printf("handle reactions %+v\n", msg.Reactions)
|
||||||
|
var totalCount int
|
||||||
|
for _, r := range msg.Reactions.Results {
|
||||||
|
totalCount += r.Count
|
||||||
|
}
|
||||||
|
|
||||||
|
reactionsList := msg.Reactions.RecentReactions
|
||||||
|
if totalCount > 0 && len(reactionsList) == 0 && !msg.Reactions.CanSeeList {
|
||||||
|
// We don't know who reacted in a channel, so we can't bridge it properly either
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO
|
||||||
|
// if self.peer_type == "channel" and not self.megagroup:
|
||||||
|
// # This should never happen with the previous if
|
||||||
|
// self.log.warning(f"Can see reaction list in channel ({data!s})")
|
||||||
|
// # return
|
||||||
|
|
||||||
|
dbMsg, err := t.main.Bridge.DB.Message.GetFirstPartByID(ctx, ids.MakeUserLoginID(t.loginID), ids.MakeMessageID(msg.ID))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
} else if dbMsg == nil {
|
||||||
|
return fmt.Errorf("no message found with ID %d", msg.ID)
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(reactionsList) < totalCount {
|
||||||
|
if user, ok := msg.PeerID.(*tg.PeerUser); ok {
|
||||||
|
reactionsList = splitDMReactionCounts(msg.Reactions.Results, user.UserID, t.loginID)
|
||||||
|
// } else if t.isBot {
|
||||||
|
// // Can't fetch exact reaction senders as a bot
|
||||||
|
// return
|
||||||
|
|
||||||
|
// TODO should calls to this be limited?
|
||||||
|
} else if peer, err := ids.InputPeerForPortalKey(ids.MakePortalKey(msg.PeerID)); err != nil {
|
||||||
|
return err
|
||||||
|
} else {
|
||||||
|
reactions, err := t.client.API().MessagesGetMessageReactionsList(ctx, &tg.MessagesGetMessageReactionsListRequest{
|
||||||
|
Peer: peer, ID: msg.ID, Limit: 100,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
reactionsList = reactions.Reactions
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, ok := t.reactionMessageLocks[msg.ID]; !ok {
|
||||||
|
t.reactionMessageLocks[msg.ID] = &sync.Mutex{}
|
||||||
|
}
|
||||||
|
t.reactionMessageLocks[msg.ID].Lock()
|
||||||
|
defer t.reactionMessageLocks[msg.ID].Unlock()
|
||||||
|
|
||||||
|
isFull := len(reactionsList) == totalCount
|
||||||
|
reactions := map[networkid.UserID][]tg.MessagePeerReaction{}
|
||||||
|
var customEmojiIDs []int64
|
||||||
|
for _, reaction := range reactionsList {
|
||||||
|
if e, ok := reaction.Reaction.(*tg.ReactionCustomEmoji); ok {
|
||||||
|
customEmojiIDs = append(customEmojiIDs, e.DocumentID)
|
||||||
|
} else if reaction.Reaction.TypeID() != tg.ReactionEmojiTypeID {
|
||||||
|
// We don't know how to process this type of emoji
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if p, ok := reaction.PeerID.(*tg.PeerUser); !ok {
|
||||||
|
return fmt.Errorf("reaction peer ID is not a user")
|
||||||
|
} else {
|
||||||
|
reactions[ids.MakeUserID(p.UserID)] = append(reactions[ids.MakeUserID(p.UserID)], reaction)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return t.handleTelegramParsedReactionsLocked(ctx, dbMsg, reactions, customEmojiIDs, isFull, nil, nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
func splitDMReactionCounts(res []tg.ReactionCount, theirUserID, myUserID int64) (reactions []tg.MessagePeerReaction) {
|
||||||
|
for _, item := range res {
|
||||||
|
if item.Count == 2 || item.ChosenOrder > 0 {
|
||||||
|
reactions = append(reactions, tg.MessagePeerReaction{
|
||||||
|
Reaction: item.Reaction,
|
||||||
|
PeerID: &tg.PeerUser{UserID: myUserID},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
if item.Count == 2 {
|
||||||
|
reactions = append(reactions, tg.MessagePeerReaction{
|
||||||
|
Reaction: item.Reaction,
|
||||||
|
PeerID: &tg.PeerUser{UserID: theirUserID},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TelegramClient) getReactionLimit(ctx context.Context, sender networkid.UserID) (int, error) {
|
||||||
|
// TODO implement this correctly (probably need to put something into metadata)
|
||||||
|
// ghost, err := t.main.Bridge.GetGhostByID(ctx, sender)
|
||||||
|
// if err != nil {
|
||||||
|
// return 0, err
|
||||||
|
// }
|
||||||
|
return 1, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *TelegramClient) handleTelegramParsedReactionsLocked(ctx context.Context, msg *database.Message, reactions map[networkid.UserID][]tg.MessagePeerReaction, customEmojiIDs []int64, isFull bool, onlyUserID *networkid.UserID, timestamp *time.Time) error {
|
||||||
|
// TODO deal with the custom emoji IDs
|
||||||
|
fmt.Printf("custom emoji IDs %v\n", customEmojiIDs)
|
||||||
|
|
||||||
|
existingReactions, err := t.main.Bridge.DB.Reaction.GetAllToMessage(ctx, msg.ID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
var removed []*database.Reaction
|
||||||
|
for _, existing := range existingReactions {
|
||||||
|
if onlyUserID != nil && existing.SenderID != *onlyUserID {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
newReactions := reactions[existing.SenderID]
|
||||||
|
var matched bool
|
||||||
|
reactions[existing.SenderID], matched, err = reactionsFilter(newReactions, existing)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
} else if !matched {
|
||||||
|
if isFull {
|
||||||
|
removed = append(removed, existing)
|
||||||
|
} else if reactionLimit, err := t.getReactionLimit(ctx, existing.SenderID); err != nil {
|
||||||
|
return err
|
||||||
|
} else if len(newReactions) == reactionLimit {
|
||||||
|
removed = append(removed, existing)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for sender, reactions := range reactions {
|
||||||
|
senderID, err := ids.ParseUserID(sender)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, reaction := range reactions {
|
||||||
|
var emojiID networkid.EmojiID
|
||||||
|
var emoji string
|
||||||
|
if rce, ok := reaction.Reaction.(*tg.ReactionCustomEmoji); ok {
|
||||||
|
emojiID = ids.MakeEmojiIDFromDocumentID(rce.DocumentID)
|
||||||
|
emoji = "custom" // TODO
|
||||||
|
} else if r, ok := reaction.Reaction.(*tg.ReactionEmoji); ok {
|
||||||
|
emojiID = ids.MakeEmojiIDFromEmoticon(r.Emoticon)
|
||||||
|
emoji = r.Emoticon
|
||||||
|
}
|
||||||
|
|
||||||
|
evt := &bridgev2.SimpleRemoteEvent[any]{
|
||||||
|
Type: bridgev2.RemoteEventReaction,
|
||||||
|
LogContext: func(c zerolog.Context) zerolog.Context {
|
||||||
|
return c.
|
||||||
|
Any("reaction", reaction.Reaction).
|
||||||
|
Str("sender_id", string(sender)).
|
||||||
|
Str("message_id", string(msg.ID))
|
||||||
|
},
|
||||||
|
Sender: bridgev2.EventSender{
|
||||||
|
IsFromMe: reaction.My,
|
||||||
|
SenderLogin: ids.MakeUserLoginID(senderID),
|
||||||
|
Sender: sender,
|
||||||
|
},
|
||||||
|
PortalKey: msg.Room,
|
||||||
|
TargetMessage: msg.ID,
|
||||||
|
EmojiID: emojiID,
|
||||||
|
Emoji: emoji,
|
||||||
|
}
|
||||||
|
if timestamp != nil {
|
||||||
|
evt.Timestamp = *timestamp
|
||||||
|
}
|
||||||
|
t.main.Bridge.QueueRemoteEvent(t.userLogin, evt)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, r := range removed {
|
||||||
|
senderID, err := ids.ParseUserID(r.SenderID)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
evt := &bridgev2.SimpleRemoteEvent[any]{
|
||||||
|
Type: bridgev2.RemoteEventReactionRemove,
|
||||||
|
LogContext: func(c zerolog.Context) zerolog.Context {
|
||||||
|
return c.
|
||||||
|
Str("emoji_id", string(r.EmojiID)).
|
||||||
|
Str("sender_id", string(r.SenderID)).
|
||||||
|
Str("message_id", string(msg.ID))
|
||||||
|
},
|
||||||
|
Sender: bridgev2.EventSender{
|
||||||
|
IsFromMe: t.loginID == senderID,
|
||||||
|
SenderLogin: ids.MakeUserLoginID(senderID),
|
||||||
|
Sender: r.SenderID,
|
||||||
|
},
|
||||||
|
PortalKey: msg.Room,
|
||||||
|
TargetMessage: r.MessageID,
|
||||||
|
EmojiID: r.EmojiID,
|
||||||
|
}
|
||||||
|
if timestamp != nil {
|
||||||
|
evt.Timestamp = *timestamp
|
||||||
|
}
|
||||||
|
t.main.Bridge.QueueRemoteEvent(t.userLogin, evt)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("%v %v %v\n", isFull, reactions, customEmojiIDs)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func reactionsFilter(reactions []tg.MessagePeerReaction, existing *database.Reaction) (newReactions []tg.MessagePeerReaction, matched bool, err error) {
|
||||||
|
if len(reactions) == 0 {
|
||||||
|
return nil, false, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
documentID, emoticon, err := ids.ParseEmojiID(existing.EmojiID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
newReactions = slices.DeleteFunc(reactions, func(r tg.MessagePeerReaction) bool {
|
||||||
|
if rce, ok := r.Reaction.(*tg.ReactionCustomEmoji); ok {
|
||||||
|
return documentID == rce.DocumentID
|
||||||
|
} else if r, ok := r.Reaction.(*tg.ReactionEmoji); ok {
|
||||||
|
return emoticon == r.Emoticon
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
})
|
||||||
|
return newReactions, len(newReactions) < len(reactions), nil
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user