properly handling side effects originating from constructors methods …

…calls
dedoc · Feb 17, 2025 · 5d0828f · 5d0828f
1 parent b297c9a
commit 5d0828f
Show file tree

Hide file tree

Showing 2 changed files with 68 additions and 2 deletions.
diff --git a/src/Infer/Services/ReferenceTypeResolver.php b/src/Infer/Services/ReferenceTypeResolver.php
@@ -545,6 +545,10 @@ private function resolveNewCallReferenceType(Scope $scope, NewCallReferenceType
                 ->map(fn (TemplateType $t) => $inferredTemplates->get($t->name, new UnknownType))
                 ->toArray(),
         );
+//        dd(
+//            $type,
+//            $this->getMethodCallsSideEffectIntroducedTypesInConstructor($type, $scope, $classDefinition, $constructorDefinition)
+//        );
 
         return $this->getMethodCallsSideEffectIntroducedTypesInConstructor($type, $scope, $classDefinition, $constructorDefinition);
     }
@@ -815,13 +819,22 @@ private function getMethodCallsSideEffectIntroducedTypesInConstructor(Generic $t
                 continue;
             }
 
-            $type = $this->getFunctionCallResult($methodDefinition, $se->arguments, $type, new MethodCallEvent(
+            $event = new MethodCallEvent(
                 instance: $type,
                 name: $se->methodName,
                 scope: $scope,
                 arguments: $se->arguments,
                 methodDefiningClassName: $type->name,
-            ));
+            );
+
+            foreach ($methodDefinition->sideEffects as $sideEffect) {
+                if (
+                    $sideEffect instanceof SelfTemplateDefinition
+                    && $type instanceof Generic
+                ) {
+                    $sideEffect->apply($type, $event);
+                }
+            }
         }
 
         return $type;

diff --git a/tests/Infer/Analyzer/ClassAnalyzerTest.php b/tests/Infer/Analyzer/ClassAnalyzerTest.php
@@ -105,6 +105,59 @@
     expect($type->toString())->toBe('Dedoc\Scramble\Tests\Infer\stubs\ChildParentSetterCalls<string(from ChildParentSetterCalls constructor), string(from ChildParentSetterCalls wow), string(some)>');
 });
 
+it('analyzes fluent setters called in constructor', function () {
+    $this->classAnalyzer->analyze(Foo_ClassAnalyzerTest::class);
+
+    $type = getStatementType('new Foo_ClassAnalyzerTest()');
+
+    expect($type->toString())->toBe('Foo_ClassAnalyzerTest<int(42), string(baz)>');
+});
+class Foo_ClassAnalyzerTest {
+    public int $foo;
+    public string $bar;
+    public function __construct()
+    {
+        $this
+            ->setFoo(42)
+            ->setBar('baz');
+    }
+    public function setFoo($number)
+    {
+        $this->foo = $number;
+        return $this;
+    }
+    public function setBar($string)
+    {
+        $this->bar = $string;
+        return $this;
+    }
+}
+
+it('analyzes not fluent setters called in constructor', function () {
+    $this->classAnalyzer->analyze(FooNotFluent_ClassAnalyzerTest::class);
+
+    $type = getStatementType('new FooNotFluent_ClassAnalyzerTest()');
+
+    expect($type->toString())->toBe('FooNotFluent_ClassAnalyzerTest<int(42), string(baz)>');
+});
+class FooNotFluent_ClassAnalyzerTest {
+    public int $foo;
+    public string $bar;
+    public function __construct()
+    {
+        $this->setFoo(42);
+        $this->setBar('baz');
+    }
+    public function setFoo($number)
+    {
+        $this->foo = $number;
+    }
+    public function setBar($string)
+    {
+        $this->bar = $string;
+    }
+}
+
 it('analyzes static method call on class constants', function () {
     $this->classAnalyzer->analyze(ConstFetchStaticCallChild_ClassAnalyzerTest::class);