Adam Van Ymeren
7a04f298d2
- update to latest telegram layer - remove some references to fields in tg.Entities that don't exist in the schema - originally added here: https://github.com/beeper/td/commit/820929062a2ba0104397bc01235ab58a9cff780e - referenced here - https://github.com/mautrix/telegramgo/commit/124f0967ed195b5a380c9bd02e170ada9710dde3 - https://github.com/mautrix/telegramgo/commit/4205047aab2e0639217148b5d125bfaab668bd8e
104 lines
2.6 KiB
Go
104 lines
2.6 KiB
Go
package peers
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.uber.org/multierr"
|
|
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/message/entity"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/updates"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
|
)
|
|
|
|
var _ updates.AccessHasher = (*Manager)(nil)
|
|
|
|
func (m *Manager) SetChannelAccessHash(ctx context.Context, forUserID, channelID, accessHash int64) error {
|
|
myID, ok := m.myID()
|
|
if !ok || myID != forUserID {
|
|
return nil
|
|
}
|
|
return m.storage.Save(ctx, Key{
|
|
Prefix: channelPrefix,
|
|
ID: channelID,
|
|
}, Value{
|
|
AccessHash: accessHash,
|
|
})
|
|
}
|
|
|
|
func (m *Manager) GetChannelAccessHash(ctx context.Context, forUserID, channelID int64) (accessHash int64, found bool, err error) {
|
|
myID, ok := m.myID()
|
|
if !ok || myID != forUserID {
|
|
return 0, false, nil
|
|
}
|
|
|
|
v, found, err := m.storage.Find(ctx, Key{
|
|
Prefix: channelPrefix,
|
|
ID: channelID,
|
|
})
|
|
return v.AccessHash, found, err
|
|
}
|
|
|
|
func (m *Manager) SetUserAccessHash(ctx context.Context, forUserID, userID, accessHash int64) error {
|
|
myID, ok := m.myID()
|
|
if !ok || myID != forUserID {
|
|
return nil
|
|
}
|
|
return m.storage.Save(ctx, Key{
|
|
Prefix: usersPrefix,
|
|
ID: userID,
|
|
}, Value{
|
|
AccessHash: accessHash,
|
|
})
|
|
}
|
|
|
|
func (m *Manager) GetUserAccessHash(ctx context.Context, forUserID, userID int64) (accessHash int64, found bool, err error) {
|
|
myID, ok := m.myID()
|
|
if !ok || myID != forUserID {
|
|
return 0, false, nil
|
|
}
|
|
|
|
v, found, err := m.storage.Find(ctx, Key{
|
|
Prefix: usersPrefix,
|
|
ID: userID,
|
|
})
|
|
return v.AccessHash, found, err
|
|
}
|
|
|
|
// UpdateHook returns update middleware hook for collecting entities.
|
|
func (m *Manager) UpdateHook(next telegram.UpdateHandler) telegram.UpdateHandler {
|
|
f := func(ctx context.Context, u tg.UpdatesClass) error {
|
|
var (
|
|
users []tg.UserClass
|
|
chats []tg.ChatClass
|
|
updates []tg.UpdateClass
|
|
)
|
|
switch u := u.(type) {
|
|
case *tg.UpdatesCombined:
|
|
users = u.GetUsers()
|
|
chats = u.GetChats()
|
|
updates = u.GetUpdates()
|
|
case *tg.Updates:
|
|
users = u.GetUsers()
|
|
chats = u.GetChats()
|
|
updates = u.GetUpdates()
|
|
}
|
|
m.applyUpdates(updates)
|
|
applyErr := m.applyEntities(ctx, users, chats)
|
|
handleErr := next.Handle(ctx, u)
|
|
return multierr.Append(handleErr, applyErr)
|
|
}
|
|
return telegram.UpdateHandlerFunc(f)
|
|
}
|
|
|
|
// UserResolveHook creates entity.UserResolver attached to this Manager.
|
|
func (m *Manager) UserResolveHook(ctx context.Context) entity.UserResolver {
|
|
return func(id int64) (tg.InputUserClass, error) {
|
|
r, err := m.getUser(ctx, &tg.InputUser{UserID: id})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return r.AsInput(), nil
|
|
}
|
|
}
|