Fix resolving mixins that mix themselves in

Author: ccutrer

lib/yard/handlers/ruby/mixin_handler.rb

@@ -26,11 +26,15 @@ def process_mixin(mixin)
     raise YARD::Parser::UndocumentableError unless mixin.ref?
     raise YARD::Parser::UndocumentableError if mixin.first.type == :ident
 
-    case obj = Proxy.new(namespace, mixin.source)
-    when ConstantObject # If a constant is included, use its value as the real object
-      obj = Proxy.new(namespace, obj.value, :module)
+    if mixin.type == :var_ref && mixin[0] == s(:kw, "self")
+      obj = namespace
     else
-      obj = Proxy.new(namespace, mixin.source, :module)
+      case obj = Proxy.new(namespace, mixin.source)
+      when ConstantObject # If a constant is included, use its value as the real object
+        obj = Proxy.new(namespace, obj.value, :module)
+      else
+        obj = Proxy.new(namespace, mixin.source, :module)
+      end
     end
 
     rec = recipient(mixin)
@@ -41,7 +45,7 @@ def process_mixin(mixin)
   end
 
   def recipient(mixin)
-    if statement[0].type == :const_path_ref
+    if statement[0].type == :const_path_ref || statement[0].type == :top_const_ref
       Proxy.new(namespace, statement[0].source)
     elsif statement[0].type == :var_ref && statement[0][0] != s(:kw, "self")
       statement[0][0].type == :const ?

spec/handlers/examples/mixin_handler_001.rb.txt

@@ -38,3 +38,16 @@ end
 
 module FromConstant; end
 FromConstant.include A
+
+module Foo
+end
+
+module MixMySelfIn
+  Foo.include(self)
+end
+
+module Nested
+  module Foo
+    ::Foo.include(self)
+  end
+end

spec/handlers/mixin_handler_spec.rb

@@ -69,4 +69,8 @@ class D1; class E1; class F1; end end end
     expect(YARD::Registry.root.instance_mixins).not_to eq [P('D1::E1::F1')]
     expect(P('A1::B1::C1').instance_mixins).to eq [P('D1::E1::F1')]
   end
+
+  it "resolves modules that mix themselves in" do
+    expect(Registry.at('Foo').mixins).to match_array [P('MixMySelfIn'), P('Nested::Foo')]
+  end
 end