2021-08-27 21:05:46 +03:00

88 lines
2.2 KiB
TypeScript

import { Api as GramJs } from '../../../lib/gramjs';
import { strippedPhotoToJpg } from '../../../lib/gramjs/Utils';
import {
ApiPhoto, ApiPhotoSize, ApiThumbnail,
} from '../../types';
import { bytesToDataUri } from './helpers';
import { pathBytesToSvg } from './pathBytesToSvg';
const DEFAULT_THUMB_SIZE = { w: 100, h: 100 };
export function buildApiThumbnailFromStripped(
sizes?: GramJs.TypePhotoSize[], mimeType?: string,
): ApiThumbnail | undefined {
if (!sizes || !sizes.length) {
return undefined;
}
const thumb = sizes.find((s: any): s is GramJs.PhotoStrippedSize => s instanceof GramJs.PhotoStrippedSize);
if (!thumb) {
return undefined;
}
const realSizes = sizes.filter((s): s is GramJs.PhotoSize => s instanceof GramJs.PhotoSize);
const { w, h } = realSizes.length ? realSizes[realSizes.length - 1] : DEFAULT_THUMB_SIZE;
const { bytes } = thumb;
const dataUri = bytesToDataUri(
!mimeType || mimeType === 'image/jpeg' ? strippedPhotoToJpg(bytes) : bytes,
undefined,
mimeType,
);
return {
dataUri,
width: w,
height: h,
};
}
export function buildApiThumbnailFromCached(photoSize: GramJs.PhotoCachedSize): ApiThumbnail | undefined {
const { w, h, bytes } = photoSize;
const dataUri = bytesToDataUri(bytes, undefined, 'image/webp');
return {
dataUri,
width: w,
height: h,
};
}
export function buildApiThumbnailFromPath(
photoSize: GramJs.PhotoPathSize,
sizeAttribute: GramJs.DocumentAttributeImageSize,
): ApiThumbnail | undefined {
const { w, h } = sizeAttribute;
const dataUri = `data:image/svg+xml;utf8,${pathBytesToSvg(photoSize.bytes, w, h)}`;
return {
dataUri,
width: w,
height: h,
};
}
export function buildApiPhoto(photo: GramJs.Photo): ApiPhoto {
const sizes = photo.sizes
.filter((s: any): s is GramJs.PhotoSize => {
return s instanceof GramJs.PhotoSize || s instanceof GramJs.PhotoSizeProgressive;
})
.map(buildApiPhotoSize);
return {
id: String(photo.id),
thumbnail: buildApiThumbnailFromStripped(photo.sizes),
sizes,
};
}
export function buildApiPhotoSize(photoSize: GramJs.PhotoSize): ApiPhotoSize {
const { w, h, type } = photoSize;
return {
width: w,
height: h,
type: type as ('m' | 'x' | 'y'),
};
}