class GraphQL::Upgrader::ResolveProcToMethodTransform::ResolveProcProcessor

def initialize
  @resolve_proc_sections = []
end

def on_block(node)
  send_node, args_node, body_node = node.children
  _receiver, method_name, _send_args_node = *send_node
  # Assume that the first three-argument proc we enter is the resolve
  if (
    method_name == :lambda && args_node.children.size == 3 &&
    !@resolve_proc_sections.empty? && @resolve_proc_sections[-1].proc_arg_names.nil?
  )
    resolve_proc_section = @resolve_proc_sections[-1]
    source_exp = body_node.loc.expression
    resolve_proc_section.proc_arg_names = args_node.children.map { |arg_node| arg_node.children[0].to_s }
    resolve_proc_section.proc_start = source_exp.begin.begin_pos
    resolve_proc_section.proc_end = source_exp.end.end_pos
  end
  super(node)
end

def on_send(node)
  receiver, method_name, _args = *node
  if method_name == :resolve && receiver.nil?
    resolve_proc_section = ResolveProcSection.new
    source_exp = node.loc.expression
    resolve_proc_section.resolve_start = source_exp.begin.begin_pos
    resolve_proc_section.resolve_end = source_exp.end.end_pos
    resolve_proc_section.resolve_indent = source_exp.column
    @resolve_proc_sections << resolve_proc_section
  end
  super(node)
end