From 55c71a16abd68471290ade4e475f83f35f83c915 Mon Sep 17 00:00:00 2001
From: robchett <robchett@gmail.com>
Date: Thu, 2 Nov 2023 11:59:42 +0000
Subject: [PATCH] Allow enum cases to be global constants

---
 .../Expression/SimpleTypeInferer.php          | 16 +++++---
 tests/EnumTest.php                            | 37 +++++++++++++++++++
 2 files changed, 48 insertions(+), 5 deletions(-)

diff --git a/src/Psalm/Internal/Analyzer/Statements/Expression/SimpleTypeInferer.php b/src/Psalm/Internal/Analyzer/Statements/Expression/SimpleTypeInferer.php
index 101196a2ff0..baead15e9da 100644
--- a/src/Psalm/Internal/Analyzer/Statements/Expression/SimpleTypeInferer.php
+++ b/src/Psalm/Internal/Analyzer/Statements/Expression/SimpleTypeInferer.php
@@ -12,6 +12,7 @@
 use Psalm\FileSource;
 use Psalm\Internal\Analyzer\ClassLikeAnalyzer;
 use Psalm\Internal\Analyzer\Statements\Expression\BinaryOp\ArithmeticOpAnalyzer;
+use Psalm\Internal\Analyzer\Statements\Expression\Fetch\ConstFetchAnalyzer;
 use Psalm\Internal\Analyzer\StatementsAnalyzer;
 use Psalm\Internal\Provider\NodeDataProvider;
 use Psalm\Internal\Type\TypeCombiner;
@@ -261,23 +262,28 @@ public static function infer(
         }
 
         if ($stmt instanceof PhpParser\Node\Expr\ConstFetch) {
-            $name = strtolower($stmt->name->getFirst());
-            if ($name === 'false') {
+            $name = $stmt->name->getFirst();
+            $name_lowercase = strtolower($name);
+            if ($name_lowercase === 'false') {
                 return Type::getFalse();
             }
 
-            if ($name === 'true') {
+            if ($name_lowercase === 'true') {
                 return Type::getTrue();
             }
 
-            if ($name === 'null') {
+            if ($name_lowercase === 'null') {
                 return Type::getNull();
             }
 
-            if ($stmt->name->getFirst() === '__NAMESPACE__') {
+            if ($name === '__NAMESPACE__') {
                 return Type::getString($aliases->namespace);
             }
 
+            if ($type = ConstFetchAnalyzer::getGlobalConstType($codebase, $name, $name)) {
+                return $type;
+            }
+
             return null;
         }
 
diff --git a/tests/EnumTest.php b/tests/EnumTest.php
index ebc67a05e1e..2d7acb238f3 100644
--- a/tests/EnumTest.php
+++ b/tests/EnumTest.php
@@ -657,6 +657,21 @@ enum BarEnum: int {
                 'ignored_issues' => [],
                 'php_version' => '8.1',
             ],
+            'backedEnumCaseValueFromGlobalConstant' => [
+                'code' => <<<'PHP'
+                    <?php
+                    enum FooEnum: string {
+                        case FOO = \PHP_BINARY;
+                    }
+
+                    enum BarEnum: int {
+                        case BAR = \UPLOAD_ERR_OK;
+                    }
+                    PHP,
+                'assertions' => [],
+                'ignored_issues' => [],
+                'php_version' => '8.1',
+            ],
         ];
     }
 
@@ -1107,6 +1122,28 @@ enum Bar: int
                 'ignored_issues' => [],
                 'php_version' => '8.1',
             ],
+            'stringBackedEnumCaseValueFromGlobalConstant' => [
+                'code' => '<?php
+                    enum Bar: string
+                    {
+                        case Foo = \UPLOAD_ERR_OK;
+                    }
+                ',
+                'error_message' => 'InvalidEnumCaseValue',
+                'ignored_issues' => [],
+                'php_version' => '8.1',
+            ],
+            'intBackedEnumCaseValueFromGlobalConstant' => [
+                'code' => '<?php
+                    enum Bar: int
+                    {
+                        case Foo = \PHP_BINARY;
+                    }
+                ',
+                'error_message' => 'InvalidEnumCaseValue',
+                'ignored_issues' => [],
+                'php_version' => '8.1',
+            ],
         ];
     }
 }