diff options
-rw-r--r-- | reader/rewrite/rewrite_functions.go | 16 | ||||
-rw-r--r-- | reader/rewrite/rewriter_test.go | 10 |
2 files changed, 22 insertions, 4 deletions
diff --git a/reader/rewrite/rewrite_functions.go b/reader/rewrite/rewrite_functions.go index 6717deb..412266d 100644 --- a/reader/rewrite/rewrite_functions.go +++ b/reader/rewrite/rewrite_functions.go @@ -22,9 +22,19 @@ func addImageTitle(entryURL, entryContent string) string { return entryContent } - imgTag := doc.Find("img").First() - if titleAttr, found := imgTag.Attr("title"); found { - return entryContent + `<blockquote cite="` + entryURL + `">` + titleAttr + "</blockquote>" + matches := doc.Find("img[src][title]") + + if matches.Length() > 0 { + matches.Each(func(i int, img *goquery.Selection) { + altAttr := img.AttrOr("alt", "") + srcAttr, _ := img.Attr("src") + titleAttr, _ := img.Attr("title") + + img.ReplaceWithHtml(`<figure><img src="` + srcAttr + `" alt="` + altAttr + `"/><figcaption><p>` + titleAttr + `</p></figcaption></figure>`) + }) + + output, _ := doc.Find("body").First().Html() + return output } return entryContent diff --git a/reader/rewrite/rewriter_test.go b/reader/rewrite/rewriter_test.go index 995508b..7a33b1f 100644 --- a/reader/rewrite/rewriter_test.go +++ b/reader/rewrite/rewriter_test.go @@ -35,7 +35,15 @@ func TestRewriteWithInexistingCustomRule(t *testing.T) { func TestRewriteWithXkcdLink(t *testing.T) { description := `<img src="https://imgs.xkcd.com/comics/thermostat.png" title="Your problem is so terrible, I worry that, if I help you, I risk drawing the attention of whatever god of technology inflicted it on you." alt="Your problem is so terrible, I worry that, if I help you, I risk drawing the attention of whatever god of technology inflicted it on you." />` output := Rewriter("https://xkcd.com/1912/", description, ``) - expected := description + `<blockquote cite="https://xkcd.com/1912/">Your problem is so terrible, I worry that, if I help you, I risk drawing the attention of whatever god of technology inflicted it on you.</blockquote>` + expected := `<figure><img src="https://imgs.xkcd.com/comics/thermostat.png" alt="Your problem is so terrible, I worry that, if I help you, I risk drawing the attention of whatever god of technology inflicted it on you."/><figcaption><p>Your problem is so terrible, I worry that, if I help you, I risk drawing the attention of whatever god of technology inflicted it on you.</p></figcaption></figure>` + if expected != output { + t.Errorf(`Not expected output: got "%s" instead of "%s"`, output, expected) + } +} +func TestRewriteWithXkcdLinkAndImageNoTitle(t *testing.T) { + description := `<img src="https://imgs.xkcd.com/comics/thermostat.png" alt="Your problem is so terrible, I worry that, if I help you, I risk drawing the attention of whatever god of technology inflicted it on you." />` + output := Rewriter("https://xkcd.com/1912/", description, ``) + expected := description if expected != output { t.Errorf(`Not expected output: got "%s" instead of "%s"`, output, expected) } |