91 lines
2.4 KiB
JavaScript
91 lines
2.4 KiB
JavaScript
require('dotenv').config();
|
|
const express = require('express');
|
|
const bodyParser = require('body-parser');
|
|
|
|
let app = express();
|
|
app.use(bodyParser.json());
|
|
app.use(bodyParser.urlencoded({ extended: true }));
|
|
|
|
// CORS
|
|
app.use((req, res, next) => {
|
|
res.set('Access-Control-Allow-Origin', '*');
|
|
next();
|
|
});
|
|
|
|
let connections = {};
|
|
|
|
app.get('/subscribe', (req, res) => {
|
|
if (!req.get("X-User-Claim")) {
|
|
res.sendStatus(401);
|
|
return;
|
|
}
|
|
|
|
try {
|
|
const user_claim = JSON.parse(req.get("X-User-Claim"));
|
|
if (!connections[user_claim.userid]) {
|
|
connections[user_claim.userid] = {};
|
|
}
|
|
|
|
console.log(`Someone subscribed to ${user_claim.userid}, ${user_claim.clientid}`);
|
|
|
|
connections[user_claim.userid][user_claim.clientid] = res;
|
|
res.writeHead(200, {
|
|
'Content-Type': 'text/event-stream',
|
|
'Cache-Control': 'no-cache'
|
|
});
|
|
|
|
// So the connection doesn't timeout and die
|
|
let timeoutID = 0;
|
|
const refresh = () => {
|
|
res.write(':\n\n');
|
|
timeoutID = setTimeout(refresh, 25000);
|
|
};
|
|
refresh();
|
|
|
|
res.on('close', () => {
|
|
connections[user_claim.userid][user_claim.clientid] = null;
|
|
clearTimeout(timeoutID);
|
|
});
|
|
} catch(e) {
|
|
res.sendStatus(401)
|
|
}
|
|
});
|
|
|
|
app.get('/user/:user/devices', (req, res) => {
|
|
if (!connections[req.params.user]) {
|
|
res.status(200).json([]);
|
|
} else {
|
|
res.status(200).json(Object.keys(connections[req.params.user]));
|
|
}
|
|
});
|
|
|
|
app.post('/user/:user/device/:device', (req, res) => {
|
|
if(!connections[req.params.user]){
|
|
res.sendStatus(404);
|
|
} else if (!connections[req.params.user][req.params.device]){
|
|
res.sendStatus(404);
|
|
} else if (!req.body.data) {
|
|
res.sendStatus(400);
|
|
} else {
|
|
try {
|
|
const user_claim = JSON.parse(req.get("X-User-Claim"));
|
|
if(!(user_claim.userid && user_claim.clientid)) {
|
|
res.sendStatus(401);
|
|
} else {
|
|
console.log('Posting:');
|
|
console.log(req.body.data);
|
|
console.log(`To ${req.params.user}, ${req.params.device}`);
|
|
req.body.data.id = `${user_claim.userid}-${user_claim.clientid}`;
|
|
connections[req.params.user][req.params.device].write(`data: ${JSON.stringify(req.body.data)}\n\n`);
|
|
res.sendStatus(200);
|
|
}
|
|
} catch(e) {
|
|
res.sendStatus(401);
|
|
}
|
|
}
|
|
});
|
|
|
|
app.listen(process.env.PORT, (err) => {
|
|
err ? console.error(err) : console.log(`Listening on port ${process.env.PORT}`);
|
|
});
|