Add pollHistory timeline property

element-hq · Nov 28, 2023 · c98b097 · c98b097
1 parent 51ddc74
commit c98b097
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/ElementX/Sources/Mocks/Generated/GeneratedMocks.swift b/ElementX/Sources/Mocks/Generated/GeneratedMocks.swift
@@ -1939,6 +1939,11 @@ class RoomProxyMock: RoomProxyProtocol {
         set(value) { underlyingTimeline = value }
     }
     var underlyingTimeline: TimelineProxyProtocol!
+    var pollHistoryTimeline: TimelineProxyProtocol {
+        get { return underlyingPollHistoryTimeline }
+        set(value) { underlyingPollHistoryTimeline = value }
+    }
+    var underlyingPollHistoryTimeline: TimelineProxyProtocol!
 
     //MARK: - subscribeForUpdates
 

diff --git a/ElementX/Sources/Services/Room/RoomProxy.swift b/ElementX/Sources/Services/Room/RoomProxy.swift
@@ -24,6 +24,7 @@ class RoomProxy: RoomProxyProtocol {
     private let roomListItem: RoomListItemProtocol
     private let room: RoomProtocol
     let timeline: TimelineProxyProtocol
+    let pollHistoryTimeline: TimelineProxyProtocol
     private let backgroundTaskService: BackgroundTaskServiceProtocol
     private let backgroundTaskName = "SendRoomEvent"
 
@@ -60,6 +61,7 @@ class RoomProxy: RoomProxyProtocol {
         self.room = room
         self.backgroundTaskService = backgroundTaskService
         timeline = await TimelineProxy(timeline: room.timeline(), backgroundTaskService: backgroundTaskService)
+        pollHistoryTimeline = await TimelineProxy(timeline: room.pollHistory(), backgroundTaskService: backgroundTaskService)
 
         Task {
             // Force the timeline to load member details so it can populate sender profiles whenever we add a timeline listener

diff --git a/ElementX/Sources/Services/Room/RoomProxyProtocol.swift b/ElementX/Sources/Services/Room/RoomProxyProtocol.swift
@@ -71,6 +71,9 @@ protocol RoomProxyProtocol {
 
     var timeline: TimelineProxyProtocol { get }
 
+    /// A timeline providing just polls related events
+    var pollHistoryTimeline: TimelineProxyProtocol { get }
+
     func subscribeForUpdates() async
 
     func loadAvatarURLForUserId(_ userId: String) async -> Result<URL?, RoomProxyError>