muffin-cli/dist/lib/db.js

The command line utility for building sites using muffin

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});

var _mongoose = require('mongoose');

var _mongoose2 = _interopRequireDefault(_mongoose);

var _utils = require('./utils');

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

const dbHost = process.env.DB_HOST || 'localhost';
const dbName = process.env.DB_NAME || 'muffin';

// Connect to DB using the credentials set in the ".env" file
_mongoose2.default.connect('mongodb://' + dbHost + '/' + dbName, {
  user: process.env.DB_USER,
  pass: process.env.DB_PASSWORD
});

const connection = _mongoose2.default.connection;

connection.on('error', function (info) {
  if (info.message.includes('ECONNREFUSED')) {
    info.message = 'Please make sure it\'s running and accessible!';
  }

  (0, _utils.log)('Couldn\'t connect to DB: ' + info.message);
  process.exit(1);
});

process.on('SIGINT', () => connection.close(() => {
  process.exit(0);
}));

exports.default = connection;