hoernschen/dendrite
1
0
Fork 0
mirror of https://github.com/hoernschen/dendrite.git synced 2025-08-02 14:12:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add query functions in the federation API

Browse source
This commit is contained in:
Neil Alexander 2021-12-08 14:47:47 +00:00
parent 61406a6747
commit 739acc1291
No known key found for this signature in database
GPG key ID: A02A2019A2BB0944
4 changed files with 256 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

123
federationapi/internal/query.go
View file

@ -25,6 +25,129 @@ func (f *FederationInternalAPI) QueryJoinedHostServerNamesInRoom(
return
}
func (f *FederationInternalAPI) QueryEventAuthFromFederation(
ctx context.Context,
request *api.QueryEventAuthFromFederationRequest,
response *api.QueryEventAuthFromFederationResponse,
) error {
joinedHosts, err := f.db.GetJoinedHostsForRooms(ctx, []string{request.RoomID})
if err != nil {
return fmt.Errorf("f.db.GetJoinedHostsForRooms: %w", err)
}
tryHost := func(serverName gomatrixserverlib.ServerName) error {
ctx, cancel := context.WithTimeout(ctx, time.Second*30)
defer cancel()
ires, err := f.doRequest(serverName, func() (interface{}, error) {
return f.federation.GetEventAuth(
ctx,
serverName,
request.RoomID,
request.EventID,
)
})
if err != nil {
return fmt.Errorf("f.doRequest: %w", err)
}
tx := ires.(gomatrixserverlib.RespEventAuth)
response.Events = tx.AuthEvents
return nil
}
var lasterr error
for _, host := range joinedHosts {
if lasterr = tryHost(host); lasterr != nil {
continue
}
break
}
return lasterr
}
func (f *FederationInternalAPI) QueryStateIDsFromFederation(
ctx context.Context,
request *api.QueryStateIDsFromFederationRequest,
response *api.QueryStateIDsFromFederationResponse,
) error {
joinedHosts, err := f.db.GetJoinedHostsForRooms(ctx, []string{request.RoomID})
if err != nil {
return fmt.Errorf("f.db.GetJoinedHostsForRooms: %w", err)
}
tryHost := func(serverName gomatrixserverlib.ServerName) error {
ctx, cancel := context.WithTimeout(ctx, time.Second*30)
defer cancel()
ires, err := f.doRequest(serverName, func() (interface{}, error) {
return f.federation.LookupStateIDs(
ctx,
serverName,
request.RoomID,
request.EventID,
)
})
if err != nil {
return fmt.Errorf("f.doRequest: %w", err)
}
tx := ires.(gomatrixserverlib.RespStateIDs)
response.AuthEventIDs = tx.AuthEventIDs
response.StateEventIDs = tx.StateEventIDs
return nil
}
var lasterr error
for _, host := range joinedHosts {
if lasterr = tryHost(host); lasterr != nil {
continue
}
break
}
return lasterr
}
func (f *FederationInternalAPI) QueryStateFromFederation(
ctx context.Context,
request *api.QueryStateFromFederationRequest,
response *api.QueryStateFromFederationResponse,
) error {
joinedHosts, err := f.db.GetJoinedHostsForRooms(ctx, []string{request.RoomID})
if err != nil {
return fmt.Errorf("f.db.GetJoinedHostsForRooms: %w", err)
}
tryHost := func(serverName gomatrixserverlib.ServerName) error {
ctx, cancel := context.WithTimeout(ctx, time.Second*30)
defer cancel()
ires, err := f.doRequest(serverName, func() (interface{}, error) {
return f.federation.LookupState(
ctx,
serverName,
request.RoomID,
request.EventID,
request.RoomVersion,
)
})
if err != nil {
return fmt.Errorf("f.doRequest: %w", err)
}
tx := ires.(gomatrixserverlib.RespState)
response.AuthEvents = tx.AuthEvents
response.StateEvents = tx.StateEvents
return nil
}
var lasterr error
for _, host := range joinedHosts {
if lasterr = tryHost(host); lasterr != nil {
continue
}
break
}
return lasterr
}
func (a *FederationInternalAPI) fetchServerKeysDirectly(ctx context.Context, serverName gomatrixserverlib.ServerName) (*gomatrixserverlib.ServerKeys, error) {
ctx, cancel := context.WithTimeout(ctx, time.Second*30)
defer cancel()