You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
99 lines
3.5 KiB
99 lines
3.5 KiB
"use strict";
|
|
exports.__esModule = true;
|
|
var NodeMediaServer = require("node-media-server");
|
|
var fs = require("fs");
|
|
var exec = require('child_process').exec;
|
|
var ircd = require("./lib/IRCDjs-six/lib/server.js").Server;
|
|
//initialize configs, eventually grab from runtime config file
|
|
var mediaconfig = {
|
|
rtmp: {
|
|
port: 1935,
|
|
chunk_size: 60000,
|
|
gop_cache: true,
|
|
ping: 30,
|
|
ping_timeout: 60
|
|
},
|
|
http: {
|
|
port: 8000,
|
|
allow_origin: '*',
|
|
mediaroot: './media'
|
|
},
|
|
trans: {
|
|
ffmpeg: '/usr/bin/ffmpeg',
|
|
tasks: [
|
|
{
|
|
app: 'live',
|
|
hls: 'true',
|
|
hlsFlags: '[hls_time=2:hls_list_size=3:hls_flags=delete_segments]'
|
|
}
|
|
]
|
|
}
|
|
};
|
|
function streamAuth(path) {
|
|
if (path.split("/").length > 3) {
|
|
console.log("[NodeMediaServer] Malformed URL, closing connection.");
|
|
return false;
|
|
}
|
|
var app = path.split("/")[1];
|
|
var key = path.split("/")[2];
|
|
console.log("[NodeMediaServer] Authenticating stream with credentials: ", "app=" + app + " key=" + key);
|
|
if (app !== "stream") {
|
|
console.log("[NodeMediaServer] Invalid app name, closing connection.");
|
|
return false;
|
|
}
|
|
console.log("[NodeMediaServer] App name ok.");
|
|
if (key !== "temp") {
|
|
console.log("[NodeMediaServer] Invalid stream key, closing connection.");
|
|
return false;
|
|
}
|
|
console.log("[NodeMediaServer] Stream key ok.");
|
|
return true;
|
|
}
|
|
var nms = new NodeMediaServer(mediaconfig);
|
|
nms.run();
|
|
ircd.boot();
|
|
nms.on('prePublish', function (id, StreamPath, args) {
|
|
console.log("[NodeMediaServer] Prepublish Hook for stream id=", id);
|
|
var session = nms.getSession(id);
|
|
if (StreamPath.split("/").length > 3) {
|
|
console.log("[NodeMediaServer] Malformed URL, closing connection.");
|
|
session.reject();
|
|
return false;
|
|
}
|
|
var app = StreamPath.split("/")[1];
|
|
var key = StreamPath.split("/")[2];
|
|
console.log("[NodeMediaServer] Authenticating stream with credentials: ", "app=" + app + " key=" + key);
|
|
if (app !== "stream") {
|
|
console.log("[NodeMediaServer] Invalid app name, closing connection.");
|
|
session.reject();
|
|
return false;
|
|
}
|
|
console.log("[NodeMediaServer] App name ok.");
|
|
//TODO: Hook up to DB and redirect from query
|
|
if (key !== "temp") {
|
|
console.log("[NodeMediaServer] Invalid stream key, closing connection.");
|
|
session.reject();
|
|
return false;
|
|
}
|
|
console.log("[NodeMediaServer] Stream key ok.");
|
|
session.publishStreamPath = "/live/amy";
|
|
});
|
|
nms.on('postPublish', function (id, StreamPath, args) {
|
|
console.log('[NodeMediaServer] Checking record flag for ', "id=" + id + " StreamPath=" + StreamPath);
|
|
//Hook up to postgres DB.
|
|
if (true) {
|
|
console.log('[NodeMediaServer] Initiating recording for ', "id=" + id + " StreamPath=" + StreamPath);
|
|
fs.mkdir('./media' + StreamPath, { recursive: true }, function (err) {
|
|
if (err)
|
|
throw err;
|
|
});
|
|
var subprocess = exec('ffmpeg -i rtmp://127.0.0.1' + StreamPath + ' -vcodec copy -acodec copy ./media' + StreamPath + '/$(date +%d%b%Y-%H%M).mp4', {
|
|
detached: true,
|
|
stdio: 'inherit'
|
|
});
|
|
subprocess.unref();
|
|
//spawn an ffmpeg process to record the stream, then detach it completely
|
|
return true;
|
|
}
|
|
console.log('[NodeMediaServer] Skipping recording for ', "id=" + id + " StreamPath=" + StreamPath);
|
|
});
|
|
|