lowdb-recursive/src/index.js

recursive extension based on lowdb

var _ = require('lodash')
var low = require('lowdb')
var FileSync = require('lowdb/adapters/FileSync')

var DataBase = function (path) {
  const adapter = new FileSync(path)
  var db = low(adapter)
  db._.mixin({
    recursive: function (type, update, array, where, value, cb, count, envio) {
      where = where.split('.')
      var that = this

      count ? count++ : count = 1

      if (_.compact(where).length > 1) {
        var state = where.shift()
        _.forEach(array, function (document) {
          if (document[state] && count === 1) {
            that.recursive(type, update, document[state],
              where.join('.'), value, cb, count, document)
          } else {
            that.recursive(type, update, document[state],
              where.join('.'), value, cb, count, envio)
          }
        })
      } else {
        var json = {}
        json[where.join()] = value

        if (type === 'UPDATE') {
          if ((!update)) {
            _.forEach(array, function (document) {
              _.assign(document, json)
            })
            cb(array)
          } else {
            _.forEach(_.filter(array, json), function (val) {
              _.assign(val, update)
            })
            cb(envio)
          }
        } else if (type === 'FIND') {
          var dato = _.filter(array, json)
          if (dato.length > 0) {
            cb(dato)
          }
        } else if (type === 'PUSH') {
          if ((!update)) {
            if (typeof value === 'object' && _.keys(value).length > 0) {
              _.forEach(array, function (document) {
                document[where.join()].push(value)
              })
            } else {
              _.forEach(array, function (document) {
                _.merge(document, json)
              })
            }
          } else {
            if (_.filter(array, json).length > 0) {
              array.push(update)
            }
          }
          cb(array)
        } else if (type === 'REMOVE') {
          var lista = []
          _.forEach(_.filter(array, json), function (val) {
            _.remove(array, val)
            lista.push(val)
          })
          cb(lista)
        } else if (type === 'WHERE') {
          if (_.filter(array, json).length > 0) {
            count === 1 ? cb(dato) : cb(envio)
          }
        }
      }
    },
    findAll: function (array, where) {
      var list = []
      var that = this
      _.forIn(where, function (value, key) {
        that.recursive('FIND', null, array, key, value, function (dato) {
          if (!(Object.prototype.toString.call(dato) === '[object Array]')) {
            list.push(dato)
          } else {
            _.forEach(dato, function (value) {
              list.push(value)
            })
          }
        })
        array = list
        list = []
      })
      var val = {}
      var listFinal = []
      _.forEach(array, function (document) {
        if (!_.isEqual(val, document)) {
          val = document
          listFinal.push(document)
        }
      })
      array = listFinal

      return array
    },
    whereAll: function (array, where) {
      var list = []
      var that = this
      _.forIn(where, function (value, key) {
        that.recursive('WHERE', null, array, key, value, function (dato) {
          if (!(Object.prototype.toString.call(dato) === '[object Array]')) {
            list.push(dato)
          } else {
            _.forEach(dato, function (value) {
              list.push(value)
            })
          }
        })
        array = list
        list = []
      })
      var val = {}
      var listFinal = []
      _.forEach(array, function (document) {
        if (!_.isEqual(val, document)) {
          val = document
          listFinal.push(document)
        }
      })
      array = listFinal

      return array
    },
    updateAll: function (array, update, where) {
      var that = this
      _.forIn(update, function (value, key) {
        that.recursive('UPDATE', where, array, key, value, function (dato) {})
      })
      return array
    },
    pushAll: function (array, where, push) {
      var list = []
      var that = this
      _.forIn(where, function (value, key) {
        that.recursive('PUSH', push, array, key, value, function (dato) {
          if (!(Object.prototype.toString.call(dato) === '[object Array]')) {
            list.push(dato)
          } else {
            _.forEach(dato, function (value) {
              list.push(value)
            })
          }
        })
        array = list
        list = []
      })
      var val = {}
      var listFinal = []
      _.forEach(array, function (document) {
        if (!_.isEqual(val, document)) {
          val = document
          listFinal.push(document)
        }
      })
      array = listFinal

      return array
    },
    removeAll: function (array, remove) {
      var list = []
      var that = this
      _.forIn(remove, function (value, key) {
        that.recursive('REMOVE', null, array, key, value, function (dato) {
          if (!(Object.prototype.toString.call(dato) === '[object Array]')) {
            list.push(dato)
          } else {
            _.forEach(dato, function (value) {
              list.push(value)
            })
          }
        })
        array = list
        list = []
      })
      var val = {}
      var listFinal = []
      _.forEach(array, function (document) {
        if (!_.isEqual(val, document)) {
          val = document
          listFinal.push(document)
        }
      })
      array = listFinal

      return array
    }
  })
  return db
}
module.exports = DataBase