r6d-web/testStream.js at master · future-thinking/r6d-web · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
process.on('uncaughtException', function(err) {
    console.log('Caught exception: ' + err);
    console.log(err.stack);
});

const express = require('express');
const raspividStream = require('raspivid-stream');

const app = express();
const wss = require('express-ws')(app);

app.get('/', (req, res) => res.send("" +
    "<html>\n" +
    "    <body>\n" +
    "        <script type=\"text/javascript\" src=\"https://rawgit.com/131/h264-live-player/master/vendor/dist/http-live-player.js\"></script>\n" +
    "        <script>\n" +
    "            var canvas = document.createElement(\"canvas\");\n" +
    "            document.body.appendChild(canvas);\n" +
    "\n" +
    "            var wsavc = new WSAvcPlayer(canvas, \"webgl\");\n" +
    "\n" +
    "            var protocol = window.location.protocol === \"https:\" ? \"wss:\" : \"ws:\"\n" +
    "            wsavc.connect(protocol + '//' + window.location.host + '/video-stream');\n" +
    "        </script>\n" +
    "    </body>\n" +
    "</html>"));

app.ws('/video-stream', (ws, req) => {
    console.log('Client connected');

    ws.send(JSON.stringify({
        action: 'init',
        width: '960',
        height: '540'
    }));

    var videoStream = raspividStream({ rotation: 180 });

    videoStream.on('data', (data) => {
        ws.send(data, { binary: true }, (error) => { if (error) console.error(error); });
    });

    ws.on('close', () => {
        console.log('Client left');
        videoStream.removeAllListeners('data');
    });
});

app.use(function (err, req, res, next) {
    console.error(err);
    next(err);
})

app.listen(80, () => console.log('Server started on 80'));