Fixes

zoonru · Dec 6, 2023 · 9c99d3a · 9c99d3a
1 parent 0a513e8
commit 9c99d3a
Show file tree

Hide file tree

Showing 5 changed files with 34 additions and 4 deletions.
diff --git a/src/Psalm/Internal/Analyzer/Statements/Expression/Assignment/ArrayAssignmentAnalyzer.php b/src/Psalm/Internal/Analyzer/Statements/Expression/Assignment/ArrayAssignmentAnalyzer.php
@@ -349,7 +349,7 @@ private static function updateTypeWithKeyValues(
             $properties = [];
             $classStrings = [];
             $current_type = $current_type->setPossiblyUndefined(
-                $current_type->possibly_undefined || count($key_values) > 1
+                $current_type->possibly_undefined || count($key_values) > 1,
             );
             foreach ($key_values as $key_value) {
                 $properties[$key_value->value] = $current_type;

diff --git a/src/Psalm/Internal/Type/TypeCombination.php b/src/Psalm/Internal/Type/TypeCombination.php
@@ -57,6 +57,9 @@ final class TypeCombination
     /** @var array<string|int, Union> */
     public array $objectlike_entries = [];
 
+    /** @var array<string, true> */
+    public array $objectlike_class_string_keys = [];
+
     public bool $objectlike_sealed = true;
 
     public ?Union $objectlike_key_type = null;

diff --git a/src/Psalm/Internal/Type/TypeCombiner.php b/src/Psalm/Internal/Type/TypeCombiner.php
@@ -662,6 +662,7 @@ private static function scrapeTypeProperties(
 
             $has_defined_keys = false;
 
+            $class_strings = $type->class_strings ?? [];
             foreach ($type->properties as $candidate_property_name => $candidate_property_type) {
                 $value_type = $combination->objectlike_entries[$candidate_property_name] ?? null;
 
@@ -700,6 +701,15 @@ private static function scrapeTypeProperties(
                     );
                 }
 
+                if (isset($combination->objectlike_class_string_keys[$candidate_property_name])) {
+                    $combination->objectlike_class_string_keys[$candidate_property_name] =
+                        $combination->objectlike_class_string_keys[$candidate_property_name]
+                        && ($class_strings[$candidate_property_name] ?? false);
+                } else {
+                    $combination->objectlike_class_string_keys[$candidate_property_name] =
+                        ($class_strings[$candidate_property_name] ?? false);
+                }
+
                 unset($missing_entries[$candidate_property_name]);
             }
 
@@ -1415,7 +1425,7 @@ private static function handleKeyedArrayEntries(
                 } else {
                     $objectlike = new TKeyedArray(
                         $combination->objectlike_entries,
-                        null,
+                        $combination->objectlike_class_string_keys,
                         $sealed || $fallback_key_type === null || $fallback_value_type === null
                             ? null
                             : [$fallback_key_type, $fallback_value_type],

diff --git a/src/Psalm/Type/Reconciler.php b/src/Psalm/Type/Reconciler.php
@@ -47,7 +47,6 @@
 use Psalm\Type\Atomic\TFalse;
 use Psalm\Type\Atomic\TInt;
 use Psalm\Type\Atomic\TKeyedArray;
-use Psalm\Type\Atomic\TList;
 use Psalm\Type\Atomic\TLiteralInt;
 use Psalm\Type\Atomic\TLiteralString;
 use Psalm\Type\Atomic\TMixed;
@@ -1130,7 +1129,7 @@ private static function adjustTKeyedArrayType(
         $base_key = implode($key_parts);
 
         $result_type = $result_type->setPossiblyUndefined(
-            $result_type->possibly_undefined || count($array_key_offsets) > 1
+            $result_type->possibly_undefined || count($array_key_offsets) > 1,
         );
 
         foreach ($array_key_offsets as $array_key_offset) {

diff --git a/tests/ArrayAssignmentTest.php b/tests/ArrayAssignmentTest.php
@@ -57,6 +57,24 @@ public function providerValidCodeParse(): iterable
                     '$resultOpt===' => 'array{a?: true, b?: true}',
                 ],
             ],
+            'assignUnionOfLiteralsClassKeys' => [
+                'code' => '<?php
+                    class a {}
+                    class b {}
+
+                    $result = [];
+
+                    foreach ([a::class, b::class] as $k) {
+                        $result[$k] = true;
+                    }
+
+                    foreach ($result as $k => $v) {
+                        $vv = new $k;
+                    }',
+                'assertions' => [
+                    '$result===' => 'array{a::class: true, b::class: true}',
+                ],
+            ],
             'genericArrayCreationWithSingleIntValue' => [
                 'code' => '<?php
                     $out = [];