From 857ae11ebd78163bb44481d662728312c588ae4b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Fri, 17 Nov 2023 14:01:46 +0900
Subject: [PATCH] optimize: limit bed to image

---
 image.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/image.go b/image.go
index 023346d..f55ba98 100644
--- a/image.go
+++ b/image.go
@@ -9,6 +9,7 @@ import (
 	"net/http"
 	"strings"
 
+	"github.com/fumiama/imgsz"
 	"github.com/pkg/errors"
 )
 
@@ -35,6 +36,10 @@ func Live(u string) bool {
 
 // Bed image to server
 func Bed(t string, b []byte) (string, uint64, uint64, error) {
+	_, _, err := imgsz.DecodeSize(bytes.NewReader(b))
+	if err != nil {
+		return "", 0, 0, errors.Wrap(err, getThisFuncName())
+	}
 	m := md5.Sum(b)
 	u := API + hex.EncodeToString(m[:])
 	req, err := http.NewRequest("PUT", u, bytes.NewReader(b))