forked from illuspas/Node-Media-Server
-
Notifications
You must be signed in to change notification settings - Fork 0
/
node_trans_server.js
86 lines (77 loc) · 2.71 KB
/
node_trans_server.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
//
// Created by Mingliang Chen on 18/3/9.
// illuspas[a]gmail.com
// Copyright (c) 2018 Nodemedia. All rights reserved.
//
const Logger = require('./node_core_logger');
const NodeTransSession = require('./node_trans_session');
const context = require('./node_core_ctx');
const { getFFmpegVersion, getFFmpegUrl } = require('./node_core_utils');
const fs = require('fs');
const _ = require('lodash');
const mkdirp = require('mkdirp');
class NodeTransServer {
constructor(config) {
this.config = config;
this.transSessions = new Map();
}
async run() {
try {
mkdirp.sync(this.config.http.mediaroot);
fs.accessSync(this.config.http.mediaroot, fs.constants.W_OK);
} catch (error) {
Logger.error(`Node Media Trans Server startup failed. MediaRoot:${this.config.http.mediaroot} cannot be written.`);
return;
}
try {
fs.accessSync(this.config.trans.ffmpeg, fs.constants.X_OK);
} catch (error) {
Logger.error(`Node Media Trans Server startup failed. ffmpeg:${this.config.trans.ffmpeg} cannot be executed.`);
return;
}
let version = await getFFmpegVersion(this.config.trans.ffmpeg);
if (version === '' || parseInt(version.split('.')[0]) < 4) {
Logger.error(`Node Media Trans Server startup failed. ffmpeg requires version 4.0.0 above`);
Logger.error('Download the latest ffmpeg static program:', getFFmpegUrl());
return;
}
let i = this.config.trans.tasks.length;
let apps = '';
while (i--) {
apps += this.config.trans.tasks[i].app;
apps += ' ';
}
context.nodeEvent.on('postPublish', this.onPostPublish.bind(this));
context.nodeEvent.on('donePublish', this.onDonePublish.bind(this));
Logger.log(`Node Media Trans Server started for apps: [ ${apps}] , MediaRoot: ${this.config.http.mediaroot}, ffmpeg version: ${version}`);
}
onPostPublish(id, streamPath, args) {
let regRes = /\/(.*)\/(.*)/gi.exec(streamPath);
let [app, stream] = _.slice(regRes, 1);
let i = this.config.trans.tasks.length;
while (i--) {
let conf = this.config.trans.tasks[i];
conf.port = this.config.rtmp.port;
conf.ffmpeg = this.config.trans.ffmpeg;
conf.mediaroot = this.config.http.mediaroot;
conf.streamPath = streamPath;
conf.stream = stream;
conf.args = args;
if (app === conf.app) {
let session = new NodeTransSession(conf);
this.transSessions.set(id, session);
session.on('end', () => {
this.transSessions.delete(id);
});
session.run();
}
}
}
onDonePublish(id, streamPath, args) {
let session = this.transSessions.get(id);
if (session) {
session.end();
}
}
}
module.exports = NodeTransServer;