36 lines
1.1 KiB
JavaScript
36 lines
1.1 KiB
JavaScript
import mongodb from 'mongodb-promisified';
|
|
const MongoClient = mongodb().MongoClient;
|
|
import debugname from 'debug';
|
|
const debug = debugname('hostr:mongo');
|
|
|
|
const configuredClient = new Promise((resolve, reject) => {
|
|
debug('Connecting to Mongodb');
|
|
return MongoClient.connect(process.env.MONGO_URL).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});
|
|
client.ObjectId = client.objectId = mongodb().ObjectId;
|
|
return resolve(client);
|
|
}).catch((e) => {
|
|
reject(e);
|
|
});
|
|
}).catch((e) => {
|
|
debug(e);
|
|
});
|
|
|
|
export default function() {
|
|
return function* dbMiddleware(next) {
|
|
try {
|
|
this.db = yield configuredClient;
|
|
} catch (e) {
|
|
debug(e);
|
|
}
|
|
yield next;
|
|
};
|
|
}
|