engagementgamelab/CivicSeed

'use strict'

var rootDir = require('app-root-path')
var config = require(rootDir + '/app/config')
var serverHelpers = require(rootDir + '/server/utils/server-helpers')

var mongoose = require('mongoose')
var winston = require('winston')

var mongooseConnected = (mongoose.connection.readyState === 1 || mongoose.connection.readyState === 2)

module.exports = {

  db: null,

  connectMongoose: function (app, callback) {
    winston.info('Starting database services ...'.yellow)

    if (!mongooseConnected) {
      this.db = mongoose.createConnection(config.get('MONGO_URL'))
      this.db.on('error', function (err) {
        // err is not a string; have to force it
        winston.error('MongoDB: '.blue + ' %s '.white.bgRed, err.toString())
      })
      this.db.once('open', function () {
        winston.info('MongoDB: '.blue + 'Connected to '.green + app.get('env').yellow.inverse)
        if (typeof callback === 'function') {
          callback({ mongooseDb: this.db })
        }
      })
    }
  },

  useModel: function (modelFilename) {
    try {
      var model = require(rootDir + '/models/' + modelFilename)
    } catch (err) {
      winston.error('CS: '.blue + 'Error preloading model '.red + modelFilename.yellow + ': '.red + err.code.red)
      return
    }

    winston.info('CS: '.blue + 'Preloading model for SS RPC: '.green + model.name.yellow.underline)
    return this.db.model(model.name, new mongoose.Schema(model.schema, { collection: model.collection }))
  },

  getAndSetNetworkIp: function (callback) {
    serverHelpers.getNetworkIPs(function (err, ips) {
      if (err || !ips.length) {
        config.set('IP', 'localhost')
        winston.warn('CS:'.blue + ' Could not find network IP. Defaulting to \'localhost.\''.red)
      } else {
        config.set('IP', ips[0])
        winston.info('CS:'.blue + ' Running on network IP: ' + ips[0].yellow)
      }

      if (typeof callback === 'function') {
        callback()
      }
    })
  }

}