TelegramPWA/src/hooks/useMedia.ts

39 lines
1.1 KiB
TypeScript

import { useEffect } from '../lib/teact/teact';
import { ApiMediaFormat } from '../api/types';
import { selectIsSynced } from '../global/selectors';
import * as mediaLoader from '../util/mediaLoader';
import useSelector from './data/useSelector';
import useForceUpdate from './useForceUpdate';
const useMedia = (
mediaHash: string | false | undefined,
noLoad = false,
mediaFormat = ApiMediaFormat.BlobUrl,
delay?: number | false,
) => {
const mediaData = mediaHash ? mediaLoader.getFromMemory(mediaHash) : undefined;
const forceUpdate = useForceUpdate();
const isSynced = useSelector(selectIsSynced);
useEffect(() => {
if (!noLoad && mediaHash && !mediaData) {
const startedAt = Date.now();
mediaLoader.fetch(mediaHash, mediaFormat).then(() => {
const spentTime = Date.now() - startedAt;
if (!delay || spentTime >= delay) {
forceUpdate();
} else {
setTimeout(forceUpdate, delay - spentTime);
}
});
}
}, [noLoad, mediaHash, mediaData, mediaFormat, delay, isSynced]);
return mediaData;
};
export default useMedia;