-
Notifications
You must be signed in to change notification settings - Fork 13
/
Copy pathindex.js
66 lines (49 loc) · 1.51 KB
/
index.js
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
55
56
57
58
59
60
61
62
63
64
65
66
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);
const Blinkt = require('node-blinkt');
const leds = new Blinkt();
leds.setup();
leds.setAllPixels(255, 255, 255, 1);
leds.sendUpdate();
setTimeout(updateLeds, 1000);
app.get('/', (req, res) => res.sendFile(__dirname + '/index.html'));
app.ws('/video-stream', (ws, req) => {
console.log('Client connected');
updateLeds();
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');
updateLeds();
videoStream.removeAllListeners('data');
});
});
function updateLeds() {
const clientCount = wss.getWss().clients.size;
console.log('Updating leds to show', clientCount, 'clients');
leds.clearAll();
for (let i = 0; i < clientCount; i++) {
let colors = [0, 0, 0];
colors[Math.floor(i / 8)] = 255;
leds.setPixel(i % 8, ...colors, 1);
}
leds.sendUpdate();
}
app.use(function (err, req, res, next) {
console.error(err);
next(err);
})
app.listen(80, () => console.log('Server started on 80'));