Improve the fileExtension expression

Author: sebprt

src/FileExtension.php

@@ -21,13 +21,26 @@ private function compile(string $value): string
     {
         return <<<PHP
             (function () use (\$input) : string {
-                 return \\pathinfo($value, \FILEINFO_EXTENSION)
+                \$validExtensions = ['jpg', 'jpeg', 'png', 'gif', 'bmp', 'svg', 'pdf', 'mp4', 'webm', 'mp3'];
+                
+                \$extension = \\pathinfo({$value}, \\FILEINFO_EXTENSION);
+                if (!\\in_array(\$extension, \$validExtensions, true)) {
+                    return null;
+                }
             })()
             PHP;
     }
 
-    private function evaluate(array $context, string $file): string
+    private function evaluate(array $context, string $file): ?string
     {
-        return pathinfo($file, \FILEINFO_EXTENSION);
+        $validExtensions = ['jpg', 'jpeg', 'png', 'gif', 'bmp', 'svg'];
+
+        $extension = pathinfo($file, \FILEINFO_EXTENSION);
+
+        if (!\in_array($extension, $validExtensions, true)) {
+            return null;
+        }
+
+        return $extension;
     }
 }