media improvements #12
+2
-1
@@ -11,6 +11,7 @@ import (
|
||||
|
||||
"git.stephensearles.com/stephen/acedoc"
|
||||
"git.stephensearles.com/stephen/caddy-hugo2/comments"
|
||||
"git.stephensearles.com/stephen/caddy-hugo2/media"
|
||||
|
||||
"github.com/gohugoio/hugo/deps"
|
||||
"github.com/gohugoio/hugo/hugolib"
|
||||
@@ -35,7 +36,7 @@ type CaddyHugo struct {
|
||||
|
||||
Dir string
|
||||
|
||||
Media *MediaSource
|
||||
Media *media.MediaSource
|
||||
Comments *comments.Service
|
||||
|
||||
docs map[string]*editSession
|
||||
|
||||
@@ -3,206 +3,14 @@ package caddyhugo
|
||||
import (
|
||||
"fmt"
|
||||
"image"
|
||||
_ "image/gif" // for processing images
|
||||
"image/jpeg"
|
||||
_ "image/png" // for processing images
|
||||
"io"
|
||||
"net/http"
|
||||
"os"
|
||||
"path"
|
||||
"path/filepath"
|
||||
"regexp"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"github.com/nfnt/resize"
|
||||
"git.stephensearles.com/stephen/caddy-hugo2/media"
|
||||
)
|
||||
|
||||
type MediaSource struct {
|
||||
StorageDir string
|
||||
ThumbDir string
|
||||
}
|
||||
|
||||
func (ms *MediaSource) LocationOrig(m Media) string {
|
||||
return path.Join(ms.StorageDir, m.Name)
|
||||
}
|
||||
|
||||
func (ms *MediaSource) ThumbPath(m Media, size image.Rectangle) string {
|
||||
w := size.Dx()
|
||||
h := size.Dy()
|
||||
|
||||
var ws, hs string
|
||||
|
||||
if w != 0 {
|
||||
ws = fmt.Sprint(w)
|
||||
}
|
||||
if h != 0 {
|
||||
hs = fmt.Sprint(h)
|
||||
}
|
||||
|
||||
thumbSlug := filepath.Join(fmt.Sprintf("%sx%s", ws, hs), m.Name)
|
||||
return path.Join("/media", thumbSlug)
|
||||
}
|
||||
|
||||
func (ms *MediaSource) ThumbFilename(m Media, size image.Rectangle) string {
|
||||
return filepath.Join(ms.ThumbDir, ms.ThumbPath(m, size))
|
||||
}
|
||||
|
||||
func (ms *MediaSource) receiveNewMedia(name string, r io.Reader) error {
|
||||
dest := path.Join(ms.StorageDir, name)
|
||||
f, err := os.OpenFile(dest, os.O_WRONLY|os.O_CREATE, 0644)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err = io.Copy(f, r)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return f.Close()
|
||||
}
|
||||
|
||||
type Media struct {
|
||||
Type string
|
||||
Name string
|
||||
Size image.Rectangle
|
||||
}
|
||||
|
||||
func (ms *MediaSource) Size(name string) (image.Rectangle, error) {
|
||||
f, err := os.Open(name)
|
||||
if err != nil {
|
||||
return image.ZR, err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
cfg, _, err := image.DecodeConfig(f)
|
||||
if err != nil {
|
||||
return image.ZR, err
|
||||
}
|
||||
|
||||
width := cfg.Width
|
||||
height := cfg.Height
|
||||
|
||||
return image.Rect(0, 0, width, height), nil
|
||||
}
|
||||
|
||||
func (ms *MediaSource) ThumbMax(m Media, maxDim int) (string, image.Rectangle, error) {
|
||||
f, err := os.Open(ms.LocationOrig(m))
|
||||
if err != nil {
|
||||
return "", image.ZR, err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
cfg, _, err := image.DecodeConfig(f)
|
||||
if err != nil {
|
||||
return "", image.ZR, err
|
||||
}
|
||||
|
||||
width := cfg.Width
|
||||
height := cfg.Height
|
||||
|
||||
if width > height {
|
||||
height = height * maxDim / width
|
||||
width = maxDim
|
||||
} else {
|
||||
width = width * maxDim / height
|
||||
height = maxDim
|
||||
}
|
||||
|
||||
size := image.Rect(0, 0, width, height)
|
||||
if ms.HasThumb(m, size) {
|
||||
return ms.ThumbPath(m, size), size, nil
|
||||
}
|
||||
|
||||
_, err = f.Seek(0, io.SeekStart)
|
||||
if err != nil {
|
||||
return "", image.ZR, err
|
||||
}
|
||||
|
||||
img, _, err := image.Decode(f)
|
||||
if err != nil {
|
||||
return "", image.ZR, err
|
||||
}
|
||||
|
||||
src, err := ms.ThumbImage(img, m, size)
|
||||
return src, size, err
|
||||
}
|
||||
|
||||
func (ms *MediaSource) HasThumb(m Media, size image.Rectangle) bool {
|
||||
_, err := os.Stat(ms.ThumbFilename(m, size))
|
||||
return err == nil
|
||||
}
|
||||
|
||||
func (ms *MediaSource) ByName(name string) *Media {
|
||||
size, _ := ms.Size(path.Join(ms.StorageDir, name))
|
||||
return &Media{
|
||||
Type: "image",
|
||||
Name: name,
|
||||
Size: size,
|
||||
}
|
||||
}
|
||||
|
||||
func (ms *MediaSource) Thumb(m Media, size image.Rectangle) (string, error) {
|
||||
if ms.HasThumb(m, size) {
|
||||
return ms.ThumbPath(m, size), nil
|
||||
}
|
||||
|
||||
f, err := os.Open(ms.LocationOrig(m))
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
img, _, err := image.Decode(f)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return ms.ThumbImage(img, m, size)
|
||||
}
|
||||
|
||||
func (ms *MediaSource) ThumbImage(img image.Image, m Media, size image.Rectangle) (string, error) {
|
||||
|
||||
thumbLoc := ms.ThumbFilename(m, size)
|
||||
os.MkdirAll(path.Dir(thumbLoc), 0755)
|
||||
fthumb, err := os.OpenFile(thumbLoc, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0655)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
img = resize.Resize(uint(size.Dx()), uint(size.Dy()), img, resize.Bilinear)
|
||||
|
||||
err = jpeg.Encode(fthumb, img, nil)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
err = fthumb.Close()
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return ms.ThumbPath(m, size), nil
|
||||
}
|
||||
|
||||
func (ms *MediaSource) Walk() ([]*Media, error) {
|
||||
var media []*Media
|
||||
|
||||
err := filepath.Walk(ms.StorageDir, func(name string, fi os.FileInfo, err error) error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if fi.IsDir() {
|
||||
return nil
|
||||
}
|
||||
media = append(media, ms.ByName(path.Base(name)))
|
||||
return nil
|
||||
})
|
||||
|
||||
return media, err
|
||||
}
|
||||
|
||||
func (ch *CaddyHugo) uploadMedia(w http.ResponseWriter, r *http.Request) (int, error) {
|
||||
if ch.Media == nil {
|
||||
http.NotFound(w, r)
|
||||
@@ -227,7 +35,7 @@ func (ch *CaddyHugo) uploadMedia(w http.ResponseWriter, r *http.Request) (int, e
|
||||
|
||||
name := part.FileName()
|
||||
if name != "" {
|
||||
err = ch.Media.receiveNewMedia(name, part)
|
||||
err = ch.Media.ReceiveNewMedia(name, part)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return 500, nil
|
||||
@@ -269,13 +77,13 @@ func (ch *CaddyHugo) serveMediaPage(w http.ResponseWriter, r *http.Request) (int
|
||||
<div style="display: inline-block; width: 100%; height: 10vh;"></div>
|
||||
`)
|
||||
if ch.Media != nil {
|
||||
media, err := ch.Media.Walk()
|
||||
mm, err := ch.Media.Walk()
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return 500, nil
|
||||
}
|
||||
|
||||
for _, m := range media {
|
||||
for _, m := range media.Set(mm).ByDate() {
|
||||
|
||||
src, size, err := ch.Media.ThumbMax(*m, 100)
|
||||
if err != nil {
|
||||
@@ -314,49 +122,6 @@ func (ch *CaddyHugo) serveMediaPage(w http.ResponseWriter, r *http.Request) (int
|
||||
return 200, nil
|
||||
}
|
||||
|
||||
var (
|
||||
sizeString = regexp.MustCompile(`([0-9]*)(x)?([0-9]*)`)
|
||||
)
|
||||
|
||||
func parseSizeString(str string, actual image.Rectangle) (image.Rectangle, error) {
|
||||
var err = fmt.Errorf("expected a size string {width}x{height}, saw %q", str)
|
||||
|
||||
strs := sizeString.FindStringSubmatch(str)
|
||||
if len(strs) < 4 {
|
||||
return image.ZR, err
|
||||
}
|
||||
|
||||
var w, h int
|
||||
var strconvErr error
|
||||
|
||||
if strs[1] != "" {
|
||||
w, strconvErr = strconv.Atoi(strs[1])
|
||||
if strconvErr != nil {
|
||||
return image.ZR, err
|
||||
}
|
||||
}
|
||||
|
||||
if strs[3] != "" {
|
||||
h, strconvErr = strconv.Atoi(strs[3])
|
||||
if strconvErr != nil {
|
||||
return image.ZR, err
|
||||
}
|
||||
}
|
||||
|
||||
if strs[2] != "x" {
|
||||
// w was the only dimension given, so set it to the greater dimension
|
||||
// of the actual image size
|
||||
if actual.Dx() > actual.Dy() {
|
||||
h = 0
|
||||
} else {
|
||||
h = w
|
||||
w = 0
|
||||
}
|
||||
}
|
||||
|
||||
return image.Rect(0, 0, w, h), nil
|
||||
}
|
||||
|
||||
func (ch *CaddyHugo) serveMedia(w http.ResponseWriter, r *http.Request) (int, error) {
|
||||
if ch.Media == nil {
|
||||
http.NotFound(w, r)
|
||||
@@ -372,7 +137,7 @@ func (ch *CaddyHugo) serveMedia(w http.ResponseWriter, r *http.Request) (int, er
|
||||
|
||||
if len(segs) >= 4 && len(segs) > 2 {
|
||||
var err error
|
||||
size, err = parseSizeString(segs[len(segs)-2], m.Size)
|
||||
size, err = media.ParseSizeString(segs[len(segs)-2], m.Size)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||
return 400, nil
|
||||
@@ -395,8 +160,3 @@ func (ch *CaddyHugo) serveMedia(w http.ResponseWriter, r *http.Request) (int, er
|
||||
|
||||
return 200, nil
|
||||
}
|
||||
|
||||
func removeExtension(name string) string {
|
||||
ext := path.Ext(name)
|
||||
return name[:len(name)-len(ext)]
|
||||
}
|
||||
|
||||
+320
@@ -0,0 +1,320 @@
|
||||
package media
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"image"
|
||||
"image/jpeg"
|
||||
"io"
|
||||
"os"
|
||||
"path"
|
||||
"path/filepath"
|
||||
"regexp"
|
||||
"sort"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
// for processing images
|
||||
_ "image/gif"
|
||||
_ "image/png"
|
||||
|
||||
"github.com/nfnt/resize"
|
||||
"github.com/tajtiattila/metadata"
|
||||
)
|
||||
|
||||
type MediaSource struct {
|
||||
StorageDir string
|
||||
ThumbDir string
|
||||
|
||||
set Set
|
||||
}
|
||||
|
||||
type Media struct {
|
||||
Type string
|
||||
Name string
|
||||
Size image.Rectangle
|
||||
FullName string
|
||||
|
||||
metadata *metadata.Metadata
|
||||
}
|
||||
|
||||
func (ms *MediaSource) LocationOrig(m Media) string {
|
||||
return path.Join(ms.StorageDir, m.Name)
|
||||
}
|
||||
|
||||
func (ms *MediaSource) ThumbPath(m Media, size image.Rectangle) string {
|
||||
w := size.Dx()
|
||||
h := size.Dy()
|
||||
|
||||
var ws, hs string
|
||||
|
||||
if w != 0 {
|
||||
ws = fmt.Sprint(w)
|
||||
}
|
||||
if h != 0 {
|
||||
hs = fmt.Sprint(h)
|
||||
}
|
||||
|
||||
thumbSlug := filepath.Join(fmt.Sprintf("%sx%s", ws, hs), m.Name)
|
||||
return path.Join("/media", thumbSlug)
|
||||
}
|
||||
|
||||
func (ms *MediaSource) ThumbFilename(m Media, size image.Rectangle) string {
|
||||
return filepath.Join(ms.ThumbDir, ms.ThumbPath(m, size))
|
||||
}
|
||||
|
||||
func (ms *MediaSource) ReceiveNewMedia(name string, r io.Reader) error {
|
||||
ms.set = nil
|
||||
|
||||
dest := path.Join(ms.StorageDir, name)
|
||||
f, err := os.OpenFile(dest, os.O_WRONLY|os.O_CREATE, 0644)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
_, err = io.Copy(f, r)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return f.Close()
|
||||
}
|
||||
|
||||
func (m *Media) Date() time.Time {
|
||||
m.getMetadata()
|
||||
|
||||
if m.metadata != nil {
|
||||
raw := m.metadata.Get(metadata.DateTimeOriginal)
|
||||
if raw != "" {
|
||||
d, err := time.Parse("2006-01-02T15:04:05", raw)
|
||||
if err == nil {
|
||||
return d
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fi, err := os.Stat(m.FullName)
|
||||
if err == nil {
|
||||
return fi.ModTime()
|
||||
}
|
||||
|
||||
return time.Time{}
|
||||
}
|
||||
|
||||
func (m *Media) getMetadata() error {
|
||||
if m.metadata != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
f, err := os.Open(m.FullName)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
md, err := metadata.Parse(f)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
m.metadata = md
|
||||
return nil
|
||||
}
|
||||
|
||||
func (ms *MediaSource) Size(name string) (image.Rectangle, error) {
|
||||
f, err := os.Open(name)
|
||||
if err != nil {
|
||||
return image.ZR, err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
cfg, _, err := image.DecodeConfig(f)
|
||||
if err != nil {
|
||||
return image.ZR, err
|
||||
}
|
||||
|
||||
width := cfg.Width
|
||||
height := cfg.Height
|
||||
|
||||
return image.Rect(0, 0, width, height), nil
|
||||
}
|
||||
|
||||
func (ms *MediaSource) ThumbMax(m Media, maxDim int) (string, image.Rectangle, error) {
|
||||
f, err := os.Open(ms.LocationOrig(m))
|
||||
if err != nil {
|
||||
return "", image.ZR, err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
cfg, _, err := image.DecodeConfig(f)
|
||||
if err != nil {
|
||||
return "", image.ZR, err
|
||||
}
|
||||
|
||||
width := cfg.Width
|
||||
height := cfg.Height
|
||||
|
||||
if width > height {
|
||||
height = height * maxDim / width
|
||||
width = maxDim
|
||||
} else {
|
||||
width = width * maxDim / height
|
||||
height = maxDim
|
||||
}
|
||||
|
||||
size := image.Rect(0, 0, width, height)
|
||||
if ms.HasThumb(m, size) {
|
||||
return ms.ThumbPath(m, size), size, nil
|
||||
}
|
||||
|
||||
_, err = f.Seek(0, io.SeekStart)
|
||||
if err != nil {
|
||||
return "", image.ZR, err
|
||||
}
|
||||
|
||||
src, err := ms.thumbReader(f, m, size)
|
||||
return src, size, err
|
||||
}
|
||||
|
||||
func (ms *MediaSource) HasThumb(m Media, size image.Rectangle) bool {
|
||||
fi, err := os.Stat(ms.ThumbFilename(m, size))
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
return m.Date().Before(fi.ModTime())
|
||||
}
|
||||
|
||||
func (ms *MediaSource) ByName(name string) *Media {
|
||||
size, _ := ms.Size(path.Join(ms.StorageDir, name))
|
||||
m := Media{
|
||||
Type: "image",
|
||||
Name: name,
|
||||
Size: size,
|
||||
}
|
||||
m.FullName = ms.LocationOrig(m)
|
||||
return &m
|
||||
}
|
||||
|
||||
func (ms *MediaSource) Thumb(m Media, size image.Rectangle) (string, error) {
|
||||
if ms.HasThumb(m, size) {
|
||||
return ms.ThumbPath(m, size), nil
|
||||
}
|
||||
|
||||
f, err := os.Open(ms.LocationOrig(m))
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
defer f.Close()
|
||||
|
||||
return ms.thumbReader(f, m, size)
|
||||
}
|
||||
|
||||
func (ms *MediaSource) thumbReader(r io.Reader, m Media, size image.Rectangle) (string, error) {
|
||||
img, _, err := image.Decode(r)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return ms.ThumbImage(img, m, size)
|
||||
}
|
||||
|
||||
func (ms *MediaSource) ThumbImage(img image.Image, m Media, size image.Rectangle) (string, error) {
|
||||
thumbLoc := ms.ThumbFilename(m, size)
|
||||
os.MkdirAll(path.Dir(thumbLoc), 0755)
|
||||
fthumb, err := os.OpenFile(thumbLoc, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0655)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
img = resize.Resize(uint(size.Dx()), uint(size.Dy()), img, resize.Bilinear)
|
||||
|
||||
err = jpeg.Encode(fthumb, img, nil)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
err = fthumb.Close()
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return ms.ThumbPath(m, size), nil
|
||||
}
|
||||
|
||||
func (ms *MediaSource) Walk() ([]*Media, error) {
|
||||
if ms.set == nil {
|
||||
var media []*Media
|
||||
|
||||
err := filepath.Walk(ms.StorageDir, func(name string, fi os.FileInfo, err error) error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if fi.IsDir() {
|
||||
return nil
|
||||
}
|
||||
media = append(media, ms.ByName(path.Base(name)))
|
||||
return nil
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
return media, err
|
||||
}
|
||||
|
||||
ms.set = media
|
||||
}
|
||||
|
||||
return ms.set, nil
|
||||
}
|
||||
|
||||
type Set []*Media
|
||||
|
||||
func (s Set) ByDate() Set {
|
||||
sort.Slice(s, func(i, j int) bool {
|
||||
return s[i].Date().After(s[j].Date())
|
||||
})
|
||||
return s
|
||||
}
|
||||
|
||||
var (
|
||||
sizeString = regexp.MustCompile(`([0-9]*)(x)?([0-9]*)`)
|
||||
)
|
||||
|
||||
func ParseSizeString(str string, actual image.Rectangle) (image.Rectangle, error) {
|
||||
var err = fmt.Errorf("expected a size string {width}x{height}, saw %q", str)
|
||||
|
||||
strs := sizeString.FindStringSubmatch(str)
|
||||
if len(strs) < 4 {
|
||||
return image.ZR, err
|
||||
}
|
||||
|
||||
var w, h int
|
||||
var strconvErr error
|
||||
|
||||
if strs[1] != "" {
|
||||
w, strconvErr = strconv.Atoi(strs[1])
|
||||
if strconvErr != nil {
|
||||
return image.ZR, err
|
||||
}
|
||||
}
|
||||
|
||||
if strs[3] != "" {
|
||||
h, strconvErr = strconv.Atoi(strs[3])
|
||||
if strconvErr != nil {
|
||||
return image.ZR, err
|
||||
}
|
||||
}
|
||||
|
||||
if strs[2] != "x" {
|
||||
// w was the only dimension given, so set it to the greater dimension
|
||||
// of the actual image size
|
||||
if actual.Dx() > actual.Dy() {
|
||||
h = 0
|
||||
} else {
|
||||
h = w
|
||||
w = 0
|
||||
}
|
||||
}
|
||||
|
||||
return image.Rect(0, 0, w, h), nil
|
||||
}
|
||||
|
||||
func removeExtension(name string) string {
|
||||
ext := path.Ext(name)
|
||||
return name[:len(name)-len(ext)]
|
||||
}
|
||||
@@ -1,4 +1,4 @@
|
||||
package caddyhugo
|
||||
package media
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
@@ -33,7 +33,7 @@ func TestThumbSizeStrings(t *testing.T) {
|
||||
|
||||
for _, c := range cases {
|
||||
t.Run(fmt.Sprint(c.input, "=>", c.actualSize), func(t *testing.T) {
|
||||
got, err := parseSizeString(c.input, image.Rect(0, 0, 100, 100))
|
||||
got, err := ParseSizeString(c.input, image.Rect(0, 0, 100, 100))
|
||||
if err != nil {
|
||||
t.Errorf("error parsing size string: %v", err)
|
||||
return
|
||||
@@ -3,11 +3,11 @@ package caddyhugo
|
||||
import (
|
||||
"fmt"
|
||||
"html/template"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"path"
|
||||
|
||||
"git.stephensearles.com/stephen/caddy-hugo2/comments"
|
||||
"git.stephensearles.com/stephen/caddy-hugo2/media"
|
||||
|
||||
"github.com/gohugoio/hugo/deps"
|
||||
"github.com/gohugoio/hugo/hugofs"
|
||||
@@ -89,14 +89,9 @@ func (ch *CaddyHugo) Setup(dir string) error {
|
||||
return fmt.Errorf("edit template invalid: %v", err)
|
||||
}
|
||||
|
||||
thumbDir, err := ioutil.TempDir("", "thumbs")
|
||||
if err != nil {
|
||||
return fmt.Errorf("couldn't initialize media: %v", err)
|
||||
}
|
||||
|
||||
ch.Media = &MediaSource{
|
||||
ch.Media = &media.MediaSource{
|
||||
StorageDir: path.Join(dir, "media"),
|
||||
ThumbDir: thumbDir,
|
||||
ThumbDir: path.Join(dir, "thumbs"),
|
||||
}
|
||||
|
||||
err = os.MkdirAll(ch.Media.StorageDir, 0755)
|
||||
|
||||
Reference in New Issue
Block a user