From b2df3da864bd5c7d9188cf3bcb7c77228aac284c Mon Sep 17 00:00:00 2001
From: "Stephen McQuay (smcquay)" <stephen@mcquay.me>
Date: Fri, 25 Nov 2016 23:34:07 -0800
Subject: [PATCH] tweak to get link href and img src

---
 spider.go | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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
 				}