Phase 2 Refactor MImageBody to MVVM and remove legacy component (#33212)

* MVVMing of MImageBody and removing legacy component + css

* Fix Prettier

* update small image to large image in test

* Update test

* Preserve MImageBody legacy class names

* Click image in custom component download test

* Update snapshots

* Update MBodyFactory snapshots

* Added new tests to pass coverage

* Fix prettier

* Remove legacy import that was removed

* Add MImageReplayBody test for coverage

* Remove legacy MImageBody selectors from image view

* Update image body selectors in Playwright tests

* Keep file panel image body spacing compact

* Update apps/web/src/viewmodels/message-body/ImageBodyViewModel.ts

Co-authored-by: Florian Duros <florian.duros@ormaz.fr>

* added documentation to component

* Fix hidden media placeholder import

---------

Co-authored-by: Florian Duros <florian.duros@ormaz.fr>
This commit is contained in:
Zack
2026-05-13 06:03:43 +00:00
committed by GitHub
co-authored by Florian Duros
parent 13dd1a0b5e
commit 1e7c9f672a
26 changed files with 3389 additions and 1608 deletions
@@ -7,9 +7,11 @@ Please see LICENSE files in the repository root for full details.
import React, { type JSX, type RefObject, useContext, useEffect, useRef } from "react";
import { MsgType } from "matrix-js-sdk/src/matrix";
import { type ImageContent } from "matrix-js-sdk/src/types";
import {
DecryptionFailureBodyView,
FileBodyView,
ImageBodyView,
RedactedBodyView,
VideoBodyView,
useCreateAutoDisposedViewModel,
@@ -21,8 +23,10 @@ import { LocalDeviceVerificationStateContext } from "../../../contexts/LocalDevi
import { useMediaVisible } from "../../../hooks/useMediaVisible";
import { DecryptionFailureBodyViewModel } from "../../../viewmodels/room/timeline/event-tile/body/DecryptionFailureBodyViewModel";
import { FileBodyViewModel } from "../../../viewmodels/message-body/FileBodyViewModel";
import { ImageBodyViewModel } from "../../../viewmodels/message-body/ImageBodyViewModel";
import { RedactedBodyViewModel } from "../../../viewmodels/message-body/RedactedBodyViewModel";
import { VideoBodyViewModel } from "../../../viewmodels/message-body/VideoBodyViewModel";
import { isMimeTypeAllowed } from "../../../utils/blobs";
type MBodyComponent = React.ComponentType<IBodyProps>;
@@ -134,6 +138,122 @@ export function VideoBodyFactory({
);
}
export function ImageBodyFactory({
mxEvent,
mediaEventHelper,
forExport,
maxImageHeight,
permalinkCreator,
showFileInfo,
}: Readonly<
Pick<
IBodyProps,
"mxEvent" | "mediaEventHelper" | "forExport" | "maxImageHeight" | "permalinkCreator" | "showFileInfo"
>
>): JSX.Element {
const { timelineRenderingType } = useContext(RoomContext);
const [mediaVisible, setMediaVisible] = useMediaVisible(mxEvent);
const imageRef = useRef<HTMLImageElement>(null);
const content = mxEvent.getContent<ImageContent>();
const shouldFallbackToFileBody =
mediaEventHelper?.media.isEncrypted === true &&
!isMimeTypeAllowed(content.info?.mimetype ?? "") &&
!content.info?.thumbnail_info;
const vm = useCreateAutoDisposedViewModel(
() =>
new ImageBodyViewModel({
mxEvent,
mediaEventHelper,
forExport,
maxImageHeight,
mediaVisible,
permalinkCreator,
timelineRenderingType,
imageRef,
setMediaVisible,
}),
);
useEffect(() => {
if (shouldFallbackToFileBody) return;
vm.loadInitialMediaIfVisible();
}, [shouldFallbackToFileBody, vm]);
useEffect(() => {
if (shouldFallbackToFileBody) return;
vm.setEvent(mxEvent, mediaEventHelper);
}, [mediaEventHelper, mxEvent, shouldFallbackToFileBody, vm]);
useEffect(() => {
if (shouldFallbackToFileBody) return;
vm.setForExport(forExport);
}, [forExport, shouldFallbackToFileBody, vm]);
useEffect(() => {
if (shouldFallbackToFileBody) return;
vm.setMaxImageHeight(maxImageHeight);
}, [maxImageHeight, shouldFallbackToFileBody, vm]);
useEffect(() => {
if (shouldFallbackToFileBody) return;
vm.setMediaVisible(mediaVisible);
}, [mediaVisible, shouldFallbackToFileBody, vm]);
useEffect(() => {
if (shouldFallbackToFileBody) return;
vm.setPermalinkCreator(permalinkCreator);
}, [permalinkCreator, shouldFallbackToFileBody, vm]);
useEffect(() => {
if (shouldFallbackToFileBody) return;
vm.setTimelineRenderingType(timelineRenderingType);
}, [shouldFallbackToFileBody, timelineRenderingType, vm]);
useEffect(() => {
if (shouldFallbackToFileBody) return;
vm.setSetMediaVisible(setMediaVisible);
}, [setMediaVisible, shouldFallbackToFileBody, vm]);
const showFileBody =
!forExport &&
timelineRenderingType !== TimelineRenderingType.Room &&
timelineRenderingType !== TimelineRenderingType.Pinned &&
timelineRenderingType !== TimelineRenderingType.Search &&
timelineRenderingType !== TimelineRenderingType.Thread &&
timelineRenderingType !== TimelineRenderingType.ThreadsList;
if (shouldFallbackToFileBody) {
return (
<FileBodyFactory
mxEvent={mxEvent}
mediaEventHelper={mediaEventHelper}
forExport={forExport}
showFileInfo={showFileInfo}
/>
);
}
return (
<ImageBodyView
vm={vm}
className="mx_ImageBody"
containerClassName="mx_ImageBody_container"
imageClassName="mx_ImageBody_image"
imageRef={imageRef}
>
{showFileBody ? (
<FileBodyFactory
mxEvent={mxEvent}
mediaEventHelper={mediaEventHelper}
forExport={forExport}
showFileInfo={false}
/>
) : null}
</ImageBodyView>
);
}
export function RedactedBodyFactory({ mxEvent, ref }: Pick<IBodyProps, "mxEvent" | "ref">): JSX.Element {
const vm = useCreateAutoDisposedViewModel(() => new RedactedBodyViewModel({ mxEvent }));
@@ -164,6 +284,7 @@ export function DecryptionFailureBodyFactory({ mxEvent, ref }: Pick<IBodyProps,
// Message body factory registry for bodies that already route through view-model-backed wrappers.
const MESSAGE_BODY_TYPES = new Map<string, MBodyComponent>([
[MsgType.Image, ImageBodyFactory],
[MsgType.File, FileBodyFactory],
[MsgType.Video, VideoBodyFactory],
]);
@@ -1,714 +0,0 @@
/*
Copyright 2024 New Vector Ltd.
Copyright 2015-2021 The Matrix.org Foundation C.I.C.
Copyright 2018, 2019 Michael Telatynski <7t3chguy@gmail.com>
SPDX-License-Identifier: AGPL-3.0-only OR GPL-3.0-only OR LicenseRef-Element-Commercial
Please see LICENSE files in the repository root for full details.
*/
import React, { type JSX, type ComponentProps, createRef, type ReactNode } from "react";
import { Blurhash } from "react-blurhash";
import classNames from "classnames";
import { CSSTransition, SwitchTransition } from "react-transition-group";
import { logger } from "matrix-js-sdk/src/logger";
import { ClientEvent } from "matrix-js-sdk/src/matrix";
import { type ImageContent } from "matrix-js-sdk/src/types";
import { Tooltip } from "@vector-im/compound-web";
import { ImageErrorIcon } from "@vector-im/compound-design-tokens/assets/web/icons";
import { HiddenMediaPlaceholder } from "@element-hq/web-shared-components";
import Modal from "../../../Modal";
import { _t } from "../../../languageHandler";
import SettingsStore from "../../../settings/SettingsStore";
import Spinner from "../elements/Spinner";
import { type Media, mediaFromContent } from "../../../customisations/Media";
import { BLURHASH_FIELD, createThumbnail } from "../../../utils/image-media";
import ImageView from "../elements/ImageView";
import { type IBodyProps } from "./IBodyProps";
import { type ImageSize, suggestedSize as suggestedImageSize } from "../../../settings/enums/ImageSize";
import { MatrixClientPeg } from "../../../MatrixClientPeg";
import RoomContext, { TimelineRenderingType } from "../../../contexts/RoomContext";
import { blobIsAnimated, mayBeAnimated } from "../../../utils/Image";
import { presentableTextForFile } from "../../../utils/FileUtils";
import { createReconnectedListener } from "../../../utils/connection";
import MediaProcessingError from "./shared/MediaProcessingError";
import { DecryptError, DownloadError } from "../../../utils/DecryptFile";
import { useMediaVisible } from "../../../hooks/useMediaVisible";
import { isMimeTypeAllowed } from "../../../utils/blobs.ts";
import { FileBodyFactory, renderMBody } from "./MBodyFactory";
enum Placeholder {
NoImage,
Blurhash,
}
interface IState {
contentUrl: string | null;
thumbUrl: string | null;
isAnimated?: boolean;
error?: unknown;
imgError: boolean;
imgLoaded: boolean;
loadedImageDimensions?: {
naturalWidth: number;
naturalHeight: number;
};
hover: boolean;
focus: boolean;
placeholder: Placeholder;
}
interface IProps extends IBodyProps {
/**
* Should the media be behind a preview.
*/
mediaVisible: boolean;
/**
* Set the visibility of the media event.
* @param visible Should the event be visible.
*/
setMediaVisible: (visible: boolean) => void;
}
/**
* @private Only use for inheritance. Use the default export for presentation.
*/
export class MImageBodyInner extends React.Component<IProps, IState> {
public static contextType = RoomContext;
declare public context: React.ContextType<typeof RoomContext>;
private unmounted = false;
private image = createRef<HTMLImageElement>();
private placeholder = createRef<HTMLDivElement>();
private timeout?: number;
private sizeWatcher?: string;
public state: IState = {
contentUrl: null,
thumbUrl: null,
imgError: false,
imgLoaded: false,
hover: false,
focus: false,
placeholder: Placeholder.NoImage,
};
protected onClick = (ev: React.MouseEvent): void => {
if (ev.button === 0 && !ev.metaKey) {
ev.preventDefault();
if (!this.props.mediaVisible) {
this.props.setMediaVisible(true);
return;
}
const content = this.props.mxEvent.getContent<ImageContent>();
let httpUrl = this.state.contentUrl;
if (
this.props.mediaEventHelper?.media.isEncrypted &&
!isMimeTypeAllowed(this.props.mediaEventHelper.sourceBlob.cachedValue?.type ?? "")
) {
// contentUrl will be a blob URI mime-type=application/octet-stream so fall back to the thumbUrl instead
httpUrl = this.state.thumbUrl;
}
if (!httpUrl) return;
const params: Omit<ComponentProps<typeof ImageView>, "onFinished"> = {
src: httpUrl,
name: content.body && content.body.length > 0 ? content.body : _t("common|attachment"),
mxEvent: this.props.mxEvent,
permalinkCreator: this.props.permalinkCreator,
};
if (content.info) {
params.width = content.info.w;
params.height = content.info.h;
params.fileSize = content.info.size;
}
if (this.image.current) {
const clientRect = this.image.current.getBoundingClientRect();
params.thumbnailInfo = {
width: clientRect.width,
height: clientRect.height,
positionX: clientRect.x,
positionY: clientRect.y,
};
}
Modal.createDialog(ImageView, params, "mx_Dialog_lightbox", undefined, true);
}
};
private get shouldAutoplay(): boolean {
return !(
!this.state.contentUrl ||
!this.props.mediaVisible ||
!this.state.isAnimated ||
SettingsStore.getValue("autoplayGifs")
);
}
protected onImageEnter = (): void => {
this.setState({ hover: true });
};
protected onImageLeave = (): void => {
this.setState({ hover: false });
};
private onFocus = (): void => {
this.setState({ focus: true });
};
private onBlur = (): void => {
this.setState({ focus: false });
};
private reconnectedListener = createReconnectedListener((): void => {
MatrixClientPeg.get()?.off(ClientEvent.Sync, this.reconnectedListener);
this.setState({ imgError: false });
});
private onImageError = (): void => {
// If the thumbnail failed to load then try again using the contentUrl
if (this.state.thumbUrl) {
this.setState({
thumbUrl: null,
});
return;
}
this.clearBlurhashTimeout();
this.setState({
imgError: true,
});
MatrixClientPeg.safeGet().on(ClientEvent.Sync, this.reconnectedListener);
};
private onImageLoad = (): void => {
this.clearBlurhashTimeout();
let loadedImageDimensions: IState["loadedImageDimensions"];
if (this.image.current) {
const { naturalWidth, naturalHeight } = this.image.current;
// this is only used as a fallback in case content.info.w/h is missing
loadedImageDimensions = { naturalWidth, naturalHeight };
}
this.setState({ imgLoaded: true, loadedImageDimensions });
};
private getContentUrl(): string | null {
// During export, the content url will point to the MSC, which will later point to a local url
if (this.props.forExport) return this.media.srcMxc;
return this.media.srcHttp;
}
private get media(): Media {
return mediaFromContent(this.props.mxEvent.getContent());
}
private getThumbUrl(): string | null {
// FIXME: we let images grow as wide as you like, rather than capped to 800x600.
// So either we need to support custom timeline widths here, or reimpose the cap, otherwise the
// thumbnail resolution will be unnecessarily reduced.
// custom timeline widths seems preferable.
const thumbWidth = 800;
const thumbHeight = 600;
const content = this.props.mxEvent.getContent<ImageContent>();
const media = mediaFromContent(content);
const info = content.info;
if (info?.mimetype === "image/svg+xml" && media.hasThumbnail) {
// Special-case to return clientside sender-generated thumbnails for SVGs, if any,
// given we deliberately don't thumbnail them serverside to prevent billion lol attacks and similar.
return media.getThumbnailHttp(thumbWidth, thumbHeight, "scale");
}
// we try to download the correct resolution for hi-res images (like retina screenshots).
// Synapse only supports 800x600 thumbnails for now though,
// so we'll need to download the original image for this to work well for now.
// First, let's try a few cases that let us avoid downloading the original, including:
// - When displaying a GIF, we always want to thumbnail so that we can
// properly respect the user's GIF autoplay setting (which relies on
// thumbnailing to produce the static preview image)
// - On a low DPI device, always thumbnail to save bandwidth
// - If there's no sizing info in the event, default to thumbnail
if (this.state.isAnimated || window.devicePixelRatio === 1.0 || !info || !info.w || !info.h || !info.size) {
return media.getThumbnailOfSourceHttp(thumbWidth, thumbHeight);
}
// We should only request thumbnails if the image is bigger than 800x600 (or 1600x1200 on retina) otherwise
// the image in the timeline will just end up resampled and de-retina'd for no good reason.
// Ideally the server would pre-gen 1600x1200 thumbnails in order to provide retina thumbnails,
// but we don't do this currently in synapse for fear of disk space.
// As a compromise, let's switch to non-retina thumbnails only if the original image is both
// physically too large and going to be massive to load in the timeline (e.g. >1MB).
const isLargerThanThumbnail = info.w > thumbWidth || info.h > thumbHeight;
const isLargeFileSize = info.size > 1 * 1024 * 1024; // 1mb
if (isLargeFileSize && isLargerThanThumbnail) {
// image is too large physically and byte-wise to clutter our timeline so,
// we ask for a thumbnail, despite knowing that it will be max 800x600
// despite us being retina (as synapse doesn't do 1600x1200 thumbs yet).
return media.getThumbnailOfSourceHttp(thumbWidth, thumbHeight);
}
// download the original image otherwise, so we can scale it client side to take pixelRatio into account.
return media.srcHttp;
}
private async downloadImage(): Promise<void> {
if (this.state.contentUrl) return; // already downloaded
let thumbUrl: string | null;
let contentUrl: string | null;
if (this.props.mediaEventHelper?.media.isEncrypted) {
try {
[contentUrl, thumbUrl] = await Promise.all([
this.props.mediaEventHelper.sourceUrl.value,
this.props.mediaEventHelper.thumbnailUrl.value,
]);
} catch (error) {
if (this.unmounted) return;
if (error instanceof DecryptError) {
logger.error("Unable to decrypt attachment: ", error);
} else if (error instanceof DownloadError) {
logger.error("Unable to download attachment to decrypt it: ", error);
} else {
logger.error("Error encountered when downloading encrypted attachment: ", error);
}
// Set a placeholder image when we can't decrypt the image.
this.setState({ error });
return;
}
} else {
thumbUrl = this.getThumbUrl();
contentUrl = this.getContentUrl();
}
const content = this.props.mxEvent.getContent<ImageContent>();
let isAnimated = content.info?.["org.matrix.msc4230.is_animated"] ?? mayBeAnimated(content.info?.mimetype);
// If there is no included non-animated thumbnail then we will generate our own, we can't depend on the server
// because 1. encryption and 2. we can't ask the server specifically for a non-animated thumbnail.
if (isAnimated && !SettingsStore.getValue("autoplayGifs")) {
if (!thumbUrl || !content?.info?.thumbnail_info || mayBeAnimated(content.info.thumbnail_info.mimetype)) {
const img = document.createElement("img");
const loadPromise = new Promise((resolve, reject) => {
img.onload = resolve;
img.onerror = reject;
});
img.crossOrigin = "Anonymous"; // CORS allow canvas access
img.src = contentUrl ?? "";
try {
await loadPromise;
} catch (error) {
logger.error("Unable to download attachment: ", error);
this.setState({ error: error as Error });
return;
}
try {
// If we didn't receive the MSC4230 is_animated flag
// then we need to check if the image is animated by downloading it.
if (
content.info?.["org.matrix.msc4230.is_animated"] === false ||
(await blobIsAnimated(await this.props.mediaEventHelper!.sourceBlob.value)) === false
) {
isAnimated = false;
}
if (isAnimated) {
const thumb = await createThumbnail(
img,
img.width,
img.height,
content.info?.mimetype ?? "image/jpeg",
false,
);
thumbUrl = URL.createObjectURL(thumb.thumbnail);
}
} catch (error) {
// This is a non-critical failure, do not surface the error or bail the method here
logger.warn("Unable to generate thumbnail for animated image: ", error);
}
}
}
if (this.unmounted) return;
this.setState({
contentUrl,
thumbUrl,
isAnimated,
});
}
private clearBlurhashTimeout(): void {
if (this.timeout) {
clearTimeout(this.timeout);
this.timeout = undefined;
}
}
public componentDidMount(): void {
this.unmounted = false;
if (this.props.mediaVisible) {
// noinspection JSIgnoredPromiseFromCall
this.downloadImage();
}
// Add a 150ms timer for blurhash to first appear.
if (this.props.mxEvent.getContent().info?.[BLURHASH_FIELD]) {
this.clearBlurhashTimeout();
this.timeout = window.setTimeout(() => {
if (!this.state.imgLoaded || !this.state.imgError) {
this.setState({
placeholder: Placeholder.Blurhash,
});
}
}, 150);
}
this.sizeWatcher = SettingsStore.watchSetting("Images.size", null, () => {
this.forceUpdate(); // we don't really have a reliable thing to update, so just update the whole thing
});
}
public componentDidUpdate(prevProps: Readonly<IProps>): void {
if (!prevProps.mediaVisible && this.props.mediaVisible) {
// noinspection JSIgnoredPromiseFromCall
this.downloadImage();
}
}
public componentWillUnmount(): void {
this.unmounted = true;
MatrixClientPeg.get()?.off(ClientEvent.Sync, this.reconnectedListener);
this.clearBlurhashTimeout();
SettingsStore.unwatchSetting(this.sizeWatcher);
if (this.state.isAnimated && this.state.thumbUrl) {
URL.revokeObjectURL(this.state.thumbUrl);
}
}
protected getBanner(content: ImageContent): ReactNode {
// Hide it for the threads list & the file panel where we show it as text anyway.
if (
[TimelineRenderingType.ThreadsList, TimelineRenderingType.File].includes(this.context.timelineRenderingType)
) {
return null;
}
return (
<span className="mx_MImageBody_banner">
{presentableTextForFile(content, _t("common|image"), true, true)}
</span>
);
}
protected messageContent(
contentUrl: string | null,
thumbUrl: string | null,
content: ImageContent,
forcedHeight?: number,
): ReactNode {
if (!thumbUrl) thumbUrl = contentUrl; // fallback
// magic number
// edge case for this not to be set by conditions below
let infoWidth = 500;
let infoHeight = 500;
let infoSvg = false;
if (content.info?.w && content.info?.h) {
infoWidth = content.info.w;
infoHeight = content.info.h;
infoSvg = content.info.mimetype === "image/svg+xml";
} else if (thumbUrl && contentUrl) {
// Whilst the image loads, display nothing. We also don't display a blurhash image
// because we don't really know what size of image we'll end up with.
//
// Once loaded, use the loaded image dimensions stored in `loadedImageDimensions`.
//
// By doing this, the image "pops" into the timeline, but is still restricted
// by the same width and height logic below.
if (!this.state.loadedImageDimensions) {
let imageElement: JSX.Element;
if (!this.props.mediaVisible) {
imageElement = (
<HiddenMediaPlaceholder onClick={this.onClick}>
{_t("timeline|m.image|show_image")}
</HiddenMediaPlaceholder>
);
} else {
imageElement = (
<img
style={{ display: "none" }}
src={thumbUrl}
ref={this.image}
alt={content.body}
onError={this.onImageError}
onLoad={this.onImageLoad}
/>
);
}
return this.wrapImage(contentUrl, imageElement);
}
infoWidth = this.state.loadedImageDimensions.naturalWidth;
infoHeight = this.state.loadedImageDimensions.naturalHeight;
}
// The maximum size of the thumbnail as it is rendered as an <img>,
// accounting for any height constraints
const { w: maxWidth, h: maxHeight } = suggestedImageSize(
SettingsStore.getValue("Images.size") as ImageSize,
{ w: infoWidth, h: infoHeight },
forcedHeight ?? this.props.maxImageHeight,
);
let img: JSX.Element | undefined;
let placeholder: JSX.Element | undefined;
let gifLabel: JSX.Element | undefined;
if (!this.props.forExport && !this.state.imgLoaded) {
const classes = classNames("mx_MImageBody_placeholder", {
"mx_MImageBody_placeholder--blurhash": this.props.mxEvent.getContent().info?.[BLURHASH_FIELD],
});
placeholder = (
<div className={classes} ref={this.placeholder}>
{this.getPlaceholder(maxWidth, maxHeight)}
</div>
);
}
let showPlaceholder = Boolean(placeholder);
const hoverOrFocus = this.state.hover || this.state.focus;
if (thumbUrl && !this.state.imgError) {
let url = thumbUrl;
if (hoverOrFocus && this.shouldAutoplay) {
url = this.state.contentUrl!;
}
// Restrict the width of the thumbnail here, otherwise it will fill the container
// which has the same width as the timeline
// mx_MImageBody_thumbnail resizes img to exactly container size
img = (
<img
className="mx_MImageBody_thumbnail"
src={url}
ref={this.image}
alt={content.body}
onError={this.onImageError}
onLoad={this.onImageLoad}
onMouseEnter={this.onImageEnter}
onMouseLeave={this.onImageLeave}
/>
);
}
if (!this.props.mediaVisible) {
img = (
<div style={{ width: maxWidth, height: maxHeight }}>
<HiddenMediaPlaceholder onClick={this.onClick}>
{_t("timeline|m.image|show_image")}
</HiddenMediaPlaceholder>
</div>
);
showPlaceholder = false; // because we're hiding the image, so don't show the placeholder.
}
if (this.state.isAnimated && !SettingsStore.getValue("autoplayGifs") && !hoverOrFocus) {
// XXX: Arguably we may want a different label when the animated image is WEBP and not GIF
gifLabel = <p className="mx_MImageBody_gifLabel">GIF</p>;
}
let banner: ReactNode | undefined;
if (this.props.mediaVisible && hoverOrFocus) {
banner = this.getBanner(content);
}
// many SVGs don't have an intrinsic size if used in <img> elements.
// due to this we have to set our desired width directly.
// this way if the image is forced to shrink, the height adapts appropriately.
const sizing = infoSvg ? { maxHeight, maxWidth, width: maxWidth } : { maxHeight, maxWidth };
if (!this.props.forExport) {
placeholder = (
<SwitchTransition mode="out-in">
<CSSTransition
classNames="mx_rtg--fade"
key={`img-${showPlaceholder}`}
timeout={300}
nodeRef={this.placeholder}
>
{
showPlaceholder ? (
placeholder
) : (
<div ref={this.placeholder} />
) /* Transition always expects a child */
}
</CSSTransition>
</SwitchTransition>
);
}
const tooltipProps = this.getTooltipProps();
let thumbnail = (
<div
className="mx_MImageBody_thumbnail_container"
style={{ maxHeight, maxWidth, aspectRatio: `${infoWidth}/${infoHeight}` }}
tabIndex={tooltipProps ? 0 : undefined}
>
{placeholder}
<div style={sizing}>
{img}
{gifLabel}
{banner}
</div>
{/* HACK: This div fills out space while the image loads, to prevent scroll jumps */}
{!this.props.forExport && !this.state.imgLoaded && !placeholder && (
<div style={{ height: maxHeight, width: maxWidth }} />
)}
</div>
);
if (tooltipProps) {
// We specify isTriggerInteractive=true and make the div interactive manually as a workaround for
// https://github.com/element-hq/compound/issues/294
thumbnail = (
<Tooltip {...tooltipProps} isTriggerInteractive={true}>
{thumbnail}
</Tooltip>
);
}
return this.wrapImage(contentUrl, thumbnail);
}
// Overridden by MStickerBody
protected wrapImage(contentUrl: string | null | undefined, children: JSX.Element): ReactNode {
if (contentUrl) {
return (
<a
href={contentUrl}
target={this.props.forExport ? "_blank" : undefined}
onClick={this.onClick}
onFocus={this.onFocus}
onBlur={this.onBlur}
>
{children}
</a>
);
}
return children;
}
// Overridden by MStickerBody
protected getPlaceholder(width: number, height: number): ReactNode {
const blurhash = this.props.mxEvent.getContent().info?.[BLURHASH_FIELD];
if (blurhash) {
if (this.state.placeholder === Placeholder.NoImage) {
return null;
} else if (this.state.placeholder === Placeholder.Blurhash) {
return <Blurhash className="mx_Blurhash" hash={blurhash} width={width} height={height} />;
}
}
return <Spinner size={32} />;
}
// Overridden by MStickerBody
protected getTooltipProps(): ComponentProps<typeof Tooltip> | null {
return null;
}
// Overridden by MStickerBody
protected getFileBody(): ReactNode {
if (this.props.forExport) return null;
/*
* In the room timeline or the thread context we don't need the download
* link as the message action bar will fulfill that
*/
const hasMessageActionBar =
this.context.timelineRenderingType === TimelineRenderingType.Room ||
this.context.timelineRenderingType === TimelineRenderingType.Pinned ||
this.context.timelineRenderingType === TimelineRenderingType.Search ||
this.context.timelineRenderingType === TimelineRenderingType.Thread ||
this.context.timelineRenderingType === TimelineRenderingType.ThreadsList;
if (!hasMessageActionBar) {
return renderMBody({ ...this.props, showFileInfo: false }, FileBodyFactory);
}
}
public render(): React.ReactNode {
const content = this.props.mxEvent.getContent<ImageContent>();
// Fall back to file-body view if we are unable to render this image e.g. in the case of a blob svg
if (
this.props.mediaEventHelper?.media.isEncrypted &&
!isMimeTypeAllowed(content.info?.mimetype ?? "") &&
!content.info?.thumbnail_info
) {
return renderMBody(this.props, FileBodyFactory);
}
if (this.state.error) {
let errorText = _t("timeline|m.image|error");
if (this.state.error instanceof DecryptError) {
errorText = _t("timeline|m.image|error_decrypting");
} else if (this.state.error instanceof DownloadError) {
errorText = _t("timeline|m.image|error_downloading");
}
return (
<MediaProcessingError className="mx_MImageBody" Icon={ImageErrorIcon}>
{errorText}
</MediaProcessingError>
);
}
let contentUrl = this.state.contentUrl;
let thumbUrl: string | null;
if (this.props.forExport) {
contentUrl = this.props.mxEvent.getContent().url ?? this.props.mxEvent.getContent().file?.url;
thumbUrl = contentUrl;
} else if (this.state.isAnimated && SettingsStore.getValue("autoplayGifs")) {
thumbUrl = contentUrl;
} else {
thumbUrl = this.state.thumbUrl ?? this.state.contentUrl;
}
const thumbnail = this.messageContent(contentUrl, thumbUrl, content);
const fileBody = this.getFileBody();
return (
<div className="mx_MImageBody">
{thumbnail}
{fileBody}
</div>
);
}
}
// Wrap MImageBody component so we can use a hook here.
const MImageBody: React.FC<IBodyProps> = (props) => {
const [mediaVisible, setVisible] = useMediaVisible(props.mxEvent);
return <MImageBodyInner mediaVisible={mediaVisible} setMediaVisible={setVisible} {...props} />;
};
export default MImageBody;
@@ -6,16 +6,621 @@ SPDX-License-Identifier: AGPL-3.0-only OR GPL-3.0-only OR LicenseRef-Element-Com
Please see LICENSE files in the repository root for full details.
*/
import React, { type JSX } from "react";
import React, { type JSX, type ComponentProps, createRef, type ReactNode } from "react";
import { Blurhash } from "react-blurhash";
import classNames from "classnames";
import { CSSTransition, SwitchTransition } from "react-transition-group";
import { logger } from "matrix-js-sdk/src/logger";
import { ClientEvent } from "matrix-js-sdk/src/matrix";
import { type ImageContent } from "matrix-js-sdk/src/types";
import { Tooltip } from "@vector-im/compound-web";
import { ImageErrorIcon } from "@vector-im/compound-design-tokens/assets/web/icons";
import { HiddenMediaPlaceholder } from "@element-hq/web-shared-components";
import { MImageBodyInner } from "./MImageBody";
import Modal from "../../../Modal";
import { _t } from "../../../languageHandler";
import SettingsStore from "../../../settings/SettingsStore";
import Spinner from "../elements/Spinner";
import { type Media, mediaFromContent } from "../../../customisations/Media";
import { BLURHASH_FIELD, createThumbnail } from "../../../utils/image-media";
import ImageView from "../elements/ImageView";
import { type IBodyProps } from "./IBodyProps";
import { type ImageSize, suggestedSize as suggestedImageSize } from "../../../settings/enums/ImageSize";
import { MatrixClientPeg } from "../../../MatrixClientPeg";
import RoomContext, { TimelineRenderingType } from "../../../contexts/RoomContext";
import { blobIsAnimated, mayBeAnimated } from "../../../utils/Image";
import { presentableTextForFile } from "../../../utils/FileUtils";
import { createReconnectedListener } from "../../../utils/connection";
import MediaProcessingError from "./shared/MediaProcessingError";
import { DecryptError, DownloadError } from "../../../utils/DecryptFile";
import { useMediaVisible } from "../../../hooks/useMediaVisible";
import { isMimeTypeAllowed } from "../../../utils/blobs.ts";
import { FileBodyFactory, renderMBody } from "./MBodyFactory";
enum Placeholder {
NoImage,
Blurhash,
}
interface IState {
contentUrl: string | null;
thumbUrl: string | null;
isAnimated?: boolean;
error?: unknown;
imgError: boolean;
imgLoaded: boolean;
loadedImageDimensions?: {
naturalWidth: number;
naturalHeight: number;
};
hover: boolean;
focus: boolean;
placeholder: Placeholder;
}
export interface ImageBodyBaseProps extends IBodyProps {
mediaVisible: boolean;
setMediaVisible: (visible: boolean) => void;
}
export class ImageBodyBaseInner extends React.Component<ImageBodyBaseProps, IState> {
public static contextType = RoomContext;
declare public context: React.ContextType<typeof RoomContext>;
private unmounted = false;
private image = createRef<HTMLImageElement>();
private placeholder = createRef<HTMLDivElement>();
private timeout?: number;
private sizeWatcher?: string;
public state: IState = {
contentUrl: null,
thumbUrl: null,
imgError: false,
imgLoaded: false,
hover: false,
focus: false,
placeholder: Placeholder.NoImage,
};
protected onClick = (ev: React.MouseEvent): void => {
if (ev.button === 0 && !ev.metaKey) {
ev.preventDefault();
if (!this.props.mediaVisible) {
this.props.setMediaVisible(true);
return;
}
const content = this.props.mxEvent.getContent<ImageContent>();
let httpUrl = this.state.contentUrl;
if (
this.props.mediaEventHelper?.media.isEncrypted &&
!isMimeTypeAllowed(this.props.mediaEventHelper.sourceBlob.cachedValue?.type ?? "")
) {
httpUrl = this.state.thumbUrl;
}
if (!httpUrl) return;
const params: Omit<ComponentProps<typeof ImageView>, "onFinished"> = {
src: httpUrl,
name: content.body && content.body.length > 0 ? content.body : _t("common|attachment"),
mxEvent: this.props.mxEvent,
permalinkCreator: this.props.permalinkCreator,
};
if (content.info) {
params.width = content.info.w;
params.height = content.info.h;
params.fileSize = content.info.size;
}
if (this.image.current) {
const clientRect = this.image.current.getBoundingClientRect();
params.thumbnailInfo = {
width: clientRect.width,
height: clientRect.height,
positionX: clientRect.x,
positionY: clientRect.y,
};
}
Modal.createDialog(ImageView, params, "mx_Dialog_lightbox", undefined, true);
}
};
private get shouldAutoplay(): boolean {
return !(
!this.state.contentUrl ||
!this.props.mediaVisible ||
!this.state.isAnimated ||
SettingsStore.getValue("autoplayGifs")
);
}
protected onImageEnter = (): void => {
this.setState({ hover: true });
};
protected onImageLeave = (): void => {
this.setState({ hover: false });
};
private onFocus = (): void => {
this.setState({ focus: true });
};
private onBlur = (): void => {
this.setState({ focus: false });
};
private reconnectedListener = createReconnectedListener((): void => {
MatrixClientPeg.get()?.off(ClientEvent.Sync, this.reconnectedListener);
this.setState({ imgError: false });
});
private onImageError = (): void => {
if (this.state.thumbUrl) {
this.setState({
thumbUrl: null,
});
return;
}
this.clearBlurhashTimeout();
this.setState({
imgError: true,
});
MatrixClientPeg.safeGet().on(ClientEvent.Sync, this.reconnectedListener);
};
private onImageLoad = (): void => {
this.clearBlurhashTimeout();
let loadedImageDimensions: IState["loadedImageDimensions"];
if (this.image.current) {
const { naturalWidth, naturalHeight } = this.image.current;
loadedImageDimensions = { naturalWidth, naturalHeight };
}
this.setState({ imgLoaded: true, loadedImageDimensions });
};
private getContentUrl(): string | null {
if (this.props.forExport) return this.media.srcMxc;
return this.media.srcHttp;
}
private get media(): Media {
return mediaFromContent(this.props.mxEvent.getContent());
}
private getThumbUrl(): string | null {
const thumbWidth = 800;
const thumbHeight = 600;
const content = this.props.mxEvent.getContent<ImageContent>();
const media = mediaFromContent(content);
const info = content.info;
if (info?.mimetype === "image/svg+xml" && media.hasThumbnail) {
return media.getThumbnailHttp(thumbWidth, thumbHeight, "scale");
}
if (this.state.isAnimated || window.devicePixelRatio === 1.0 || !info || !info.w || !info.h || !info.size) {
return media.getThumbnailOfSourceHttp(thumbWidth, thumbHeight);
}
const isLargerThanThumbnail = info.w > thumbWidth || info.h > thumbHeight;
const isLargeFileSize = info.size > 1 * 1024 * 1024;
if (isLargeFileSize && isLargerThanThumbnail) {
return media.getThumbnailOfSourceHttp(thumbWidth, thumbHeight);
}
return media.srcHttp;
}
private async downloadImage(): Promise<void> {
if (this.state.contentUrl) return;
let thumbUrl: string | null;
let contentUrl: string | null;
if (this.props.mediaEventHelper?.media.isEncrypted) {
try {
[contentUrl, thumbUrl] = await Promise.all([
this.props.mediaEventHelper.sourceUrl.value,
this.props.mediaEventHelper.thumbnailUrl.value,
]);
} catch (error) {
if (this.unmounted) return;
if (error instanceof DecryptError) {
logger.error("Unable to decrypt attachment: ", error);
} else if (error instanceof DownloadError) {
logger.error("Unable to download attachment to decrypt it: ", error);
} else {
logger.error("Error encountered when downloading encrypted attachment: ", error);
}
this.setState({ error });
return;
}
} else {
thumbUrl = this.getThumbUrl();
contentUrl = this.getContentUrl();
}
const content = this.props.mxEvent.getContent<ImageContent>();
let isAnimated = content.info?.["org.matrix.msc4230.is_animated"] ?? mayBeAnimated(content.info?.mimetype);
if (isAnimated && !SettingsStore.getValue("autoplayGifs")) {
if (!thumbUrl || !content?.info?.thumbnail_info || mayBeAnimated(content.info.thumbnail_info.mimetype)) {
const img = document.createElement("img");
const loadPromise = new Promise((resolve, reject) => {
img.onload = resolve;
img.onerror = reject;
});
img.crossOrigin = "Anonymous";
img.src = contentUrl ?? "";
try {
await loadPromise;
} catch (error) {
logger.error("Unable to download attachment: ", error);
this.setState({ error: error as Error });
return;
}
try {
if (
content.info?.["org.matrix.msc4230.is_animated"] === false ||
(await blobIsAnimated(await this.props.mediaEventHelper!.sourceBlob.value)) === false
) {
isAnimated = false;
}
if (isAnimated) {
const thumb = await createThumbnail(
img,
img.width,
img.height,
content.info?.mimetype ?? "image/jpeg",
false,
);
thumbUrl = URL.createObjectURL(thumb.thumbnail);
}
} catch (error) {
logger.warn("Unable to generate thumbnail for animated image: ", error);
}
}
}
if (this.unmounted) return;
this.setState({
contentUrl,
thumbUrl,
isAnimated,
});
}
private clearBlurhashTimeout(): void {
if (this.timeout) {
clearTimeout(this.timeout);
this.timeout = undefined;
}
}
public componentDidMount(): void {
this.unmounted = false;
if (this.props.mediaVisible) {
void this.downloadImage();
}
if (this.props.mxEvent.getContent().info?.[BLURHASH_FIELD]) {
this.clearBlurhashTimeout();
this.timeout = window.setTimeout(() => {
if (!this.state.imgLoaded || !this.state.imgError) {
this.setState({
placeholder: Placeholder.Blurhash,
});
}
}, 150);
}
this.sizeWatcher = SettingsStore.watchSetting("Images.size", null, () => {
this.forceUpdate();
});
}
public componentDidUpdate(prevProps: Readonly<ImageBodyBaseProps>): void {
if (!prevProps.mediaVisible && this.props.mediaVisible) {
void this.downloadImage();
}
}
public componentWillUnmount(): void {
this.unmounted = true;
MatrixClientPeg.get()?.off(ClientEvent.Sync, this.reconnectedListener);
this.clearBlurhashTimeout();
SettingsStore.unwatchSetting(this.sizeWatcher);
if (this.state.isAnimated && this.state.thumbUrl) {
URL.revokeObjectURL(this.state.thumbUrl);
}
}
protected getBanner(content: ImageContent): ReactNode {
if (
[TimelineRenderingType.ThreadsList, TimelineRenderingType.File].includes(this.context.timelineRenderingType)
) {
return null;
}
return (
<span className="mx_MImageBody_banner">
{presentableTextForFile(content, _t("common|image"), true, true)}
</span>
);
}
protected messageContent(
contentUrl: string | null,
thumbUrl: string | null,
content: ImageContent,
forcedHeight?: number,
): ReactNode {
if (!thumbUrl) thumbUrl = contentUrl;
let infoWidth = 500;
let infoHeight = 500;
let infoSvg = false;
if (content.info?.w && content.info?.h) {
infoWidth = content.info.w;
infoHeight = content.info.h;
infoSvg = content.info.mimetype === "image/svg+xml";
} else if (thumbUrl && contentUrl) {
if (!this.state.loadedImageDimensions) {
let imageElement: JSX.Element;
if (!this.props.mediaVisible) {
imageElement = (
<HiddenMediaPlaceholder onClick={this.onClick}>
{_t("timeline|m.image|show_image")}
</HiddenMediaPlaceholder>
);
} else {
imageElement = (
<img
style={{ display: "none" }}
src={thumbUrl}
ref={this.image}
alt={content.body}
onError={this.onImageError}
onLoad={this.onImageLoad}
/>
);
}
return this.wrapImage(contentUrl, imageElement);
}
infoWidth = this.state.loadedImageDimensions.naturalWidth;
infoHeight = this.state.loadedImageDimensions.naturalHeight;
}
const { w: maxWidth, h: maxHeight } = suggestedImageSize(
SettingsStore.getValue("Images.size") as ImageSize,
{ w: infoWidth, h: infoHeight },
forcedHeight ?? this.props.maxImageHeight,
);
let img: JSX.Element | undefined;
let placeholder: JSX.Element | undefined;
let gifLabel: JSX.Element | undefined;
if (!this.props.forExport && !this.state.imgLoaded) {
const classes = classNames("mx_MImageBody_placeholder", {
"mx_MImageBody_placeholder--blurhash": this.props.mxEvent.getContent().info?.[BLURHASH_FIELD],
});
placeholder = (
<div className={classes} ref={this.placeholder}>
{this.getPlaceholder(maxWidth, maxHeight)}
</div>
);
}
let showPlaceholder = Boolean(placeholder);
const hoverOrFocus = this.state.hover || this.state.focus;
if (thumbUrl && !this.state.imgError) {
let url = thumbUrl;
if (hoverOrFocus && this.shouldAutoplay) {
url = this.state.contentUrl!;
}
img = (
<img
className="mx_MImageBody_thumbnail"
src={url}
ref={this.image}
alt={content.body}
onError={this.onImageError}
onLoad={this.onImageLoad}
onMouseEnter={this.onImageEnter}
onMouseLeave={this.onImageLeave}
/>
);
}
if (!this.props.mediaVisible) {
img = (
<div style={{ width: maxWidth, height: maxHeight }}>
<HiddenMediaPlaceholder onClick={this.onClick}>
{_t("timeline|m.image|show_image")}
</HiddenMediaPlaceholder>
</div>
);
showPlaceholder = false;
}
if (this.state.isAnimated && !SettingsStore.getValue("autoplayGifs") && !hoverOrFocus) {
gifLabel = <p className="mx_MImageBody_gifLabel">GIF</p>;
}
let banner: ReactNode | undefined;
if (this.props.mediaVisible && hoverOrFocus) {
banner = this.getBanner(content);
}
const sizing = infoSvg ? { maxHeight, maxWidth, width: maxWidth } : { maxHeight, maxWidth };
if (!this.props.forExport) {
placeholder = (
<SwitchTransition mode="out-in">
<CSSTransition
classNames="mx_rtg--fade"
key={`img-${showPlaceholder}`}
timeout={300}
nodeRef={this.placeholder}
>
{showPlaceholder ? placeholder : <div ref={this.placeholder} />}
</CSSTransition>
</SwitchTransition>
);
}
const tooltipProps = this.getTooltipProps();
let thumbnail = (
<div
className="mx_MImageBody_thumbnail_container"
style={{ maxHeight, maxWidth, aspectRatio: `${infoWidth}/${infoHeight}` }}
tabIndex={tooltipProps ? 0 : undefined}
>
{placeholder}
<div style={sizing}>
{img}
{gifLabel}
{banner}
</div>
{!this.props.forExport && !this.state.imgLoaded && !placeholder && (
<div style={{ height: maxHeight, width: maxWidth }} />
)}
</div>
);
if (tooltipProps) {
thumbnail = (
<Tooltip {...tooltipProps} isTriggerInteractive={true}>
{thumbnail}
</Tooltip>
);
}
return this.wrapImage(contentUrl, thumbnail);
}
protected wrapImage(contentUrl: string | null | undefined, children: JSX.Element): ReactNode {
if (contentUrl) {
return (
<a
href={contentUrl}
target={this.props.forExport ? "_blank" : undefined}
onClick={this.onClick}
onFocus={this.onFocus}
onBlur={this.onBlur}
>
{children}
</a>
);
}
return children;
}
protected getPlaceholder(width: number, height: number): ReactNode {
const blurhash = this.props.mxEvent.getContent().info?.[BLURHASH_FIELD];
if (blurhash) {
if (this.state.placeholder === Placeholder.NoImage) {
return null;
} else if (this.state.placeholder === Placeholder.Blurhash) {
return <Blurhash className="mx_Blurhash" hash={blurhash} width={width} height={height} />;
}
}
return <Spinner size={32} />;
}
protected getTooltipProps(): ComponentProps<typeof Tooltip> | null {
return null;
}
protected getFileBody(): ReactNode {
if (this.props.forExport) return null;
const hasMessageActionBar =
this.context.timelineRenderingType === TimelineRenderingType.Room ||
this.context.timelineRenderingType === TimelineRenderingType.Pinned ||
this.context.timelineRenderingType === TimelineRenderingType.Search ||
this.context.timelineRenderingType === TimelineRenderingType.Thread ||
this.context.timelineRenderingType === TimelineRenderingType.ThreadsList;
if (!hasMessageActionBar) {
return renderMBody({ ...this.props, showFileInfo: false }, FileBodyFactory);
}
}
public render(): React.ReactNode {
const content = this.props.mxEvent.getContent<ImageContent>();
if (
this.props.mediaEventHelper?.media.isEncrypted &&
!isMimeTypeAllowed(content.info?.mimetype ?? "") &&
!content.info?.thumbnail_info
) {
return renderMBody(this.props, FileBodyFactory);
}
if (this.state.error) {
let errorText = _t("timeline|m.image|error");
if (this.state.error instanceof DecryptError) {
errorText = _t("timeline|m.image|error_decrypting");
} else if (this.state.error instanceof DownloadError) {
errorText = _t("timeline|m.image|error_downloading");
}
return (
<MediaProcessingError className="mx_MImageBody" Icon={ImageErrorIcon}>
{errorText}
</MediaProcessingError>
);
}
let contentUrl = this.state.contentUrl;
let thumbUrl: string | null;
if (this.props.forExport) {
contentUrl = this.props.mxEvent.getContent().url ?? this.props.mxEvent.getContent().file?.url;
thumbUrl = contentUrl;
} else if (this.state.isAnimated && SettingsStore.getValue("autoplayGifs")) {
thumbUrl = contentUrl;
} else {
thumbUrl = this.state.thumbUrl ?? this.state.contentUrl;
}
const thumbnail = this.messageContent(contentUrl, thumbUrl, content);
const fileBody = this.getFileBody();
return (
<div className="mx_MImageBody">
{thumbnail}
{fileBody}
</div>
);
}
}
const FORCED_IMAGE_HEIGHT = 44;
class MImageReplyBodyInner extends MImageBodyInner {
class MImageReplyBodyInner extends ImageBodyBaseInner {
public onClick = (ev: React.MouseEvent): void => {
ev.preventDefault();
};
@@ -37,6 +642,7 @@ class MImageReplyBodyInner extends MImageBodyInner {
return <div className="mx_MImageReplyBody">{thumbnail}</div>;
}
}
const MImageReplyBody: React.FC<IBodyProps> = (props) => {
const [mediaVisible, setVisible] = useMediaVisible(props.mxEvent);
return <MImageReplyBodyInner mediaVisible={mediaVisible} setMediaVisible={setVisible} {...props} />;
@@ -9,13 +9,13 @@ import React, { type JSX, type ComponentProps, type ReactNode } from "react";
import { type Tooltip } from "@vector-im/compound-web";
import { type MediaEventContent } from "matrix-js-sdk/src/types";
import { MImageBodyInner } from "./MImageBody";
import { ImageBodyBaseInner } from "./MImageReplyBody";
import { BLURHASH_FIELD } from "../../../utils/image-media";
import IconsShowStickersSvg from "../../../../res/img/icons-show-stickers.svg";
import { type IBodyProps } from "./IBodyProps";
import { useMediaVisible } from "../../../hooks/useMediaVisible";
class MStickerBodyInner extends MImageBodyInner {
class MStickerBodyInner extends ImageBodyBaseInner {
// Mostly empty to prevent default behaviour of MImageBody
protected onClick = (ev: React.MouseEvent): void => {
ev.preventDefault();
@@ -25,7 +25,6 @@ import { Mjolnir } from "../../../mjolnir/Mjolnir";
import { type IMediaBody } from "./IMediaBody";
import { MediaEventHelper } from "../../../utils/MediaEventHelper";
import { type IBodyProps } from "./IBodyProps";
import MImageBody from "./MImageBody";
import MVoiceOrAudioBody from "./MVoiceOrAudioBody";
import MStickerBody from "./MStickerBody";
import MPollBody from "./MPollBody";
@@ -36,6 +35,7 @@ import { MjolnirBodyViewModel } from "../../../viewmodels/room/timeline/event-ti
import {
DecryptionFailureBodyFactory,
FileBodyFactory,
ImageBodyFactory,
RedactedBodyFactory,
VideoBodyFactory,
renderMBody,
@@ -67,7 +67,7 @@ const baseBodyTypes = new Map<string, React.ComponentType<IBodyProps>>([
[MsgType.Text, TextualBodyFactory],
[MsgType.Notice, TextualBodyFactory],
[MsgType.Emote, TextualBodyFactory],
[MsgType.Image, MImageBody],
[MsgType.Image, ImageBodyFactory],
[MsgType.File, (props: IBodyProps) => renderMBody(props, FileBodyFactory)!],
[MsgType.Audio, MVoiceOrAudioBody],
[MsgType.Video, VideoBodyFactory],
@@ -283,7 +283,7 @@ export default class MessageEvent extends React.Component<IProps> implements IMe
}
if (
((BodyType === MImageBody || BodyType === VideoBodyFactory) &&
((BodyType === ImageBodyFactory || BodyType === VideoBodyFactory) &&
!this.validateImageOrVideoMimetype(content)) ||
(BodyType === MStickerBody && !this.validateStickerMimetype(content))
) {
@@ -0,0 +1,674 @@
/*
* Copyright 2026 Element Creations Ltd.
*
* SPDX-License-Identifier: AGPL-3.0-only OR GPL-3.0-only OR LicenseRef-Element-Commercial
* Please see LICENSE files in the repository root for full details.
*/
import { type ComponentProps, type MouseEvent, type RefObject } from "react";
import { logger } from "matrix-js-sdk/src/logger";
import { ClientEvent, type MatrixEvent } from "matrix-js-sdk/src/matrix";
import { type ImageContent } from "matrix-js-sdk/src/types";
import {
BaseViewModel,
ImageBodyViewPlaceholder,
ImageBodyViewState,
type ImageBodyViewModel as ImageBodyViewModelInterface,
type ImageBodyViewSnapshot,
} from "@element-hq/web-shared-components";
import Modal from "../../Modal";
import { MatrixClientPeg } from "../../MatrixClientPeg";
import { _t } from "../../languageHandler";
import { mediaFromContent } from "../../customisations/Media";
import { TimelineRenderingType } from "../../contexts/RoomContext";
import SettingsStore from "../../settings/SettingsStore";
import { type ImageSize, suggestedSize as suggestedImageSize } from "../../settings/enums/ImageSize";
import { presentableTextForFile } from "../../utils/FileUtils";
import { type MediaEventHelper } from "../../utils/MediaEventHelper";
import { blobIsAnimated, mayBeAnimated } from "../../utils/Image";
import { type RoomPermalinkCreator } from "../../utils/permalinks/Permalinks";
import { createReconnectedListener } from "../../utils/connection";
import { DecryptError, DownloadError } from "../../utils/DecryptFile";
import { BLURHASH_FIELD, createThumbnail } from "../../utils/image-media";
import { isMimeTypeAllowed } from "../../utils/blobs";
import ImageView from "../../components/views/elements/ImageView";
export interface ImageBodyViewModelProps {
/**
* Image event being rendered.
*/
mxEvent: MatrixEvent;
/**
* Helper for resolving encrypted media sources.
*/
mediaEventHelper?: MediaEventHelper;
/**
* Whether the image is being rendered for export instead of the live timeline.
*/
forExport?: boolean;
/**
* Optional maximum height applied when computing the rendered image dimensions.
*/
maxImageHeight?: number;
/**
* Whether the media should currently be shown instead of the hidden-media preview.
*/
mediaVisible: boolean;
/**
* Permalink helper passed to the image lightbox.
*/
permalinkCreator?: RoomPermalinkCreator;
/**
* Timeline context used to decide which labels and supplemental content should be shown.
*/
timelineRenderingType: TimelineRenderingType;
/**
* Ref to the underlying image element used for load dimensions and lightbox animation.
*/
imageRef: RefObject<HTMLImageElement | null>;
/**
* Callback invoked when hidden media is revealed.
*/
setMediaVisible?: (visible: boolean) => void;
}
interface LoadedImageDimensions {
naturalWidth: number;
naturalHeight: number;
}
interface InternalState {
contentUrl: string | null;
thumbUrl: string | null;
isAnimated: boolean;
error: unknown | null;
imgError: boolean;
imgLoaded: boolean;
loadedImageDimensions?: LoadedImageDimensions;
placeholder: ImageBodyViewPlaceholder;
imageSize: ImageSize;
generatedThumbnailUrl: string | null;
}
type ImageInfoWithAnimationFlag = NonNullable<ImageContent["info"]> & {
"org.matrix.msc4230.is_animated"?: boolean;
};
/**
* View model for the image message body, encapsulating media loading, sizing,
* visibility, animated-image previews, and lightbox interactions.
*/
export class ImageBodyViewModel
extends BaseViewModel<ImageBodyViewSnapshot, ImageBodyViewModelProps>
implements ImageBodyViewModelInterface
{
private state: InternalState;
private blurhashTimeout?: number;
private readonly reconnectedListener = createReconnectedListener((): void => {
MatrixClientPeg.get()?.off(ClientEvent.Sync, this.reconnectedListener);
if (!this.state.imgError) {
return;
}
this.state = {
...this.state,
imgError: false,
};
this.updateSnapshotFromState();
});
public constructor(props: ImageBodyViewModelProps) {
const initialState = ImageBodyViewModel.createInitialState(props.mxEvent);
super(props, ImageBodyViewModel.computeSnapshot(props, initialState));
this.state = initialState;
const imageSizeWatcherRef = SettingsStore.watchSetting("Images.size", null, (_s, _r, _l, _nvl, value) => {
this.setImageSize(value as ImageSize);
});
this.disposables.track(() => SettingsStore.unwatchSetting(imageSizeWatcherRef));
}
private static createInitialState(mxEvent: MatrixEvent): InternalState {
return {
contentUrl: null,
thumbUrl: null,
isAnimated: false,
error: null,
imgError: false,
imgLoaded: false,
loadedImageDimensions: undefined,
placeholder: mxEvent.getContent<ImageContent>().info?.[BLURHASH_FIELD]
? ImageBodyViewPlaceholder.NONE
: ImageBodyViewPlaceholder.SPINNER,
imageSize: SettingsStore.getValue("Images.size") as ImageSize,
generatedThumbnailUrl: null,
};
}
private static getImageDimensions(
props: ImageBodyViewModelProps,
state: InternalState,
): Pick<ImageBodyViewSnapshot, "maxWidth" | "maxHeight" | "aspectRatio" | "isSvg"> {
const content = props.mxEvent.getContent<ImageContent>();
const info = content.info;
const naturalWidth = info?.w ?? state.loadedImageDimensions?.naturalWidth;
const naturalHeight = info?.h ?? state.loadedImageDimensions?.naturalHeight;
if (!naturalWidth || !naturalHeight) {
return {
maxWidth: undefined,
maxHeight: undefined,
aspectRatio: undefined,
isSvg: info?.mimetype === "image/svg+xml",
};
}
const { w: maxWidth, h: maxHeight } = suggestedImageSize(
state.imageSize,
{ w: naturalWidth, h: naturalHeight },
props.maxImageHeight,
);
return {
maxWidth,
maxHeight,
aspectRatio: `${naturalWidth}/${naturalHeight}`,
isSvg: info?.mimetype === "image/svg+xml",
};
}
private static computeErrorLabel(error: unknown, imgError: boolean): string {
if (error instanceof DecryptError) return _t("timeline|m.image|error_decrypting");
if (error instanceof DownloadError) return _t("timeline|m.image|error_downloading");
if (imgError || error) return _t("timeline|m.image|error");
return _t("timeline|m.image|error");
}
private static shouldShowBanner(timelineRenderingType: TimelineRenderingType): boolean {
return ![TimelineRenderingType.ThreadsList, TimelineRenderingType.File].includes(timelineRenderingType);
}
private static computeSnapshot(props: ImageBodyViewModelProps, state: InternalState): ImageBodyViewSnapshot {
const content = props.mxEvent.getContent<ImageContent>();
const dimensions = ImageBodyViewModel.getImageDimensions(props, state);
const autoplayGifs = SettingsStore.getValue("autoplayGifs") as boolean;
const contentUrl = ImageBodyViewModel.getContentUrl(props, state);
const thumbnailSrc = props.forExport
? (contentUrl ?? undefined)
: state.isAnimated && autoplayGifs
? (contentUrl ?? undefined)
: (state.thumbUrl ?? contentUrl ?? undefined);
if (state.error || state.imgError) {
return {
state: ImageBodyViewState.ERROR,
errorLabel: ImageBodyViewModel.computeErrorLabel(state.error, state.imgError),
...dimensions,
};
}
if (!props.mediaVisible) {
return {
state: ImageBodyViewState.HIDDEN,
hiddenButtonLabel: _t("timeline|m.image|show_image"),
...dimensions,
};
}
return {
state: ImageBodyViewState.READY,
alt: content.body,
src: contentUrl ?? undefined,
thumbnailSrc,
showAnimatedContentOnHover: state.isAnimated && !autoplayGifs && !!contentUrl,
placeholder: !props.forExport && !state.imgLoaded ? state.placeholder : ImageBodyViewPlaceholder.NONE,
blurhash: content.info?.[BLURHASH_FIELD],
gifLabel: state.isAnimated && !autoplayGifs ? "GIF" : undefined,
bannerLabel: ImageBodyViewModel.shouldShowBanner(props.timelineRenderingType)
? presentableTextForFile(content, _t("common|image"), true, true)
: undefined,
linkUrl: contentUrl ?? undefined,
linkTarget: props.forExport ? "_blank" : undefined,
...dimensions,
};
}
private static getContentUrl(props: ImageBodyViewModelProps, state: InternalState): string | null {
if (props.forExport) {
return (
props.mxEvent.getContent<ImageContent>().url ??
props.mxEvent.getContent<ImageContent>().file?.url ??
null
);
}
return state.contentUrl;
}
public loadInitialMediaIfVisible(): void {
if (!this.props.mediaVisible) {
return;
}
this.scheduleBlurhashPlaceholder();
void this.downloadImage();
}
private updateSnapshotFromState(): void {
this.snapshot.set(ImageBodyViewModel.computeSnapshot(this.props, this.state));
}
private resetState(mxEvent: MatrixEvent): void {
this.clearBlurhashTimeout();
MatrixClientPeg.get()?.off(ClientEvent.Sync, this.reconnectedListener);
this.revokeGeneratedThumbnailUrl();
this.state = ImageBodyViewModel.createInitialState(mxEvent);
}
private revokeGeneratedThumbnailUrl(): void {
if (!this.state.generatedThumbnailUrl) {
return;
}
URL.revokeObjectURL(this.state.generatedThumbnailUrl);
this.state = {
...this.state,
generatedThumbnailUrl: null,
};
}
private clearBlurhashTimeout(): void {
if (!this.blurhashTimeout) {
return;
}
clearTimeout(this.blurhashTimeout);
this.blurhashTimeout = undefined;
}
private scheduleBlurhashPlaceholder(): void {
if (
!this.props.mxEvent.getContent<ImageContent>().info?.[BLURHASH_FIELD] ||
this.state.imgLoaded ||
this.state.imgError
) {
return;
}
this.clearBlurhashTimeout();
this.blurhashTimeout = window.setTimeout(() => {
if (this.isDisposed || this.state.imgLoaded || this.state.imgError) {
return;
}
this.state = {
...this.state,
placeholder: ImageBodyViewPlaceholder.BLURHASH,
};
this.snapshot.merge({ placeholder: ImageBodyViewPlaceholder.BLURHASH });
}, 150);
}
private getThumbUrl(): string | null {
const thumbWidth = 800;
const thumbHeight = 600;
const content = this.props.mxEvent.getContent<ImageContent>();
const media = mediaFromContent(content);
const info = content.info;
if (info?.mimetype === "image/svg+xml" && media.hasThumbnail) {
return media.getThumbnailHttp(thumbWidth, thumbHeight, "scale");
}
if (this.state.isAnimated || window.devicePixelRatio === 1.0 || !info || !info.w || !info.h || !info.size) {
return media.getThumbnailOfSourceHttp(thumbWidth, thumbHeight);
}
const isLargerThanThumbnail = info.w > thumbWidth || info.h > thumbHeight;
const isLargeFileSize = info.size > 1 * 1024 * 1024;
if (isLargeFileSize && isLargerThanThumbnail) {
return media.getThumbnailOfSourceHttp(thumbWidth, thumbHeight);
}
return media.srcHttp;
}
private async downloadImage(): Promise<void> {
if (this.state.contentUrl || this.props.forExport) {
return;
}
let thumbUrl: string | null;
let contentUrl: string | null;
if (this.props.mediaEventHelper?.media.isEncrypted) {
try {
[contentUrl, thumbUrl] = await Promise.all([
this.props.mediaEventHelper.sourceUrl.value,
this.props.mediaEventHelper.thumbnailUrl.value,
]);
} catch (error) {
if (this.isDisposed) {
return;
}
if (error instanceof DecryptError) {
logger.error("Unable to decrypt attachment: ", error);
} else if (error instanceof DownloadError) {
logger.error("Unable to download attachment to decrypt it: ", error);
} else {
logger.error("Error encountered when downloading encrypted attachment: ", error);
}
this.state = {
...this.state,
error: error as Error,
};
this.updateSnapshotFromState();
return;
}
} else {
contentUrl = mediaFromContent(this.props.mxEvent.getContent<ImageContent>()).srcHttp;
thumbUrl = this.getThumbUrl();
}
const content = this.props.mxEvent.getContent<ImageContent>();
let generatedThumbnailUrl: string | null = null;
let isAnimated = (content.info as ImageInfoWithAnimationFlag | undefined)?.["org.matrix.msc4230.is_animated"];
if (isAnimated === undefined) {
isAnimated = mayBeAnimated(content.info?.mimetype);
}
const autoplayGifs = SettingsStore.getValue("autoplayGifs") as boolean;
if (isAnimated && !autoplayGifs) {
if (!thumbUrl || !content.info?.thumbnail_info || mayBeAnimated(content.info.thumbnail_info.mimetype)) {
const image = document.createElement("img");
const loadPromise = new Promise<void>((resolve, reject) => {
image.onload = (): void => resolve();
image.onerror = (): void => reject(new Error("Unable to load image"));
});
image.crossOrigin = "Anonymous";
image.src = contentUrl ?? "";
try {
await loadPromise;
} catch (error) {
logger.error("Unable to download attachment: ", error);
this.state = {
...this.state,
error: error as Error,
};
this.updateSnapshotFromState();
return;
}
try {
if (
(content.info as ImageInfoWithAnimationFlag | undefined)?.["org.matrix.msc4230.is_animated"] ===
false ||
(this.props.mediaEventHelper &&
(await blobIsAnimated(await this.props.mediaEventHelper.sourceBlob.value)) === false)
) {
isAnimated = false;
}
if (isAnimated) {
const thumbnail = await createThumbnail(
image,
image.width,
image.height,
content.info?.mimetype ?? "image/jpeg",
false,
);
generatedThumbnailUrl = URL.createObjectURL(thumbnail.thumbnail);
thumbUrl = generatedThumbnailUrl;
}
} catch (error) {
logger.warn("Unable to generate thumbnail for animated image: ", error);
}
}
}
if (this.isDisposed) {
if (generatedThumbnailUrl) {
URL.revokeObjectURL(generatedThumbnailUrl);
}
return;
}
this.revokeGeneratedThumbnailUrl();
this.state = {
...this.state,
contentUrl,
thumbUrl,
isAnimated,
error: null,
generatedThumbnailUrl,
};
this.updateSnapshotFromState();
}
private openImageViewer(event: MouseEvent<HTMLAnchorElement>): void {
if (event.button !== 0 || event.metaKey) {
return;
}
event.preventDefault();
if (!this.props.mediaVisible) {
this.props.setMediaVisible?.(true);
return;
}
const content = this.props.mxEvent.getContent<ImageContent>();
let httpUrl = this.state.contentUrl;
if (
this.props.mediaEventHelper?.media.isEncrypted &&
!isMimeTypeAllowed(this.props.mediaEventHelper.sourceBlob.cachedValue?.type ?? "")
) {
httpUrl = this.state.thumbUrl;
}
if (!httpUrl) {
return;
}
const params: Omit<ComponentProps<typeof ImageView>, "onFinished"> = {
src: httpUrl,
name: content.body && content.body.length > 0 ? content.body : _t("common|attachment"),
mxEvent: this.props.mxEvent,
permalinkCreator: this.props.permalinkCreator,
};
if (content.info) {
params.width = content.info.w;
params.height = content.info.h;
params.fileSize = content.info.size;
}
if (this.props.imageRef.current) {
const clientRect = this.props.imageRef.current.getBoundingClientRect();
params.thumbnailInfo = {
width: clientRect.width,
height: clientRect.height,
positionX: clientRect.x,
positionY: clientRect.y,
};
}
Modal.createDialog(ImageView, params, "mx_Dialog_lightbox", undefined, true);
}
public onLinkClick = (event: MouseEvent<HTMLAnchorElement>): void => {
this.openImageViewer(event);
};
public onHiddenButtonClick = (): void => {
this.props.setMediaVisible?.(true);
};
public onImageError = (): void => {
if (this.state.thumbUrl && this.state.thumbUrl !== this.state.contentUrl) {
this.state = {
...this.state,
thumbUrl: null,
};
this.updateSnapshotFromState();
return;
}
this.clearBlurhashTimeout();
if (this.state.imgError) {
return;
}
this.state = {
...this.state,
imgError: true,
};
MatrixClientPeg.safeGet().on(ClientEvent.Sync, this.reconnectedListener);
this.updateSnapshotFromState();
};
public onImageLoad = (): void => {
this.clearBlurhashTimeout();
let loadedImageDimensions: LoadedImageDimensions | undefined;
if (this.props.imageRef.current) {
const { naturalWidth, naturalHeight } = this.props.imageRef.current;
loadedImageDimensions = { naturalWidth, naturalHeight };
}
this.state = {
...this.state,
imgLoaded: true,
loadedImageDimensions,
placeholder: ImageBodyViewPlaceholder.NONE,
};
this.updateSnapshotFromState();
};
public setEvent(mxEvent: MatrixEvent, mediaEventHelper?: MediaEventHelper): void {
if (this.props.mxEvent === mxEvent && this.props.mediaEventHelper === mediaEventHelper) {
return;
}
const previousVisible = this.props.mediaVisible;
this.props = {
...this.props,
mxEvent,
mediaEventHelper,
};
this.resetState(mxEvent);
this.updateSnapshotFromState();
if (previousVisible) {
this.scheduleBlurhashPlaceholder();
void this.downloadImage();
}
}
public setForExport(forExport?: boolean): void {
if (this.props.forExport === forExport) {
return;
}
this.props = {
...this.props,
forExport,
};
this.updateSnapshotFromState();
}
public setMaxImageHeight(maxImageHeight?: number): void {
if (this.props.maxImageHeight === maxImageHeight) {
return;
}
this.props = {
...this.props,
maxImageHeight,
};
this.updateSnapshotFromState();
}
public setMediaVisible(mediaVisible: boolean): void {
if (this.props.mediaVisible === mediaVisible) {
return;
}
const wasVisible = this.props.mediaVisible;
this.props = {
...this.props,
mediaVisible,
};
this.updateSnapshotFromState();
if (!wasVisible && mediaVisible) {
this.scheduleBlurhashPlaceholder();
void this.downloadImage();
}
}
public setPermalinkCreator(permalinkCreator?: RoomPermalinkCreator): void {
if (this.props.permalinkCreator === permalinkCreator) {
return;
}
this.props = {
...this.props,
permalinkCreator,
};
}
public setTimelineRenderingType(timelineRenderingType: TimelineRenderingType): void {
if (this.props.timelineRenderingType === timelineRenderingType) {
return;
}
this.props = {
...this.props,
timelineRenderingType,
};
this.snapshot.merge(ImageBodyViewModel.computeSnapshot(this.props, this.state));
}
public setSetMediaVisible(setMediaVisible?: (visible: boolean) => void): void {
if (this.props.setMediaVisible === setMediaVisible) {
return;
}
this.props = {
...this.props,
setMediaVisible,
};
}
private setImageSize(imageSize: ImageSize): void {
if (this.state.imageSize === imageSize) {
return;
}
this.state = {
...this.state,
imageSize,
};
this.updateSnapshotFromState();
}
public dispose(): void {
this.clearBlurhashTimeout();
MatrixClientPeg.get()?.off(ClientEvent.Sync, this.reconnectedListener);
this.revokeGeneratedThumbnailUrl();
super.dispose();
}
}