From 6382af1c97a5561948a27d257143f3a5b6026488 Mon Sep 17 00:00:00 2001
From: Vincent Langlet <vincentlanglet@hotmail.fr>
Date: Thu, 3 Nov 2022 14:21:18 +0100
Subject: [PATCH] Add support for phpstan assertions

---
 .../Reflector/FunctionLikeDocblockParser.php  | 56 +++++++++++--------
 1 file changed, 34 insertions(+), 22 deletions(-)

diff --git a/src/Psalm/Internal/PhpVisitor/Reflector/FunctionLikeDocblockParser.php b/src/Psalm/Internal/PhpVisitor/Reflector/FunctionLikeDocblockParser.php
index 2eb713b9458..a7900376992 100644
--- a/src/Psalm/Internal/PhpVisitor/Reflector/FunctionLikeDocblockParser.php
+++ b/src/Psalm/Internal/PhpVisitor/Reflector/FunctionLikeDocblockParser.php
@@ -176,7 +176,7 @@ public static function parse(
             }
         }
 
-        foreach (['psalm-self-out', 'psalm-this-out'] as $alias) {
+        foreach (['psalm-self-out', 'psalm-this-out', 'phpstan-self-out', 'phpstan-this-out'] as $alias) {
             if (isset($parsed_docblock->tags[$alias])) {
                 foreach ($parsed_docblock->tags[$alias] as $offset => $param) {
                     $line_parts = CommentAnalyzer::splitDocLine($param);
@@ -465,36 +465,48 @@ public static function parse(
             }
         }
 
-        if (isset($parsed_docblock->tags['psalm-assert'])) {
-            foreach ($parsed_docblock->tags['psalm-assert'] as $assertion) {
-                $line_parts = self::sanitizeAssertionLineParts(CommentAnalyzer::splitDocLine($assertion));
+        foreach (['psalm-assert', 'phpstan-assert'] as $assert) {
+            if (isset($parsed_docblock->tags[$assert])) {
+                foreach ($parsed_docblock->tags[$assert] as $assertion) {
+                    $line_parts = self::sanitizeAssertionLineParts(CommentAnalyzer::splitDocLine($assertion));
 
-                $info->assertions[] = [
-                    'type' => $line_parts[0],
-                    'param_name' => $line_parts[1][0] === '$' ? substr($line_parts[1], 1) : $line_parts[1],
-                ];
+                    $info->assertions[] = [
+                        'type'       => $line_parts[0],
+                        'param_name' => $line_parts[1][0] === '$' ? substr($line_parts[1], 1) : $line_parts[1],
+                    ];
+                }
+
+                break;
             }
         }
 
-        if (isset($parsed_docblock->tags['psalm-assert-if-true'])) {
-            foreach ($parsed_docblock->tags['psalm-assert-if-true'] as $assertion) {
-                $line_parts = self::sanitizeAssertionLineParts(CommentAnalyzer::splitDocLine($assertion));
+        foreach (['psalm-assert-if-true', 'phpstan-assert-if-true'] as $assert) {
+            if (isset($parsed_docblock->tags[$assert])) {
+                foreach ($parsed_docblock->tags[$assert] as $assertion) {
+                    $line_parts = self::sanitizeAssertionLineParts(CommentAnalyzer::splitDocLine($assertion));
 
-                $info->if_true_assertions[] = [
-                    'type' => $line_parts[0],
-                    'param_name' => $line_parts[1][0] === '$' ? substr($line_parts[1], 1) : $line_parts[1],
-                ];
+                    $info->if_true_assertions[] = [
+                        'type'       => $line_parts[0],
+                        'param_name' => $line_parts[1][0] === '$' ? substr($line_parts[1], 1) : $line_parts[1],
+                    ];
+                }
+
+                break;
             }
         }
 
-        if (isset($parsed_docblock->tags['psalm-assert-if-false'])) {
-            foreach ($parsed_docblock->tags['psalm-assert-if-false'] as $assertion) {
-                $line_parts = self::sanitizeAssertionLineParts(CommentAnalyzer::splitDocLine($assertion));
+        foreach (['psalm-assert-if-false', 'phpstan-assert-if-false'] as $assert) {
+            if (isset($parsed_docblock->tags[$assert])) {
+                foreach ($parsed_docblock->tags[$assert] as $assertion) {
+                    $line_parts = self::sanitizeAssertionLineParts(CommentAnalyzer::splitDocLine($assertion));
 
-                $info->if_false_assertions[] = [
-                    'type' => $line_parts[0],
-                    'param_name' => $line_parts[1][0] === '$' ? substr($line_parts[1], 1) : $line_parts[1],
-                ];
+                    $info->if_false_assertions[] = [
+                        'type'       => $line_parts[0],
+                        'param_name' => $line_parts[1][0] === '$' ? substr($line_parts[1], 1) : $line_parts[1],
+                    ];
+                }
+
+                break;
             }
         }