forked from arkime/arkime
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsource.file.js
84 lines (75 loc) · 2.61 KB
/
source.file.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
/******************************************************************************/
/*
*
* Copyright 2012-2016 AOL Inc. All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this Software except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
'use strict';
var fs = require('fs');
var util = require('util');
var simpleSource = require('./simpleSource.js');
// ----------------------------------------------------------------------------
function FileSource (api, section) {
FileSource.super_.call(this, api, section);
this.file = api.getConfig(section, 'file');
this.cacheTimeout = -1;
if (this.file === undefined) {
console.log(this.section, '- ERROR not loading since no file specified in config file');
return;
}
if (!fs.existsSync(this.file)) {
console.log(this.section, '- ERROR not loading since', this.file, "doesn't exist");
return;
}
if (!this.initSimple()) {
return;
}
setImmediate(this.load.bind(this));
// Watch file for changes, combine multiple changes into one, on move restart watch after a pause
this.watchTimeout = null;
let watchCb = (event, filename) => {
clearTimeout(this.watchTimeout);
if (event === 'rename') {
this.watch.close();
setTimeout(() => {
this.load();
this.watch = fs.watch(this.file, watchCb);
}, 500);
} else {
this.watchTimeout = setTimeout(() => {
this.watchTimeout = null;
this.load();
}, 2000);
}
};
this.watch = fs.watch(this.file, watchCb);
}
util.inherits(FileSource, simpleSource);
// ----------------------------------------------------------------------------
FileSource.prototype.simpleSourceLoad = function (setFunc, cb) {
fs.readFile(this.file, (err, body) => {
if (err) {
return cb(err);
}
this.parse(body, setFunc, cb);
});
};
// ----------------------------------------------------------------------------
exports.initSource = function (api) {
var sections = api.getConfigSections().filter((e) => { return e.match(/^file:/); });
sections.forEach((section) => {
return new FileSource(api, section);
});
};
// ----------------------------------------------------------------------------