1 files changed, 81 insertions, 0 deletions
diff --git a/pkg/worker/file_scanner.go b/pkg/worker/file_scanner.go
new file mode 100644
index 0000000..321fbca
--- /dev/null
+++ b/pkg/worker/file_scanner.go
@@ -0,0 +1,81 @@
+package worker
+
+import (
+	"context"
+	"crypto/md5"
+	"encoding/hex"
+	"io/fs"
+	"path/filepath"
+
+	"github.com/gabriel-vasile/mimetype"
+
+	"git.sr.ht/~gabrielgio/img/pkg/components/media"
+)
+
+type (
+	FileScanner struct {
+		root       string
+		repository media.Repository
+	}
+)
+
+var _ ChanProcessor[string] = &FileScanner{}
+
+func NewFileScanner(root string, repository media.Repository) *FileScanner {
+	return &FileScanner{
+		root:       root,
+		repository: repository,
+	}
+}
+
+func (f *FileScanner) Query(ctx context.Context) (<-chan string, error) {
+	c := make(chan string)
+	go func() {
+		defer close(c)
+		_ = filepath.Walk(f.root, func(path string, info fs.FileInfo, err error) error {
+			if info.IsDir() && filepath.Base(info.Name())[0] == '.' {
+				return filepath.SkipDir
+			}
+
+			if info.IsDir() {
+				return nil
+			}
+
+			if filepath.Ext(info.Name()) != ".jpg" &&
+				filepath.Ext(info.Name()) != ".jpeg" &&
+				filepath.Ext(info.Name()) != ".png" {
+				return nil
+			}
+			c <- path
+			return nil
+		})
+	}()
+	return c, nil
+}
+
+func (f *FileScanner) Process(ctx context.Context, path string) error {
+	hash := md5.Sum([]byte(path))
+	str := hex.EncodeToString(hash[:])
+	name := filepath.Base(path)
+
+	exists, errResp := f.repository.Exists(ctx, str)
+	if errResp != nil {
+		return errResp
+	}
+
+	if exists {
+		return nil
+	}
+
+	mime, errResp := mimetype.DetectFile(path)
+	if errResp != nil {
+		return errResp
+	}
+
+	return f.repository.Create(ctx, &media.CreateMedia{
+		Name:     name,
+		Path:     path,
+		PathHash: str,
+		MIMEType: mime.String(),
+	})
+}