codecombat/app/lib/LocalMongo.coffee

51 lines
2.4 KiB
CoffeeScript
Raw Normal View History

2014-05-13 15:09:54 -04:00
LocalMongo = module.exports
# Checks whether func(l, r) is true for at least one value of left for at least one value of right
mapred = (left, right, func) ->
_.reduce(left, ((result, singleLeft) ->
result or (_.reduce (_.map right, (singleRight) -> func(singleLeft, singleRight)),
((intermediate, value) -> intermediate or value), false)), false)
doQuerySelector = (originalValue, operatorObj) ->
value = if _.isArray originalValue then originalValue else [originalValue] # left hand can be an array too
for operator, originalBody of operatorObj
body = if _.isArray originalBody then originalBody else [originalBody] # right hand can be an array too
2014-05-13 15:09:54 -04:00
switch operator
when '$gt' then return false unless mapred value, body, (l, r) -> l > r
when '$gte' then return false unless mapred value, body, (l, r) -> l >= r
when '$lt' then return false unless mapred value, body, (l, r) -> l < r
when '$lte' then return false unless mapred value, body, (l, r) -> l <= r
when '$ne' then return false if mapred value, body, (l, r) -> l == r
2014-05-13 18:08:22 -04:00
when '$in' then return false unless _.reduce value, ((result, val) -> result or val in body), false
when '$nin' then return false if _.reduce value, ((result, val) -> result or val in body), false
when '$exists' then return false if value[0]? isnt body[0]
else
trimmedOperator = _.pick(operatorObj, operator)
return false unless _.isObject(originalValue) and matchesQuery(originalValue, trimmedOperator)
2014-05-13 15:09:54 -04:00
true
2014-05-14 11:12:33 -04:00
matchesQuery = (target, queryObj) ->
return true unless queryObj
throw new Error 'Expected an object to match a query against, instead got null' unless target
2014-05-13 18:08:22 -04:00
for prop, query of queryObj
2014-05-14 11:12:33 -04:00
if prop[0] == '$'
switch prop
when '$or' then return false unless _.reduce query, ((res, obj) -> res or matchesQuery target, obj), false
when '$and' then return false unless _.reduce query, ((res, obj) -> res and matchesQuery target, obj), true
else return false
else
# Do nested properties
pieces = prop.split('.')
obj = target
for piece in pieces
unless piece of obj
obj = null
break
obj = obj[piece]
2014-05-14 11:12:33 -04:00
if typeof query != 'object' or _.isArray query
return false unless obj == query or (query in obj if _.isArray obj)
else return false unless doQuerySelector obj, query
2014-05-14 11:12:33 -04:00
true
LocalMongo.matchesQuery = matchesQuery