From 450923be2fe0970ffe3cd7ae02fa7980d60b984a Mon Sep 17 00:00:00 2001
From: Stephen Searles <stephen.searles@gmail.com>
Date: Thu, 7 Sep 2017 19:11:00 -0500
Subject: [PATCH] sorting images by date; acquiring metadata

---
 media.go       |  4 +--
 media/media.go | 95 ++++++++++++++++++++++++++++++++++++++++++--------
 2 files changed, 83 insertions(+), 16 deletions(-)

diff --git a/media.go b/media.go
index c33570d..c63e293 100644
--- a/media.go
+++ b/media.go
@@ -77,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 {
diff --git a/media/media.go b/media/media.go
index 16ae2e2..0674c3c 100644
--- a/media/media.go
+++ b/media/media.go
@@ -9,18 +9,23 @@ import (
 	"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
 }
 
 func (ms *MediaSource) LocationOrig(m Media) string {
@@ -49,6 +54,8 @@ func (ms *MediaSource) ThumbFilename(m Media, size image.Rectangle) string {
 }
 
 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 {
@@ -64,9 +71,50 @@ func (ms *MediaSource) ReceiveNewMedia(name string, r io.Reader) error {
 }
 
 type Media struct {
-	Type string
-	Name string
-	Size image.Rectangle
+	Type     string
+	Name     string
+	Size     image.Rectangle
+	FullName string
+
+	metadata *metadata.Metadata
+}
+
+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) {
@@ -136,11 +184,13 @@ func (ms *MediaSource) HasThumb(m Media, size image.Rectangle) bool {
 
 func (ms *MediaSource) ByName(name string) *Media {
 	size, _ := ms.Size(path.Join(ms.StorageDir, name))
-	return &Media{
+	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) {
@@ -187,20 +237,37 @@ func (ms *MediaSource) ThumbImage(img image.Image, m Media, size image.Rectangle
 }
 
 func (ms *MediaSource) Walk() ([]*Media, error) {
-	var media []*Media
+	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
+		})
 
-	err := filepath.Walk(ms.StorageDir, func(name string, fi os.FileInfo, err error) error {
 		if err != nil {
-			return err
-		}
-		if fi.IsDir() {
-			return nil
+			return media, err
 		}
-		media = append(media, ms.ByName(path.Base(name)))
-		return 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 (