40 lines
851 B
Ruby
40 lines
851 B
Ruby
require_relative "normalizer"
|
|
|
|
module Ruby
|
|
class IfStatement < Statement
|
|
include Normalizer
|
|
|
|
attr_reader :condition , :if_true , :if_false
|
|
|
|
def initialize( cond , if_true , if_false = nil)
|
|
@condition = cond
|
|
@if_true = if_true
|
|
@if_false = if_false
|
|
end
|
|
|
|
def to_vool
|
|
cond , rest = *normalize_name(@condition)
|
|
fals = @if_false ? @if_false.normalize : nil
|
|
me = IfStatement.new(cond , @if_true.normalize, fals)
|
|
return me unless rest
|
|
rest << me
|
|
rest
|
|
end
|
|
|
|
def has_false?
|
|
@if_false != nil
|
|
end
|
|
|
|
def has_true?
|
|
@if_true != nil
|
|
end
|
|
|
|
def to_s(depth = 0)
|
|
parts = ["if (#{@condition})" , @body.to_s(depth + 1) ]
|
|
parts += ["else" , "@if_false.to_s(depth + 1)"] if(@false)
|
|
parts << "end"
|
|
at_depth(depth , *parts )
|
|
end
|
|
end
|
|
end
|