refactored code
This commit is contained in:
parent
5b8ae127b5
commit
9ee35f725e
44
src/services/ffmpegService.ts
Normal file
44
src/services/ffmpegService.ts
Normal file
|
@ -0,0 +1,44 @@
|
||||||
|
import { createFFmpeg, FFmpeg } from '@ffmpeg/ffmpeg';
|
||||||
|
import { getUint8ArrayView } from './upload/readFileService';
|
||||||
|
|
||||||
|
class FFmpegService {
|
||||||
|
private ffmpeg: FFmpeg = null;
|
||||||
|
async init() {
|
||||||
|
try {
|
||||||
|
this.ffmpeg = createFFmpeg({
|
||||||
|
log: true,
|
||||||
|
corePath: '/js/ffmpeg-core.js',
|
||||||
|
});
|
||||||
|
console.log('Loading ffmpeg-core.js');
|
||||||
|
await this.ffmpeg.load();
|
||||||
|
} catch (e) {
|
||||||
|
throw Error('ffmpeg load failed');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async generateThumbnail(file: File) {
|
||||||
|
if (!this.ffmpeg) {
|
||||||
|
await this.init();
|
||||||
|
}
|
||||||
|
this.ffmpeg.FS(
|
||||||
|
'writeFile',
|
||||||
|
file.name,
|
||||||
|
await getUint8ArrayView(new FileReader(), file)
|
||||||
|
);
|
||||||
|
|
||||||
|
await this.ffmpeg.run(
|
||||||
|
'-i',
|
||||||
|
file.name,
|
||||||
|
'-ss',
|
||||||
|
'00:00:01.000',
|
||||||
|
'-vframes',
|
||||||
|
'1',
|
||||||
|
'thumb.png'
|
||||||
|
);
|
||||||
|
console.log('Complete transcoding');
|
||||||
|
const thumb = this.ffmpeg.FS('readFile', 'thumb.png');
|
||||||
|
return thumb;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export default new FFmpegService();
|
|
@ -4,7 +4,7 @@ import { convertHEIC2JPEG } from 'utils/file';
|
||||||
import { logError } from 'utils/sentry';
|
import { logError } from 'utils/sentry';
|
||||||
import { BLACK_THUMBNAIL_BASE64 } from '../../../public/images/black-thumbnail-b64';
|
import { BLACK_THUMBNAIL_BASE64 } from '../../../public/images/black-thumbnail-b64';
|
||||||
import { getUint8ArrayView } from './readFileService';
|
import { getUint8ArrayView } from './readFileService';
|
||||||
import { createFFmpeg, FFmpeg } from '@ffmpeg/ffmpeg';
|
import FFmpegService from 'services/ffmpegService';
|
||||||
|
|
||||||
const THUMBNAIL_HEIGHT = 720;
|
const THUMBNAIL_HEIGHT = 720;
|
||||||
const MAX_ATTEMPTS = 3;
|
const MAX_ATTEMPTS = 3;
|
||||||
|
@ -27,7 +27,7 @@ export async function generateThumbnail(
|
||||||
canvas = await generateImageThumbnail(file, isHEIC);
|
canvas = await generateImageThumbnail(file, isHEIC);
|
||||||
} else {
|
} else {
|
||||||
try {
|
try {
|
||||||
const thumb = await ffmpegThumbnailGenerator(file);
|
const thumb = await FFmpegService.generateThumbnail(file);
|
||||||
return { thumbnail: thumb, hasStaticThumbnail: false };
|
return { thumbnail: thumb, hasStaticThumbnail: false };
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
canvas = await generateVideoThumbnail(file);
|
canvas = await generateVideoThumbnail(file);
|
||||||
|
@ -189,36 +189,3 @@ async function thumbnailCanvasToBlob(canvas: HTMLCanvasElement) {
|
||||||
|
|
||||||
return thumbnailBlob;
|
return thumbnailBlob;
|
||||||
}
|
}
|
||||||
|
|
||||||
async function ffmpegThumbnailGenerator(file: File) {
|
|
||||||
let ffmpeg: FFmpeg = null;
|
|
||||||
try {
|
|
||||||
ffmpeg = createFFmpeg({
|
|
||||||
log: true,
|
|
||||||
corePath: '/js/ffmpeg-core.js',
|
|
||||||
});
|
|
||||||
console.log('Loading ffmpeg-core.js');
|
|
||||||
await ffmpeg.load();
|
|
||||||
} catch (e) {
|
|
||||||
throw Error('ffmpeg load failed');
|
|
||||||
}
|
|
||||||
console.log('Start transcoding');
|
|
||||||
ffmpeg.FS(
|
|
||||||
'writeFile',
|
|
||||||
file.name,
|
|
||||||
await getUint8ArrayView(new FileReader(), file)
|
|
||||||
);
|
|
||||||
|
|
||||||
await ffmpeg.run(
|
|
||||||
'-i',
|
|
||||||
file.name,
|
|
||||||
'-ss',
|
|
||||||
'00:00:01.000',
|
|
||||||
'-vframes',
|
|
||||||
'1',
|
|
||||||
'thumb.png'
|
|
||||||
);
|
|
||||||
console.log('Complete transcoding');
|
|
||||||
const thumb = ffmpeg.FS('readFile', 'thumb.png');
|
|
||||||
return thumb;
|
|
||||||
}
|
|
||||||
|
|
Loading…
Reference in a new issue