sorbet-runtime 0.5.5855 → 0.5.5865

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 05211daadd19374c8196176c89361938ef05fd1c5fa4c02f9e9d581bcf4114bf
-  data.tar.gz: 33f7feb752c822c281276f990b99c3ceeab66a4dd63359c00cfad66643680231
+  metadata.gz: 356642261e3c9f5451beb62ee6fd4cd4e4a346b2686954ce87fe60df0796013a
+  data.tar.gz: f9884e720101c6ec2f90601ea470ae36dc4afc4728ff5015a088bb6cc063782a
 SHA512:
-  metadata.gz: 908de44e4d49035037a83f508074c273af88c8ff465147e569ecb062e6a19c818bdd0e8812830f5ce23b582fec0010e04dd9191e24095b024e04bcb07ed9ef69
-  data.tar.gz: 8a77c89234429113ac224c877a02a0fefa81fc4cc066e1c2a3b0e1b7b94129372396523e260e5280a6487f7cc41b8de2cc0efb1af63214108b664407657fa97b
+  metadata.gz: 98a17fa21e3662b1b1c7c0626faa39a3dcb4f5b45059cc3f891ecfecb51a5556279e21e70f1f8aadb394786ed874f74ef577bc145fe8e0d4450efc4c6a4b3cfc
+  data.tar.gz: 50ebe42e01b78d7d390c55dbb50f439818470b022bdf8293a0dc903584456becee442654c052f65096e7247e5881b777efd37a91ab64138af7de08160b40cc98

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.5855
+  version: 0.5.5865
 platform: ruby
 authors:
 - Stripe
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-07-30 00:00:00.000000000 Z
+date: 2020-08-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: minitest