From bcd6a38a05fb3df0af795d74c2b8d91ab0e5cd56 Mon Sep 17 00:00:00 2001
From: Feross Aboukhadijeh <feross@feross.org>
Date: Fri, 16 Sep 2016 10:14:39 -0700
Subject: [PATCH] Use capture-frame package

See: https://github.com/feross/capture-frame
Capture video screenshot from a `<video>` tag (at the current time)

Changes from our version:

- Added tests in Chrome/Firefox browsers.
- Use built-in TypeError (which is meant for bad arguments) instead of
custom IllegalArgumentError.
---
 package.json                            |  1 +
 src/renderer/lib/capture-video-frame.js | 32 -------------------------
 src/renderer/lib/errors.js              |  7 +-----
 src/renderer/lib/torrent-poster.js      |  4 ++--
 4 files changed, 4 insertions(+), 40 deletions(-)
 delete mode 100644 src/renderer/lib/capture-video-frame.js

diff --git a/package.json b/package.json
index 26bcb575..8cac942d 100644
--- a/package.json
+++ b/package.json
@@ -18,6 +18,7 @@
     "application-config": "^1.0.0",
     "auto-launch": "^4.0.1",
     "bitfield": "^1.0.2",
+    "capture-frame": "^1.0.0",
     "chromecasts": "^1.8.0",
     "create-torrent": "^3.24.5",
     "deep-equal": "^1.0.1",
diff --git a/src/renderer/lib/capture-video-frame.js b/src/renderer/lib/capture-video-frame.js
deleted file mode 100644
index 06ff8b6a..00000000
--- a/src/renderer/lib/capture-video-frame.js
+++ /dev/null
@@ -1,32 +0,0 @@
-module.exports = captureVideoFrame
-
-const {IllegalArgumentError} = require('./errors')
-
-function captureVideoFrame (video, format) {
-  if (typeof video === 'string') {
-    video = document.querySelector(video)
-  }
-
-  if (video == null || video.nodeName !== 'VIDEO') {
-    throw new IllegalArgumentError('First argument must be a <video> element or selector')
-  }
-
-  if (format == null) {
-    format = 'png'
-  }
-
-  if (format !== 'png' && format !== 'jpg' && format !== 'webp') {
-    throw new IllegalArgumentError('Second argument must be one of "png", "jpg", or "webp"')
-  }
-
-  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)
-  const data = dataUri.split(',')[1]
-
-  return new Buffer(data, 'base64')
-}
diff --git a/src/renderer/lib/errors.js b/src/renderer/lib/errors.js
index e4e4c29f..4fbaefbf 100644
--- a/src/renderer/lib/errors.js
+++ b/src/renderer/lib/errors.js
@@ -35,10 +35,6 @@ class TorrentKeyNotFoundError extends TorrentError {
 
 class InvalidTorrentError extends TorrentError {}
 
-/* Miscellaneous */
-
-class IllegalArgumentError extends ExtendableError {}
-
 module.exports = {
   CastingError,
   PlaybackError,
@@ -49,6 +45,5 @@ module.exports = {
   PlaybackTimedOutError,
   InvalidSoundNameError,
   TorrentKeyNotFoundError,
-  InvalidTorrentError,
-  IllegalArgumentError
+  InvalidTorrentError
 }
diff --git a/src/renderer/lib/torrent-poster.js b/src/renderer/lib/torrent-poster.js
index feb59fab..f05d1bb5 100644
--- a/src/renderer/lib/torrent-poster.js
+++ b/src/renderer/lib/torrent-poster.js
@@ -1,6 +1,6 @@
 module.exports = torrentPoster
 
-const captureVideoFrame = require('./capture-video-frame')
+const captureFrame = require('capture-frame')
 const path = require('path')
 
 function torrentPoster (torrent, cb) {
@@ -61,7 +61,7 @@ function torrentPosterFromVideo (file, torrent, cb) {
     function onSeeked () {
       video.removeEventListener('seeked', onSeeked)
 
-      const buf = captureVideoFrame(video)
+      const buf = captureFrame(video)
 
       // unload video element
       video.pause()