38 lines
1.2 KiB
JavaScript
38 lines
1.2 KiB
JavaScript
|
import mongodb from 'mongodb-promisified';
|
||
|
|
const MongoClient = mongodb().MongoClient;
|
||
|
|
import debugname from 'debug';
|
||
|
|
const debug = debugname('hostr:mongo');
|
||
|
|
|
||
|
|
const uristring = process.env.MONGO_URL || process.env.MONGOLAB_URI || 'mongodb://localhost:27017/hostr';
|
||
|
|
|
||
|
|
let configuredClient = new Promise(function (resolve, reject) {
|
||
|
|
debug('Connecting to Mongodb');
|
||
|
|
return MongoClient.connect(uristring).then((client) => {
|
||
|
|
debug('Successfully connected to Mongodb');
|
||
|
|
client.Users = client.collection('users');
|
||
|
|
client.Files = client.collection('files');
|
||
|
|
client.Transactions = client.collection('transactions');
|
||
|
|
client.Logins = client.collection('logins');
|
||
|
|
client.Remember = client.collection('remember');
|
||
|
|
client.Reset = client.collection('reset');
|
||
|
|
client.Remember.ensureIndex({'created': 1}, {expireAfterSeconds: 2592000});
|
||
|
|
client.Files.ensureIndex({'owner': 1, 'status': 1, 'time_added': -1});
|
||
|
|
return resolve(client);
|
||
|
|
}).catch((e) => {
|
||
|
|
reject(e)
|
||
|
|
});
|
||
|
|
}).catch((e) => {
|
||
|
|
debug(e);
|
||
|
|
});
|
||
|
|
|
||
|
|
export default function() {
|
||
|
|
return function* (next) {
|
||
|
|
try {
|
||
|
|
this.db = yield configuredClient;
|
||
|
|
} catch (e) {
|
||
|
|
debug(e);
|
||
|
|
}
|
||
|
|
yield next;
|
||
|
|
}
|
||
|
|
}
|