module RipperRubyParser::SexpHandlers::Blocks

# File lib/ripper_ruby_parser/sexp_handlers/blocks.rb, line 53
def process_begin exp
  _, body = exp.shift 2

  body = process(body)
  strip_typeless_sexp(body)
end

# File lib/ripper_ruby_parser/sexp_handlers/blocks.rb, line 45
def process_block_var exp
  _, args, _ = exp.shift 3

  names = process(args)

  convert_special_args names
end

# File lib/ripper_ruby_parser/sexp_handlers/blocks.rb, line 84
def process_bodystmt exp
  _, body, rescue_block, else_block, ensure_block = exp.shift 5

  body = map_body body

  body = wrap_in_block(body)

  body = if body.nil?
           s()
         else
           s(body)
         end

  if rescue_block
    body.push(*process(rescue_block))
    body << process(else_block) if else_block
    body = s(s(:rescue, *body))
  elsif else_block
    body << process(else_block)
  end

  if ensure_block
    body << process(ensure_block)
    body = s(s(:ensure, *body))
  end

  body = wrap_in_block(body)

  body = if body.nil?
           s()
         else
           s(body)
         end

  body
end

# File lib/ripper_ruby_parser/sexp_handlers/blocks.rb, line 13
def process_brace_block exp
  handle_generic_block exp
end

# File lib/ripper_ruby_parser/sexp_handlers/blocks.rb, line 140
def process_break exp
  _, args = exp.shift 2
  if args.empty?
    s(:break)
  else
    s(:break, handle_return_argument_list(args))
  end
end

# File lib/ripper_ruby_parser/sexp_handlers/blocks.rb, line 17
def process_do_block exp
  handle_generic_block exp
end

# File lib/ripper_ruby_parser/sexp_handlers/blocks.rb, line 126
def process_ensure exp
  _, block = exp.shift 2
  strip_typeless_sexp safe_wrap_in_block s(*map_body(block))
end

# File lib/ripper_ruby_parser/sexp_handlers/blocks.rb, line 149
def process_lambda exp
  _, args, statements = exp.shift 3
  args = convert_special_args(process(args))
  args = 0 if args == s(:args)
  make_iter(s(:call, nil, :lambda),
            args,
            *handle_potentially_typeless_sexp(statements))
end

# File lib/ripper_ruby_parser/sexp_handlers/blocks.rb, line 4
def process_method_add_block exp
  _, call, block = exp.shift 3
  block = process(block)
  args = block[1]
  stmt = block[2].first
  call = process(call)
  make_iter call, args, stmt
end

# File lib/ripper_ruby_parser/sexp_handlers/blocks.rb, line 131
def process_next exp
  _, args = exp.shift 2
  if args.empty?
    s(:next)
  else
    s(:next, handle_return_argument_list(args))
  end
end

# File lib/ripper_ruby_parser/sexp_handlers/blocks.rb, line 21
def process_params exp
  if exp.size == 6
    _, normal, defaults, splat, rest, block = exp.shift 6
  else
    _, normal, defaults, splat, rest, _, _, block = exp.shift 8
  end

  args = [*normal].map do |id|
    process(id)
  end

  [*defaults].each do |pair|
    sym = process(pair[0])
    val = process(pair[1])
    args << s(:lasgn, sym[1], val)
  end

  args << process(splat) unless splat.nil? || splat == 0
  [*rest].each {|arg| args << process(arg)}
  args << process(block) unless block.nil?

  s(:args, *args)
end

# File lib/ripper_ruby_parser/sexp_handlers/blocks.rb, line 60
def process_rescue exp
  _, eclass, evar, block, after = exp.shift 5
  rescue_block = map_body(block)
  rescue_block << nil if rescue_block.empty?

  arr = []
  if eclass
    eclass = handle_potentially_typeless_sexp eclass
    if eclass.first.is_a? Symbol
      arr += eclass[1..-1]
    else
      arr << eclass[0]
    end
  end

  if evar
    arr << create_assignment_sub_type(process(evar), s(:gvar, :$!))
  end

  s(
    s(:resbody, s(:array, *arr), *rescue_block),
    *process(after))
end

# File lib/ripper_ruby_parser/sexp_handlers/blocks.rb, line 121
def process_rescue_mod exp
  _, scary, safe = exp.shift 3
  s(:rescue, process(scary), s(:resbody, s(:array), process(safe)))
end