Add FileExtension::mediaType() & FileExtension::mediaTypes()

src/FileExtension.php

@@ -55,4 +55,24 @@ enum FileExtension: string
             self::HEIF => Format::HEIC,
         };
     }
+
+    /**
+     * Return media types for the current format
+     *
+     * @return array<MediaType>
+     */
+    public function mediaTypes(): array
+    {
+        return $this->format()->mediaTypes();
+    }
+
+    /**
+     * Return the first found media type for the current format
+     *
+     * @return MediaType
+     */
+    public function mediaType(): MediaType
+    {
+        return $this->format()->mediaType();
+    }
 }

tests/Unit/FileExtensionTest.php

@@ -6,7 +6,9 @@ namespace Intervention\Image\Tests\Unit;
 
 use Intervention\Image\FileExtension;
 use Intervention\Image\Format;
+use Intervention\Image\MediaType;
 use Intervention\Image\Tests\BaseTestCase;
+use PHPUnit\Framework\Attributes\DataProvider;
 
 final class FileExtensionTest extends BaseTestCase
 {
@@ -81,4 +83,27 @@ final class FileExtensionTest extends BaseTestCase
         $ext = FileExtension::HEIF;
         $this->assertEquals(Format::HEIC, $ext->format());
     }
+
+    #[DataProvider('mediaTypesDataProvider')]
+    public function testMediatypes(FileExtension $extension, int $mediaTypeCount, MediaType $mediaType): void
+    {
+        $this->assertCount($mediaTypeCount, $extension->mediaTypes());
+        $this->assertEquals($mediaType, $extension->mediaType());
+    }
+
+    public static function mediaTypesDataProvider(): array
+    {
+        return [
+            [FileExtension::JPEG, 4, MediaType::IMAGE_JPEG],
+            [FileExtension::WEBP, 2, MediaType::IMAGE_WEBP],
+            [FileExtension::GIF, 1, MediaType::IMAGE_GIF],
+            [FileExtension::PNG, 2, MediaType::IMAGE_PNG],
+            [FileExtension::AVIF, 2, MediaType::IMAGE_AVIF],
+            [FileExtension::BMP, 8, MediaType::IMAGE_BMP],
+            [FileExtension::TIFF, 1, MediaType::IMAGE_TIFF],
+            [FileExtension::TIF, 1, MediaType::IMAGE_TIFF],
+            [FileExtension::JP2, 3, MediaType::IMAGE_JP2],
+            [FileExtension::HEIC, 3, MediaType::IMAGE_HEIC],
+        ];
+    }
 }