add: Basic match implementation.

dedoc · May 22, 2024 · ddc97fb · ddc97fb
1 parent dbbf57b
commit ddc97fb
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/Infer/Scope/Scope.php b/src/Infer/Scope/Scope.php
@@ -60,6 +60,17 @@ public function getType(Node $node): Type
             ]);
         }
 
+        if ($node instanceof Node\Expr\Match_) {
+            return match (get_class($node->cond)) {
+                Node\Expr\ArrayDimFetch::class, Node\Expr\Variable::class => Union::wrap(
+                    collect($node->arms)
+                        ->map(fn (Node\MatchArm $arm) => $this->getType($arm->body))
+                        ->toArray()
+                ),
+                default => new UnknownType('Cannot infer type of property fetch: not supported yet.'),
+            };
+        }
+
         if ($node instanceof Node\Expr\ClassConstFetch) {
             return (new ClassConstFetchTypeGetter)($node, $this);
         }