diff --git a/src/matrix/room/Room.js b/src/matrix/room/Room.js index 20a3efca..578a8e7f 100644 --- a/src/matrix/room/Room.js +++ b/src/matrix/room/Room.js @@ -112,14 +112,17 @@ export class Room extends BaseRoom { async fetchContext(eventId, log = null) { const response = await this._hsApi.context(this._roomId, eventId, {}, {log}).response(); + let contextEvent = null; await this._fetchEvents(async (txn, gapWriter) => { // Just in case we somehow receive remote echoes during event fetch const beforeExtra = await this._writeGapFill(response.events_before, txn, log); const afterExtra = await this._writeGapFill(response.events_after, txn, log); beforeExtra.push(...afterExtra); const gapResult = await gapWriter.writeContext(response, txn, log); + contextEvent = gapResult.contextEvent; return { extraGapFillChanges: beforeExtra, gapResult }; }, log); + return contextEvent; } /** @package */ diff --git a/src/matrix/room/timeline/persistence/GapWriter.js b/src/matrix/room/timeline/persistence/GapWriter.js index b03626b4..67fee305 100644 --- a/src/matrix/room/timeline/persistence/GapWriter.js +++ b/src/matrix/room/timeline/persistence/GapWriter.js @@ -230,7 +230,7 @@ export class GapWriter { const eventEntry = await txn.timelineEvents.getByEventId(this._roomId, event.event_id); if (eventEntry) { // If we have the current event, eary return. - return { entries: [], updatedEntries: [], fragments: [] } + return { entries: [], updatedEntries: [], fragments: [], contextEvent: new EventEntry(eventEntry, this._fragmentIdComparer) } } const maxFragmentKey = await txn.timelineFragments.getMaxFragmentId(this._roomId); @@ -256,6 +256,7 @@ export class GapWriter { // may as well modify one of them in-place instead of returning a third. startFill.entries.push(...endFill.entries); startFill.updatedEntries.push(...endFill.updatedEntries); + startFill.contextEvent = startFill.entries.find(e => e.id === event.event_id); startFill.fragments.push(...endFill.fragments); if (!startFill.fragments.includes(newFragment)) { // We created the fragment so it is updated.