mirror/ditto
mirror/ditto
1
0
Fork 0
mirror of https://gitlab.com/soapbox-pub/ditto.git synced 2025-12-06 11:29:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Extract a poster image from videos (not efficient yet)

Browse source
This commit is contained in:
Alex Gleason 2025-02-28 09:58:14 -06:00
parent 813026e734
commit 43ec58085a
No known key found for this signature in database
GPG key ID: 7211D1F99744FBB7
1 changed files with 19 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
packages/ditto/utils/upload.ts
View file

@ -1,4 +1,4 @@
import { transcodeVideo } from '@ditto/transcode'; import { extractVideoFrame, transcodeVideo } from '@ditto/transcode';
import { HTTPException } from '@hono/hono/http-exception'; import { HTTPException } from '@hono/hono/http-exception';
import { logi } from '@soapbox/logi'; import { logi } from '@soapbox/logi';
import { crypto } from '@std/crypto'; import { crypto } from '@std/crypto';
@ -62,6 +62,8 @@ export async function uploadFile(
const m = tags.find(([key]) => key === 'm')?.[1]; const m = tags.find(([key]) => key === 'm')?.[1];
const dim = tags.find(([key]) => key === 'dim')?.[1]; const dim = tags.find(([key]) => key === 'dim')?.[1];
const size = tags.find(([key]) => key === 'size')?.[1]; const size = tags.find(([key]) => key === 'size')?.[1];
const image = tags.find(([key]) => key === 'image')?.[1];
const thumb = tags.find(([key]) => key === 'thumb')?.[1];
const blurhash = tags.find(([key]) => key === 'blurhash')?.[1]; const blurhash = tags.find(([key]) => key === 'blurhash')?.[1];
if (!x) { if (!x) {
@ -77,6 +79,22 @@ export async function uploadFile(
tags.push(['size', file.size.toString()]); tags.push(['size', file.size.toString()]);
} }
if (baseType === 'video' && (!image || !thumb)) {
const tmp = new URL('file://' + await Deno.makeTempFile());
await Deno.writeFile(tmp, file.stream());
const bytes = await extractVideoFrame(tmp);
const [[, url]] = await uploader.upload(new File([bytes], 'thumb.jpg', { type: 'image/jpeg' }), { signal });
await Deno.remove(tmp);
if (!image) {
tags.push(['image', url]);
}
if (!thumb) {
tags.push(['thumb', url]);
}
}
// If the uploader didn't already, try to get a blurhash and media dimensions. // If the uploader didn't already, try to get a blurhash and media dimensions.
// This requires `MEDIA_ANALYZE=true` to be configured because it comes with security tradeoffs. // This requires `MEDIA_ANALYZE=true` to be configured because it comes with security tradeoffs.
if (conf.mediaAnalyze && (!blurhash || !dim)) { if (conf.mediaAnalyze && (!blurhash || !dim)) {