hls.js

'use strict';

const config = require('../../config');
const errors = require('../../components/errors');

const _ = require('lodash');
const path = require('path');
const express = require('express');
const router = express.Router();

const VideoLib = require('node-video-lib');
const Movie = require('../../components/movie');

router.get(/^(.*)\/playlist\.m3u8$/, Movie.openMovie, (req, res) => {
    let streamAttributes = {
        'program-id': 1,
    };

    let duration = req.fragmentList.relativeDuration();
    if (duration > 0) {
        streamAttributes.bandwidth = Math.round(8 * req.fragmentList.size() / duration);
    }

    if (req.fragmentList.video) {
        streamAttributes.resolution = [req.fragmentList.video.width, req.fragmentList.video.height].join('x');
        let codecs = [req.fragmentList.video, req.fragmentList.audio]
            .filter(data => data !== null)
            .map(data => data.codec)
            .filter(codec => Boolean(codec))
            .join(',');
        if (codecs !== '') {
            streamAttributes.codecs = `"${codecs}"`;
        }
    }

    let streamAttributesPairs = _.toPairs(streamAttributes).map(pair => `${pair[0].toUpperCase()}=${pair[1]}`);
    let playlist = [
        '#EXTM3U',
        '#EXT-X-VERSION:3',
        `#EXT-X-STREAM-INF:${streamAttributesPairs.join(',')}`,
        path.join(req.baseUrl, req.params[0], 'chunklist.m3u8').replace(/\\/g, '/'),
        '',
    ];
    res.header('Content-Type', 'application/x-mpegURL');
    res.send(playlist.join('\n'));
});

router.get(/^(.*)\/chunklist\.m3u8$/, Movie.openMovie, (req, res) => {
    let keyUrl = path.join(req.baseUrl, req.params[0], 'drm.key').replace(/\\/g, '/');
    let playlist = [
        '#EXTM3U',
        '#EXT-X-VERSION:3',
        `#EXT-X-TARGETDURATION:${req.fragmentList.fragmentDuration}`,
        '#EXT-X-MEDIA-SEQUENCE:1',
    ];
    if (config.drmEnabled) {
        playlist.push(`#EXT-X-KEY:METHOD=AES-128,URI="${keyUrl}",IV=0x${Movie.movieIv(req.params[0]).toString('hex')}`);
    }
    for (let i = 0, l = req.fragmentList.count(); i < l; i++) {
        playlist.push(`#EXTINF:${req.fragmentList.get(i).relativeDuration().toFixed(1)},`);
        playlist.push(path.join(req.baseUrl, req.params[0], `media-${i + 1}.ts`).replace(/\\/g, '/'));
    }
    playlist.push('#EXT-X-ENDLIST');
    playlist.push('');
    res.header('Content-Type', 'application/x-mpegURL');
    res.send(playlist.join('\n'));
});

router.get(/^(.*)\/media-(\d+)\.ts$/, Movie.openMovie, (req, res) => {
    let index = parseInt(req.params[1], 10);
    if (req.fragmentList.count() < index) {
        throw new errors.NotFoundError('Chunk Not Found');
    }
    let fragment = req.fragmentList.get(index - 1);
    let sampleBuffers = VideoLib.FragmentReader.readSamples(fragment, req.file);
    let buffer = VideoLib.HLSPacketizer.packetize(fragment, sampleBuffers);
    res.header('Content-Type', 'video/MP2T');
    if (config.drmEnabled) {
        res.send(Movie.encryptChunk(req.params[0], buffer));
    } else {
        res.send(buffer);
    }
});

router.get(/^(.*)\/drm\.key$/, (req, res) => {
    if (config.drmEnabled) {
        res.send(Movie.movieKey(req.params[0]));
    } else {
        throw new errors.NotFoundError('Key Not Found');
    }
});

module.exports = router;