Initial commit: docker compose config

server/torr/utils/webImageChecker.go

@@ -0,0 +1,56 @@
+package utils
+
+import (
+	"context"
+	"image"
+	_ "image/gif"
+	_ "image/jpeg"
+	_ "image/png"
+	"io"
+	"net/http"
+	"strings"
+	"time"
+
+	"golang.org/x/image/webp"
+
+	"server/log"
+)
+
+func CheckImgUrl(link string) bool {
+	if link == "" {
+		return false
+	}
+
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	req, err := http.NewRequestWithContext(ctx, "GET", link, nil)
+	if err != nil {
+		log.TLogln("Error create request for image:", err)
+		return false
+	}
+
+	client := &http.Client{
+		Timeout: 5 * time.Second,
+	}
+
+	resp, err := client.Do(req)
+	if err != nil {
+		log.TLogln("Error check image:", err)
+		return false
+	}
+	defer resp.Body.Close()
+
+	limitedReader := io.LimitReader(resp.Body, 2*1024*1024)
+
+	if strings.HasSuffix(link, ".webp") {
+		_, err = webp.Decode(limitedReader)
+	} else {
+		_, _, err = image.Decode(limitedReader)
+	}
+	if err != nil {
+		log.TLogln("Error decode image:", err)
+		return false
+	}
+	return true
+}