From 299eca4daa08664cdd4af8127b3934663d974f63 Mon Sep 17 00:00:00 2001
From: Pete Walker <pete.walker@ourpeople.com>
Date: Fri, 21 Jan 2022 17:31:48 +0000
Subject: [PATCH] fix: Add PHP version checks / more tests

---
 .../Reflector/ClassLikeNodeScanner.php        |  17 +--
 .../Reflector/FunctionLikeNodeScanner.php     |   8 ++
 .../PhpVisitor/Reflector/TypeHintResolver.php |  35 ++++++
 tests/NativeIntersectionsTest.php             |  42 +++++++
 tests/NativeUnionsTest.php                    | 119 ++++++++++++++++++
 5 files changed, 213 insertions(+), 8 deletions(-)
 create mode 100644 tests/NativeUnionsTest.php

diff --git a/src/Psalm/Internal/PhpVisitor/Reflector/ClassLikeNodeScanner.php b/src/Psalm/Internal/PhpVisitor/Reflector/ClassLikeNodeScanner.php
index 4ae30b5f4db..4c77a4ae1be 100644
--- a/src/Psalm/Internal/PhpVisitor/Reflector/ClassLikeNodeScanner.php
+++ b/src/Psalm/Internal/PhpVisitor/Reflector/ClassLikeNodeScanner.php
@@ -1488,22 +1488,23 @@ private function visitPropertyDeclaration(
             $parser_property_type = $stmt->type;
             /** @var Identifier|IntersectionType|Name|NullableType|UnionType $parser_property_type */
 
+            $signature_type_location = new CodeLocation(
+                $this->file_scanner,
+                $parser_property_type,
+                null,
+                false,
+                CodeLocation::FUNCTION_RETURN_TYPE
+            );
+
             $signature_type = TypeHintResolver::resolve(
                 $parser_property_type,
+                $signature_type_location,
                 $this->codebase,
                 $this->file_storage,
                 $this->storage,
                 $this->aliases,
                 $this->codebase->analysis_php_version_id
             );
-
-            $signature_type_location = new CodeLocation(
-                $this->file_scanner,
-                $parser_property_type,
-                null,
-                false,
-                CodeLocation::FUNCTION_RETURN_TYPE
-            );
         }
 
         $doc_var_group_type = $var_comment->type ?? null;
diff --git a/src/Psalm/Internal/PhpVisitor/Reflector/FunctionLikeNodeScanner.php b/src/Psalm/Internal/PhpVisitor/Reflector/FunctionLikeNodeScanner.php
index 077445c09e8..b67940138ef 100644
--- a/src/Psalm/Internal/PhpVisitor/Reflector/FunctionLikeNodeScanner.php
+++ b/src/Psalm/Internal/PhpVisitor/Reflector/FunctionLikeNodeScanner.php
@@ -430,6 +430,10 @@ public function start(PhpParser\Node\FunctionLike $stmt, bool $fake_method = fal
 
             $storage->return_type = TypeHintResolver::resolve(
                 $original_type,
+                new CodeLocation(
+                    $this->file_scanner,
+                    $original_type
+                ),
                 $this->codebase,
                 $this->file_storage,
                 $this->classlike_storage,
@@ -825,6 +829,10 @@ private function getTranslatedFunctionParam(
 
             $param_type = TypeHintResolver::resolve(
                 $param_typehint,
+                new CodeLocation(
+                    $this->file_scanner,
+                    $param_typehint
+                ),
                 $this->codebase,
                 $this->file_storage,
                 $this->classlike_storage,
diff --git a/src/Psalm/Internal/PhpVisitor/Reflector/TypeHintResolver.php b/src/Psalm/Internal/PhpVisitor/Reflector/TypeHintResolver.php
index f11b97a87c9..8293f70b315 100644
--- a/src/Psalm/Internal/PhpVisitor/Reflector/TypeHintResolver.php
+++ b/src/Psalm/Internal/PhpVisitor/Reflector/TypeHintResolver.php
@@ -10,7 +10,10 @@
 use PhpParser\Node\UnionType;
 use Psalm\Aliases;
 use Psalm\Codebase;
+use Psalm\CodeLocation;
 use Psalm\Internal\Analyzer\ClassLikeAnalyzer;
+use Psalm\Issue\ParseError;
+use Psalm\IssueBuffer;
 use Psalm\Storage\ClassLikeStorage;
 use Psalm\Storage\FileStorage;
 use Psalm\Type;
@@ -31,6 +34,7 @@ class TypeHintResolver
      */
     public static function resolve(
         PhpParser\NodeAbstract $hint,
+        CodeLocation $code_location,
         Codebase $codebase,
         FileStorage $file_storage,
         ?ClassLikeStorage $classlike_storage,
@@ -44,9 +48,19 @@ public static function resolve(
                 throw new UnexpectedValueException('bad');
             }
 
+            if ($analysis_php_version_id < 8_00_00) {
+                IssueBuffer::add(
+                    new ParseError(
+                        'Union types are not supported in PHP < 8',
+                        $code_location
+                    )
+                );
+            }
+
             foreach ($hint->types as $atomic_typehint) {
                 $resolved_type = self::resolve(
                     $atomic_typehint,
+                    $code_location,
                     $codebase,
                     $file_storage,
                     $classlike_storage,
@@ -67,9 +81,20 @@ public static function resolve(
                 throw new UnexpectedValueException('bad');
             }
 
+            if ($analysis_php_version_id < 8_10_00) {
+                IssueBuffer::add(
+                    new ParseError(
+                        'Intersection types are not supported in PHP < 8.1',
+                        $code_location
+                    )
+                );
+                return false;
+            }
+
             foreach ($hint->types as $atomic_typehint) {
                 $resolved_type = self::resolve(
                     $atomic_typehint,
+                    $code_location,
                     $codebase,
                     $file_storage,
                     $classlike_storage,
@@ -77,6 +102,16 @@ public static function resolve(
                     $analysis_php_version_id
                 );
 
+                if ($resolved_type->hasScalarType()) {
+                    IssueBuffer::add(
+                        new ParseError(
+                            'Intersection types cannot contain scalar types',
+                            $code_location
+                        )
+                    );
+                    return null;
+                }
+
                 $type = Type::intersectUnionTypes($resolved_type, $type, $codebase);
             }
 
diff --git a/tests/NativeIntersectionsTest.php b/tests/NativeIntersectionsTest.php
index 443cf2ba329..f16f57fe51f 100644
--- a/tests/NativeIntersectionsTest.php
+++ b/tests/NativeIntersectionsTest.php
@@ -103,6 +103,48 @@ function test(A&B $in): void {
                 'ignored_issues' => [],
                 'php_version' => '8.1'
             ],
+            'intersectionsNotAllowedWithUnions' => [
+                '<?php
+                    interface A {
+                    }
+                    interface B {
+                    }
+                    interface C {
+                    }
+                    function foo (A&B|C $test): A&B|C {
+                        return $test;
+                    }',
+                'error_message' => 'ParseError',
+                [],
+                false,
+                '8.1'
+            ],
+            'intersectionsNotAllowedWithNonClasses' => [
+                '<?php
+                    interface A {
+                    }
+                    function foo (A&string $test): A&string {
+                        return $test;
+                    }',
+                'error_message' => 'ParseError',
+                [],
+                false,
+                '8.1'
+            ],
+            'intersectionsNotAllowedInPHP80' => [
+                '<?php
+                    interface A {
+                    }
+                    interface B {
+                    }
+                    function foo (A&B $test): A&B {
+                        return $test;
+                    }',
+                'error_message' => 'ParseError',
+                [],
+                false,
+                '8.0'
+            ],
         ];
     }
 }
diff --git a/tests/NativeUnionsTest.php b/tests/NativeUnionsTest.php
new file mode 100644
index 00000000000..6ee8ef7129c
--- /dev/null
+++ b/tests/NativeUnionsTest.php
@@ -0,0 +1,119 @@
+<?php
+
+namespace Psalm\Tests;
+
+use Psalm\Tests\Traits\InvalidCodeAnalysisTestTrait;
+use Psalm\Tests\Traits\ValidCodeAnalysisTestTrait;
+
+class NativeUnionsTest extends TestCase
+{
+    use InvalidCodeAnalysisTestTrait;
+    use ValidCodeAnalysisTestTrait;
+
+    /**
+     * @return iterable<string,array{string,assertions?:array<string,string>,error_levels?:string[],php_version?:string}>
+     */
+    public function providerValidCodeParse(): iterable
+    {
+        return [
+            'nativeTypeUnionInConstructor' => [
+                '<?php
+                    interface A {
+                    }
+                    interface B {
+                    }
+                    class Foo {
+                        public function __construct(private A|B $self) {}
+
+                        public function self(): A|B
+                        {
+                            return $this->self;
+                        }
+                    }',
+                'assertions' => [],
+                'error_levels' => [],
+                'php_version' => '8.0'
+            ],
+            'nativeTypeUnionAsArgument' => [
+                '<?php
+                    interface A {
+                        function foo(): void;
+                    }
+                    interface B {
+                        function foo(): void;
+                    }
+                    class C implements A {
+                        function foo(): void {
+                        }
+                    }
+                    function test(A|B $in): void {
+                        $in->foo();
+                    }
+                    test(new C());
+                ',
+                'assertions' => [],
+                'error_levels' => [],
+                'php_version' => '8.0'
+            ],
+            'unionAndNullableEquivalent' => [
+                '<?php
+                    function test(string|null $in): ?string {
+                        return $in;
+                    }
+                ',
+                'assertions' => [],
+                'error_levels' => [],
+                'php_version' => '8.0'
+            ],
+        ];
+    }
+
+    /**
+     * @return iterable<string,array{string,error_message:string,1?:string[],2?:bool,3?:string}>
+     */
+    public function providerInvalidCodeParse(): iterable
+    {
+        return [
+            'invalidNativeUnionArgument' => [
+                '<?php
+                    function test(string|null $in): string|null {
+                        return $in;
+                    }
+                    test(2);
+                ',
+                'error_message' => 'InvalidScalarArgument',
+                [],
+                false,
+                '8.0'
+            ],
+            'mismatchDocblockNativeUnionArgument' => [
+                '<?php
+                    /**
+                     * @param string|null $in
+                     */
+                    function test(int|bool $in): bool {
+                        return !!$in;
+                    }
+                ',
+                'error_message' => 'MismatchingDocblockParamType',
+                [],
+                false,
+                '8.0'
+            ],
+            'unionsNotAllowedInPHP74' => [
+                '<?php
+                    interface A {
+                    }
+                    interface B {
+                    }
+                    function foo (A|B $test): A&B {
+                        return $test;
+                    }',
+                'error_message' => 'ParseError',
+                [],
+                false,
+                '7.4'
+            ],
+        ];
+    }
+}