From 23543218dcb7090593ccb92441d53dac879c7b77 Mon Sep 17 00:00:00 2001
From: Daniil Gentili <daniil@daniil.it>
Date: Thu, 18 Jan 2024 12:59:56 +0100
Subject: [PATCH] Fix template, conditional array keys

---
 src/Psalm/Internal/Type/TypeParser.php |  9 +++-
 tests/ArrayKeysTest.php                | 58 ++++++++++++++++++++++++++
 2 files changed, 66 insertions(+), 1 deletion(-)

diff --git a/src/Psalm/Internal/Type/TypeParser.php b/src/Psalm/Internal/Type/TypeParser.php
index e908c4e3a..ecdd5910c 100644
--- a/src/Psalm/Internal/Type/TypeParser.php
+++ b/src/Psalm/Internal/Type/TypeParser.php
@@ -672,8 +672,8 @@ final class TypeParser
                 // PHP 8 values with whitespace after number are counted as numeric
                 // and filter_var treats them as such too
                 if ($atomic_type instanceof TLiteralString
-                    && trim($atomic_type->value) === $atomic_type->value
                     && ($string_to_int = filter_var($atomic_type->value, FILTER_VALIDATE_INT)) !== false
+                    && trim($atomic_type->value) === $atomic_type->value
                 ) {
                     $builder = $generic_params[0]->getBuilder();
                     $builder->removeType($key);
@@ -688,7 +688,14 @@ final class TypeParser
                     || $atomic_type instanceof TMixed
                     || $atomic_type instanceof TNever
                     || $atomic_type instanceof TTemplateParam
+                    || $atomic_type instanceof TTemplateIndexedAccess
+                    || $atomic_type instanceof TTemplateValueOf
+                    || $atomic_type instanceof TTemplateKeyOf
+                    || $atomic_type instanceof TTemplateParamClass
+                    || $atomic_type instanceof TTypeAlias
                     || $atomic_type instanceof TValueOf
+                    || $atomic_type instanceof TConditional
+                    || $atomic_type instanceof TKeyOf
                     || !$from_docblock
                 ) {
                     continue;
diff --git a/tests/ArrayKeysTest.php b/tests/ArrayKeysTest.php
index e3f9cc897..d2c363f94 100644
--- a/tests/ArrayKeysTest.php
+++ b/tests/ArrayKeysTest.php
@@ -124,6 +124,64 @@ class ArrayKeysTest extends TestCase
                         }
                     }',
             ],
+            'variousArrayKeys' => [
+                'code' => '<?php
+                    /**
+                     * @psalm-type TAlias = 123
+                     */
+                    class a {}
+                    
+                    /**
+                     * @psalm-import-type TAlias from a
+                     * @template TKey as array-key
+                     * @template TValue as array-key
+                     * @template T as array<TKey, TValue>
+                     * @template TT
+                     * @template TBool as bool
+                     */
+                    class b {
+                        /** 
+                         * @var array<TAlias, int> 
+                         */
+                        private array $a = [];
+                        
+                        /** @var array<value-of<T>, int> */
+                        private array $c = [];
+                        
+                        /** @var array<key-of<T>, int> */
+                        private array $d = [];
+                        
+                        /** @var array<TT, int> */
+                        private array $e = [];
+                        
+                        /** @var array<key-of<array<int, int>>, int> */
+                        private array $f = [];
+                        
+                        /** @var array<value-of<array<int, int>>, int> */
+                        private array $g = [];
+
+                        /** @var array<TBool is true ? string : int, int> */
+                        private array $h = [];
+
+                        /**
+                         * @param T $arr
+                         * @param class-string<TT> $b
+                         * @param TBool $c
+                         */
+                        public function __construct(
+                            array $arr,
+                            string $b,
+                            bool $c
+                        ) {}
+
+                        /**
+                         * @return array<$v is true ? "a" : 123, 123>
+                         */
+                        public function test(bool $v): array {
+                            return $v ? ["a" => 123] : [123 => 123];
+                        }
+                    }',
+            ],
         ];
     }