const { getHtml } = require('./html');
const { Lbryio } = require('lbryinc/dist/bundle.es.js');
const { generateStreamUrl, CONTINENT_COOKIE } = require('../../ui/util/lbrytv');
const fetch = require('node-fetch');
const Router = require('@koa/router');
// So any code from 'lbry-redux'/'lbryinc' that uses `fetch` can be run on the server
global.fetch = fetch;
const router = new Router();
function getStreamUrl(ctx) {
const { claimName, claimId } = ctx.params;
// hack to get around how we managing the continent cookie
// defaulting to "NA" becasue saved-passwords.js assumes it's in the browser and won't work properly
// changes need to be made to that to better work with the server
const streamingContinentCookie = ctx.cookies.get(CONTINENT_COOKIE) || 'NA';
const streamUrl = generateStreamUrl(claimName, claimId, undefined, streamingContinentCookie);
return streamUrl;
}
function getSupportedCDN(continent) {
switch (continent) {
case 'NA':
case 'AS':
case 'EU':
return continent;
default:
return 'NA';
}
}
async function fetchContinentCookie(ip) {
return Lbryio.call('locale', 'get', { ip }, 'post').then(result => {
const userContinent = getSupportedCDN(result.continent);
return userContinent;
});
}
router.get(`/$/download/:claimName/:claimId`, async ctx => {
const streamUrl = getStreamUrl(ctx);
const downloadUrl = `${streamUrl}?download=1`;
ctx.redirect(downloadUrl);
});
router.get(`/$/stream/:claimName/:claimId`, async ctx => {
const streamUrl = getStreamUrl(ctx);
ctx.redirect(streamUrl);
});
router.get('*', async ctx => {
const hasContinentCookie = ctx.cookies.get(CONTINENT_COOKIE);
const ip = ctx.ip;
if (!hasContinentCookie) {
try {
const continentValue = await fetchContinentCookie(ip);
ctx.cookies.set(CONTINENT_COOKIE, continentValue, { httpOnly: false });
} catch (e) {}
}
const html = await getHtml(ctx);
ctx.body = html;
});
module.exports = router;