move gotd fork into repo. (#111)
- 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
This commit is contained in:
@@ -0,0 +1,74 @@
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user