Add inline parser support for module-self annotation

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: soutaro

lib/rbs/ast/ruby/declarations.rb

@@ -137,7 +137,17 @@ def each_decl(&block)
 
           def type_params = []
 
-          def self_types = []
+          def self_types
+            members.filter_map do |member|
+              if member.is_a?(Members::ModuleSelfMember)
+                AST::Declarations::Module::Self.new(
+                  name: member.name,
+                  args: member.args,
+                  location: member.location
+                )
+              end
+            end
+          end
 
           def location
             rbs_location(node.location)

lib/rbs/ast/ruby/members.rb

@@ -717,6 +717,34 @@ def type_fingerprint
             ]
           end
         end
+
+        class ModuleSelfMember < Base
+          attr_reader :annotation
+
+          def initialize(buffer, annotation)
+            super(buffer)
+            @annotation = annotation
+          end
+
+          def name
+            annotation.name
+          end
+
+          def args
+            annotation.args
+          end
+
+          def location
+            annotation.location
+          end
+
+          def type_fingerprint
+            [
+              "members/module_self",
+              annotation.type_fingerprint
+            ]
+          end
+        end
       end
     end
   end

lib/rbs/inline_parser.rb

@@ -134,6 +134,32 @@ def visit_module_node(node)
         end
 
         module_decl = AST::Ruby::Declarations::ModuleDecl.new(buffer, module_name, node)
+
+        if leading_ref = comments.leading_block(node)
+          unused_annotations = [] #: Array[AST::Ruby::CommentBlock::AnnotationSyntaxError | AST::Ruby::Annotations::leading_annotation]
+
+          leading_ref.block.each_paragraph([]) do |paragraph|
+            case paragraph
+            when AST::Ruby::Annotations::ModuleSelfAnnotation
+              module_decl.members << AST::Ruby::Members::ModuleSelfMember.new(buffer, paragraph)
+            when Location
+              # Skip
+            when AST::Ruby::CommentBlock::AnnotationSyntaxError
+              unused_annotations << paragraph
+            when AST::Ruby::Annotations::SkipAnnotation
+              # Already handled by skip_node?
+            else
+              unused_annotations << paragraph
+            end
+          end
+
+          unless unused_annotations.empty?
+            report_unused_annotation(*unused_annotations)
+          end
+
+          leading_ref.associate!
+        end
+
         insert_declaration(module_decl)
         visit_class_or_module_body(module_decl, node)
       end

sig/ast/ruby/members.rbs

@@ -16,6 +16,7 @@ module RBS
                | IncludeMember | ExtendMember | PrependMember
                | AttrReaderMember | AttrWriterMember | AttrAccessorMember
                | InstanceVariableMember
+               | ModuleSelfMember
 
         class MethodTypeAnnotation
           class DocStyle
@@ -172,6 +173,20 @@ module RBS
 
           def type_fingerprint: () -> untyped
         end
+
+        class ModuleSelfMember < Base
+          attr_reader annotation: Annotations::ModuleSelfAnnotation
+
+          def initialize: (Buffer, Annotations::ModuleSelfAnnotation) -> void
+
+          def name: () -> TypeName
+
+          def args: () -> Array[Types::t]
+
+          def location: () -> Location
+
+          def type_fingerprint: () -> untyped
+        end
       end
     end
   end

test/rbs/inline_parser_test.rb

@@ -1564,6 +1564,58 @@ def foo
     end
   end
 
+  def test_parse__module_self
+    result = parse(<<~RUBY)
+      # @rbs module-self: _Each[String]
+      module Enumerable2
+      end
+    RUBY
+
+    assert_empty result.diagnostics
+
+    result.declarations[0].tap do |decl|
+      assert_instance_of RBS::AST::Ruby::Declarations::ModuleDecl, decl
+      assert_equal RBS::TypeName.parse("Enumerable2"), decl.module_name
+
+      assert_equal 1, decl.members.size
+      assert_equal 1, decl.self_types.size
+
+      decl.members[0].tap do |member|
+        assert_instance_of RBS::AST::Ruby::Members::ModuleSelfMember, member
+        assert_equal "_Each", member.name.to_s
+        assert_equal 1, member.args.size
+        assert_equal "String", member.args[0].to_s
+      end
+    end
+  end
+
+  def test_parse__module_self_multiple
+    result = parse(<<~RUBY)
+      # @rbs module-self: _Each[String]
+      # @rbs module-self: Comparable
+      module StringCollection
+      end
+    RUBY
+
+    assert_empty result.diagnostics
+
+    result.declarations[0].tap do |decl|
+      assert_instance_of RBS::AST::Ruby::Declarations::ModuleDecl, decl
+
+      assert_equal 2, decl.self_types.size
+
+      decl.self_types[0].tap do |member|
+        assert_equal "_Each", member.name.to_s
+        assert_equal 1, member.args.size
+      end
+
+      decl.self_types[1].tap do |member|
+        assert_equal "Comparable", member.name.to_s
+        assert_equal 0, member.args.size
+      end
+    end
+  end
+
   def test_parse__constant_declaration_basic
     result = parse(<<~RUBY)
       class Example