Merge pull request #194 from soutaro/or_asgn_and_asgn

Support or_asgn/and_asgn with send

lib/steep/type_construction.rb

@@ -1800,13 +1800,34 @@ def synthesize(node, hint: nil)
         when :or_asgn, :and_asgn
           yield_self do
             asgn, rhs = node.children
-            type, constr = synthesize(rhs, hint: hint)
 
             case asgn.type
             when :lvasgn
+              type, constr = synthesize(rhs, hint: hint)
               constr.lvasgn(asgn, type)
             when :ivasgn
+              type, constr = synthesize(rhs, hint: hint)
               constr.ivasgn(asgn, type)
+            when :send
+              rhs_ = node.updated(:send,
+                                  [
+                                    asgn.children[0],
+                                    :"#{asgn.children[1]}=",
+                                    asgn.children[2],
+                                    rhs
+                                  ])
+              node_type = case node.type
+                          when :or_asgn
+                            :or
+                          when :and_asgn
+                            :and
+                          end
+              node_ = node.updated(node_type, [asgn, rhs_])
+
+              synthesize(node_, hint: hint)
+            else
+              Steep.logger.error { "#{node.type} with #{asgn.type} lhs is not supported"}
+              fallback_to_any(node)
             end
           end
 

test/type_construction_test.rb

@@ -5173,4 +5173,21 @@ def set
       end
     end
   end
+
+  def test_orasgn_call
+    with_checker do |checker|
+      source = parse_ruby(<<-RUBY)
+a = [1,2,3]
+
+a[0] ||= 3
+a[1] &&= 4
+        RUBY
+
+      with_standard_construction(checker, source) do |construction, typing|
+        construction.synthesize(source.node)
+
+        assert_no_error typing
+      end
+    end
+  end
 end