summary refs log tree commit diff
path: root/synapse/storage/state.py
diff options
context:
space:
mode:
authorAndrew Morgan <1342360+anoadragon453@users.noreply.github.com>2019-01-25 10:49:58 +0000
committerGitHub <noreply@github.com>2019-01-25 10:49:58 +0000
commitb1b6dba2d2c0690157cdd620063d4e16a583264a (patch)
tree2a5f4a1d51dced1405e08b17f0dcd5fb669b0a65 /synapse/storage/state.py
parentMerge pull request #4447 from matrix-org/erikj/msc_1813 (diff)
parentMerge branch 'anoa/full_search_upgraded_rooms' of github.com:matrix-org/synap... (diff)
downloadsynapse-b1b6dba2d2c0690157cdd620063d4e16a583264a.tar.xz
Merge pull request #4415 from matrix-org/anoa/full_search_upgraded_rooms
Ability to search entire room history after upgrading room
Diffstat (limited to 'synapse/storage/state.py')
-rw-r--r--synapse/storage/state.py24
1 files changed, 24 insertions, 0 deletions
diff --git a/synapse/storage/state.py b/synapse/storage/state.py
index a134e9b3e8..c3ab7db7ae 100644
--- a/synapse/storage/state.py
+++ b/synapse/storage/state.py
@@ -437,6 +437,30 @@ class StateGroupWorkerStore(EventsWorkerStore, SQLBaseStore):
         create_event = yield self.get_event(create_id)
         defer.returnValue(create_event.content.get("room_version", "1"))
 
+    @defer.inlineCallbacks
+    def get_room_predecessor(self, room_id):
+        """Get the predecessor room of an upgraded room if one exists.
+        Otherwise return None.
+
+        Args:
+            room_id (str)
+
+        Returns:
+            Deferred[unicode|None]: predecessor room id
+        """
+        state_ids = yield self.get_current_state_ids(room_id)
+        create_id = state_ids.get((EventTypes.Create, ""))
+
+        # If we can't find the create event, assume we've hit a dead end
+        if not create_id:
+            defer.returnValue(None)
+
+        # Retrieve the room's create event
+        create_event = yield self.get_event(create_id)
+
+        # Return predecessor if present
+        defer.returnValue(create_event.content.get("predecessor", None))
+
     @cached(max_entries=100000, iterable=True)
     def get_current_state_ids(self, room_id):
         """Get the current state event ids for a room based on the