Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 16 additions & 1 deletion src/app/features/room/RoomTimeline.test.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ const {
windowFocused,
rowItemIndex,
rowRenders,
eventRedacted,
} = vi.hoisted(() => ({
vListHandle: {
scrollSize: 1000,
Expand Down Expand Up @@ -55,6 +56,7 @@ const {
windowFocused: { current: false },
rowItemIndex: { current: 0 },
rowRenders: { count: 0 },
eventRedacted: { current: false },
}));

let lastOnScroll: ((offset: number) => void) | undefined;
Expand Down Expand Up @@ -152,10 +154,11 @@ vi.mock('$hooks/timeline/useProcessedTimeline', async (importOriginal) => {
getTs: () => Date.now(),
getSender: () => '@me:example.org',
getId: () => '$evt1',
isRedacted: () => false,
isRedacted: () => eventRedacted.current,
},
timelineSet: undefined,
eventSender: '@me:example.org',
isRedacted: eventRedacted.current,
editId: undefined,
reactionsKey: '',
content: undefined,
Expand All @@ -166,6 +169,7 @@ vi.mock('$hooks/timeline/useProcessedTimeline', async (importOriginal) => {
processedTimelineOptions.current = options;
// Same object every call so the row memo can compare eventData by identity.
fakeEvent.itemIndex = rowItemIndex.current;
fakeEvent.isRedacted = eventRedacted.current;
return (options.items as number[]).length === 0 ? [] : [fakeEvent];
},
};
Expand Down Expand Up @@ -292,6 +296,7 @@ beforeEach(() => {
windowFocused.current = false;
rowItemIndex.current = 0;
rowRenders.count = 0;
eventRedacted.current = false;
timelineSync.eventsLength = 1;
timelineSync.focusItem = undefined;
timelineSync.canPaginateBack = false;
Expand Down Expand Up @@ -504,6 +509,16 @@ describe('MemoizedTimelineItem', () => {

expect(rowRenders.count).toBe(before);
});

it('re-renders a row when its event is redacted in place', () => {
const { rerender } = renderTimeline();
const before = rowRenders.count;

eventRedacted.current = true;
rerender(<RoomTimeline room={room} editor={{} as Editor} />);

expect(rowRenders.count).toBeGreaterThan(before);
});
});

describe('unread read marker (normal sync)', () => {
Expand Down
7 changes: 6 additions & 1 deletion src/app/features/room/RoomTimeline.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -291,6 +291,7 @@ const MemoizedTimelineItem = memo(
prev.eventData.id === next.eventData.id &&
// A filtered mid-timeline insert shifts this without changing `index`.
prev.eventData.itemIndex === next.eventData.itemIndex &&
prev.eventData.isRedacted === next.eventData.isRedacted &&
prev.eventData.collapsed === next.eventData.collapsed &&
prev.eventData.willRenderNewDivider === next.eventData.willRenderNewDivider &&
prev.eventData.willRenderDayDivider === next.eventData.willRenderDayDivider &&
Expand Down Expand Up @@ -1281,7 +1282,11 @@ export function RoomTimeline({
>
{(eventData, index) => (
<MemoizedTimelineItem
key={eventData ? eventData.id : `placeholder-${index}`}
key={
eventData
? `${eventData.id}:${eventData.isRedacted ? 'redacted' : 'message'}`
: `placeholder-${index}`
}
eventData={eventData}
index={index}
showLoadingPlaceholders={showLoadingPlaceholders}
Expand Down
1 change: 1 addition & 0 deletions src/app/features/room/ThreadDrawer.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -231,6 +231,7 @@ export function ThreadDrawer({ room, threadRootId, onClose, overlay }: ThreadDra
id: ev.getId() ?? `thread-reply-${idx}`,
itemIndex: idx,
mEvent: ev,
isRedacted: ev.isRedacted(),
timelineSet,
eventSender: ev.getSender() ?? null,
collapsed: false,
Expand Down
24 changes: 22 additions & 2 deletions src/app/hooks/timeline/useProcessedTimeline.test.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -870,13 +870,16 @@ const bodyOf = (processed: ProcessedEvent[], id: string) =>
describe('useProcessedTimeline decryption', () => {
// Stable so the append-only fast path is reachable.
const ignoredUsersSet = new Set<string>();
const renderTimeline = (getEvents: () => MatrixEvent[]) =>
const renderTimeline = (
getEvents: () => MatrixEvent[],
timelineHiddenEvents: ResolvedHiddenEventSettings = hiddenEvents
) =>
renderHook(() =>
useProcessedTimeline({
items: getEvents().map((_, i) => i),
linkedTimelines: [createTimeline(getEvents())],
ignoredUsersSet,
hiddenEvents,
hiddenEvents: timelineHiddenEvents,
mxUserId: MY_USER,
readUptoEventId: undefined,
hideMembershipEvents: true,
Expand Down Expand Up @@ -913,6 +916,23 @@ describe('useProcessedTimeline decryption', () => {
expect(bodyOf(result.current, '$enc')).toBe('the secret');
});

it('refreshes a row whose event was redacted since it was cached', () => {
let redacted = false;
const event = createEvent({ id: '$redacted' });
event.isRedacted = () => redacted;
const { result, rerender } = renderTimeline(() => [event], {
...hiddenEvents,
showTombstoneEvents: true,
});

expect(result.current[0]?.isRedacted).toBe(false);

redacted = true;
rerender();

expect(result.current[0]?.isRedacted).toBe(true);
});

it('keeps the append-only fast path for unencrypted events', () => {
let events: MatrixEvent[] = [createEvent({ id: '$a', ts: 999_000 })];
const { result, rerender } = renderTimeline(() => events);
Expand Down
16 changes: 11 additions & 5 deletions src/app/hooks/timeline/useProcessedTimeline.ts
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,7 @@ export interface ProcessedEvent {
id: string;
itemIndex: number;
mEvent: MatrixEvent;
isRedacted: boolean;
timelineSet: EventTimelineSet;
eventSender: string | null;
collapsed: boolean;
Expand Down Expand Up @@ -110,6 +111,7 @@ type ProcessedEventDraft = Omit<
type TimelineEventEntry = {
mEvent: MatrixEvent;
timelineSet: EventTimelineSet;
isRedacted: boolean;
// Decryption rewrites a MatrixEvent in place, so identity alone does not prove a cached
// row still matches it. Undefined for unencrypted events.
clearType: string | undefined;
Expand All @@ -125,6 +127,7 @@ const flattenTimelineEvents = (linkedTimelines: EventTimeline[]): TimelineEventE
entries.push({
mEvent,
timelineSet,
isRedacted: mEvent.isRedacted(),
clearType: encrypted ? mEvent.getType() : undefined,
clearContent: encrypted ? mEvent.getContent() : undefined,
});
Expand All @@ -138,6 +141,7 @@ const isCachedEntryCurrent = (
current: TimelineEventEntry | undefined
): boolean =>
cached.mEvent === current?.mEvent &&
cached.isRedacted === current.isRedacted &&
cached.clearType === current.clearType &&
cached.clearContent === current.clearContent;

Expand Down Expand Up @@ -233,6 +237,7 @@ const mergeDraftsAndExtras = (
id: mEvent.getId()!,
itemIndex,
mEvent,
isRedacted: mEvent.isRedacted(),
timelineSet,
eventSender: mEvent.getSender() ?? null,
},
Expand Down Expand Up @@ -433,7 +438,7 @@ const processTimelineItems = (
for (const item of items) {
const entry = timelineEvents[item];
if (!entry) continue;
const { mEvent, timelineSet } = entry;
const { mEvent, timelineSet, isRedacted } = entry;
const { threadRootId } = mEvent;
const mEventId = mEvent.getId();
if (!mEventId) continue;
Expand Down Expand Up @@ -461,7 +466,7 @@ const processTimelineItems = (
}
}

if (mEvent.isRedacted()) {
if (isRedacted) {
const showMessageTombstone = showTombstoneEvents && isRedactableMessageType(type);
const showReactionTombstone = hiddenEventReactionTombstone && isReaction;
if (!showMessageTombstone && !showReactionTombstone) continue;
Expand All @@ -476,8 +481,8 @@ const processTimelineItems = (

const allowSpecificHiddenEvent =
(isEdit && hiddenEventEdits) ||
(isReaction && !mEvent.isRedacted() && hiddenEventReactions) ||
(isReaction && mEvent.isRedacted() && hiddenEventReactionTombstone) ||
(isReaction && !isRedacted && hiddenEventReactions) ||
(isReaction && isRedacted && hiddenEventReactionTombstone) ||
(isRedactionEvt &&
shouldShowRedactionTimelineEvent(
mEvent,
Expand All @@ -504,7 +509,7 @@ const processTimelineItems = (

if (isEdit && !hiddenEventEdits) continue;
if (isReaction) {
if (mEvent.isRedacted()) {
if (isRedacted) {
if (!hiddenEventReactionTombstone) continue;
} else if (!hiddenEventReactions) {
continue;
Expand Down Expand Up @@ -555,6 +560,7 @@ const processTimelineItems = (
id: mEventId,
itemIndex: item,
mEvent,
isRedacted,
timelineSet,
eventSender,
collapsed,
Expand Down
Loading