Automatic temp cleaning

Author: MartinMystikJonas

src/Temp/TempDirResolver.php

@@ -4,19 +4,33 @@
 
 namespace Efabrica\PHPStanLatte\Temp;
 
+use FilesystemIterator;
+use Nette\Utils\FileSystem;
+use RecursiveDirectoryIterator;
+use RecursiveIteratorIterator;
 use RuntimeException;
+use SplFileInfo;
 
 final class TempDirResolver
 {
+    private const MAX_AGE = 7 * 24 * 60 * 60; // one week
+    private const MAX_SIZE = 100 * 1024 * 1024; // 100 MB
+
     private string $tmpDir;
 
     public function __construct(?string $tmpDir)
     {
         $tmpDir = $tmpDir ? rtrim($tmpDir, DIRECTORY_SEPARATOR) : sys_get_temp_dir() . '/phpstan-latte';
+
+        if (is_dir($tmpDir) &&
+           (time() - (int)filemtime($tmpDir) > self::MAX_AGE ||
+           $this->getDirTotalSize($tmpDir) > self::MAX_SIZE)
+        ) {
+            $this->pruneDir($tmpDir);
+        }
+
         if (!is_dir($tmpDir)) {
-            if (!@mkdir($tmpDir) && !is_dir($tmpDir)) {
-                throw new RuntimeException(sprintf('Cannot create temp dir "%s"', $tmpDir));
-            }
+            Filesystem::createDir($tmpDir);
         }
         $tmpDir = realpath($tmpDir) ?: $tmpDir;
         if (!is_writable($tmpDir)) {
@@ -39,4 +53,26 @@ public function resolveCollectorDir(): string
     {
         return $this->tmpDir . DIRECTORY_SEPARATOR . 'collector-cache' . DIRECTORY_SEPARATOR;
     }
+
+    private function pruneDir(string $tmpDir): void
+    {
+        $ri = new RecursiveIteratorIterator(
+            new RecursiveDirectoryIterator($tmpDir, FilesystemIterator::SKIP_DOTS),
+            RecursiveIteratorIterator::CHILD_FIRST
+        );
+        /** @var SplFileInfo $file */
+        foreach ($ri as $file) {
+            Filesystem::delete($file->getPathname());
+        }
+    }
+
+    private function getDirTotalSize(string $tmpDir): int
+    {
+        $size = 0;
+        /** @var SplFileInfo $file */
+        foreach (new RecursiveIteratorIterator(new RecursiveDirectoryIterator($tmpDir, FilesystemIterator::SKIP_DOTS)) as $file) {
+            $size += $file->getSize();
+        }
+        return $size;
+    }
 }

tests/config.neon

@@ -1,5 +1,6 @@
 parameters:
     latte:
+        debugMode: true
         resolveAllPossiblePaths: true
         features:
             transformDynamicFormControlNamesToString: true