Extract AnyScopeNode type alias

Author: amomchilov

lib/ext/prism.rb

@@ -0,0 +1,10 @@
+# typed: strict
+# frozen_string_literal: true
+
+module Prism
+  # Ideally this would just be in a shim in `sorbet/rbi/shims/prism.rbi`, but that causes
+  # `bundle exec tapioca gem spoom` to fail. That command uses Spoom's translator to rewrite the RBS signature comments
+  # into Sigs, which try to access the `Prism::AnyScopeNode` constant.
+  # Because shims aren't executed, no such constant exists at runtime, and that fails with a NameError.
+  #: type anyScopeNode = ::Prism::ClassNode | ::Prism::ModuleNode | ::Prism::SingletonClassNode
+end

lib/spoom/sorbet/metrics/code_metrics_visitor.rb

@@ -147,7 +147,7 @@ def visit_call_node(node)
 
         private
 
-        #: (Prism::ClassNode | Prism::ModuleNode | Prism::SingletonClassNode) { -> void } -> void
+        #: (Prism::anyScopeNode) { -> void } -> void
         def visit_scope(node, &block)
           key = node_key(node)
           @counters.increment(key)
@@ -216,7 +216,7 @@ def collect_last_srb_sigs
           sigs
         end
 
-        #: (Prism::ClassNode | Prism::ModuleNode | Prism::SingletonClassNode) -> String
+        #: (Prism::anyScopeNode) -> String
         def node_key(node)
           case node
           when Prism::ClassNode

lib/spoom/sorbet/translate/rbs_comments_to_sorbet_sigs/base_translator.rb

@@ -199,7 +199,7 @@ def apply_overloads_strategy(signatures, method_name:, location:)
             end
           end
 
-          #: (Prism::ClassNode | Prism::ModuleNode | Prism::SingletonClassNode) -> void
+          #: (Prism::anyScopeNode) -> void
           def apply_class_annotations(node)
             comments = node_rbs_comments(node)
             return if comments.empty?
@@ -324,7 +324,7 @@ def apply_member_annotations(annotations, sig)
             end
           end
 
-          #: (Prism::ClassNode | Prism::ModuleNode | Prism::SingletonClassNode, Regexp) -> bool
+          #: (Prism::anyScopeNode, Regexp) -> bool
           def already_extends?(node, constant_regex)
             node.child_nodes.any? do |c|
               next false unless c.is_a?(Prism::CallNode)

lib/spoom/sorbet/translate/sorbet_sigs_to_rbs_comments.rb

@@ -134,7 +134,7 @@ def visit_constant_write_node(node)
 
         private
 
-        #: (Prism::ClassNode | Prism::ModuleNode | Prism::SingletonClassNode) { -> void } -> void
+        #: (Prism::anyScopeNode) { -> void } -> void
         def visit_scope(node, &block)
           old_class_annotations = @class_annotations
           @class_annotations = []
@@ -229,7 +229,7 @@ def visit_extend(node)
           end
         end
 
-        #: (Prism::ClassNode | Prism::ModuleNode | Prism::SingletonClassNode, Prism::CallNode) -> void
+        #: (Prism::anyScopeNode, Prism::CallNode) -> void
         def apply_class_annotation(parent, node)
           unless node.message == "abstract!" || node.message == "interface!" || node.message == "sealed!" ||
               node.message == "final!" || node.message == "requires_ancestor"