Scan Plex for ${config.PLEX_PLAYLIST_IDENTIFIER} playlists.
+Any changes made to config.yml won't take effect until pseudotv-plex is restarted.
+Author: Dan Ferguson
+ ${createScript(counter)} + + ` + return str.split(' ').join('') +} + +function createConfigDetails() { + var str = `
+ Host: ${config.HOST}
+ Port: ${config.PORT}
+
| # | +Name | +Icon | +Shuffle | +
|---|---|---|---|
| ${channels[i].channel} | +${channels[i].name} | +${channels[i].icon ? ` |
+ ${channels[i].shuffle} | +
Restart Plex client apps and refresh Plex web sessions to view changes..
+ ${config.EPG_UPDATE === 0 ? '' : `Next EPG Refresh: (hh:mm:ss)
`} + ` + return str.split(' ').join('') +} + +var createScript = function (counter) { + var str = ` + ` + return str +} \ No newline at end of file diff --git a/src/vlc.js b/src/vlc.js index 39705ae..3b9f731 100644 --- a/src/vlc.js +++ b/src/vlc.js @@ -5,77 +5,143 @@ const config = require('config-yml') const xmltv = require('./xmltv') -module.exports = { router: vlcRouter } +module.exports = vlcRouter -function vlcRouter() { +function vlcRouter(client) { var router = Router() - var streams = [] - + var inUse = false router.get('/video', (req, res) => { - var programs = xmltv.readXMLPrograms() - if (!req.query.channel) - return res.status(422).send("No channel queried") - - req.query.channel = req.query.channel.split('?')[0] - var streamIndex = -1 - for (var i = 0; i < streams.length; i++) { - if (streams[i].channel === req.query.channel) { - streamIndex = i - break - } + if (inUse) + return res.status(409).send("Error: Another user is currently viewing a stream. One one active stream is allowed.") + inUse = true + var channel = req.query.channel + if (!channel) { + inUse = false + res.status(400).send("Error: No channel queried") + return } + channel = channel.split('?')[0] + + res.writeHead(200, { + 'Content-Type': 'application/octet-stream', + 'Content-disposition': 'attachment; filename=video.ts' + }) + startStreaming(channel, res) + }) - if (streamIndex != -1) { - streams[streamIndex].viewers++ - request('http://' + config.HOST + ':' + streams[streamIndex].port + '/').on('error', (err) => {/* ignore errors */}).pipe(res) - } else { - var args = [] - var startPos = 0 - var programIndex = 0 - for (var i = 0; i < programs.length; i++) { - var date = new Date() - if (programs[i].start <= date && programs[i].stop >= date && programs[i].channel == req.query.channel) { - var dif = date.getTime() - programs[i].start.getTime() - startPos = dif / 1000 + return router + + function startStreaming(channel, res) { + var programs = xmltv.readXMLPrograms() + var startPos = -1 + var programIndex = -1 + var channelExists = false + for (var i = 0; i < programs.length; i++) { + var date = new Date() + if (programs[i].channel == channel) { + channelExists = true + if (programs[i].start <= date && programs[i].stop >= date) { + startPos = date.getTime() - programs[i].start.getTime() programIndex = i break } } - for (var i = programIndex; i < programs.length; i++) - if (programs[i].channel == req.query.channel) - args.push(programs[i].video) - - if (args.length == 0) - return res.status(422).send("Channel not found") - - var vlcPort = config.PORT + streams.length + 1 - - args.push("--start-time=" + startPos) - if (programs.optimized) - args.push(`--sout=#http{mux=ts,dst=:${vlcPort}/}`) - else - args.push(`--sout=#${config.VLC_TRANSCODE_SETTINGS}:http{mux=ts,dst=:${vlcPort}/}`) - if (config.VLC_HIDDEN) - args.push("--intf=dummy") - - - var vlcExe = spawn(config.VLC_EXECUTABLE, args) - var stream = { vlcExe: vlcExe, channel: req.query.channel, viewers: 1, port: vlcPort } - streamIndex = streams.length - streams.push(stream) - setTimeout(() => { - request(`http://${config.HOST}:${vlcPort}/`).on('error', function (err) {/* ignore errors */}).pipe(res) - }, config.VLC_STARTUP_DELAY) } - - res.on('close', () => { - streams[streamIndex].viewers-- - if (streams[streamIndex].viewers == 0) { - streams[streamIndex].vlcExe.kill() - streams.splice(streamIndex, 1) + // End session if any errors. + if (!channelExists) { + inUse = false + res.status(403).send(`Error: Channel doesn't exist. Channel: ${channel}`) + return + } + if (programIndex === -1) { + inUse = false + res.status(403).send(`Error: No scheduled programming available. Channel: ${channel}`) + return + } + if (startPos === -1) { + inUse = false + res.status(403).send(`Error: How the fuck did you get here?. Channel: ${channel}`) + return + } + // Query plex for current program + client.Get(programs[programIndex].key, (result) => { + if (result.err) { + inUse = false + res.status(403).send(`Error: Failed to fetch program info from Plex`) + return } - }) - }) + var fetchedItem = result.result.MediaContainer.Metadata[0] + // Transcode it + client.Transcode(fetchedItem, startPos, (result) => { + if (result.err) { + inUse = false + res.status(403).send(`Error: Failed to add program to playQueue`) + return + } + // Update server timeline every 10 seconds + var stream = result.result + var msElapsed = startPos + var timelineInterval = setInterval(() => { + stream.update(msElapsed) + msElapsed += 10000 + }, 10000) + var args = [ + stream.url, + `--start-time=${(startPos + config.VLC_OPTIONS.DELAY) / 1000}`, + `--sout=#http{mux=ts,dst=:${config.VLC_OPTIONS.PORT}/}` + ] + if (config.VLC_OPTIONS.HIDDEN) + args.push("--intf=dummy") + // Fire up VLC + var vlc = spawn(config.VLC_OPTIONS.PATH, args) + // Wait for VLC to open before we request anything. + setTimeout(() => { + request(`http://${config.HOST}:${config.VLC_OPTIONS.PORT}/`) + .on('error', (err) => { + vlc.kill() + if (err.code === 'ECONNRESET') { + var end = programs[programIndex].stop + var now = new Date() + var timeUntilDone = end.valueOf() - now.valueOf() + timeUntilDone = timeUntilDone > 0 ? timeUntilDone : 0 + setTimeout(() => { + res.removeListener('close', httpEnd) + startStreaming(channel, res) + }, timeUntilDone) + } + }) + .on('data', (chunk) => { + res.write(chunk) + }) + .on("complete", () => { + vlc.kill() + var end = programs[programIndex].stop + var now = new Date() + var timeUntilDone = end.valueOf() - now.valueOf() + timeUntilDone = timeUntilDone > 0 ? timeUntilDone : 0 + setTimeout(() => { + res.removeListener('close', httpEnd) + startStreaming(channel, res) + }, timeUntilDone) + }) + }, config.VLC_OPTIONS.DELAY) + + // When the http session ends: kill vlc + var httpEnd = function () { + vlc.kill() + inUse = false + } + res.on('close', httpEnd) - return router + vlc.on('close', (code) => { + clearInterval(timelineInterval) + stream.stop() + if (code !== 0 && !res.headersSent) { + res.status(400).send(`Error: VLC closed unexpectedly`) + } + }) + }) + }) + + } } \ No newline at end of file diff --git a/src/xmltv.js b/src/xmltv.js index 9402c60..b4209c5 100644 --- a/src/xmltv.js +++ b/src/xmltv.js @@ -3,8 +3,15 @@ const XMLReader = require('xml-reader') const fs = require('fs') const config = require('config-yml') +module.exports = { + WriteXMLTV: WriteXMLTV, + UpdateXMLTV: UpdateXMLTV, + readXMLPrograms: readXMLPrograms, + readXMLChannels: readXMLChannels +} + function readXMLPrograms() { - var data = fs.readFileSync(config.XMLTV_OUTPUT) + var data = fs.readFileSync(config.XMLTV_FILE) var xmltv = XMLReader.parseSync(data.toString()) var programs = [] var tv = xmltv.children @@ -15,16 +22,42 @@ function readXMLPrograms() { channel: tv[i].attributes.channel, start: createDate(tv[i].attributes.start), stop: createDate(tv[i].attributes.stop), - video: tv[i].attributes.video, - optimized: tv[i].attributes.optimized == "true" ? true : false + key: tv[i].attributes['plex-key'] } programs.push(program) } return programs } +function readXMLChannels() { + var data = fs.readFileSync(config.XMLTV_FILE) + var xmltv = XMLReader.parseSync(data.toString()) + var channels = [] + var tv = xmltv.children + for (var i = 0; i < tv.length; i++) { + if (tv[i].name == 'programme') + continue; + //console.log(tv[i]) + var channel = { + channel: tv[i].attributes.id, + shuffle: tv[i].attributes.shuffle + } + for (var y = 0; y < tv[i].children.length; y++) + { + if (tv[i].children[y].name === 'display-name') { + channel.name = tv[i].children[y].children[0].value + } + if (tv[i].children[y].name === 'icon') { + channel.icon = tv[i].children[y].attributes.src + } + } + channels.push(channel) + } + return channels +} + function WriteXMLTV(channels, cb) { - var xw = new XMLWriter(true); + var xw = new XMLWriter(true) var time = new Date() // Build XMLTV and M3U files xw.startDocument() @@ -32,57 +65,45 @@ function WriteXMLTV(channels, cb) { xw.startElement('tv') xw.writeAttribute('generator-info-name', 'psuedotv-plex') writeChannels(xw, channels) - // Programmes + // For each channel for (var i = 0; i < channels.length; i++) { var future = new Date() future.setHours(time.getHours() + config.EPG_CACHE) var tempDate = new Date(time.valueOf()) - while (tempDate < future && channels[i].playlist.length > 0) { - for (var y = 0; y < channels[i].playlist.length; y++) { + // Loop items until EPG_CACHE is satisfied, starting time of first show is NOW. + while (tempDate < future && channels[i].items.length > 0) { + for (var y = 0; y < channels[i].items.length && tempDate < future; y++) { var stopDate = new Date(tempDate.valueOf()) - stopDate.setMilliseconds(stopDate.getMilliseconds() + channels[i].playlist[y].duration) - var plexURL = channels[i].playlist[y].Media[0].Part[0].key - var optimizedForStreaming = false - for (var z = 0; z < channels[i].playlist[y].Media.length; z++) { - var part = channels[i].playlist[y].Media[z].Part[0] - if (typeof part.optimizedForStreaming !== 'undefined' && part.optimizedForStreaming) { - plexURL = part.key - optimizedForStreaming = part.optimizedForStreaming - break; - } - } - plexURL = `http://${config.PLEX_OPTIONS.hostname}:${config.PLEX_OPTIONS.port}${plexURL}?X-Plex-Token=${config.PLEX_OPTIONS.token}` + stopDate.setMilliseconds(stopDate.getMilliseconds() + channels[i].items[y].duration) var program = { - info: channels[i].playlist[y], + info: channels[i].items[y], channel: channels[i].channel, start: new Date(tempDate.valueOf()), - stop: stopDate, - plexURL: plexURL, - optimizedForStreaming: optimizedForStreaming.toString() + stop: stopDate } writeProgramme(xw, program) - tempDate.setMilliseconds(tempDate.getMilliseconds() + channels[i].playlist[y].duration) + tempDate.setMilliseconds(tempDate.getMilliseconds() + channels[i].items[y].duration) } } } // End TV xw.endElement() xw.endDocument() - fs.writeFileSync(config.XMLTV_OUTPUT, xw.toString()) + fs.writeFileSync(config.XMLTV_FILE, xw.toString()) if (typeof cb == 'function') cb() } function UpdateXMLTV(channels, cb) { var xw = new XMLWriter(true) - var data = fs.readFileSync(config.XMLTV_OUTPUT) + var data = fs.readFileSync(config.XMLTV_FILE) var xml = XMLReader.parseSync(data.toString()) var time = new Date() xw.startDocument() xw.startElement('tv') xw.writeAttribute('generator-info-name', 'psuedotv-plex') writeChannels(xw, channels) - // Programmes + // Foreach channel for (var i = 0; i < channels.length; i++) { // get non-expired programmes for channel var validPrograms = [] @@ -94,97 +115,67 @@ function UpdateXMLTV(channels, cb) { } } // If Channel doesnt exists.. - if (validPrograms.length == 0) { - // write out programs from plex + if (validPrograms.length === 0) { var future = new Date() future.setHours(time.getHours() + config.EPG_CACHE) var tempDate = new Date(time.valueOf()) + // Loop items until EPG_CACHE is satisfied, starting time of first show is NOW. while (tempDate < future) { - for (var y = 0; y < channels[i].playlist.length; y++) { // foreach item in playlist + for (var y = 0; y < channels[i].items.length && tempDate < future; y++) { // foreach item in playlist var stopDate = new Date(tempDate.valueOf()) - stopDate.setMilliseconds(stopDate.getMilliseconds() + channels[i].playlist[y].duration) - var plexURL = channels[i].playlist[y].Media[0].Part[0].key - var optimizedForStreaming = false - for (var z = 0; z < channels[i].playlist[y].Media.length; z++) { // get optimed video if there is one - var part = channels[i].playlist[y].Media[z].Part[0] - if (typeof part.optimizedForStreaming !== 'undefined' && part.optimizedForStreaming) { - plexURL = part.key - optimizedForStreaming = part.optimizedForStreaming - break; - } - } - plexURL = `http://${config.PLEX_OPTIONS.hostname}:${config.PLEX_OPTIONS.port}${plexURL}?X-Plex-Token=${config.PLEX_OPTIONS.token}` + stopDate.setMilliseconds(stopDate.getMilliseconds() + channels[i].items[y].duration) var program = { - info: channels[i].playlist[y], + info: channels[i].items[y], channel: channels[i].channel, start: new Date(tempDate.valueOf()), - stop: stopDate, - plexURL: plexURL, - optimizedForStreaming: optimizedForStreaming.toString() + stop: stopDate } writeProgramme(xw, program) - tempDate.setMilliseconds(tempDate.getMilliseconds() + channels[i].playlist[y].duration) + tempDate.setMilliseconds(tempDate.getMilliseconds() + channels[i].items[y].duration) } } - } else { - var playlistStartIndex = 0 - var isFirstItemFound = false + } else { // Otherwise the channel already exists.. + var playlistStartIndex = -1 var startingDate = new Date(time.valueOf()) var endDate = new Date(time.valueOf()) endDate.setHours(endDate.getHours() + config.EPG_CACHE) // rewrite first valid xml programmes, if it still exists in the plex playlist.. - for (var z = 0; z < channels[i].playlist.length; z++) { - if (channels[i].playlist[z].guid == validPrograms[0].attributes.guid) { - - isFirstItemFound = true + for (var z = 0; z < channels[i].items.length; z++) { + if (channels[i].items[z].key == validPrograms[0].attributes['plex-key']) { playlistStartIndex = z var program = { channel: validPrograms[0].attributes.channel, start: createDate(validPrograms[0].attributes.start), stop: createDate(validPrograms[0].attributes.stop), - plexURL: validPrograms[0].attributes.video, - optimizedForStreaming: validPrograms[0].attributes.optimized, - info: channels[i].playlist[z] + info: channels[i].items[z] } startingDate = new Date(program.stop.valueOf()) writeProgramme(xw, program) break; } } - if (isFirstItemFound) { + if (playlistStartIndex !== -1) { playlistStartIndex++ - if (channels[i].playlist.length == playlistStartIndex) + if (playlistStartIndex === channels[i].items.length) playlistStartIndex = 0 + } else { + playlistStartIndex = 0 } - // write programs from plex, starting at the live playlist index. while (startingDate < endDate) { - for (var y = playlistStartIndex; y < channels[i].playlist.length; y++) { + for (var y = playlistStartIndex; y < channels[i].items.length && startingDate < endDate; y++) { var stopDate = new Date(startingDate.valueOf()) - stopDate.setMilliseconds(stopDate.getMilliseconds() + channels[i].playlist[y].duration) - var plexURL = channels[i].playlist[y].Media[0].Part[0].key - var optimizedForStreaming = false - for (var z = 0; z < channels[i].playlist[y].Media.length; z++) { - var part = channels[i].playlist[y].Media[z].Part[0] - if (typeof part.optimizedForStreaming !== 'undefined' && part.optimizedForStreaming) { - plexURL = part.key - optimizedForStreaming = part.optimizedForStreaming - break; - } - } - plexURL = `http://${config.PLEX_OPTIONS.hostname}:${config.PLEX_OPTIONS.port}${plexURL}?X-Plex-Token=${config.PLEX_OPTIONS.token}` + stopDate.setMilliseconds(stopDate.getMilliseconds() + channels[i].items[y].duration) var program = { - info: channels[i].playlist[y], + info: channels[i].items[y], channel: channels[i].channel, start: new Date(startingDate.valueOf()), - stop: stopDate, - plexURL: plexURL, - optimizedForStreaming: optimizedForStreaming.toString() + stop: stopDate } writeProgramme(xw, program) - startingDate.setMilliseconds(startingDate.getMilliseconds() + channels[i].playlist[y].duration) - playlistStartIndex = 0 + startingDate.setMilliseconds(startingDate.getMilliseconds() + channels[i].items[y].duration) } + playlistStartIndex = 0 } } } @@ -192,23 +183,17 @@ function UpdateXMLTV(channels, cb) { xw.endElement() // End Doc xw.endDocument() - fs.writeFileSync(config.XMLTV_OUTPUT, xw.toString()) + fs.writeFileSync(config.XMLTV_FILE, xw.toString()) if (typeof cb == 'function') cb() } - - -module.exports = { - WriteXMLTV: WriteXMLTV, - UpdateXMLTV: UpdateXMLTV, - readXMLPrograms: readXMLPrograms -} function writeChannels(xw, channels) { // Channels for (var i = 0; i < channels.length; i++) { xw.startElement('channel') xw.writeAttribute('id', channels[i].channel) + xw.writeAttribute('shuffle', channels[i].shuffle ? 'yes': 'no') xw.startElement('display-name') xw.writeAttribute('lang', 'en') xw.text(channels[i].name) @@ -221,16 +206,14 @@ function writeChannels(xw, channels) { xw.endElement() } } + function writeProgramme(xw, program) { // Programme xw.startElement('programme') xw.writeAttribute('start', createXMLTVDate(program.start)) xw.writeAttribute('stop', createXMLTVDate(program.stop)) xw.writeAttribute('channel', program.channel) - // For VLC to handle... - xw.writeAttribute('video', program.plexURL) - xw.writeAttribute('optimized', program.optimizedForStreaming) - xw.writeAttribute('guid', program.info.guid) + xw.writeAttribute('plex-key', program.info.key) // Used to link this programme to Plex.. // Title xw.startElement('title') xw.writeAttribute('lang', 'en') @@ -255,9 +238,9 @@ function writeProgramme(xw, program) { // Icon xw.startElement('icon') if (program.info.type == 'movie') - xw.writeAttribute('src', 'http://' + config.PLEX_OPTIONS.hostname + ':' + config.PLEX_OPTIONS.port + program.info.thumb + '?X-Plex-Token=' + config.PLEX_OPTIONS.token) + xw.writeAttribute('src', 'http://' + config.PLEX_OPTIONS.hostname + ':' + config.PLEX_OPTIONS.port + program.info.thumb) else if (program.info.type == 'episode') - xw.writeAttribute('src', 'http://' + config.PLEX_OPTIONS.hostname + ':' + config.PLEX_OPTIONS.port + program.info.parentThumb + '?X-Plex-Token=' + config.PLEX_OPTIONS.token) + xw.writeAttribute('src', 'http://' + config.PLEX_OPTIONS.hostname + ':' + config.PLEX_OPTIONS.port + program.info.parentThumb) xw.endElement() // Desc xw.startElement('desc')