From d6475f61341713042004cd5569060c728bf2080f Mon Sep 17 00:00:00 2001 From: Jordan Date: Sun, 27 Mar 2022 14:23:37 -0700 Subject: util:go: HTML parsing patch; generalize article location --- util.go | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/util.go b/util.go index 0843cca..8fa9c5f 100644 --- a/util.go +++ b/util.go @@ -269,20 +269,17 @@ func getPaper(client *http.Client, u string) (string, error) { var dl *url.URL var f func(*html.Node) f = func(n *html.Node) { - if n.Type == html.ElementNode && n.Data == "embed" { + if n.Type == html.ElementNode { for _, a := range n.Attr { if a.Key == "src" { - _u, err := url.Parse(u) - if err != nil { - continue - } _v, err := url.Parse(a.Val) if err != nil { continue } + fmt.Println(_v.Path) if strings.HasSuffix(_v.Path, "pdf") { - _u.Path = _v.Path - dl = _u + dl = _v + break } } } -- cgit v1.2.3-54-g00ecf