在块中拦截def

时间:2013-04-17 08:12:31

标签: ruby metaprogramming

我有一个方法需要在附加的块上做一些巫术。这样一个块的样本可能是

myMethod do 

   somemethod x
   someother y
   def name(a,b)
     a+b
   end
end

前两个方法调用(somemethod xsomeother y)应该正常执行。但是我想在不实际定义新方法的情况下拦截方法定义(作为S表达式)。如果我将整个块转换为S表达式然后搜索AST,我可以这样做。然而,我需要弄清楚如何调用方法。任何一种解决方案都可以。那是

  • 拦截定义,转换为S-expression(Ruby2Ruby可以理解)
  • 将块转换为S表达式并找到方法调用并执行这些

修改 我正在寻找的AST类似于

[:defn,
 :name,
 [:args,:a,:b],
 [:call,
     [lvar,:a],
     :+
     [lvar,:b]]]

1 个答案:

答案 0 :(得分:0)

如果我理解正确,您希望能够在传递给另一个方法的代码块中定义一个方法,但是拦截该内部方法定义,以便它实际上不会被定义,而是转换为S表达式?所以你希望它的行为几乎就像它被注释掉了,一些外部进程通过源代码来解析并将实现解析成一个S表达式?

类似的东西:

myMethod do 

   somemethod x
   someother y
   # def name(a,b)
   #   a+b
   # end
end

somemethodsomeother仍然执行,name的实现已定义,但Ruby解释器会忽略它。然后你想以某种方式将此实现捕获为S表达式。显然不会这样做,通过评论它,但它是一个很好的方式来描绘行为。

嗯,RubyParser gem可能会做你想要的。它利用了将“here document”作为字符串传递给方法的能力,如下例所示:

def x (str)
  str
end

x( <<-EOF )
this is
   a here
document
EOF

# => "this is\n   a here\ndocument\n"

使用RubyParser,您可以执行以下操作:

require 'ruby_parser'
require 'pp'

pp RubyParser.new.parse( <<-EOF )
def plus(x,y)
  x+y
end
EOF
# => s(:defn, :name, s(:args, :a, :b), s(:call, s(:lvar, :a), :+, s(:lvar, :b)))

然后,将它与您想要的代码合并将是一件小事,如下:

require 'ruby_parser'

def myMethod
  x = yield
end

def somemethod( x )
  puts x
end

def someother( y )
  puts y
end

x = 'magic'
y = 'output'

sexp = myMethod do
  somemethod x
  someother y
  RubyParser.new.parse( <<-EOF )
  def name(a,b)
    a+b
  end
  EOF
end

pp sexp
name(1,2)
# magic
# output
# s(:defn, :name, s(:args, :a, :b), s(:call, s(:lvar, :a), :+, s(:lvar, :b)))
# NoMethodError: undefined local variable or method 'name' for main:Object

如您所见,方法定义基本上只是一个字符串,可以这样操作。