import _ from 'lodash' import knex from 'knex' import bookshelf from 'bookshelf' import config from '../config' import log from '../log' let host = config.get('knex:connection') /* istanbul ignore if */ if (host.match && host.match(/@[^/]+/)) { host = host.match(/@[^/]+/)[0] } log.info(host, 'Connecting to DB') const client = knex(config.get('knex')) // Check if we're running tests while connected to // potential production environment. /* istanbul ignore if */ if (config.get('NODE_ENV') === 'test' && config.get('knex:connection:database') !== 'test' || config.get('knex:connection:connection')) { // There is an offchance that we're running tests on // production database. Exit NOW! log.error('Critical: potentially running test on production enviroment. Shutting down.') process.exit(1) } let shelf = bookshelf(client) // Helper method to create models shelf.createModel = (attr, opts) => { // Create default attributes to all models let attributes = _.defaults(attr, { initialize() { this.on('fetching', this.checkFetching) }, checkFetching(model, columns, options) { options.query.where({ is_deleted: false }) }, }) // Create default options for all models let options = _.defaults(opts, { create(data) { return this.forge(data).save() }, getSingle(id, withRelated = [], require = true) { let where = { id: Number(id) || 0 } return this.query({ where }) .fetch({ require, withRelated }) }, getAll(where = {}, withRelated = []) { where.is_deleted = false return this.query({ where }) .fetchAll({ withRelated }) }, }) return shelf.Model.extend(attributes, options) } export default shelf