forked from mrvautin/adminMongo
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathconnections.js
83 lines (71 loc) · 2.5 KB
/
connections.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
var MongoClient = require('mongodb').MongoClient;
exports.addConnection = function (connection, app, callback){
if(!app.locals.dbConnections){
app.locals.dbConnections = {};
}
if(app.locals.dbConnections[connection.name]){
return callback(null, app.locals.dbConnections[connection.name].connection);
}
if(!connection.connOptions){
connection.connOptions = {};
}
var dbObj = {};
dbObj.connection = null;
dbObj.connect = (callback) => {
if(dbObj.connection){
callback(null, dbObj.connection);
} else {
MongoClient.connect(dbObj.connString, dbObj.connOptions, function(err, database){
dbObj.connection = database;
callback(err, database);
});
}
};
dbObj.connString = connection.connString;
dbObj.connOptions = connection.connOptions;
app.locals.dbConnections[connection.connName] = null;
app.locals.dbConnections[connection.connName] = dbObj;
callback(null, null);
};
exports.getConnection = function(request, response, connectionName) {
// Get request and check if user has required roles
var connection_list = request.app.locals.dbConnections;
var connection = connection_list[connectionName];
if(!connection){
throw new Error(req.i18n.__('Invalid connection name'));
}
if(connection.requiredRoles){
var role = request.get("x-role");
if(connection.requiredRoles.map(r => r.toLowerCase()).indexOf(role.toLowerCase()) === -1){
throw new Error(req.i18n.__('Insufficient permissions'));
}
}
return connection
}
exports.getConnections = function(request, response){
var connection_list = request.app.locals.dbConnections;
var result = {};
if(connection_list){
Object.keys(connection_list).forEach(name => {
let connection = connection_list[name];
if(connection.requiredRoles){
var role = request.get("x-role");
if(connection.requiredRoles.map(r => r.toLowerCase()).indexOf(role.toLowerCase()) === -1){
return;
}
}
result[name] = connection;
});
}
return result;
}
exports.removeConnection = function (connection, app){
if(!app.locals.dbConnections){
app.locals.dbConnections = [];
}
try{
app.locals.dbConnections[connection].connection.close();
}catch(e){}
delete app.locals.dbConnections[connection];
return;
};