satyr/src/server.ts

import NodeMediaServer = require('node-media-server');
import fs = require('fs');
const { exec } = require('child_process');

//initialize configs, eventually grab from runtime config file
const 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: string){
	if (path.split("/").length > 3){
		console.log("[NodeMediaServer] Malformed URL, closing connection.");
		return false;
	}
	let app: string = path.split("/")[1];
	let key: string = 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;
}


const nms = new NodeMediaServer(mediaconfig);

nms.run();

nms.on('prePublish', (id, StreamPath, args) => {
	console.log("[NodeMediaServer] Prepublish Hook for stream id=",id);
	let session = nms.getSession(id);
	if (StreamPath.split("/").length > 3){
		console.log("[NodeMediaServer] Malformed URL, closing connection.");
		session.reject();
		return false;
	}
	let app: string = StreamPath.split("/")[1];
	let key: string = 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', (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}`);
});