module PP::PPMethods

# File lib/pp.rb, line 161
def check_inspect_key(id)
  recursive_state = Thread.current[:__recursive_key__] or return false
  recursive_state[:inspect]&.include?(id)
end

# File lib/pp.rb, line 238
def comma_breakable
  text ','
  breakable
end

# File lib/pp.rb, line 147
def guard_inspect_key
  recursive_state = Thread.current[:__recursive_key__] ||= {}.compare_by_identity
  save = recursive_state[:inspect] ||= {}.compare_by_identity
  begin
    recursive_state[:inspect] = {}.compare_by_identity
    yield
  ensure
    recursive_state[:inspect] = save
  end
end

# File lib/pp.rb, line 228
def object_address_group(obj, &block)
  str = Kernel.instance_method(:to_s).bind_call(obj)
  str.chomp!('>')
  group(1, str, '>', &block)
end

# File lib/pp.rb, line 222
def object_group(obj, &block) # :yield:
  group(1, '#<' + obj.class.name, '>', &block)
end

# File lib/pp.rb, line 173
def pop_inspect_key(id)
  Thread.current[:__recursive_key__][:inspect].delete id
end

# File lib/pp.rb, line 200
def pp(obj)
  # If obj is a Delegator then use the object being delegated to for cycle
  # detection
  obj = obj.__getobj__ if defined?(::Delegator) and ::Delegator === obj

  if check_inspect_key(obj)
    group {obj.pretty_print_cycle self}
    return
  end

  guard_inspect(obj) do
    group do
      obj.pretty_print self
    rescue NoMethodError
      text Kernel.instance_method(:inspect).bind_call(obj)
    end
  end
end

# File lib/pp.rb, line 302
def pp_hash(obj)
  group(1, '{', '}') {
    seplist(obj, nil, :each_pair) {|k, v|
      group {
        pp_hash_pair k, v
      }
    }
  }
end

# File lib/pp.rb, line 314
def pp_hash_pair(k, v)
  if Symbol === k
    if k.inspect.match?(%r[\A:["$@!]|[%&*+\-\/<=>@\]^`|~]\z])
      k = k.to_s.inspect
    end
    text "#{k}:"
  else
    pp k
    text ' '
    text '=>'
  end
  group(1) {
    breakable
    pp v
  }
end

# File lib/pp.rb, line 286
def pp_object(obj)
  object_address_group(obj) {
    seplist(obj.pretty_print_instance_variables, lambda { text ',' }) {|v|
      breakable
      v = v.to_s if Symbol === v
      text v
      text '='
      group(1) {
        breakable ''
        pp(obj.instance_eval(v))
      }
    }
  }
end

# File lib/pp.rb, line 168
def push_inspect_key(id)
  Thread.current[:__recursive_key__][:inspect][id] = true
end

# File lib/pp.rb, line 267
def seplist(list, sep=nil, iter_method=:each) # :yield: element
  sep ||= lambda { comma_breakable }
  first = true
  kwsplat = EMPTY_KWHASH
  list.__send__(iter_method) {|*v|
    if first
      first = false
    else
      sep.call
    end
    kwsplat ? yield(*v, **kwsplat) : yield(*v)
  }
end