mirror of
https://github.com/MisskeyIO/misskey
synced 2024-12-23 19:18:44 +09:00
46 lines
1.1 KiB
TypeScript
46 lines
1.1 KiB
TypeScript
import { Inject, Injectable } from '@nestjs/common';
|
|
import FFmpeg from 'fluent-ffmpeg';
|
|
import { DI } from '@/di-symbols.js';
|
|
import type { Config } from '@/config.js';
|
|
import { ImageProcessingService } from '@/core/ImageProcessingService.js';
|
|
import type { IImage } from '@/core/ImageProcessingService.js';
|
|
import { createTempDir } from '@/misc/create-temp.js';
|
|
import { bindThis } from '@/decorators.js';
|
|
|
|
@Injectable()
|
|
export class VideoProcessingService {
|
|
constructor(
|
|
@Inject(DI.config)
|
|
private config: Config,
|
|
|
|
private imageProcessingService: ImageProcessingService,
|
|
) {
|
|
}
|
|
|
|
@bindThis
|
|
public async generateVideoThumbnail(source: string): Promise<IImage> {
|
|
const [dir, cleanup] = await createTempDir();
|
|
|
|
try {
|
|
await new Promise((res, rej) => {
|
|
FFmpeg({
|
|
source,
|
|
})
|
|
.on('end', res)
|
|
.on('error', rej)
|
|
.screenshot({
|
|
folder: dir,
|
|
filename: 'out.png', // must have .png extension
|
|
count: 1,
|
|
timestamps: ['5%'],
|
|
});
|
|
});
|
|
|
|
return await this.imageProcessingService.convertToWebp(`${dir}/out.png`, 498, 280);
|
|
} finally {
|
|
cleanup();
|
|
}
|
|
}
|
|
}
|
|
|