mirrors/dendrite
1
0
Fork 0
mirror of https://github.com/matrix-org/dendrite.git synced 2025-12-26 08:13:09 -06:00
Code Issues Packages Projects Releases Wiki Activity

WIP: Make room members filterable

Browse source
This commit is contained in:
Alex Chan 2021-06-22 14:26:31 -04:00
parent 3a77bca28b
commit e50dbe61be
8 changed files with 76 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
clientapi/routing/memberships.go
View file

@ -51,16 +51,47 @@ type databaseJoinedMember struct {
AvatarURL string `json:"avatar_url"`
}
// // MembershipFilter is an enum representing kinds of membership to a room
// type MembershipFilter string
// const (
// MembershipDefault MembershipFilter = "none"
// MembershipJoin MembershipFilter = "join"
// MembershipInvite MembershipFilter = "invite"
// MembershipLeave MembershipFilter = "leave"
// MembershipBan MembershipFilter = "ban"
// )
// GetMemberships implements GET /rooms/{roomId}/members
func GetMemberships(
req *http.Request, device *userapi.Device, roomID string, joinedOnly bool,
_ *config.ClientAPI,
rsAPI api.RoomserverInternalAPI,
) util.JSONResponse {
// how would i unpack this nicely into
// type queryParams struct {
// membership string
// notMembership string
// }
membership := req.URL.Query().Get("membership")
notMembership := req.URL.Query().Get("not_membership")
membershipStatusFilter := []string{"join", "invite", "leave", "ban"}
if len(notMembership) > 0 {
for i, v := range membershipStatusFilter {
if v == notMembership {
membershipStatusFilter = append(membershipStatusFilter[:i], membershipStatusFilter[i+1:]...)
}
}
} else if len(membership) > 0 {
membershipStatusFilter = []string{membership}
} else {
membershipStatusFilter = []string{}
}
queryReq := api.QueryMembershipsForRoomRequest{
JoinedOnly: joinedOnly,
RoomID: roomID,
Sender: device.UserID,
JoinedOnly: joinedOnly,
MembershipStatusFilter: membershipStatusFilter,
RoomID: roomID,
Sender: device.UserID,
}
var queryRes api.QueryMembershipsForRoomResponse
if err := rsAPI.QueryMembershipsForRoom(req.Context(), &queryReq, &queryRes); err != nil {

3
roomserver/api/query.go
View file

@ -148,7 +148,8 @@ type QueryMembershipForUserResponse struct {
// QueryMembershipsForRoomRequest is a request to QueryMembershipsForRoom
type QueryMembershipsForRoomRequest struct {
// If true, only returns the membership events of "join" membership
JoinedOnly bool `json:"joined_only"`
JoinedOnly bool `json:"joined_only"`
MembershipStatusFilter []string `json:"membership_status_filter"`
// ID of the room to fetch memberships from
RoomID string `json:"room_id"`
// Optional - ID of the user sending the request, for checking if the

2
roomserver/internal/helpers/helpers.go
View file

@ -59,7 +59,7 @@ func IsServerCurrentlyInRoom(ctx context.Context, db storage.Database, serverNam
return false, fmt.Errorf("unknown room %s", roomID)
}
eventNIDs, err := db.GetMembershipEventNIDsForRoom(ctx, info.RoomNID, true, false)
eventNIDs, err := db.GetMembershipEventNIDsForRoom(ctx, info.RoomNID, []string{"join"}, false)
if err != nil {
return false, err
}

2
roomserver/internal/input/input_events.go
View file

@ -256,7 +256,7 @@ func (r *Inputer) calculateAndSetState(
stateAtEvent.Overwrite = true
var joinEventNIDs []types.EventNID
// Request join memberships only for local users only.
if joinEventNIDs, err = r.DB.GetMembershipEventNIDsForRoom(ctx, roomInfo.RoomNID, true, true); err == nil {
if joinEventNIDs, err = r.DB.GetMembershipEventNIDsForRoom(ctx, roomInfo.RoomNID, []string{"join"}, true); err == nil {
// If we have no local users that are joined to the room then any state about
// the room that we have is quite possibly out of date. Therefore in that case
// we should overwrite it rather than merge it.

2
roomserver/internal/perform/perform_backfill.go
View file

@ -537,7 +537,7 @@ func joinEventsFromHistoryVisibility(
if err != nil {
return nil, err
}
joinEventNIDs, err := db.GetMembershipEventNIDsForRoom(ctx, info.RoomNID, true, false)
joinEventNIDs, err := db.GetMembershipEventNIDsForRoom(ctx, info.RoomNID, []string{"join"}, false)
if err != nil {
return nil, err
}

6
roomserver/internal/query/query.go
View file

@ -251,7 +251,7 @@ func (r *Queryer) QueryMembershipsForRoom(
if request.Sender == "" {
var events []types.Event
var eventNIDs []types.EventNID
eventNIDs, err = r.DB.GetMembershipEventNIDsForRoom(ctx, info.RoomNID, request.JoinedOnly, false)
eventNIDs, err = r.DB.GetMembershipEventNIDsForRoom(ctx, info.RoomNID, request.MembershipStatusFilter, false)
if err != nil {
return fmt.Errorf("r.DB.GetMembershipEventNIDsForRoom: %w", err)
}
@ -286,7 +286,7 @@ func (r *Queryer) QueryMembershipsForRoom(
var stateEntries []types.StateEntry
if stillInRoom {
var eventNIDs []types.EventNID
eventNIDs, err = r.DB.GetMembershipEventNIDsForRoom(ctx, info.RoomNID, request.JoinedOnly, false)
eventNIDs, err = r.DB.GetMembershipEventNIDsForRoom(ctx, info.RoomNID, request.MembershipStatusFilter, false)
if err != nil {
return err
}
@ -328,7 +328,7 @@ func (r *Queryer) QueryServerJoinedToRoom(
}
response.RoomExists = true
eventNIDs, err := r.DB.GetMembershipEventNIDsForRoom(ctx, info.RoomNID, true, false)
eventNIDs, err := r.DB.GetMembershipEventNIDsForRoom(ctx, info.RoomNID, []string{"join"}, false)
if err != nil {
return fmt.Errorf("r.DB.GetMembershipEventNIDsForRoom: %w", err)
}

2
roomserver/storage/interface.go
View file

@ -131,7 +131,7 @@ type Database interface {
// been members of a given room. Only lookup events of "join" membership if
// joinOnly is set to true.
// Returns an error if there was a problem talking to the database.
GetMembershipEventNIDsForRoom(ctx context.Context, roomNID types.RoomNID, joinOnly bool, localOnly bool) ([]types.EventNID, error)
GetMembershipEventNIDsForRoom(ctx context.Context, roomNID types.RoomNID, membershipStatusFilter []string, localOnly bool) ([]types.EventNID, error)
// EventsFromIDs looks up the Events for a list of event IDs. Does not error if event was
// not found.
// Returns an error if the retrieval went wrong.

38
roomserver/storage/shared/storage.go
View file

@ -343,12 +343,40 @@ func (d *Database) GetMembership(ctx context.Context, roomNID types.RoomNID, req
}
func (d *Database) GetMembershipEventNIDsForRoom(
ctx context.Context, roomNID types.RoomNID, joinOnly bool, localOnly bool,
ctx context.Context, roomNID types.RoomNID, membershipStatusFilter []string, localOnly bool,
) ([]types.EventNID, error) {
if joinOnly {
return d.MembershipTable.SelectMembershipsFromRoomAndMembership(
ctx, roomNID, tables.MembershipStateJoin, localOnly,
)
var eventNIDs []types.EventNID
var filteredEventNIDs []types.EventNID
var err error
if len(membershipStatusFilter) > 0 {
for _, filter := range membershipStatusFilter {
if filter == "join" {
filteredEventNIDs, err = d.MembershipTable.SelectMembershipsFromRoomAndMembership(
ctx, roomNID, tables.MembershipStateJoin, localOnly,
)
if err != nil {
return eventNIDs, err
}
eventNIDs = append(eventNIDs, filteredEventNIDs...)
} else if filter == "invite" {
filteredEventNIDs, err = d.MembershipTable.SelectMembershipsFromRoomAndMembership(
ctx, roomNID, tables.MembershipStateInvite, localOnly,
)
if err != nil {
return eventNIDs, err
}
eventNIDs = append(eventNIDs, filteredEventNIDs...)
} else {
filteredEventNIDs, err = d.MembershipTable.SelectMembershipsFromRoomAndMembership(
ctx, roomNID, tables.MembershipStateLeaveOrBan, localOnly,
)
if err != nil {
return eventNIDs, err
}
eventNIDs = append(eventNIDs, filteredEventNIDs...)
}
}
return eventNIDs, err
}
return d.MembershipTable.SelectMembershipsFromRoom(ctx, roomNID, localOnly)