Fix shaped array class string key combination

Author: danog

src/Psalm/Internal/Analyzer/Statements/Expression/Assignment/ArrayAssignmentAnalyzer.php

@@ -351,7 +351,9 @@ private static function updateTypeWithKeyValues(
         if (!$has_matching_objectlike_property && !$has_matching_string) {
             $properties = [];
             $classStrings = [];
-            $current_type = $current_type->setPossiblyUndefined(count($key_values) > 1);
+            $current_type = $current_type->setPossiblyUndefined(
+                $current_type->possibly_undefined || count($key_values) > 1,
+            );
             foreach ($key_values as $key_value) {
                 $properties[$key_value->value] = $current_type;
                 if ($key_value instanceof TLiteralClassString) {

src/Psalm/Internal/Type/TypeCombination.php

@@ -55,6 +55,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;

src/Psalm/Internal/Type/TypeCombiner.php

@@ -667,6 +667,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;
 
@@ -705,6 +706,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]);
             }
 
@@ -1421,7 +1431,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],

src/Psalm/Type/Reconciler.php

@@ -1132,7 +1132,9 @@ private static function adjustTKeyedArrayType(
 
         $base_key = implode($key_parts);
 
-        $result_type = $result_type->setPossiblyUndefined(count($array_key_offsets) > 1);
+        $result_type = $result_type->setPossiblyUndefined(
+            $result_type->possibly_undefined || count($array_key_offsets) > 1,
+        );
 
         foreach ($array_key_offsets as $array_key_offset) {
             if (isset($existing_types[$base_key]) && $array_key_offset !== false) {

tests/ArrayAssignmentTest.php

@@ -55,6 +55,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 = [];