Added skeleton function for recording.

merge-requests/1/merge
knotteye 2019-09-14 21:22:07 -05:00
parent 5a637ce866
commit e128bdcdaa
3 changed files with 69 additions and 2 deletions

1
.gitignore vendored
View File

@ -1 +1,2 @@
node_modules
media

View File

@ -1,6 +1,8 @@
"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/lib/server.js").Server;
//initialize configs, eventually grab from runtime config file
var mediaconfig = {
@ -13,7 +15,18 @@ var mediaconfig = {
},
http: {
port: 8000,
allow_origin: '*'
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) {
@ -56,6 +69,7 @@ nms.on('prePublish', function (id, StreamPath, args) {
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();
@ -64,3 +78,23 @@ nms.on('prePublish', function (id, StreamPath, args) {
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;
});
//we kinda stanitize it in prePublish? :blobshrug:
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);
});

View File

@ -1,4 +1,6 @@
import NodeMediaServer = require("node-media-server");
import fs = require ("fs");
const { exec } = require('child_process');
const ircd = require("./lib/ircdjs/lib/server.js").Server;
//initialize configs, eventually grab from runtime config file
@ -12,7 +14,18 @@ const mediaconfig = {
},
http: {
port:8000,
allow_origin: '*'
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]'
}
]
}
};
@ -69,3 +82,22 @@ nms.on('prePublish', (id, StreamPath, args) => {
console.log("[NodeMediaServer] Stream key ok.");
session.publishStreamPath = "/live/amy";
});
nms.on('postPublish', (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 }, (err) => {
if (err) throw err;
});
let 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}`);
});