Matrix-Mirrors/forked-synapse
1
0
Fork 0
mirror of https://mau.dev/maunium/synapse.git synced 2024-10-01 01:36:05 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add very basic filter API to /publicRooms

Browse Source
This commit is contained in:
Erik Johnston 2016-09-15 17:50:16 +01:00
parent c33b25fd8d
commit 098db4aa52
1 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
synapse/handlers/room_list.py
View File

@ -45,7 +45,7 @@ class RoomListHandler(BaseHandler):
if not result: if not result:
result = self.response_cache.set( result = self.response_cache.set(
(limit, since_token), (limit, since_token),
self._get_public_room_list(limit, since_token) self._get_public_room_list(limit, since_token, search_filter)
) )
return result return result
@ -119,13 +119,13 @@ class RoomListHandler(BaseHandler):
rooms_to_scan = sorted_rooms rooms_to_scan = sorted_rooms
if limit and not search_filter: if limit and not search_filter:
rooms_to_scan = sorted_rooms[:limit] rooms_to_scan = sorted_rooms[:limit + 1]
chunk = [] chunk = []
@defer.inlineCallbacks @defer.inlineCallbacks
def handle_room(room_id): def handle_room(room_id):
if limit and len(chunk) > limit: if limit and len(chunk) > limit + 1:
# We've already got enough, so lets just drop it. # We've already got enough, so lets just drop it.
return return
@ -208,6 +208,16 @@ class RoomListHandler(BaseHandler):
if avatar_url: if avatar_url:
result["avatar_url"] = avatar_url result["avatar_url"] = avatar_url
logger.info("search_filter: %r", search_filter)
if search_filter and search_filter.get("generic_search_term", None):
generic_search_term = search_filter["generic_search_term"]
if generic_search_term in result.get("name", ""):
chunk.append(result)
elif generic_search_term in result.get("topic", ""):
chunk.append(result)
elif generic_search_term in result.get("canonical_alias", ""):
chunk.append(result)
else:
chunk.append(result) chunk.append(result)
yield concurrently_execute(handle_room, rooms_to_scan, 10) yield concurrently_execute(handle_room, rooms_to_scan, 10)
@ -215,7 +225,10 @@ class RoomListHandler(BaseHandler):
chunk.sort(key=lambda e: (-e["num_joined_members"], e["room_id"])) chunk.sort(key=lambda e: (-e["num_joined_members"], e["room_id"]))
new_limit = None new_limit = None
if chunk: if chunk and (not limit or len(chunk) > limit):
if limit:
chunk = chunk[:limit]
addition = 1 addition = 1
if since_token: if since_token:
addition += since_token.current_limit addition += since_token.current_limit