stephen/caddy-hugo2
1
0
Fork 0
Code Issues 8 Pull Requests Releases Wiki Activity

including counts of how many pages media is referenced on

Browse Source
This commit is contained in:
stephen
2017-09-20 10:18:50 -04:00
parent fb93746ba6
commit e75784599e
1 changed files with 47 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
media.go
+47 -2
View File
@@ -1,13 +1,48 @@
package caddyhugo package caddyhugo
import ( import (
"bytes"
"fmt" "fmt"
"io" "io"
"net/http" "net/http"
"net/url"
"path"
"github.com/PuerkitoBio/goquery"
"github.com/gohugoio/hugo/hugolib"
"git.stephensearles.com/stephen/caddy-hugo2/media" "git.stephensearles.com/stephen/caddy-hugo2/media"
) )
func (ch *CaddyHugo) ReferencedMedia() map[string]map[*hugolib.Page]struct{} {
found := map[string]map[*hugolib.Page]struct{}{}
for _, page := range ch.HugoSites.Pages() {
r := bytes.NewBufferString(string(page.Render()))
doc, err := goquery.NewDocumentFromReader(r)
if err != nil {
continue
}
doc.Find("img,video").Map(func(i int, s *goquery.Selection) string {
u, ok := s.Attr("src")
if ok {
u = path.Base(u)
if ud, err := url.QueryUnescape(u); err == nil {
u = ud
}
if m := found[u]; m == nil {
found[u] = make(map[*hugolib.Page]struct{})
}
found[u][page] = struct{}{}
}
return ""
})
}
return found
}
func (ch *CaddyHugo) uploadMedia(w http.ResponseWriter, r *http.Request) (int, error) { func (ch *CaddyHugo) uploadMedia(w http.ResponseWriter, r *http.Request) (int, error) {
if ch.Media == nil { if ch.Media == nil {
http.NotFound(w, r) http.NotFound(w, r)
@@ -49,6 +84,8 @@ func (ch *CaddyHugo) serveMediaPage(w http.ResponseWriter, r *http.Request) (int
return 404, nil return 404, nil
} }
referenced := ch.ReferencedMedia()
io.WriteString(w, `<html> io.WriteString(w, `<html>
<head><style> <head><style>
iframe { height: 100%; } iframe { height: 100%; }
@@ -87,12 +124,20 @@ func (ch *CaddyHugo) serveMediaPage(w http.ResponseWriter, r *http.Request) (int
fmt.Fprintf(w, `<div class="img">error rendering %q: %v</div>`, m.Name, err) fmt.Fprintf(w, `<div class="img">error rendering %q: %v</div>`, m.Name, err)
continue continue
} }
refs := len(referenced[m.Name])
plural := "s"
if refs == 1 {
plural = ""
}
refLine := fmt.Sprintf("included on %d page%s", refs, plural)
switch m.Type { switch m.Type {
case media.TypeImage: case media.TypeImage:
fmt.Fprintf(w, `<div class="img"><img width=%d height=%d src=%q data-filename=%q /><br /><input type="text" readonly value=%q /><span class="copy">&#x1F4CB;</span></div>`, size.Dx(), size.Dy(), m.ThumbPath(size), m.Name, m.ThumbPath(size)) fmt.Fprintf(w, `<div class="img"><img width=%d height=%d src=%q data-filename=%q /><br /><input type="text" readonly value=%q /><span class="copy">&#x1F4CB;</span><br />%s</div>`, size.Dx(), size.Dy(), m.ThumbPath(size), m.Name, m.ThumbPath(size), refLine)
case media.TypeVideo: case media.TypeVideo:
// TODO: onmouseover sucks for mobile // TODO: onmouseover sucks for mobile
fmt.Fprintf(w, `<div class="img"><video width=%d height=%d src=%q data-filename=%q onmouseover="this.play()" onmouseout="this.pause();this.currentTime=0;"></video><br /><input type="text" readonly value=%q /><span class="copy">&#x1F4CB;</span></div>`, size.Dx(), size.Dy(), m.ThumbPath(size), m.Name, m.ThumbPath(size)) fmt.Fprintf(w, `<div class="img"><video width=%d height=%d src=%q data-filename=%q onmouseover="this.play()" onmouseout="this.pause();this.currentTime=0;"></video><br /><input type="text" readonly value=%q /><span class="copy">&#x1F4CB;</span><br />%s</div>`, size.Dx(), size.Dy(), m.ThumbPath(size), m.Name, m.ThumbPath(size), refLine)
} }
} }
} }