Merge pull request #2079 from tk0miya/support_self_CONST

prototype rb: Fix crashed by self::CONST

Author: soutaro

lib/rbs/prototype/rb.rb

@@ -368,7 +368,7 @@ def process(node, decls:, comments:, context:)
                        when node.children[0].is_a?(Symbol)
                          TypeName.new(name: node.children[0], namespace: Namespace.empty)
                        else
-                         const_to_name!(node.children[0])
+                         const_to_name!(node.children[0], context: context)
                        end
 
           value_node = node.children.last
@@ -429,20 +429,24 @@ def process_children(node, decls:, comments:, context:)
         end
       end
 
-      def const_to_name!(node)
+      def const_to_name!(node, context: nil)
         case node.type
         when :CONST
           TypeName.new(name: node.children[0], namespace: Namespace.empty)
         when :COLON2
           if node.children[0]
-            namespace = const_to_name!(node.children[0]).to_namespace
+            namespace = const_to_name!(node.children[0], context: context).to_namespace
           else
             namespace = Namespace.empty
           end
 
           TypeName.new(name: node.children[1], namespace: namespace)
         when :COLON3
           TypeName.new(name: node.children[0], namespace: Namespace.root)
+        when :SELF
+          raise if context.nil?
+
+          context.namespace.to_type_name
         else
           raise
         end

sig/prototype/rb.rbs

@@ -44,7 +44,7 @@ module RBS
       # Returns a type name that represents the name of the constant.
       # `node` must be _constant_ node, `CONST`, `COLON2`, or `COLON3` node.
       #
-      def const_to_name!: (RubyVM::AbstractSyntaxTree::Node node) -> TypeName
+      def const_to_name!: (RubyVM::AbstractSyntaxTree::Node node, ?context: Context?) -> TypeName
 
       # Returns a type name that represents the name of the constant.
       # `node` can be `SELF` for `extend self` pattern.

test/rbs/rb_prototype_test.rb

@@ -688,6 +688,7 @@ module Foo
   VERSION = '0.1.1'
   FROZEN = 'str'.freeze
   ::Hello::World = :foo
+  self::MAX = 42
 end
     EOR
 
@@ -700,6 +701,8 @@ module Foo
   FROZEN: "str"
 
   ::Hello::World: :foo
+
+  ::Foo::MAX: 42
 end
     EOF
   end