rubyx/lib/ruby/method_statement.rb
Torsten Rüger 32f908c127 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.
2019-08-17 23:27:55 +03:00

44 lines
1.3 KiB
Ruby

module Ruby
class MethodStatement < Statement
attr_reader :name, :args , :body , :clazz
def initialize( name , args , body , clazz = nil)
@name , @args , @body = name , args , body
raise "no bod" unless @body
@clazz = clazz
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")
end
end
end