-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathindex.js
More file actions
88 lines (76 loc) · 2.34 KB
/
index.js
File metadata and controls
88 lines (76 loc) · 2.34 KB
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
var fs = require('fs');
var os = require('os');
var path = require('path');
var yaml = require('js-yaml');
var log4js = require('log4js');
var telepathy = require('telepathymq');
var asynk = require('asynk');
var sftp = require('./backups/sftp');
var ftp = require('./backups/ftp');
var folder = require('./backups/src/folder');
process.on('uncaughtException', function(err) {
console.log(err);
});
var config_folder;
switch(os.platform()) {
case 'linux':
config_folder = '/etc/backup-agent/';
break;
}
if (!config_folder) {
return console.log('Plateform ' + os.platform() + ' is not supported');
}
var config = yaml.safeLoad(fs.readFileSync(path.join(config_folder, 'config.yml')));
log4js.configure({
appenders: { agent: { type: 'file', filename: config.log } },
categories: { default: { appenders: ['agent'], level: 'error' } }
});
var logger = log4js.getLogger('agent');
var socket = new telepathy(config.name);
asynk.add(fs.readFile).args(config.key, asynk.callback)
.add(fs.readFile).args(config.cert, asynk.callback)
.add(fs.readFile).args(config.manager.cert, asynk.callback)
.parallel().asCallback(function(err, files) {
if (err) {
return logger.error(err);
}
var tlsOptions = {
requestCert: true,
rejectUnauthorized: true,
key: files.shift(),
cert: files.shift(),
ca: files
};
var connectString = 'tls://' + config.manager.ip + ':' + config.manager.port;
socket.register('manager', connectString, tlsOptions);
socket.on('register', function() {
logger.info('agent connected to manager');
});
socket.on('request', function(req, data) {
if (!data || !data.backup || !data.backup.type) {
req.reject('INVALID_REQUEST: no backup type');
return logger.error('no backup type found: ' + JSON.stringify(data));
}
var files = null;
switch(data.type) {
case 'folder':
files = folder(data.src);
}
if (!files) {
req.reject('INVALID_SOURCE_TYPE');
return logger.error('invalid source type: ' + JSON.stringify(data));
}
switch(data.backup.type) {
case 'sftp':
sftp(req, data, files, logger);
break;
case 'ftp':
ftp(req, data, files, logger);
break;
default:
req.reject('INVALID_BACKUP_TYPE');
logger.error('invalid backup type: ' + JSON.stringify(data));
break;
}
});
});