更新

html.go

@@ -2,76 +2,50 @@ package mailx
 
 import (
 	"bytes"
+	"fmt"
 
 	"github.com/PuerkitoBio/goquery"
 )
 
-// html路径替换
-func ParseHtmlResource(html string) ([]string, error) {
-	resp := []string{}
-
-	b := bytes.NewBufferString(html)
-	doc, err := goquery.NewDocumentFromReader(b)
+// ParseHTMLResource 解析 HTML 中引用的静态资源地址（css/js/img/video/audio）。
+// 返回资源 URL 列表；忽略空属性与 dns-prefetch 预请求。
+func ParseHTMLResource(html string) ([]string, error) {
+	doc, err := goquery.NewDocumentFromReader(bytes.NewBufferString(html))
 	if err != nil {
-		return nil, err
+		return nil, fmt.Errorf("mailx: parse html: %w", err)
 	}
 
-	// 找到所有的 css 标签，并且打印它们的 href 属性
-	doc.Find("link").Each(func(i int, s *goquery.Selection) {
-		// 忽略dns预请求
-		r, ok := s.Attr("rel")
-		if ok && r == "dns-prefetch" {
+	var res []string
+	collect := func(sel, attr string) {
+		doc.Find(sel).Each(func(_ int, s *goquery.Selection) {
+			if v, ok := s.Attr(attr); ok && v != "" {
+				res = append(res, v)
+			}
+		})
+	}
+
+	// link：跳过 dns-prefetch
+	doc.Find("link").Each(func(_ int, s *goquery.Selection) {
+		if rel, ok := s.Attr("rel"); ok && rel == "dns-prefetch" {
 			return
 		}
-
-		href, ok := s.Attr("href")
-		if ok && href != "" {
-			resp = append(resp, href)
+		if href, ok := s.Attr("href"); ok && href != "" {
+			res = append(res, href)
 		}
 	})
+	collect("script", "src")
+	collect("img", "src")
+	collect("img", "data-src")
+	collect("video", "src")
+	collect("video", "data-src")
+	collect("audio", "src")
 
-	// 找到所有的 script 标签，并且打印它们的 src 属性
-	doc.Find("script").Each(func(i int, s *goquery.Selection) {
-		src, ok := s.Attr("src")
-		if ok && src != "" {
-			resp = append(resp, src)
-		}
-	})
-
-	// 找到所有的 img 标签，并且打印它们的 src 属性
-	doc.Find("img").Each(func(i int, s *goquery.Selection) {
-		src, ok := s.Attr("src")
-		if ok && src != "" {
-			resp = append(resp, src)
-		}
-		data_src, ok := s.Attr("data-src")
-		if ok && data_src != "" {
-			resp = append(resp, data_src)
-		}
-	})
-
-	// 找到所有的 video 标签，并且打印它们的 src 属性
-	doc.Find("video").Each(func(i int, s *goquery.Selection) {
-		src, ok := s.Attr("src")
-		if ok && src != "" {
-			resp = append(resp, src)
-		}
-
-		data_src, ok := s.Attr("data-src")
-		if ok && data_src != "" {
-			resp = append(resp, data_src)
-		}
-	})
-
-	// 找到所有的 audio 标签，并且打印它们的 src 属性
-	doc.Find("audio").Each(func(i int, s *goquery.Selection) {
-		src, ok := s.Attr("src")
-
-		if ok && src != "" {
-			resp = append(resp, src)
-		}
-	})
-
-	return resp, nil
-
+	return res, nil
+}
+
+// ParseHtmlResource 是 ParseHTMLResource 的旧名称，保留以兼容历史调用，建议使用新名。
+//
+// Deprecated: 请使用 ParseHTMLResource。
+func ParseHtmlResource(html string) ([]string, error) {
+	return ParseHTMLResource(html)
 }