feat: Add thumbnailer

1 files changed, 62 insertions, 0 deletions

diff --git a/pkg/worker/thumbnail_scanner.go b/pkg/worker/thumbnail_scanner.go
new file mode 100644
index 0000000..cc201b8
--- /dev/null
+++ b/pkg/worker/thumbnail_scanner.go
@@ -0,0 +1,62 @@
+package worker
+
+import (
+	"context"
+	"math"
+	"os"
+	"path"
+
+	"git.sr.ht/~gabrielgio/img/pkg/database/repository"
+	"git.sr.ht/~gabrielgio/img/pkg/fileop"
+)
+
+type (
+	ThumbnailScanner struct {
+		repository repository.MediaRepository
+		cachePath  string
+	}
+)
+
+var _ BatchProcessor[*repository.Media] = &EXIFScanner{}
+
+func NewThumbnailScanner(cachePath string, repository repository.MediaRepository) *ThumbnailScanner {
+	return &ThumbnailScanner{
+		repository: repository,
+		cachePath:  cachePath,
+	}
+}
+
+func (t *ThumbnailScanner) Query(ctx context.Context) ([]*repository.Media, error) {
+	return t.repository.ListEmptyThumbnail(ctx, &repository.Pagination{
+		Page: 0,
+		Size: 100,
+	})
+}
+
+func (t *ThumbnailScanner) Process(ctx context.Context, media *repository.Media) error {
+	split := media.PathHash[:2]
+	filename := media.PathHash[2:]
+	folder := path.Join(t.cachePath, split)
+	output := path.Join(folder, filename+".jpeg")
+
+	err := os.MkdirAll(folder, os.ModePerm)
+	if err != nil {
+		return err
+	}
+
+	if media.IsVideo() {
+		err := fileop.EncodeVideoThumbnail(media.Path, output, 1080, 1080)
+		if err != nil {
+			return err
+		}
+	} else {
+		err := fileop.EncodeImageThumbnail(media.Path, output, 1080, math.MaxInt)
+		if err != nil {
+			return err
+		}
+	}
+
+	return t.repository.CreateThumbnail(ctx, media.ID, &repository.MediaThumbnail{
+		Path: output,
+	})
+}