solargraph 0.58.0 → 0.59.0.dev.1

data/lib/solargraph/pin/block.rb

@@ -1,104 +1,126 @@
-# frozen_string_literal: true
-
-module Solargraph
-  module Pin
-    class Block < Callable
-      include Breakable
-
-      # @return [Parser::AST::Node]
-      attr_reader :receiver
-
-      # @return [Parser::AST::Node]
-      attr_reader :node
-
-      # @param receiver [Parser::AST::Node, nil]
-      # @param node [Parser::AST::Node, nil]
-      # @param context [ComplexType, nil]
-      # @param args [::Array<Parameter>]
-      def initialize receiver: nil, args: [], context: nil, node: nil, **splat
-        super(**splat, parameters: args)
-        @receiver = receiver
-        @context = context
-        @return_type = ComplexType.parse('::Proc')
-        @node = node
-      end
-
-      # @param api_map [ApiMap]
-      # @return [void]
-      def rebind api_map
-        @rebind ||= maybe_rebind(api_map)
-      end
-
-      def binder
-        @rebind&.defined? ? @rebind : closure.binder
-      end
-
-      # @param yield_types [::Array<ComplexType>]
-      # @param parameters [::Array<Parameter>]
-      #
-      # @return [::Array<ComplexType>]
-      def destructure_yield_types(yield_types, parameters)
-        # yielding a tuple into a block will destructure the tuple
-        if yield_types.length == 1
-          yield_type = yield_types.first
-          return yield_type.all_params if yield_type.tuple? && yield_type.all_params.length == parameters.length
-        end
-        parameters.map.with_index { |_, idx| yield_types[idx] || ComplexType::UNDEFINED }
-      end
-
-      # @param api_map [ApiMap]
-      # @return [::Array<ComplexType>]
-      def typify_parameters(api_map)
-        chain = Parser.chain(receiver, filename, node)
-        clip = api_map.clip_at(location.filename, location.range.start)
-        locals = clip.locals - [self]
-        meths = chain.define(api_map, closure, locals)
-        # @todo Convert logic to use signatures
-        # @param meth [Pin::Method]
-        meths.each do |meth|
-          next if meth.block.nil?
-
-          yield_types = meth.block.parameters.map(&:return_type)
-          # 'arguments' is what the method says it will yield to the
-          # block; 'parameters' is what the block accepts
-          argument_types = destructure_yield_types(yield_types, parameters)
-          param_types = argument_types.each_with_index.map do |arg_type, idx|
-            param = parameters[idx]
-            param_type = chain.base.infer(api_map, param, locals)
-            unless arg_type.nil?
-              if arg_type.generic? && param_type.defined?
-                namespace_pin = api_map.get_namespace_pins(meth.namespace, closure.namespace).first
-                arg_type.resolve_generics(namespace_pin, param_type)
-              else
-                arg_type.self_to_type(chain.base.infer(api_map, self, locals)).qualify(api_map, *meth.gates)
-              end
-            end
-          end
-          return param_types if param_types.all?(&:defined?)
-        end
-        parameters.map { ComplexType::UNDEFINED }
-      end
-
-      private
-
-      # @param api_map [ApiMap]
-      # @return [ComplexType]
-      def maybe_rebind api_map
-        return ComplexType::UNDEFINED unless receiver
-
-        chain = Parser.chain(receiver, location.filename)
-        locals = api_map.source_map(location.filename).locals_at(location)
-        receiver_pin = chain.define(api_map, closure, locals).first
-        return ComplexType::UNDEFINED unless receiver_pin
-
-        types = receiver_pin.docstring.tag(:yieldreceiver)&.types
-        return ComplexType::UNDEFINED unless types&.any?
-
-        target = chain.base.infer(api_map, receiver_pin, locals)
-        target = full_context unless target.defined?
-
-        ComplexType.try_parse(*types).qualify(api_map, *receiver_pin.gates).self_to_type(target)
-      end
-    end
-  end
-end
+# frozen_string_literal: true
+
+module Solargraph
+  module Pin
+    class Block < Callable
+      include Breakable
+
+      # @return [Parser::AST::Node]
+      attr_reader :receiver
+
+      # @return [Parser::AST::Node]
+      attr_reader :node
+
+      # @param receiver [Parser::AST::Node, nil]
+      # @param node [Parser::AST::Node, nil]
+      # @param context [ComplexType, nil]
+      # @param args [::Array<Parameter>]
+      def initialize receiver: nil, args: [], context: nil, node: nil, **splat
+        super(**splat, parameters: args)
+        @receiver = receiver
+        @context = context
+        @return_type = ComplexType.parse('::Proc')
+        @node = node
+        @name = '<block>'
+      end
+
+      # @param api_map [ApiMap]
+      # @return [void]
+      def rebind api_map
+        @rebind ||= maybe_rebind(api_map)
+      end
+
+      def binder
+        out = @rebind if @rebind&.defined?
+        out ||= super
+      end
+
+      def context
+        @context = @rebind if @rebind&.defined?
+        super
+      end
+
+      # @param yield_types [::Array<ComplexType>]
+      # @param parameters [::Array<Parameter>]
+      #
+      # @return [::Array<ComplexType>]
+      def destructure_yield_types(yield_types, parameters)
+        # yielding a tuple into a block will destructure the tuple
+        if yield_types.length == 1
+          yield_type = yield_types.first
+          return yield_type.all_params if yield_type.tuple? && yield_type.all_params.length == parameters.length
+        end
+        parameters.map.with_index { |_, idx| yield_types[idx] || ComplexType::UNDEFINED }
+      end
+
+      # @param api_map [ApiMap]
+      # @return [::Array<ComplexType>]
+      def typify_parameters(api_map)
+        chain = Parser.chain(receiver, filename, node)
+        # @sg-ignore Need to add nil check here
+        clip = api_map.clip_at(location.filename, location.range.start)
+        locals = clip.locals - [self]
+        # @sg-ignore Need to add nil check here
+        meths = chain.define(api_map, closure, locals)
+        # @todo Convert logic to use signatures
+        # @param meth [Pin::Method]
+        meths.each do |meth|
+          next if meth.block.nil?
+
+          # @sg-ignore flow sensitive typing needs to handle attrs
+          yield_types = meth.block.parameters.map(&:return_type)
+          # 'arguments' is what the method says it will yield to the
+          # block; 'parameters' is what the block accepts
+          argument_types = destructure_yield_types(yield_types, parameters)
+          param_types = argument_types.each_with_index.map do |arg_type, idx|
+            param = parameters[idx]
+            param_type = chain.base.infer(api_map, param, locals)
+            unless arg_type.nil?
+              if arg_type.generic? && param_type.defined?
+                # @sg-ignore Need to add nil check here
+                namespace_pin = api_map.get_namespace_pins(meth.namespace, closure.namespace).first
+                arg_type.resolve_generics(namespace_pin, param_type)
+              else
+                arg_type.self_to_type(chain.base.infer(api_map, self, locals)).qualify(api_map, *meth.gates)
+              end
+            end
+          end
+          return param_types if param_types.all?(&:defined?)
+        end
+        parameters.map { ComplexType::UNDEFINED }
+      end
+
+      private
+
+      # @param api_map [ApiMap]
+      # @return [ComplexType]
+      def maybe_rebind api_map
+        return ComplexType::UNDEFINED unless receiver
+
+        # @sg-ignore Need to add nil check here
+        chain = Parser.chain(receiver, location.filename, node)
+        # @sg-ignore Need to add nil check here
+        locals = api_map.source_map(location.filename).locals_at(location)
+        # @sg-ignore Need to add nil check here
+        receiver_pin = chain.define(api_map, closure, locals).first
+        return ComplexType::UNDEFINED unless receiver_pin
+
+        types = receiver_pin.docstring.tag(:yieldreceiver)&.types
+        return ComplexType::UNDEFINED unless types&.any?
+
+        name_pin = self
+        # if we have Foo.bar { |x| ... }, and the bar method references self...
+        target = if chain.base.defined?
+                   # figure out Foo
+                   chain.base.infer(api_map, name_pin, locals)
+                 else
+                   # if not, any self there must be the context of our closure
+                   # @sg-ignore Need to add nil check here
+                   closure.full_context
+                 end
+
+        ComplexType.try_parse(*types).qualify(api_map, *receiver_pin.gates).self_to_type(target)
+      end
+    end
+  end
+end

data/lib/solargraph/pin/breakable.rb

@@ -1,9 +1,13 @@
-module Solargraph
-  module Pin
-    # Mix-in for pins which enclose code which the 'break' statement works with-in - e.g., blocks, when, until, ...
-    module Breakable
-      # @return [Parser::AST::Node]
-      attr_reader :node
-    end
-  end
-end
+module Solargraph
+  module Pin
+    # Mix-in for pins which enclose code which the 'break' statement
+    # works with-in - e.g., blocks, when, until, ...
+    module Breakable
+      # @return [Parser::AST::Node]
+      attr_reader :node
+
+      # @return [Location, nil]
+      attr_reader :location
+    end
+  end
+end