From 37fb9e7dfda2d5d0a8d4e07ca6527369f8fe05ed Mon Sep 17 00:00:00 2001
From: SBiOSoftWhare <steveb@matrix.org>
Date: Fri, 20 Dec 2019 10:43:07 +0100
Subject: [PATCH] RoomVC: Handle DM key verification cells.

---
 Riot/Modules/Room/RoomViewController.m | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/Riot/Modules/Room/RoomViewController.m b/Riot/Modules/Room/RoomViewController.m
index 2c4fac795..03ccc1656 100644
--- a/Riot/Modules/Room/RoomViewController.m
+++ b/Riot/Modules/Room/RoomViewController.m
@@ -353,6 +353,10 @@
     [self.bubblesTableView registerClass:RoomSelectedStickerBubbleCell.class forCellReuseIdentifier:RoomSelectedStickerBubbleCell.defaultReuseIdentifier];
     [self.bubblesTableView registerClass:RoomPredecessorBubbleCell.class forCellReuseIdentifier:RoomPredecessorBubbleCell.defaultReuseIdentifier];
     
+    [self.bubblesTableView registerClass:KeyVerificationIncomingRequestApprovalBubbleCell.class forCellReuseIdentifier:KeyVerificationIncomingRequestApprovalBubbleCell.defaultReuseIdentifier];
+    [self.bubblesTableView registerClass:KeyVerificationRequestStatusBubbleCell.class forCellReuseIdentifier:KeyVerificationRequestStatusBubbleCell.defaultReuseIdentifier];
+    [self.bubblesTableView registerClass:KeyVerificationConclusionBubbleCell.class forCellReuseIdentifier:KeyVerificationConclusionBubbleCell.defaultReuseIdentifier];
+    
     // Prepare expanded header
     expandedHeader = [ExpandedRoomTitleView roomTitleView];
     expandedHeader.delegate = self;
@@ -2039,6 +2043,18 @@
         {
             cellViewClass = RoomPredecessorBubbleCell.class;
         }
+        else if (bubbleData.tag == RoomBubbleCellDataTagDeviceKeyVerificationRequestIncomingApproval)
+        {
+            cellViewClass = KeyVerificationIncomingRequestApprovalBubbleCell.class;
+        }
+        else if (bubbleData.tag == RoomBubbleCellDataTagDeviceKeyVerificationRequest)
+        {
+            cellViewClass = KeyVerificationRequestStatusBubbleCell.class;
+        }
+        else if (bubbleData.tag == RoomBubbleCellDataTagDeviceKeyVerificationConclusion)
+        {
+            cellViewClass = KeyVerificationConclusionBubbleCell.class;
+        }
         else if (bubbleData.tag == RoomBubbleCellDataTagMembership)
         {
             if (bubbleData.collapsed)