Filter to get only photos

creekorful · Aug 17, 2021 · e1b24fe · e1b24fe
1 parent 46048b9
commit e1b24fe
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/gallery.go b/gallery.go
@@ -16,6 +16,7 @@ import (
 	"os"
 	"path/filepath"
 	"sort"
+	"strings"
 	"time"
 )
 
@@ -168,7 +169,7 @@ func processImages(photosDir, outputDir string) ([]map[string]interface{}, error
 	var photos []map[string]interface{}
 
 	if err := filepath.Walk(photosDir, func(path string, info fs.FileInfo, err error) error {
-		if info.IsDir() {
+		if !isJpegFile(info) {
 			return nil
 		}
 
@@ -269,3 +270,7 @@ func isPhotoProcessed(photoBytes []byte, targetPath string) bool {
 
 	return bytes.Equal(photoBytes, targetPhotoBytes)
 }
+
+func isJpegFile(file fs.FileInfo) bool {
+	return !file.IsDir() && (strings.HasSuffix(file.Name(), ".jpg") || strings.HasSuffix(file.Name(), ".jpeg"))
+}