express-query-params/formats/mongo.js

Express.js middleware implementing the API Query Spec, converting the params to SQL or a Mongo query

const format = require('../lib/format')
const { trimOperators, typeCast } = require('../lib/utils')

module.exports = (options = {}) => (
  rawQuery => {
    const mongo = {}
    const cast = (options.typeCast || typeCast)({ returnJSDate: true, ...options })

    for (const key in rawQuery) {
      if (!Object.prototype.hasOwnProperty.call(rawQuery, key)) continue

      if (
        Array.isArray(options.blacklistParams)
        && options.blacklistParams.length > 0
        && options.blacklistParams.indexOf(key) > -1
      ) continue

      if (
        Array.isArray(options.whitelistParams)
        && options.whitelistParams.length > 0
        && options.whitelistParams.indexOf(key) === -1
      ) continue

      const value = rawQuery[key]
      const insertValue = cast(trimOperators(value))

      if (format.stringContains(value)) mongo[key] = new RegExp(insertValue)
      else if (format.stringIsCaseInsensitive(value)) mongo[key] = new RegExp(`${insertValue}$i`)
      else if (format.isGreaterThan(value)) mongo[key] = { $gt: insertValue }
      else if (format.isGreaterThanOrEqual(value)) mongo[key] = { $gte: insertValue }
      else if (format.isLesserThan(value)) mongo[key] = { $lt: insertValue }
      else if (format.isLesserThanOrEqual(value)) mongo[key] = { $lte: insertValue }
      else if (format.oneOf(value)) mongo[key] = { $in: value.split(',') }
      else if (format.isRange(value)) {
        const [ from, to ] = insertValue.split('...')
        mongo[key] = { $gte: cast(from), $lte: cast(to) }
      } else mongo[key] = insertValue

      if (format.negated(value)) mongo[key] = { $not: mongo[key] }
    }

    return mongo
  }
)