allanfreitas / adonis-mongorito Goto Github PK

'use strict'

const fs = require('fs')

const Env = use('Env')

let ssl    = Env.get('MONGO_SSL', false);
let crt    = Env.get('MONGO_CRT', '');
let key    = Env.get('MONGO_KEY', '');
let ca     = Env.get('MONGO_CA', '');

if (ssl) {
  if (crt && fs.existsSync(crt)) {
    crt = fs.readFileSync(crt)
  }

  if (key && fs.existsSync(key)) {
    key = fs.readFileSync(key)
  }

  if (ca && fs.existsSync(ca)) {
    ca = [fs.readFileSync(ca)]
  }
}

module.exports = {
  host:   Env.get('MONGO_HOST'),
  port:   Env.get('MONGO_PORT'),
  user:   Env.get('MONGO_USER', ''),
  pass:   Env.get('MONGO_PASS', ''),
  db:     Env.get('MONGO_DATABASE'),
  verify: Env.get('MONGO_VERIFY_HOSTNAME', false),
  ssl:    ssl,
  crt:    crt,
  key:    key,
  ca:     ca
}

class MongoritoProvider extends ServiceProvider {

  * register () {
    this.app.singleton('Adonis/Addons/MongoritoModel', function (app) {
      let Config = app.use('Adonis/Src/Config')

      const mongoHost = Config.get('mongo.host')
      const mongoPort = Config.get('mongo.port')
      const mongoUser = Config.get('mongo.user')
      const mongoPass = Config.get('mongo.pass')

      let mongoDb = Config.get('mongo.db')
      let mongoVerify = Config.get('mongo.verify')

      let mongoSSL = Config.get('mongo.ssl')
      let mongoKey = Config.get('mongo.key')
      let mongoCrt = Config.get('mongo.crt')
      let mongoCa = Config.get('mongo.ca')

      let options = { server: {} }

      if (mongoSSL) {
        mongoDb = `${mongoDb}?ssl=true`

        options.server.sslValidate = true;
        options.server.sslKey      = mongoKey;
        options.server.sslCert     = mongoCrt;
        options.server.sslCA       = mongoCa;
      }

      options.server.checkServerIdentity = mongoVerify;

      const connectUri = `${mongoHost}:${mongoPort}/${mongoDb}`
      const connectionString = (mongoUser !== '' || mongoPass !== '') ? `${mongoUser}:${mongoPass}@${connectUri}` : connectUri

      logger.verbose('connection string %s', connectionString)
      Mongorito.connect(connectionString, options)

      return MongoritoModel
    })
  }
}

const token = yield request.auth.generate(user)