aboutsummaryrefslogtreecommitdiffstats
path: root/models
diff options
context:
space:
mode:
Diffstat (limited to 'models')
-rw-r--r--models/item/item.go24
1 files changed, 24 insertions, 0 deletions
diff --git a/models/item/item.go b/models/item/item.go
index 77ee02c..1e7e86a 100644
--- a/models/item/item.go
+++ b/models/item/item.go
@@ -3,11 +3,14 @@ package item
import (
"adammathes.com/neko/models"
"adammathes.com/neko/vlog"
+ "encoding/base64"
"fmt"
+ "github.com/PuerkitoBio/goquery"
"github.com/advancedlogic/GoOse"
"github.com/microcosm-cc/bluemonday"
"github.com/russross/blackfriday"
"log"
+ "strings"
)
type Item struct {
@@ -176,6 +179,7 @@ func Filter(max_id int64, feed_id int64, category string, unread_only bool, star
// but still may need to adjust rules
i.Title = p.Sanitize(i.Title)
i.Description = p.Sanitize(i.Description)
+ i.Description = rewriteImages(i.Description)
i.Url = p.Sanitize(i.Url)
i.FeedTitle = p.Sanitize(i.FeedTitle)
i.FeedUrl = p.Sanitize(i.FeedUrl)
@@ -196,3 +200,23 @@ func (i *Item) CleanHeaderImage() {
i.HeaderImage = ""
}
}
+
+func rewriteImages(s string) string {
+ doc, err := goquery.NewDocumentFromReader(strings.NewReader(s))
+ if err != nil {
+ panic("can not parse doc to rewrite")
+ }
+
+ doc.Find("img").Each(func(i int, img *goquery.Selection) {
+ if src, ok := img.Attr("src"); ok {
+ img.SetAttr("src", proxyURL(src))
+ }
+ })
+
+ output, _ := doc.Html()
+ return output
+}
+
+func proxyURL(url string) string {
+ return "/image/" + base64.URLEncoding.EncodeToString([]byte(url))
+}