Migrating dm and room tags work for migrator

This commit is contained in:
Andrew Morgan 2019-01-18 15:12:47 +00:00
parent 887ca93a1b
commit ea8903fcc9

View file

@ -260,12 +260,16 @@ class RoomCreationHandler(BaseHandler):
} }
} }
# Copy over whether this room is considered a direct message by this # Copy over room account data for this user
# user or not
user_account_data = yield self.store.get_account_data_for_user( user_account_data = yield self.store.get_account_data_for_user(
user_id, user_id,
) )
room_tags = yield self.store.get_tags_for_room(
user_id, old_room_id,
)
# Copy direct message state if applicable
if user_account_data and "m.direct" in user_account_data[0]: if user_account_data and "m.direct" in user_account_data[0]:
direct_rooms = user_account_data[0]["m.direct"] direct_rooms = user_account_data[0]["m.direct"]
# Check if this room was a DM # Check if this room was a DM
@ -278,6 +282,13 @@ class RoomCreationHandler(BaseHandler):
user_id, "m.direct", direct_rooms, user_id, "m.direct", direct_rooms,
) )
# Copy room tags if applicable
if room_tags:
# Copy each room tag to the new room
for tag in room_tags.keys():
tag_content = room_tags[tag]
yield self.store.add_tag_to_room(user_id, new_room_id, tag, tag_content)
initial_state = dict() initial_state = dict()
# Replicate relevant room events # Replicate relevant room events