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,64 @@
|
||||
package peer
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/go-faster/errors"
|
||||
|
||||
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
|
||||
)
|
||||
|
||||
// Resolver is an abstraction to resolve domains and Telegram deeplinks.
|
||||
type Resolver interface {
|
||||
ResolveDomain(ctx context.Context, domain string) (tg.InputPeerClass, error)
|
||||
ResolvePhone(ctx context.Context, phone string) (tg.InputPeerClass, error)
|
||||
}
|
||||
|
||||
// DefaultResolver creates and returns default resolver.
|
||||
func DefaultResolver(raw *tg.Client) Resolver {
|
||||
return NewLRUResolver(SingleflightResolver(Plain(raw)), 10)
|
||||
}
|
||||
|
||||
// Plain creates plain resolver.
|
||||
func Plain(raw *tg.Client) Resolver {
|
||||
return plainResolver{raw: raw}
|
||||
}
|
||||
|
||||
type plainResolver struct {
|
||||
raw *tg.Client
|
||||
}
|
||||
|
||||
func (p plainResolver) ResolveDomain(ctx context.Context, domain string) (tg.InputPeerClass, error) {
|
||||
peer, err := p.raw.ContactsResolveUsername(ctx, &tg.ContactsResolveUsernameRequest{
|
||||
Username: domain,
|
||||
})
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "resolve")
|
||||
}
|
||||
|
||||
return EntitiesFromResult(peer).ExtractPeer(peer.Peer)
|
||||
}
|
||||
|
||||
func (p plainResolver) ResolvePhone(ctx context.Context, phone string) (tg.InputPeerClass, error) {
|
||||
r, err := p.raw.ContactsGetContacts(ctx, 0)
|
||||
if err != nil {
|
||||
return nil, errors.Wrap(err, "get contacts")
|
||||
}
|
||||
|
||||
switch c := r.(type) {
|
||||
case *tg.ContactsContacts:
|
||||
for _, u := range c.Users {
|
||||
user, ok := u.AsNotEmpty()
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
if user.Phone == phone {
|
||||
return user.AsInputPeer(), nil
|
||||
}
|
||||
}
|
||||
|
||||
return nil, errors.Errorf("can't resolve phone %q", phone)
|
||||
default:
|
||||
return nil, errors.Errorf("unexpected type %T", r)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user