Add signatures for lib/dry/monads/do/all.rb

Author: aaronmallen

lib/dry/monads/do/all.rb

@@ -57,8 +57,9 @@ module Do
       module All
         # @private
         class MethodTracker < ::Module
-          attr_reader :wrappers
+          attr_reader :wrappers #: ::Hash[::Class | ::Module, ::Module]
 
+          # @rbs (::Hash[::Class | ::Module, ::Module] wrappers) -> void
           def initialize(wrappers)
             super()
 
@@ -86,11 +87,13 @@ def included(base)
             end
           end
 
+          # @rbs (::Class | ::Module target) -> void
           def extend_object(target)
             super
             target.prepend(wrappers[target])
           end
 
+          # @rbs (::Class | ::Module target, ::Symbol method) -> void
           def wrap_method(target, method)
             visibility = Do.method_visibility(target, method)
             Do.wrap_method(wrappers[target], method, visibility)
@@ -99,6 +102,7 @@ def wrap_method(target, method)
 
         class << self
           # @api private
+          # @rbs (::Class | ::Module base) -> void
           def included(base)
             super
 
@@ -110,6 +114,7 @@ def included(base)
           end
 
           # @api private
+          # @rbs (::Class | ::Module klass, ::Module target) -> void
           def wrap_defined_methods(klass, target)
             klass.public_instance_methods(false).each do |m|
               Do.wrap_method(target, m, :public)
@@ -128,6 +133,7 @@ def wrap_defined_methods(klass, target)
         # @api private
         module InstanceMixin
           # @api private
+          # @rbs (::Class | ::Module object) -> void
           def extended(object)
             super
 
@@ -156,6 +162,7 @@ def extended(object)
 
     if ::Gem::Version.new(::RUBY_VERSION) >= ::Gem::Version.new("3.4.0")
       ::Warning.singleton_class.prepend(::Module.new {
+        # @rbs (::String message, ?category: ::Symbol?, **__todo__) -> void
         def warn(message, category: nil, **)
           if message.include?("lib/dry/monads/do.rb") &&
             message.include?("warning: the block passed to")

sig/dry/monads/do/all.rbs

@@ -56,32 +56,32 @@ module Dry
       module All
         # @private
         class MethodTracker < ::Module
-          attr_reader wrappers: untyped
+          attr_reader wrappers: ::Hash[::Class | ::Module, ::Module]
 
-          def initialize: (untyped wrappers) -> untyped
+          def initialize: (::Hash[::Class | ::Module, ::Module] wrappers) -> void
 
-          def extend_object: (untyped target) -> untyped
+          def extend_object: (::Class | ::Module target) -> void
 
-          def wrap_method: (untyped target, untyped method) -> untyped
+          def wrap_method: (::Class | ::Module target, ::Symbol method) -> void
         end
 
         # @api private
-        def self.included: (untyped base) -> untyped
+        def self.included: (::Class | ::Module base) -> void
 
         # @api private
-        def self.wrap_defined_methods: (untyped klass, untyped target) -> untyped
+        def self.wrap_defined_methods: (::Class | ::Module klass, ::Module target) -> void
 
         # @api private
         module InstanceMixin
           # @api private
-          def extended: (untyped object) -> untyped
+          def extended: (::Class | ::Module object) -> void
         end
 
         extend InstanceMixin
       end
     end
 
-    def warn: (untyped message, ?category: untyped, **untyped) -> untyped
+    def warn: (::String message, ?category: ::Symbol?, **__todo__) -> void
   end
 end