# Copyright (c) 2015 Sqreen. All Rights Reserved.
# Please refer to our terms for more information: https://www.sqreen.com/terms.html

require 'sqreen/util'

class Sqreen::Util::CappedArray < Array
  attr_reader :size_cap, :depth_cap

  def initialize(*args, size_cap: 150, depth_cap: 10, &block)
    @size_cap = size_cap
    @depth_cap = depth_cap

    super(*args, &block)
  end

  def <<(value)
    keep?(size, value) ? super : self
  end
  alias_method :append, :<<

  def []=(index, value)
    super if keep?(index, value)
  end

  private

  def keep?(index, value)
    index < size_cap && (depth_cap > 0 || !value.is_a?(Hash) && !value.is_a?(Array))
  end
end