Compare commits

...

3 Commits

  1. 39
      media/media.go
  2. 8
      setup.go

@ -28,6 +28,15 @@ type MediaSource struct {
set Set set Set
} }
type Media struct {
Type string
Name string
Size image.Rectangle
FullName string
metadata *metadata.Metadata
}
func (ms *MediaSource) LocationOrig(m Media) string { func (ms *MediaSource) LocationOrig(m Media) string {
return path.Join(ms.StorageDir, m.Name) return path.Join(ms.StorageDir, m.Name)
} }
@ -70,15 +79,6 @@ func (ms *MediaSource) ReceiveNewMedia(name string, r io.Reader) error {
return f.Close() return f.Close()
} }
type Media struct {
Type string
Name string
Size image.Rectangle
FullName string
metadata *metadata.Metadata
}
func (m *Media) Date() time.Time { func (m *Media) Date() time.Time {
m.getMetadata() m.getMetadata()
@ -168,18 +168,16 @@ func (ms *MediaSource) ThumbMax(m Media, maxDim int) (string, image.Rectangle, e
return "", image.ZR, err return "", image.ZR, err
} }
img, _, err := image.Decode(f) src, err := ms.thumbReader(f, m, size)
if err != nil {
return "", image.ZR, err
}
src, err := ms.ThumbImage(img, m, size)
return src, size, err return src, size, err
} }
func (ms *MediaSource) HasThumb(m Media, size image.Rectangle) bool { func (ms *MediaSource) HasThumb(m Media, size image.Rectangle) bool {
_, err := os.Stat(ms.ThumbFilename(m, size)) fi, err := os.Stat(ms.ThumbFilename(m, size))
return err == nil if err != nil {
return false
}
return m.Date().Before(fi.ModTime())
} }
func (ms *MediaSource) ByName(name string) *Media { func (ms *MediaSource) ByName(name string) *Media {
@ -204,7 +202,11 @@ func (ms *MediaSource) Thumb(m Media, size image.Rectangle) (string, error) {
} }
defer f.Close() defer f.Close()
img, _, err := image.Decode(f) 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 { if err != nil {
return "", err return "", err
} }
@ -213,7 +215,6 @@ func (ms *MediaSource) Thumb(m Media, size image.Rectangle) (string, error) {
} }
func (ms *MediaSource) ThumbImage(img image.Image, m Media, size image.Rectangle) (string, error) { func (ms *MediaSource) ThumbImage(img image.Image, m Media, size image.Rectangle) (string, error) {
thumbLoc := ms.ThumbFilename(m, size) thumbLoc := ms.ThumbFilename(m, size)
os.MkdirAll(path.Dir(thumbLoc), 0755) os.MkdirAll(path.Dir(thumbLoc), 0755)
fthumb, err := os.OpenFile(thumbLoc, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0655) fthumb, err := os.OpenFile(thumbLoc, os.O_CREATE|os.O_WRONLY|os.O_TRUNC, 0655)

@ -3,7 +3,6 @@ package caddyhugo
import ( import (
"fmt" "fmt"
"html/template" "html/template"
"io/ioutil"
"os" "os"
"path" "path"
@ -90,14 +89,9 @@ func (ch *CaddyHugo) Setup(dir string) error {
return fmt.Errorf("edit template invalid: %v", err) 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 = &media.MediaSource{ ch.Media = &media.MediaSource{
StorageDir: path.Join(dir, "media"), StorageDir: path.Join(dir, "media"),
ThumbDir: thumbDir, ThumbDir: path.Join(dir, "thumbs"),
} }
err = os.MkdirAll(ch.Media.StorageDir, 0755) err = os.MkdirAll(ch.Media.StorageDir, 0755)

Loading…
Cancel
Save