muffin/dist/utils/db.js

The 21st century way of building websites

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.fs = exports.goose = exports.rope = undefined;

var _mongoose = require('mongoose');

var _mongoose2 = _interopRequireDefault(_mongoose);

var _gridfsStream = require('gridfs-stream');

var _gridfsStream2 = _interopRequireDefault(_gridfsStream);

var _ = require('./');

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, _.log)('Couldn\'t connect to DB: ' + info.message);
  process.exit(1);
});

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

exports.rope = connection;
exports.goose = _mongoose2.default;

// Tell gridfs where to find the files

_gridfsStream2.default.mongo = _mongoose2.default.mongo;
const gridConnection = (0, _gridfsStream2.default)(connection.db);

exports.fs = gridConnection;