rubyx/lib/mom/statement.rb

module Mom
  class Statement
    # flattening will change the structure from a tree to a linked list (and use
    # nekst to do so)
    def flatten(options = {})
      raise "not implemented for #{self}"
    end
  end

  class Statements < Statement
    include Common::Statements

    def flatten( options = {} )
      flat = @statements.shift.flatten(options)
      while( nekst = @statements.shift )
        flat.append nekst.flatten(options)
      end
      flat
    end

  end

end

require_relative "if_statement"
require_relative "while_statement"