rubyx/lib/vool/statements/while_statement.rb

require_relative "hoister"

module Vool
  class WhileStatement < Statement
    include Hoister
    attr_reader :condition , :statements

    def initialize( condition , statements )
      @condition = condition
      @statements = statements
      simplify_condition
    end

    def to_mom( method )
      statements =   @statements.to_mom( method )
      condition , hoisted  = hoist_condition( method )
      cond = Mom::TruthCheck.new(condition.to_mom(method))
      check = Mom::WhileStatement.new(  cond , statements  )
      check.hoisted = hoisted.to_mom(method) if hoisted
      check
    end

    def simplify_condition
      return unless @condition.is_a?(ScopeStatement)
      @condition = @condition.first if @condition.single?
    end

    def collect(arr)
      @condition.collect(arr)
      @statements.collect(arr)
      super
    end

  end
end