70 lines
1.7 KiB
JavaScript
70 lines
1.7 KiB
JavaScript
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
|