2017-02-11 12:27:03 +00:00
|
|
|
const bluebird = require('bluebird');
|
|
|
|
const fs = require('fs');
|
|
|
|
const mime = require('mime-types');
|
|
|
|
const strongDataUri = require('strong-data-uri');
|
|
|
|
|
2017-08-14 12:25:27 +00:00
|
|
|
const util = require('./util');
|
|
|
|
|
2017-02-11 12:27:03 +00:00
|
|
|
bluebird.promisifyAll(fs);
|
|
|
|
|
2017-08-14 12:37:57 +00:00
|
|
|
function getFrameFromVideo(video, format) {
|
2017-02-11 12:27:03 +00:00
|
|
|
const canvas = document.createElement('canvas');
|
|
|
|
canvas.width = video.videoWidth;
|
|
|
|
canvas.height = video.videoHeight;
|
|
|
|
|
|
|
|
canvas.getContext('2d').drawImage(video, 0, 0);
|
|
|
|
|
|
|
|
const dataUri = canvas.toDataURL(`image/${format}`);
|
|
|
|
|
|
|
|
return strongDataUri.decode(dataUri);
|
|
|
|
}
|
|
|
|
|
2017-08-14 12:37:57 +00:00
|
|
|
function captureFrame(customOutDir, filePath, video, currentTime, captureFormat) {
|
|
|
|
const buf = getFrameFromVideo(video, captureFormat);
|
2017-08-14 12:25:27 +00:00
|
|
|
|
2017-02-11 12:27:03 +00:00
|
|
|
const ext = mime.extension(buf.mimetype);
|
2017-08-14 12:25:27 +00:00
|
|
|
const time = util.formatDuration(currentTime);
|
|
|
|
|
|
|
|
const outPath = util.getOutPath(customOutDir, filePath, `${time}.${ext}`);
|
2017-02-11 12:27:03 +00:00
|
|
|
return fs.writeFileAsync(outPath, buf);
|
|
|
|
}
|
|
|
|
|
|
|
|
module.exports = captureFrame;
|