definesmethod

Author: kddnewton

lib/syntax_tree/visitor/compiler.rb

@@ -418,7 +418,7 @@ def serialize(insn)
             [insn[0], iseq.local_table.offset(insn[1]), *insn[2..]]
           when :defineclass
             [insn[0], insn[1], insn[2].to_a, insn[3]]
-          when :definemethod
+          when :definemethod, :definesmethod
             [insn[0], insn[1], insn[2].to_a]
           when :send
             # For any instructions that push instruction sequences onto the
@@ -511,6 +511,11 @@ def definemethod(name, method_iseq)
           iseq.push([:definemethod, name, method_iseq])
         end
 
+        def definesmethod(name, method_iseq)
+          stack.change_by(-1)
+          iseq.push([:definesmethod, name, method_iseq])
+        end
+
         def dup
           stack.change_by(-1 + 2)
           iseq.push([:dup])
@@ -1390,7 +1395,14 @@ def visit_def(node)
           end
 
         name = node.name.value.to_sym
-        builder.definemethod(name, method_iseq)
+
+        if node.target
+          visit(node.target)
+          builder.definesmethod(name, method_iseq)
+        else
+          builder.definemethod(name, method_iseq)
+        end
+
         builder.putobject(name)
       end
 

test/compiler_test.rb

@@ -364,6 +364,8 @@ class CompilerTest < Minitest::Test
       "def foo(bar, ...); end",
       "def foo(...); bar(...); end",
       "def foo(bar, ...); baz(1, 2, 3, ...); end",
+      "def self.foo; end",
+      "def foo.bar(baz); end",
       # Class/module definitions
       "module Foo; end",
       "module ::Foo; end",