Improve types

Author: AydinHassan

src/CodePatcher.php

@@ -121,6 +121,9 @@ private function applyPatch(Patch $patch, string $code): string
         return $this->printer->prettyPrintFile($statements);
     }
 
+    /**
+     * @param array<Stmt> $statements
+     */
     public function isFirstStatementStrictTypesDeclare(array $statements): bool
     {
         return isset($statements[0])

src/Patch/ForceStrictTypes.php

@@ -3,15 +3,20 @@
 namespace PhpSchool\PhpWorkshop\Patch;
 
 use PhpParser\Node\Scalar\LNumber;
+use PhpParser\Node\Stmt;
 use PhpParser\Node\Stmt\Declare_;
 use PhpParser\Node\Stmt\DeclareDeclare;
 
 class ForceStrictTypes implements Transformer
 {
-    public function transform(array $ast): array
+    /**
+     * @param array<Stmt> $statements
+     * @return array<Stmt>
+     */
+    public function transform(array $statements): array
     {
-        if ($this->isFirstStatementStrictTypesDeclare($ast)) {
-            return $ast;
+        if ($this->isFirstStatementStrictTypesDeclare($statements)) {
+            return $statements;
         }
 
         $declare = new \PhpParser\Node\Stmt\Declare_([
@@ -21,11 +26,14 @@ public function transform(array $ast): array
             )
         ]);
 
-        return array_merge([$declare], $ast);
+        return array_merge([$declare], $statements);
     }
 
+    /**
+     * @param array<Stmt> $statements
+     */
     public function isFirstStatementStrictTypesDeclare(array $statements): bool
     {
         return isset($statements[0]) && $statements[0] instanceof Declare_;
     }
-}
+}

src/Patch/Transformer.php

@@ -2,7 +2,13 @@
 
 namespace PhpSchool\PhpWorkshop\Patch;
 
+use PhpParser\Node\Stmt;
+
 interface Transformer
 {
-    public function transform(array $ast): array;
+    /**
+     * @param array<Stmt> $statements
+     * @return array<Stmt>
+     */
+    public function transform(array $statements): array;
 }