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:
2019-08-17 23:27:55 +03:00
parent 4c76ff3388
commit 32f908c127
4 changed files with 29 additions and 2 deletions

View File

@ -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")