37 lines
1.2 KiB
JavaScript
37 lines
1.2 KiB
JavaScript
import mongodb from 'mongodb-promisified';
|
|
const MongoClient = mongodb().MongoClient;
|
|
import debugname from 'debug';
|
|
const debug = debugname('hostr:mongo');
|
|
|
|
/* eslint no-param-reassign: ["error", { "props": false }] */
|
|
export const mongo = 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 mongo;
|
|
} catch (e) {
|
|
debug(e);
|
|
}
|
|
yield next;
|
|
};
|
|
}
|