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
75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
package peers
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/go-faster/errors"
|
|
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/constant"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/tgerr"
|
|
)
|
|
|
|
// ResolveTDLibID creates Peer using given constant.TDLibPeerID.
|
|
func (m *Manager) ResolveTDLibID(ctx context.Context, peerID constant.TDLibPeerID) (p Peer, err error) {
|
|
switch {
|
|
case peerID.IsUser():
|
|
p, err = m.ResolveUserID(ctx, peerID.ToPlain())
|
|
case peerID.IsChat():
|
|
p, err = m.ResolveChatID(ctx, peerID.ToPlain())
|
|
case peerID.IsChannel():
|
|
p, err = m.ResolveChannelID(ctx, peerID.ToPlain())
|
|
default:
|
|
return nil, errors.Errorf("invalid ID %d", peerID)
|
|
}
|
|
return p, err
|
|
}
|
|
|
|
// ResolveUserID creates User using given id.
|
|
func (m *Manager) ResolveUserID(ctx context.Context, id int64) (User, error) {
|
|
v, ok, err := m.storage.Find(ctx, Key{
|
|
Prefix: usersPrefix,
|
|
ID: id,
|
|
})
|
|
if err != nil {
|
|
return User{}, err
|
|
}
|
|
u, err := m.GetUser(ctx, &tg.InputUser{
|
|
UserID: id,
|
|
AccessHash: v.AccessHash,
|
|
})
|
|
if !ok && tgerr.Is(err, tg.ErrUserIDInvalid) {
|
|
return User{}, &PeerNotFoundError{
|
|
Peer: &tg.PeerUser{UserID: id},
|
|
}
|
|
}
|
|
return u, err
|
|
}
|
|
|
|
// ResolveChatID creates Chat using given id.
|
|
func (m *Manager) ResolveChatID(ctx context.Context, id int64) (Chat, error) {
|
|
c, err := m.GetChat(ctx, id)
|
|
return c, err
|
|
}
|
|
|
|
// ResolveChannelID creates Channel using given id.
|
|
func (m *Manager) ResolveChannelID(ctx context.Context, id int64) (Channel, error) {
|
|
v, ok, err := m.storage.Find(ctx, Key{
|
|
Prefix: channelPrefix,
|
|
ID: id,
|
|
})
|
|
if err != nil {
|
|
return Channel{}, err
|
|
}
|
|
c, err := m.GetChannel(ctx, &tg.InputChannel{
|
|
ChannelID: id,
|
|
AccessHash: v.AccessHash,
|
|
})
|
|
if !ok && tgerr.Is(err, tg.ErrChannelInvalid) {
|
|
return Channel{}, &PeerNotFoundError{
|
|
Peer: &tg.PeerChannel{ChannelID: id},
|
|
}
|
|
}
|
|
return c, err
|
|
}
|