sorbet-runtime 0.5.5858 → 0.5.5866

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ddae56dee53b7a4ae6c91b0ff2a279c9238255402a97a12d5c1cff9efd6e5777
-  data.tar.gz: 8de90de6b27854ae87ac9671d799e7d2fa2a2b4a4bde1a18d1574975b47f34a3
+  metadata.gz: 06f33bba34b08fbaf66e39ea5e8db2cee1d70f5f84340b85e6ed43582cebc745
+  data.tar.gz: dd8398d38b60baae1f7547d4bf8f1e39bf74e3c24bee18d72dd531011d44a35e
 SHA512:
-  metadata.gz: 33be9484fb679eacae0992574ff30492e4549ae0a63a4c17d3348bf7e514bf6f62162f58290c6b9a674e38a27343be3a13f9b9bdcd20d99d804a47719bf4206c
-  data.tar.gz: 560140944b988f084d31c7b04ff32d9f8a02f87e6a0a99d3956493f0cf7ae87fad51f032b0aced1ba6a48bf1eaa7c1b89bfc15a10330c752229dc541a2b4ae17
+  metadata.gz: eea13fe538193883ed4d03053106273d8dfabf654b9db38d18c5fae05960743ede7ea777b3f63ac40f6a3951d5e3148693b930c5f1beffb3dd9c50a3cc264706
+  data.tar.gz: b3c358a3a9e0c25d53667e37140f3d92c68a0be5b60dd0f32f80ad52e598789f93e657e804356fe0e204f2e973946ab6b55630ff6611cb6745f760aff728799c

data/lib/types/private/methods/_methods.rb

@@ -208,7 +208,7 @@ module T::Private::Methods
     T::Private::ClassUtils.replace_method(mod, method_name) do |*args, &blk|
       method_sig = T::Private::Methods.maybe_run_sig_block_for_method(new_method)
       method_sig ||= T::Private::Methods._handle_missing_method_signature(
-        is_singleton_method ? self.singleton_class : self.class,
+        self,
         original_method,
         __callee__,
       )
@@ -242,30 +242,40 @@ module T::Private::Methods
     end
   end
 
-  def self._handle_missing_method_signature(klass, original_method, callee)
+  def self._handle_missing_method_signature(receiver, original_method, callee)
     method_sig = T::Private::Methods.signature_for_method(original_method)
-
-    aliasing_method = klass.instance_method(callee)
-    aliasing_mod = aliasing_method.owner
-
-    if method_sig && aliasing_method != original_method && aliasing_method.original_name == original_method.name
-      # We're handling a case where `alias` or `alias_method` was called for a
-      # method which had already had a `sig` applied.
-      #
-      # Note, this logic is duplicated above, make sure to keep changes in sync.
-      if method_sig.check_level == :always || (method_sig.check_level == :tests && T::Private::RuntimeLevels.check_tests?)
-        # Checked, so copy the original signature to the aliased copy.
-        T::Private::Methods.unwrap_method(aliasing_mod, method_sig, aliasing_method)
-      else
-        # Unchecked, so just make `alias_method` behave as if it had been called pre-sig.
-        aliasing_mod.send(:alias_method, callee, original_method.name)
-      end
-    else
-      raise "`sig` not present for method `#{aliasing_method.name}` but you're trying to run it anyways. " \
+    if !method_sig
+      raise "`sig` not present for method `#{callee}` on #{receiver.inspect} but you're trying to run it anyways. " \
         "This should only be executed if you used `alias_method` to grab a handle to a method after `sig`ing it, but that clearly isn't what you are doing. " \
         "Maybe look to see if an exception was thrown in your `sig` lambda or somehow else your `sig` wasn't actually applied to the method."
     end
 
+    if receiver.class <= original_method.owner
+      receiving_class = receiver.class
+    elsif receiver.singleton_class <= original_method.owner
+      receiving_class = receiver.singleton_class
+    elsif receiver.is_a?(Module) && receiver <= original_method.owner
+      receiving_class = receiver
+    else
+      raise "#{receiver} is not related to #{original_method} - how did we get here?"
+    end
+
+    # Check for a case where `alias` or `alias_method` was called for a
+    # method which had already had a `sig` applied. In that case, we want
+    # to avoid hitting this slow path again, by moving to a faster validator
+    # just like we did or will for the original method.
+    #
+    # If this isn't an `alias` or `alias_method` case, we're probably in the
+    # middle of some metaprogramming using a Method object, e.g. a pattern like
+    # `arr.map(&method(:foo))`. There's nothing really we can do to optimize
+    # that here.
+    receiving_method = receiving_class.instance_method(callee)
+    if receiving_method != original_method && receiving_method.original_name == original_method.name
+      aliasing_mod = receiving_method.owner
+      method_sig = method_sig.as_alias(callee)
+      unwrap_method(aliasing_mod, method_sig, original_method)
+    end
+
     method_sig
   end
 
@@ -287,7 +297,7 @@ module T::Private::Methods
         Signature.new_untyped(method: original_method)
       end
 
-    unwrap_method(hook_mod, signature, original_method)
+    unwrap_method(signature.method.owner, signature, original_method)
     signature
   end
 
@@ -339,8 +349,8 @@ module T::Private::Methods
     end
   end
 
-  def self.unwrap_method(hook_mod, signature, original_method)
-    maybe_wrapped_method = CallValidation.wrap_method_if_needed(signature.method.owner, signature, original_method)
+  def self.unwrap_method(mod, signature, original_method)
+    maybe_wrapped_method = CallValidation.wrap_method_if_needed(mod, signature, original_method)
     @signatures_by_method[method_to_key(maybe_wrapped_method)] = signature
   end
 

data/lib/types/private/methods/signature.rb

@@ -121,6 +121,15 @@ class T::Private::Methods::Signature
     end
   end
 
+  attr_writer :method_name
+  protected :method_name=
+
+  def as_alias(alias_name)
+    new_sig = clone
+    new_sig.method_name = alias_name
+    new_sig
+  end
+
   def arg_count
     @arg_types.length
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sorbet-runtime
 version: !ruby/object:Gem::Version
-  version: 0.5.5858
+  version: 0.5.5866
 platform: ruby
 authors:
 - Stripe
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-07-31 00:00:00.000000000 Z
+date: 2020-08-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: minitest