diff --git a/spider.go b/spider.go index b3d1094..35eae71 100644 --- a/spider.go +++ b/spider.go @@ -24,9 +24,10 @@ func URLs(url string) []Link { paths := []Link{} var f func(*html.Node) f = func(n *html.Node) { - if n.Type == html.ElementNode && n.Data == "a" { + if n.Type == html.ElementNode { for _, a := range n.Attr { - if a.Key == "href" { + switch a.Key { + case "href", "src": paths = append(paths, Link{From: url, To: a.Val}) break }