Contents

module RipperRubyParser
  module SexpHandlers
    # Sexp handlers for loops
    module Loops
      def process_until(exp)
        handle_conditional_loop :until, :while, exp
      end

      def process_until_mod(exp)
        handle_conditional_loop_mod :until, :while, exp
      end

      def process_while(exp)
        handle_conditional_loop :while, :until, exp
      end

      def process_while_mod(exp)
        handle_conditional_loop_mod :while, :until, exp
      end

      def process_for(exp)
        _, var, coll, block = exp.shift 4
        coll = process(coll)
        assgn = s(:lasgn, process(var)[1])
        block = unwrap_nil process(block)
        if block
          s(:for, coll, assgn, block)
        else
          s(:for, coll, assgn)
        end
      end

      private

      def check_at_start?(block)
        block.sexp_type != :begin
      end

      def handle_conditional_loop(type, negated_type, exp)
        _, cond, body = exp.shift 3

        construct_conditional_loop(type, negated_type,
                                   process(cond),
                                   unwrap_nil(process(body)),
                                   true)
      end

      def handle_conditional_loop_mod(type, negated_type, exp)
        _, cond, body = exp.shift 3

        check_at_start = check_at_start?(body)
        construct_conditional_loop(type, negated_type,
                                   process(cond),
                                   process(body),
                                   check_at_start)
      end

      def construct_conditional_loop(type, negated_type, cond, body, check_at_start)
        if cond.sexp_type == :not
          _, inner = cond
          s(negated_type, inner, body, check_at_start)
        else
          s(type, cond, body, check_at_start)
        end
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
ripper_ruby_parser-1.4.2	lib/ripper_ruby_parser/sexp_handlers/loops.rb
ripper_ruby_parser-1.4.1	lib/ripper_ruby_parser/sexp_handlers/loops.rb
ripper_ruby_parser-1.4.0	lib/ripper_ruby_parser/sexp_handlers/loops.rb
ripper_ruby_parser-1.3.0	lib/ripper_ruby_parser/sexp_handlers/loops.rb
ripper_ruby_parser-1.2.0	lib/ripper_ruby_parser/sexp_handlers/loops.rb