forked from DeviaVir/zenbot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
boot.js
81 lines (67 loc) · 2.31 KB
/
boot.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
var glob = require('glob')
, path = require('path')
module.exports = function (cb) {
var zenbot = require('./')()
var c = getConfiguration()
var defaults = require('./conf-sample')
Object.keys(defaults).forEach(function (k) {
if (typeof c[k] === 'undefined') {
c[k] = defaults[k]
}
})
zenbot.set('@zenbot:conf', c)
function withMongo () {
//searches all directorys in {workingdir}/extensions/ for files called '_codemap.js'
glob('extensions/**/_codemap.js', {cwd: __dirname, absolute: true}, function (err, results) {
if (err) return cb(err)
results.forEach(function (result) {
var ext = require(result) //load the _codemap for the extension
zenbot.use(ext) //load the extension into zenbot
})
cb(null, zenbot)
})
}
var authStr = '', authMechanismStr, authMechanism;
if(c.mongo.username){
authStr = encodeURIComponent(c.mongo.username)
if(c.mongo.password) authStr += ':' + encodeURIComponent(c.mongo.password)
authStr += '@'
// authMechanism could be a conf.js parameter to support more mongodb authentication methods
authMechanism = 'DEFAULT'
}
var u = 'mongodb://' + authStr + c.mongo.host + ':' + c.mongo.port + '/' + c.mongo.db + '?' + (c.mongo.replicaSet ? '&replicaSet=' + c.mongo.replicaSet : '' ) + (authMechanism ? '&authMechanism=' + authMechanism : '' )
require('mongodb').MongoClient.connect(u, function (err, db) {
if (err) {
zenbot.set('zenbot:db.mongo', null)
console.error('WARNING: MongoDB Connection Error: ', err)
console.error('WARNING: without MongoDB some features (such as backfilling/simulation) may be disabled.')
console.error('Attempted authentication string: ' + u);
return withMongo()
}
zenbot.set('zenbot:db.mongo', db)
withMongo()
})
function getConfiguration() {
var conf = undefined
try {
var _allArgs = process.argv.slice();
var found = false
while (!found && _allArgs.length > 0) {
found = (_allArgs.shift() == '--conf');
}
if (found) {
try {
conf = require(_allArgs[0])
} catch (ee) {
conf = require('./conf')
}
} else {
conf = require('./conf')
}
}
catch (e) {
conf = {}
}
return conf
}
}