From 208644c9aff43c9c683a7e758599fea05432bc37 Mon Sep 17 00:00:00 2001
From: Vincent Langlet <vincentlanglet@hotmail.fr>
Date: Thu, 25 Jul 2024 13:53:44 +0200
Subject: [PATCH] From stmt

---
 .../Statements/Expression/AssignmentAnalyzer.php     |  7 ++++---
 .../Analyzer/Statements/ExpressionAnalyzer.php       | 12 ++++++------
 2 files changed, 10 insertions(+), 9 deletions(-)

diff --git a/src/Psalm/Internal/Analyzer/Statements/Expression/AssignmentAnalyzer.php b/src/Psalm/Internal/Analyzer/Statements/Expression/AssignmentAnalyzer.php
index 679abaa347a..95f17603e4b 100644
--- a/src/Psalm/Internal/Analyzer/Statements/Expression/AssignmentAnalyzer.php
+++ b/src/Psalm/Internal/Analyzer/Statements/Expression/AssignmentAnalyzer.php
@@ -896,9 +896,10 @@ public static function analyzeAssignmentOperation(
     public static function analyzeAssignmentRef(
         StatementsAnalyzer $statements_analyzer,
         PhpParser\Node\Expr\AssignRef $stmt,
-        Context $context
+        Context $context,
+        ?PhpParser\Node\Stmt $from_stmt
     ): bool {
-        ExpressionAnalyzer::analyze($statements_analyzer, $stmt->expr, $context, false, null, false, null, true);
+        ExpressionAnalyzer::analyze($statements_analyzer, $stmt->expr, $context, false, null, null, null, true);
 
         $lhs_var_id = ExpressionIdentifier::getExtendedVarId(
             $stmt->var,
@@ -912,7 +913,7 @@ public static function analyzeAssignmentRef(
             $statements_analyzer,
         );
 
-        $doc_comment = $stmt->getDocComment();
+        $doc_comment = $stmt->getDocComment() ?? ($from_stmt ? $from_stmt->getDocComment() : null);
         if ($doc_comment) {
             try {
                 $var_comments = CommentAnalyzer::getTypeFromComment(
diff --git a/src/Psalm/Internal/Analyzer/Statements/ExpressionAnalyzer.php b/src/Psalm/Internal/Analyzer/Statements/ExpressionAnalyzer.php
index 5474cdb03a3..41c1ea8fd4a 100644
--- a/src/Psalm/Internal/Analyzer/Statements/ExpressionAnalyzer.php
+++ b/src/Psalm/Internal/Analyzer/Statements/ExpressionAnalyzer.php
@@ -78,7 +78,7 @@ public static function analyze(
         Context $context,
         bool $array_assignment = false,
         ?Context $global_context = null,
-        bool $from_stmt = false,
+        ?PhpParser\Node\Stmt $from_stmt = null,
         ?TemplateResult $template_result = null,
         bool $assigned_to_reference = false
     ): bool {
@@ -183,7 +183,7 @@ private static function handleExpression(
         Context $context,
         bool $array_assignment,
         ?Context $global_context,
-        bool $from_stmt,
+        ?PhpParser\Node\Stmt $from_stmt,
         ?TemplateResult $template_result = null,
         bool $assigned_to_reference = false
     ): bool {
@@ -293,7 +293,7 @@ private static function handleExpression(
                 $stmt,
                 $context,
                 0,
-                $from_stmt,
+                $from_stmt !== null,
             );
         }
 
@@ -380,7 +380,7 @@ private static function handleExpression(
         }
 
         if ($stmt instanceof PhpParser\Node\Expr\AssignRef) {
-            if (!AssignmentAnalyzer::analyzeAssignmentRef($statements_analyzer, $stmt, $context)) {
+            if (!AssignmentAnalyzer::analyzeAssignmentRef($statements_analyzer, $stmt, $context, $from_stmt)) {
                 IssueBuffer::maybeAdd(
                     new UnsupportedReferenceUsage(
                         "This reference cannot be analyzed by Psalm",
@@ -499,7 +499,7 @@ private static function analyzeAssignment(
         StatementsAnalyzer $statements_analyzer,
         PhpParser\Node\Expr $stmt,
         Context $context,
-        bool $from_stmt
+        ?PhpParser\Node\Stmt $from_stmt
     ): bool {
         $assignment_type = AssignmentAnalyzer::analyze(
             $statements_analyzer,
@@ -507,7 +507,7 @@ private static function analyzeAssignment(
             $stmt->expr,
             null,
             $context,
-            $stmt->getDocComment(),
+            $stmt->getDocComment() ?? ($from_stmt ? $from_stmt->getDocComment() : null),
             [],
             !$from_stmt ? $stmt : null,
         );