mailx/html.go

package mailx

import (
	"bytes"
	"fmt"

	"github.com/PuerkitoBio/goquery"
)

// 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, fmt.Errorf("mailx: parse html: %w", err)
	}

	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
		}
		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")

	return res, nil
}

// ParseHtmlResource 是 ParseHTMLResource 的旧名称，保留以兼容历史调用，建议使用新名。
//
// Deprecated: 请使用 ParseHTMLResource。
func ParseHtmlResource(html string) ([]string, error) {
	return ParseHTMLResource(html)
}