Merge pull request #2401 from Shopify/at-runtime-helpers

Move runtime helpers under Tapioca::Runtime

Author: Morriar

lib/tapioca.rb

@@ -9,42 +9,6 @@ module Tapioca
 
   @traces = [] #: Array[TracePoint]
 
-  NOOP_METHOD = ->(*_args, **_kwargs, &_block) {} #: ^() -> void
-  private_constant :NOOP_METHOD
-
-  class << self
-    extend T::Sig
-
-    #: [Result] { -> Result } -> Result
-    def silence_warnings(&blk)
-      original_verbosity = $VERBOSE
-      $VERBOSE = nil
-      ::Gem::DefaultUserInteraction.use_ui(::Gem::SilentUI.new) do
-        blk.call
-      end
-    ensure
-      $VERBOSE = original_verbosity
-    end
-
-    #: [Result] { -> Result } -> Result
-    def with_disabled_exits(&block)
-      original_abort = Kernel.instance_method(:abort)
-      original_exit = Kernel.instance_method(:exit)
-
-      begin
-        Kernel.define_method(:abort, NOOP_METHOD)
-        Kernel.define_method(:exit, NOOP_METHOD)
-
-        Tapioca.silence_warnings do
-          block.call
-        end
-      ensure
-        Kernel.define_method(:exit, original_exit)
-        Kernel.define_method(:abort, original_abort)
-      end
-    end
-  end
-
   class Error < StandardError; end
 
   LIB_ROOT_DIR = T.must(__dir__) #: String

lib/tapioca/commands/abstract_gem.rb

@@ -70,7 +70,7 @@ def compile_gem_rbi(gem)
           reason: "types exported from the `#{gem.name}` gem",
         ) if @file_header
 
-        rbi.root = Tapioca.with_disabled_exits do
+        rbi.root = Runtime.with_disabled_exits do
           Tapioca::Gem::Pipeline.new(
             gem,
             include_doc: @include_doc,

lib/tapioca/commands/command.rb

@@ -21,7 +21,7 @@ def initialize
       # @final
       #: -> void
       def run
-        Tapioca.silence_warnings do
+        Runtime.silence_warnings do
           execute
         end
       end

lib/tapioca/helpers/test/content.rb

@@ -28,7 +28,7 @@ def remove_tmp_path
         #: (String name, String content, ?require_file: bool) -> String
         def add_ruby_file(name, content, require_file: true)
           add_content_file(name, content).tap do |file_name|
-            Tapioca.silence_warnings { require(file_name) } if require_file
+            Runtime.silence_warnings { require(file_name) } if require_file
           end
         end
 

lib/tapioca/internal.rb

@@ -4,6 +4,7 @@
 require "set"
 
 require "tapioca"
+require "tapioca/runtime/helpers"
 require "tapioca/runtime/reflection"
 require "tapioca/runtime/trackers"
 

lib/tapioca/loaders/gem.rb

@@ -47,7 +47,7 @@ def require_gem_file
 
         bundle_loaded = true #: bool
 
-        Tapioca.with_disabled_exits do
+        Runtime.with_disabled_exits do
           load_bundle(@bundle, @prerequire, @postrequire, @halt_upon_load_error)
         rescue LoadError => e
           bundle_loaded = false

lib/tapioca/runtime/helpers.rb

@@ -0,0 +1,40 @@
+# typed: strict
+# frozen_string_literal: true
+
+module Tapioca
+  module Runtime
+    NOOP_METHOD = ->(*_args, **_kwargs, &_block) {} #: ^() -> void
+    private_constant :NOOP_METHOD
+
+    class << self
+      #: [Result] { -> Result } -> Result
+      def silence_warnings(&blk)
+        original_verbosity = $VERBOSE
+        $VERBOSE = nil
+        ::Gem::DefaultUserInteraction.use_ui(::Gem::SilentUI.new) do
+          blk.call
+        end
+      ensure
+        $VERBOSE = original_verbosity
+      end
+
+      #: [Result] { -> Result } -> Result
+      def with_disabled_exits(&block)
+        original_abort = Kernel.instance_method(:abort)
+        original_exit = Kernel.instance_method(:exit)
+
+        begin
+          Kernel.define_method(:abort, NOOP_METHOD)
+          Kernel.define_method(:exit, NOOP_METHOD)
+
+          silence_warnings do
+            block.call
+          end
+        ensure
+          Kernel.define_method(:exit, original_exit)
+          Kernel.define_method(:abort, original_abort)
+        end
+      end
+    end
+  end
+end

lib/tapioca/runtime/trackers/autoload.rb

@@ -15,7 +15,7 @@ class << self
 
           #: -> void
           def eager_load_all!
-            Tapioca.with_disabled_exits do
+            Runtime.with_disabled_exits do
               until @constant_names_registered_for_autoload.empty?
                 # Grab the next constant name
                 constant_name = T.must(@constant_names_registered_for_autoload.shift)

spec/tapioca/dsl/compilers/protobuf_spec.rb

@@ -598,7 +598,7 @@ def add_proto_file(name, content, require_file: true)
             _, stderr, status = Open3.capture3("protoc --proto_path=#{proto_dir} --ruby_out=#{lib_path} #{proto_path}")
             raise "Error executing protoc: #{stderr}" unless status.success?
 
-            Tapioca.silence_warnings { require("#{lib_path}/#{name}_pb.rb") } if require_file
+            Runtime.silence_warnings { require("#{lib_path}/#{name}_pb.rb") } if require_file
           end
         end
       end