contact list: implement fetching

Signed-off-by: Sumner Evans <sumner.evans@automattic.com>
This commit is contained in:
Sumner Evans
2024-08-22 12:46:42 -06:00
parent dcf43ca9d9
commit 91ce540a66
2 changed files with 45 additions and 1 deletions
+4 -1
View File
@@ -47,6 +47,9 @@ type TelegramClient struct {
telegramFmtParams *telegramfmt.FormatParams
matrixParser *matrixfmt.HTMLParser
cachedContacts *tg.ContactsContacts
cachedContactsHash int64
}
var (
@@ -60,8 +63,8 @@ var (
_ bridgev2.BackfillingNetworkAPIWithLimits = (*TelegramClient)(nil)
_ bridgev2.IdentifierResolvingNetworkAPI = (*TelegramClient)(nil)
_ bridgev2.UserSearchingNetworkAPI = (*TelegramClient)(nil)
_ bridgev2.ContactListingNetworkAPI = (*TelegramClient)(nil)
// _ bridgev2.GroupCreatingNetworkAPI = (*TelegramClient)(nil)
// _ bridgev2.ContactListingNetworkAPI = (*TelegramClient)(nil)
)
type UpdateDispatcher struct {
+41
View File
@@ -7,6 +7,7 @@ import (
"strconv"
"strings"
"github.com/gotd/td/telegram/query/hasher"
"github.com/gotd/td/tg"
"maunium.net/go/mautrix/bridgev2"
@@ -130,3 +131,43 @@ func (t *TelegramClient) SearchUsers(ctx context.Context, query string) (resp []
}
return resp, nil
}
func (t *TelegramClient) GetContactList(ctx context.Context) (resp []*bridgev2.ResolveIdentifierResponse, err error) {
contacts, err := APICallWithUpdates(ctx, t, func() (*tg.ContactsContacts, error) {
c, err := t.client.API().ContactsGetContacts(ctx, t.cachedContactsHash)
if err != nil {
return nil, err
}
if c.TypeID() == tg.ContactsContactsTypeID {
t.cachedContacts = c.(*tg.ContactsContacts)
var h hasher.Hasher
for _, contact := range t.cachedContacts.Contacts {
h.Update(uint32(contact.UserID))
}
t.cachedContactsHash = h.Sum()
} else if c.TypeID() != tg.ContactsContactsNotModifiedTypeID {
return nil, fmt.Errorf("unexpected contacts type: %T", c)
}
return t.cachedContacts, nil
})
if err != nil {
return nil, err
}
users := map[int64]tg.UserClass{}
for _, user := range contacts.GetUsers() {
users[user.GetID()] = user
}
for _, contact := range contacts.Contacts {
if user, ok := users[contact.UserID]; ok {
if r, err := t.getResolveIdentifierResponseForUserID(ctx, user); err != nil {
return nil, err
} else {
resp = append(resp, r)
}
} else {
return nil, fmt.Errorf("contact user not found in contact list response")
}
}
return resp, nil
}