diff options
Diffstat (limited to 'background.js')
| -rw-r--r-- | background.js | 224 |
1 files changed, 129 insertions, 95 deletions
diff --git a/background.js b/background.js index 253e81e..5895f19 100644 --- a/background.js +++ b/background.js @@ -4,110 +4,144 @@ let recorder = null; let filename = null; let ws; let liveSteam = false; +let ffmpegServer; +let doDownload = true; + chrome.runtime.onConnect.addListener(port => { - port.onMessage.addListener(msg => { - console.log(msg); - switch (msg.type) { - case 'SET_EXPORT_PATH': - filename = msg.filename - break - case 'FFMPEG_SERVER': - let ffmpegServer = msg.ffmpegServer - ws = new WebSocket(ffmpegServer); - liveSteam = true; - break - case 'REC_STOP': - recorder.stop() - break - case 'REC_START': - if(liveSteam){ - recorder.start(1000); - }else{ - recorder.start(); - } - break - case 'REC_CLIENT_PLAY': - if(recorder){ - return - } - const tab = port.sender.tab - tab.url = msg.data.url - chrome.desktopCapture.chooseDesktopMedia(['tab', 'audio'], streamId => { - // Get the stream - navigator.webkitGetUserMedia({ - audio: { - mandatory: { - chromeMediaSource: 'system' - } - }, - video: { - mandatory: { - chromeMediaSource: 'desktop', - chromeMediaSourceId: streamId, - minWidth: 1280, - maxWidth: 1280, - minHeight: 720, - maxHeight: 720, - minFrameRate: 60, - } - } - }, stream => { - var chunks = []; - recorder = new MediaRecorder(stream, { - videoBitsPerSecond: 2500000, - ignoreMutedMedia: true, - mimeType: 'video/webm' - }); - recorder.ondataavailable = function (event) { - if (event.data.size > 0) { - chunks.push(event.data); - if(liveSteam){ - ws.send(event.data); - } + port.onMessage.addListener(msg => { + console.log(msg); + switch (msg.type) { + + case 'SET_EXPORT_PATH': + filename = msg.filename + break + + case 'FFMPEG_SERVER': + ffmpegServer = msg.ffmpegServer + startWebsock(); + break + + case 'REC_STOP': + doDownload = true; + recorder.stop() + break + + case 'REC_START': + if (liveSteam) { + recorder.start(1000); + } else { + recorder.start(); } - }; + break - recorder.onstop = function () { - if(liveSteam){ - ws.close(); + case 'REC_CLIENT_PLAY': + if (recorder) { + return } + const tab = port.sender.tab + tab.url = msg.data.url + chrome.desktopCapture.chooseDesktopMedia(['tab', 'audio'], streamId => { + // Get the stream + navigator.webkitGetUserMedia({ + audio: { + mandatory: { + chromeMediaSource: 'system' + } + }, + video: { + mandatory: { + chromeMediaSource: 'desktop', + chromeMediaSourceId: streamId, + minWidth: 1280, + maxWidth: 1280, + minHeight: 720, + maxHeight: 720, + minFrameRate: 60, + } + } + }, stream => { + var chunks = []; + recorder = new MediaRecorder(stream, { + videoBitsPerSecond: 2500000, + ignoreMutedMedia: true, + mimeType: 'video/webm;codecs=h264' + }); + recorder.ondataavailable = function (event) { + if (event.data.size > 0) { + chunks.push(event.data); + if (liveSteam) { + ws.send(event.data); + } + } + }; - var superBuffer = new Blob(chunks, { - type: 'video/webm' - }); + recorder.onstop = function () { + if (liveSteam) { + ws.close(); + } - var url = URL.createObjectURL(superBuffer); - // var a = document.createElement('a'); - // document.body.appendChild(a); - // a.style = 'display: none'; - // a.href = url; - // a.download = 'test.webm'; - // a.click(); + if(!doDownload){ + chunks = []; + return; + } - chrome.downloads.download({ - url: url, - filename: filename - }, ()=>{ - }); - } + var superBuffer = new Blob(chunks, { + type: 'video/webm' + }); - }, error => console.log('Unable to get user media', error)) - }) - break - default: - console.log('Unrecognized message', msg) - } - }) + var url = URL.createObjectURL(superBuffer); + // var a = document.createElement('a'); + // document.body.appendChild(a); + // a.style = 'display: none'; + // a.href = url; + // a.download = 'test.webm'; + // a.click(); - chrome.downloads.onChanged.addListener(function(delta) { - if (!delta.state ||(delta.state.current != 'complete')) { - return; - } - try{ - port.postMessage({downloadComplete: true}) - } - catch(e){} - }); + chrome.downloads.download({ + url: url, + filename: filename + }, () => { + }); + } + + }, error => console.log('Unable to get user media', error)) + }) + break + default: + console.log('Unrecognized message', msg) + } + }) + + chrome.downloads.onChanged.addListener(function (delta) { + if (!delta.state || (delta.state.current != 'complete')) { + return; + } + try { + port.postMessage({ downloadComplete: true }) + } + catch (e) { } + }); }) + +function startWebsock() { + + ws = new WebSocket(ffmpegServer); + liveSteam = true; + + ws.onmessage = function (e) { + console.log(e.data); + + if (e.data == "ffmpegClosed") { + + doDownload = false; + recorder.stop(); + + setTimeout(function () { + startWebsock(); + recorder.start(1000); + }, 500) + } + } +} |