From 28bba4952b5d10fec01952b9c150de18f8ecfcb4 Mon Sep 17 00:00:00 2001
From: Bruno Windels <brunow@matrix.org>
Date: Tue, 2 Apr 2019 15:12:51 +0200
Subject: [PATCH] fix detecting e2e message in timeline

---
 src/usecases/timeline.js | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/usecases/timeline.js b/src/usecases/timeline.js
index dce0203660..610d1f4e9b 100644
--- a/src/usecases/timeline.js
+++ b/src/usecases/timeline.js
@@ -117,11 +117,13 @@ function getLastEventTile(session) {
 }
 
 function getAllEventTiles(session) {
-    return session.queryAll(".mx_RoomView_MessageList > *");
+    return session.queryAll(".mx_RoomView_MessageList .mx_EventTile");
 }
 
 async function getMessageFromEventTile(eventTile) {
     const senderElement = await eventTile.$(".mx_SenderProfile_name");
+    const className = await (await eventTile.getProperty("className")).jsonValue();
+    const classNames = className.split(" ");
     const bodyElement = await eventTile.$(".mx_EventTile_body");
     let sender = null;
     if (senderElement) {
@@ -131,11 +133,10 @@ async function getMessageFromEventTile(eventTile) {
         return null;
     }
     const body = await(await bodyElement.getProperty("innerText")).jsonValue();
-    const e2eIcon = await eventTile.$(".mx_EventTile_e2eIcon");
 
     return {
         sender,
         body,
-        encrypted: !!e2eIcon
+        encrypted: classNames.includes("mx_EventTile_verified")
     };
 }