+ From 868a5106e62b7963c6e656e3d0b1f406526679de Mon Sep 17 00:00:00 2001 From: Yun <995116474@qq.com> Date: Fri, 23 Aug 2024 15:09:37 +0800 Subject: [PATCH] =?UTF-8?q?=E6=A0=B9=E6=8D=AEhtml=E6=8F=90=E5=8F=96?= =?UTF-8?q?=E8=B5=84=E6=BA=90=E9=93=BE=E6=8E=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- go.mod | 7 +++++ go.sum | 40 +++++++++++++++++++++++++++ html.go | 77 ++++++++++++++++++++++++++++++++++++++++++++++++++++ html_test.go | 60 ++++++++++++++++++++++++++++++++++++++++ mailx.go | 44 ++++-------------------------- 5 files changed, 190 insertions(+), 38 deletions(-) create mode 100644 go.sum create mode 100644 html.go create mode 100644 html_test.go diff --git a/go.mod b/go.mod index d32dbe1..4fdbf9c 100644 --- a/go.mod +++ b/go.mod @@ -1,3 +1,10 @@ module code.yun.ink/pkg/mailx go 1.20 + +require github.com/PuerkitoBio/goquery v1.9.2 + +require ( + github.com/andybalholm/cascadia v1.3.2 // indirect + golang.org/x/net v0.24.0 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..4a8159a --- /dev/null +++ b/go.sum @@ -0,0 +1,40 @@ +github.com/PuerkitoBio/goquery v1.9.2 h1:4/wZksC3KgkQw7SQgkKotmKljk0M6V8TUvA8Wb4yPeE= +github.com/PuerkitoBio/goquery v1.9.2/go.mod h1:GHPCaP0ODyyxqcNoFGYlAprUFH81NuRPd0GX3Zu2Mvk= +github.com/andybalholm/cascadia v1.3.2 h1:3Xi6Dw5lHF15JtdcmAHD3i1+T8plmv7BQ/nsViSLyss= +github.com/andybalholm/cascadia v1.3.2/go.mod h1:7gtRlve5FxPPgIgX36uWBX58OdBsSS6lUvCFb+h7KvU= +github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc= +golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4= +golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs= +golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= +golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c= +golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs= +golang.org/x/net v0.9.0/go.mod h1:d48xBJpPfHeWQsugry2m+kC02ZBRGRgulfHnEXEuWns= +golang.org/x/net v0.24.0 h1:1PcaxkF854Fu3+lvBIx5SYn9wRlBzzcnHZSiaFFAb0w= +golang.org/x/net v0.24.0/go.mod h1:2Q7sJY5mzlzWjKtYUEXSlBWCdyaioyXzRB2RtU8KVE8= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.7.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8= +golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k= +golang.org/x/term v0.7.0/go.mod h1:P32HKFT3hSsZrRxla30E9HqToFYAQPCMs/zFMBUFqPY= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= +golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8= +golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc= +golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU= +golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= diff --git a/html.go b/html.go new file mode 100644 index 0000000..b40298a --- /dev/null +++ b/html.go @@ -0,0 +1,77 @@ +package mailx + +import ( + "bytes" + + "github.com/PuerkitoBio/goquery" +) + +// 解析HTML资源,响应资源链接 +func ParseHtmlResource(html string) ([]string, error) { + resp := []string{} + + b := bytes.NewBufferString(html) + doc, err := goquery.NewDocumentFromReader(b) + if err != nil { + return nil, err + } + + // 找到所有的 css 标签,并且打印它们的 href 属性 + doc.Find("link").Each(func(i int, s *goquery.Selection) { + // 忽略dns预请求 + r, ok := s.Attr("rel") + if ok && r == "dns-prefetch" { + return + } + + href, ok := s.Attr("href") + if ok && href != "" { + resp = append(resp, href) + } + }) + + // 找到所有的 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 + +} diff --git a/html_test.go b/html_test.go new file mode 100644 index 0000000..80c6a6f --- /dev/null +++ b/html_test.go @@ -0,0 +1,60 @@ +package mailx_test + +import ( + "testing" + + "code.yun.ink/pkg/mailx" +) + +func TestParseHtmlResource(t *testing.T) { + html := ` + +
+ + +
+
+