From 418d08a7e8342533106520224cd7611516b9bc3b Mon Sep 17 00:00:00 2001
From: fantasticbin <fantasticbin@vip.qq.com>
Date: Tue, 4 Mar 2025 21:23:53 +0800
Subject: [PATCH] =?UTF-8?q?=E7=AE=80=E5=8C=96=E6=96=87=E4=BB=B6=E5=90=8D?=
 =?UTF-8?q?=E7=BB=84=E8=A3=85=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 crawler.go | 59 +++++++++++++++++++++++-------------------------------
 1 file changed, 25 insertions(+), 34 deletions(-)

diff --git a/crawler.go b/crawler.go
index 7fa8b96..cbcb4d9 100644
--- a/crawler.go
+++ b/crawler.go
@@ -27,6 +27,11 @@ type Crawler struct {
 	client     *http.Client
 }
 
+type coverCode struct {
+	letters string
+	number  int
+}
+
 func NewCrawler(avPath, outputPath string) *Crawler {
 	config := viper.New()
 	config.SetConfigName("config")
@@ -99,7 +104,7 @@ func (c *Crawler) getCodeNum(s string) int {
 }
 
 // 获取封面代码列表
-func (c *Crawler) getCoverCodeList(files []string) (coverList []string) {
+func (c *Crawler) getCoverCodeList(files []string) (coverList []coverCode) {
 	for _, file := range files {
 		// 去除域名部分
 		if strings.IndexRune(file, '@') > 0 {
@@ -116,46 +121,32 @@ func (c *Crawler) getCoverCodeList(files []string) (coverList []string) {
 			continue
 		}
 
-		format := "%s%05d"
-		if len(nameSlice[0]) > 4 {
-			format = "1%s%05d"
-		}
-
-		coverList = append(coverList, fmt.Sprintf(format, strings.ToLower(nameSlice[0]), num))
+		coverList = append(coverList, coverCode{
+			letters: strings.ToLower(nameSlice[0]),
+			number:  num,
+		})
 	}
 	return coverList
 }
 
 // 获取封面图片
-func (c *Crawler) fetchCoverImg(code string) error {
-	imgUrl := strings.ReplaceAll(c.config.GetString("crawler.url"), `*`, code)
+func (c *Crawler) fetchCoverImg(code coverCode) error {
+	if len(code.letters) < 2 || code.number < 1 {
+		return nil
+	}
+
+	format := "%s%05d"
+	if len(code.letters) > 4 {
+		format = "1%s%05d"
+	}
+
+	codeStr := fmt.Sprintf(format, code.letters, code.number)
+	imgUrl := strings.ReplaceAll(c.config.GetString("crawler.url"), `*`, codeStr)
 	suffix := filepath.Ext(imgUrl)
 
-	startOffset := 0
-	// 如果第一个字符为 '1'，则从下一个字符开始查找
-	if len(code) > 0 && code[0] == '1' {
-		startOffset = 1
-	}
-
-	// 获取号码所在的位置
-	splitIndex := strings.Index(code[startOffset:], "00")
-	if splitIndex == -1 || splitIndex+startOffset < 3 {
-		return nil
-	}
-	// 计算原始字符串中的真实位置
-	splitIndex += startOffset
-
-	// 分隔字母部分及数字部分
-	letters := code[startOffset:splitIndex]
-	numPart := code[splitIndex+2:]
-
-	if len(letters) == 0 || len(numPart) == 0 {
-		return nil
-	}
-
-	fileName := filepath.Join(c.outputPath, fmt.Sprintf("%s-%s%s",
-		strings.ToUpper(letters),
-		numPart,
+	fileName := filepath.Join(c.outputPath, fmt.Sprintf("%s-%d%s",
+		strings.ToUpper(code.letters),
+		code.number,
 		suffix,
 	))