rainbownapkin/fore.st
1
0
Fork 0
Code Issues 13 Pull requests Projects Releases Packages Wiki Activity Actions

Fix ustream

Browse source
This commit is contained in:
Calvin Montgomery 2018-01-14 15:02:15 -08:00
parent fec1372e4e
commit d706bf63b1
6 changed files with 11 additions and 57 deletions
Download patch file Download diff file Expand all files Collapse all files

54
src/web/routes/ustream_bypass.js
View file

@ -1,54 +0,0 @@
import { HTTPError } from '../../errors';
import * as HTTPStatus from '../httpstatus';
import https from 'https';
import Promise from 'bluebird';
const INJECTION = `
<head>
<style type="text/css">
#PlayerOne.ui-enabled #ScreensContainer.shown {
display: none !important;
}
</style>`;
function ustreamSucks(channelId) {
const url = `https://www.ustream.tv/embed/${channelId}`;
return new Promise((resolve, reject) => {
const req = https.get(url, (res) => {
if (res.statusCode !== HTTPStatus.OK) {
res.resume();
return reject(new HTTPError(res.statusMessage, { status: res.statusCode }));
}
res.setEncoding('utf8');
let buffer = '';
res.on('data', data => buffer += data);
res.on('end', () => {
buffer = buffer.replace(/<head>/, INJECTION);
resolve(buffer);
});
});
req.on('error', error => {
reject(error);
});
});
}
export default function initialize(app) {
app.get('/ustream_bypass/embed/:channelId', (req, res) => {
if (!req.params.channelId || !/^\d+$/.test(req.params.channelId)) {
throw new HTTPError(`Invalid channel ID "${req.params.channelId}".`, {
status: HTTPStatus.BAD_REQUEST
});
}
ustreamSucks(req.params.channelId).then(buffer => {
res.type('html').send(buffer);
}).catch(HTTPError, error => {
res.status(error.status).send(error.message);
}).catch(error => {
res.status(HTTPStatus.INTERNAL_SERVER_ERROR).send('Internal Server Error');
});
});
};

1
src/web/webserver.js
View file

@ -208,7 +208,6 @@ module.exports = {
require('./acp').init(app, ioConfig);
require('../google2vtt').attach(app);
require('./routes/google_drive_userscript')(app);
require('./routes/ustream_bypass')(app);
if (process.env.UNFINISHED_FEATURE) {
const { AccountDataRoute } = require('./routes/account/data');