Adding simple implicit return
Ruby return semantics are easy to grasp, not so easy to code. So many cases. Added support for common cases, return const/variable or call.
This commit is contained in:
@ -9,9 +9,31 @@ module Ruby
|
||||
end
|
||||
|
||||
def to_vool
|
||||
if @body.is_a?(Statements)
|
||||
@body << replace_return( @body.pop )
|
||||
else
|
||||
@body = replace_return( @body )
|
||||
end
|
||||
Vool::MethodStatement.new( @name , @args.dup , @body.to_vool)
|
||||
end
|
||||
|
||||
def replace_return(statement)
|
||||
case statement
|
||||
when SendStatement , YieldStatement, Variable , Constant
|
||||
return ReturnStatement.new( statement )
|
||||
when IvarAssignment
|
||||
ret = ReturnStatement.new( InstanceVariable.new(statement.name) )
|
||||
return Statements.new([statement , ret])
|
||||
when LocalAssignment
|
||||
ret = ReturnStatement.new( LocalVariable.new(statement.name) )
|
||||
return Statements.new([statement , ret])
|
||||
when ReturnStatement , IfStatement , WhileStatement ,BlockStatement
|
||||
return statement
|
||||
else
|
||||
raise "Not implemented implicit return #{statement.class}"
|
||||
end
|
||||
end
|
||||
|
||||
def to_s(depth = 0)
|
||||
arg_str = @args.collect{|a| a.to_s}.join(', ')
|
||||
at_depth(depth , "def #{name}(#{arg_str})" , @body.to_s(depth + 1) , "end")
|
||||
|
Reference in New Issue
Block a user