WeakAura Helper: Count Enemies Within Distance (nameplates)

function aura_env.CountEnemiesWithin( distance )
   if GetCVar("nameplateShowEnemies") == "0" then
      return 0
   end

   local count = 0

   if GetCVar("nameplateShowFriends") == "1" then
      for i, plate in ipairs(C_NamePlate.GetNamePlates()) do
         local unit = plate.namePlateUnitToken

         if UnitCanAttack("player", unit) and WeakAuras.CheckRange(unit, distance, "<=") then
            count = count + 1
         end
      end

   else
      for i, plate in ipairs(C_NamePlate.GetNamePlates()) do
         local unit = plate.namePlateUnitToken

         if WeakAuras.CheckRange(unit, distance, "<=") then
            count = count + 1
         end
      end
   end

   return count
end