From 37cea0fab3f45fb58fe90dbab103bc24e09aa3d8 Mon Sep 17 00:00:00 2001
From: Yamagishi Kazutoshi <ykzts@desire.sh>
Date: Sat, 15 May 2021 08:24:56 +0900
Subject: [PATCH] feat(transprot): add flag for avif (#441)

---
 transport.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/transport.go b/transport.go
index c93a3fb2..f0127bb3 100644
--- a/transport.go
+++ b/transport.go
@@ -29,6 +29,7 @@ import (
 	"net"
 	"net/http"
 	"net/url"
+	"os"
 	"path"
 	"strconv"
 	"strings"
@@ -72,12 +73,13 @@ func (t *Transport) makeRequest(r *http.Request) (*http.Request, error) {
 }
 
 func scanAcceptHeader(r *http.Request) string {
+	f := os.Getenv("MANAEL_ENABLE_AVIF")
 	accepts := r.Header.Get("Accept")
 
 	for _, v := range strings.Split(accepts, ",") {
 		t := strings.TrimSpace(v)
 
-		if strings.HasPrefix(t, "image/avif") {
+		if f == "true" && strings.HasPrefix(t, "image/avif") {
 			return "image/avif"
 		} else if strings.HasPrefix(t, "image/webp") {
 			return "image/webp"