Tweak recording example

Author: markandrus

README.md

@@ -15,8 +15,8 @@ This project presents a few example applications using node-webrtc.
 - [video-compositing](examples/video-compositing): uses RTCVideoSink,
   [node-canvas](https://github.com/Automattic/node-canvas), and RTCVideoSource
   to draw spinning text on top of an incoming video.
-- [Stream recording](examples/stream-record) using ffmpeg and RTCVideoSink
-- Broadcast example with one ["broadcaster"](examples/broadcaster) forwarding to many ["viewers"](examples/viewers)
+- [record-audio-video-stream](examples/record-audio-video-stream) using [FFmpeg](https://www.ffmpeg.org) and RTCVideoSink.
+- Broadcast example with one [broadcaster](examples/broadcaster) forwarding to many [viewers](examples/viewers).
 
 Usage
 -----

examples/stream-record/client.js renamed to examples/record-audio-video-stream/client.js

@@ -31,7 +31,7 @@ async function beforeAnswer(peerConnection) {
   };
 }
 
-createExample('stream-record', description, { beforeAnswer });
+createExample('record-audio-video-stream', description, { beforeAnswer });
 
 const videos = document.createElement('div');
 videos.className = 'grid';

examples/stream-record/server.js renamed to examples/record-audio-video-stream/server.js

@@ -1,12 +1,12 @@
 'use strict';
 
 const { PassThrough } = require('stream')
+const fs = require('fs')
 
 const { RTCAudioSink, RTCVideoSink } = require('wrtc').nonstandard;
 
 const ffmpeg = require('fluent-ffmpeg')
 const { StreamInput } = require('fluent-ffmpeg-multistream')
-const fs = require('fs')
 
 const VIDEO_OUTPUT_SIZE = '320x240'
 const VIDEO_OUTPUT_FILE = './recording.mp4'
@@ -22,9 +22,9 @@ function beforeOffer(peerConnection) {
 
   const streams = [];
 
-  videoSink.addEventListener('frame', function({ frame: { width, height, data }}){
+  videoSink.addEventListener('frame', ({ frame: { width, height, data }}) => {
     const size = width + 'x' + height;
-    if(!streams[0] || (streams[0] && streams[0].size !== size)) {
+    if (!streams[0] || (streams[0] && streams[0].size !== size)) {
       UID++;
 
       const stream = {
@@ -34,24 +34,24 @@ function beforeOffer(peerConnection) {
         audio: new PassThrough()
       };
 
-      const onAudioData = function({ samples: { buffer } }) {
-        if(!stream.end) {
+      const onAudioData = ({ samples: { buffer } }) => {
+        if (!stream.end) {
           stream.audio.push(Buffer.from(buffer));
         }
       };
 
-      audioSink.addEventListener('data', onAudioData)
+      audioSink.addEventListener('data', onAudioData);
 
-      stream.audio.on('end', ()=>{
-        audioSink.removeEventListener('data', onAudioData)
-      })
+      stream.audio.on('end', () => {
+        audioSink.removeEventListener('data', onAudioData);
+      });
 
-      streams.unshift(stream)
+      streams.unshift(stream);
 
       streams.forEach(item=>{
-        if(item !== stream && !item.end) {
+        if (item !== stream && !item.end) {
           item.end = true;
-          if(item.audio) {
+          if (item.audio) {
             item.audio.end();
           }
           item.video.end();
@@ -80,34 +80,34 @@ function beforeOffer(peerConnection) {
           console.log('Stop recording >> ', stream.recordPath)
         })
         .size(VIDEO_OUTPUT_SIZE)
-        .output(stream.recordPath)
+        .output(stream.recordPath);
 
-        stream.proc.run()
+        stream.proc.run();
     }
 
     streams[0].video.push(Buffer.from(data));
-  })
+  });
 
   const { close } = peerConnection;
   peerConnection.close = function() {
     audioSink.stop();
     videoSink.stop();
 
     streams.forEach(({ audio, video, end, proc, recordPath })=>{
-      if(!end) {
-        if(audio) {
+      if (!end) {
+        if (audio) {
           audio.end();
         }
         video.end();
       }
-    })
+    });
 
     let totalEnd = 0;
     const timer = setInterval(()=>{
       streams.forEach(stream=>{
-        if(stream.recordEnd) {
+        if (stream.recordEnd) {
           totalEnd++;
-          if(totalEnd  === streams.length) {
+          if (totalEnd === streams.length) {
             clearTimeout(timer);
 
             const mergeProc = ffmpeg()
@@ -123,15 +123,15 @@ function beforeOffer(peerConnection) {
 
             streams.forEach(({ recordPath })=>{
               mergeProc.addInput(recordPath)
-            })
+            });
 
             mergeProc
               .output(VIDEO_OUTPUT_FILE)
               .run();
           }
         }
-      })
-    },1000)
+      });
+    }, 1000)
 
     return close.apply(this, arguments);
   }

html/index.html

@@ -47,6 +47,6 @@ <h1>node-webrtc examples</h1>
   <li><a href="/sine-wave/index.html">sine-wave
   <li><a href="/sine-wave-stereo/index.html">sine-wave-stereo
   <li><a href="/video-compositing/index.html">video-compositing
-  <li><a href="/stream-record/index.html">record-audio-video-stream
+  <li><a href="/record-audio-video-stream/index.html">record-audio-video-stream
   <li><a href="/broadcaster/index.html">broadcaster</a> &amp; <a href="/viewer/index.html">viewer
 <script src="index.js"></script>