Merge pull request #14 from Katalam/memory-limit

[2.X] feat: add the support of the memory limit cli command of phpstan
pestphp · Jan 10, 2024 · 1b5ae0d · 1b5ae0d
2 parents 751f0c1 + fbf36fb
commit 1b5ae0d
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/src/Plugin.php b/src/Plugin.php
@@ -70,6 +70,28 @@ public function handleArguments(array $arguments): array
                 $this->coverageMin = (float) explode('=', $argument)[1];
             }
 
+            if (str_starts_with($argument, '--memory-limit')) {
+                $memoryLimit = explode('=', $argument)[1] ?? '';
+
+                if (preg_match('/^-?\d+[kMG]?$/', $memoryLimit) !== 1) {
+                    View::render('components.badge', [
+                        'type' => 'ERROR',
+                        'content' => 'Invalid memory limit: '.$memoryLimit,
+                    ]);
+
+                    $this->exit(1);
+                }
+
+                if (ini_set('memory_limit', $memoryLimit) === false) {
+                    View::render('components.badge', [
+                        'type' => 'ERROR',
+                        'content' => 'Failed to set memory limit: '.$memoryLimit,
+                    ]);
+
+                    $this->exit(1);
+                }
+            }
+
             if (str_starts_with($argument, '--type-coverage-json')) {
                 $outputPath = explode('=', $argument)[1] ?? null;