hostr/lib/mongo.js

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';

const configuredClient = new Promise((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});
    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;
  };
}