module Mongoid::Contextual::Atomic

def add_each_to_set(adds)
  view.update_many("$addToSet" => collect_each_operations(adds))
end

def add_to_set(adds)
  view.update_many("$addToSet" => collect_operations(adds))
end

def bit(bits)
  view.update_many("$bit" => collect_operations(bits))
end

def collect_each_operations(ops)
  ops.each_with_object({}) do |(field, value), operations|
    operations[database_field_name(field)] = { "$each" => Array.wrap(value).mongoize }
  end
end

def collect_operations(ops)
  ops.each_with_object({}) do |(field, value), operations|
    operations[database_field_name(field)] = value.mongoize
  end
end

def inc(incs)
  view.update_many("$inc" => collect_operations(incs))
end

def pop(pops)
  view.update_many("$pop" => collect_operations(pops))
end

def pull(pulls)
  view.update_many("$pull" => collect_operations(pulls))
end

def pull_all(pulls)
  view.update_many("$pullAll" => collect_operations(pulls))
end

def push(pushes)
  view.update_many("$push" => collect_operations(pushes))
end

def push_all(pushes)
  view.update_many("$push" => collect_each_operations(pushes))
end

def rename(renames)
  operations = renames.inject({}) do |ops, (old_name, new_name)|
    ops[old_name] = new_name.to_s
    ops
  end
  view.update_many("$rename" => collect_operations(operations))
end

def set(sets)
  view.update_many("$set" => collect_operations(sets))
end

def unset(*args)
  fields = args.map { |a| a.is_a?(Hash) ? a.keys : a }
               .__find_args__
               .map { |f| [database_field_name(f), true] }
  view.update_many("$unset" => Hash[fields])
end