class Airbrake::Truncator

def initialize(max_size)
  @max_size = max_size
end

def reduce_max_size
  @max_size /= 2
end

def replace_invalid_characters(str)
  utf8_string = SUPPORTED_ENCODINGS.include?(str.encoding)
  return str if utf8_string && str.valid_encoding?
  temp_str = str.dup
  temp_str.encode!(TEMP_ENCODING, **ENCODING_OPTIONS) if utf8_string
  temp_str.encode!('utf-8', **ENCODING_OPTIONS)
end

def stringify_object(object)
  object.to_json
rescue *Notice::JSON_EXCEPTIONS
  object.to_s
end

def truncate(object, seen = Set.new)
  if seen.include?(object.object_id)
    return CIRCULAR if CIRCULAR_TYPES.any? { |t| object.is_a?(t) }
    return object
  end
  truncate_object(object, seen << object.object_id)
end

def truncate_array(array, seen)
  array.slice(0, @max_size).map! { |elem| truncate(elem, seen) }.freeze
end

def truncate_hash(hash, seen)
  truncated_hash = {}
  hash.each_with_index do |(key, val), idx|
    break if idx + 1 > @max_size
    truncated_hash[key] = truncate(val, seen)
  end
  truncated_hash.freeze
end

def truncate_object(object, seen)
  case object
  when Hash then truncate_hash(object, seen)
  when Array then truncate_array(object, seen)
  when Set then truncate_set(object, seen)
  when String then truncate_string(object)
  when Numeric, TrueClass, FalseClass, Symbol, NilClass then object
  else
    truncate_string(stringify_object(object))
  end
end

def truncate_set(set, seen)
  truncated_set = Set.new
  set.each do |elem|
    truncated_set << truncate(elem, seen)
    break if truncated_set.size >= @max_size
  end
  truncated_set.freeze
end

def truncate_string(str)
  fixed_str = replace_invalid_characters(str)
  return fixed_str if fixed_str.length <= @max_size
  (fixed_str.slice(0, @max_size) + TRUNCATED).freeze
end