-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathserver.js
125 lines (111 loc) · 3.1 KB
/
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
var express = require('express');
var app = express();
var http = require("http");
var server = http.createServer(app);
var WebSocketServer = require("ws").Server;
var wss = new WebSocketServer({server: server});
var path = require('path');
var serveIndex = require('serve-index');
var net = require('net');
var bodyParser = require('body-parser');
var multer = require('multer');
var upload = multer(); // for parsing multipart/form-data
var basicAuth = require('basic-auth-connect');
var winston = require('winston');
var papertrail = require('winston-papertrail').Papertrail;
var port = process.env.PORT || 8080;
var _ = require('lodash');
var moment = require('moment');
var timeFormat = 'M/D/YYYY, h:mm:ss:SSS';
var tcpSocket = new net.Socket();
var webSocket;
var clientConnected = false;
var logPath = path.join(__dirname, '/admin/log/');
var fileSettings = {
name: 'fileLogger',
filename: path.join(logPath, 'test.log'),
json: false,
formatter: function (options) {
return moment().format(timeFormat) + '; ' + options.message;
}
};
var ngrokSet = false;
var auth = basicAuth(function (user, pass) {
return (user == "allen" && pass == "thesis");
}, 'Admin Login');
server.listen(port);
console.log("http server listening on port %d", port);
app.use(bodyParser.json());
app.use(bodyParser.urlencoded({extended: true}));
//Serve files
app.use(express.static('public'));
app.use('/admin/log', auth, express.static('admin/log'), serveIndex(logPath, {icons: true, view: 'details'}));
app.all('/admin', auth);
app.get('/admin', function (req, res) {
res.sendFile('/public/admin.html', {root: __dirname});
});
//Process admin panel commands
app.post('/admin', upload.array(), function (req, res) {
var fps = req.body.fps;
var logname = req.body.logname;
var port = req.body.port;
var end = req.body.end;
if (fps) {
if (clientConnected) {
webSocket.send(fps);
winston.log('info', fps);
if (ngrokSet) {
tcpSocket.write(moment().format(timeFormat) + ';' + fps);
}
res.end();
}
else {
res.send('fail');
}
}
else if (logname) {
winston
.remove('fileLogger')
.log('info', logname)
.add(winston.transports.File, _.extend(fileSettings, {
filename: path.join(logPath, logname)
}));
res.end();
}
else if (port) {
tcpSocket.connect(port, '0.tcp.ngrok.io')
.on('connect', function () {
ngrokSet = true;
res.end();
})
.on('error', function () {
res.send('fail');
//Needed to try connecting again.
tcpSocket = new net.Socket();
});
}
else if (end) {
tcpSocket.destroy();
winston.log('info', 'END');
res.end();
}
});
winston
.add(papertrail, {
host: 'logs3.papertrailapp.com',
port: '10066'
})
.add(winston.transports.File, fileSettings)
.remove(winston.transports.Console);
wss.on("connection", function (ws) {
if (!clientConnected) {
clientConnected = true;
webSocket = ws;
}
ws.on("message", function (message) {
winston.log('info', message);
});
ws.on("close", function () {
clientConnected = false;
});
});