From 6e8cd8403f0c28013a1c78db9b62aedee0b8d2d7 Mon Sep 17 00:00:00 2001
From: fantasticbin <fantasticbin@vip.qq.com>
Date: Fri, 27 Jan 2023 11:17:59 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E7=88=AC=E5=8F=96=E9=80=BB?=
 =?UTF-8?q?=E8=BE=91-=E5=90=8E=E7=BC=80=E5=90=8D=E5=8A=A8=E6=80=81?=
 =?UTF-8?q?=E8=8E=B7=E5=8F=96=EF=BC=8C=E5=8F=8A=E6=B7=BB=E5=8A=A0=E5=9C=B0?=
 =?UTF-8?q?=E5=9D=80=E5=85=BC=E5=AE=B9=E5=88=A4=E6=96=AD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/cmd/start.go | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/cmd/start.go b/src/cmd/start.go
index f59b411..db62f22 100644
--- a/src/cmd/start.go
+++ b/src/cmd/start.go
@@ -30,9 +30,16 @@ func start() {
 
 func get(num int) {
 	defer wg.Done()
-	fileName := strconv.Itoa(num) + ".jpg"
-	imgSlice := strings.Split(url, "_")
-	imgUrl := host + imgSlice[0] + "_" + fmt.Sprintf("%02d", num) + ".jpg"
+
+	// 兼容未携带斜杆的地址
+	if url[0] != '/' {
+		url = "/" + url
+	}
+
+	urlSlice := strings.Split(url, "_")                                                // 取URL组装
+	fileSlice := strings.Split(urlSlice[1], ".")                                       // 取后缀名
+	fileName := strconv.Itoa(num) + "." + fileSlice[1]                                 // 组装文件名
+	imgUrl := host + urlSlice[0] + "_" + fmt.Sprintf("%02d", num) + "." + fileSlice[1] // 组装图片URL
 	fmt.Println("获取图片:" + imgUrl)
 
 	resp, err := http.Get(imgUrl)