Merge pull request #127 from krlv/add_image_avif_support

feat: Add image/avif support
h2non · Jan 23, 2023 · cfcd7d0 · cfcd7d0
2 parents 06ca62d + bcb25e2
commit cfcd7d0
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -197,6 +197,7 @@ func main() {
 - **psd** - `image/vnd.adobe.photoshop`
 - **ico** - `image/vnd.microsoft.icon`
 - **dwg** - `image/vnd.dwg`
+- **avif** - `image/avif`
 
 #### Video
 

diff --git a/fixtures/sample.avif b/fixtures/sample.avif
diff --git a/match_test.go b/match_test.go
@@ -53,6 +53,7 @@ func TestMatchFile(t *testing.T) {
 		{"dwg"},
 		{"zst"},
 		{"exr"},
+		{"avif"},
 	}
 
 	for _, test := range cases {

diff --git a/matchers/image.go b/matchers/image.go
@@ -17,6 +17,7 @@ var (
 	TypeHeif     = newType("heif", "image/heif")
 	TypeDwg      = newType("dwg", "image/vnd.dwg")
 	TypeExr      = newType("exr", "image/x-exr")
+	TypeAvif     = newType("avif", "image/avif")
 )
 
 var Image = Map{
@@ -34,6 +35,7 @@ var Image = Map{
 	TypeHeif:     Heif,
 	TypeDwg:      Dwg,
 	TypeExr:      Exr,
+	TypeAvif:     Avif,
 }
 
 func Jpeg(buf []byte) bool {
@@ -149,3 +151,24 @@ func Exr(buf []byte) bool {
 		buf[0] == 0x76 && buf[1] == 0x2f &&
 		buf[2] == 0x31 && buf[3] == 0x01
 }
+
+func Avif(buf []byte) bool {
+	if !isobmff.IsISOBMFF(buf) {
+		return false
+	}
+
+	majorBrand, _, compatibleBrands := isobmff.GetFtyp(buf)
+	if majorBrand == "avif" {
+		return true
+	}
+
+	if majorBrand == "mif1" || majorBrand == "msf1" {
+		for _, compatibleBrand := range compatibleBrands {
+			if compatibleBrand == "avif" {
+				return true
+			}
+		}
+	}
+
+	return false
+}