refac: decouple persistence and image storage behind driver registries

internal/infrastructure/storage/localStorage.go

@@ -0,0 +1,74 @@
+package storage
+
+import (
+	domain "RenewCMS/internal/domain/image"
+	"errors"
+	"io"
+	"os"
+	"path/filepath"
+
+	"github.com/google/uuid"
+)
+
+var contentTypeExtensions = map[string]string{
+	"image/png":     ".png",
+	"image/jpeg":    ".jpeg",
+	"image/webp":    ".webp",
+	"image/svg+xml": ".svg",
+}
+
+type LocalStorage struct {
+	directory string
+}
+
+var _ domain.Storage = &LocalStorage{}
+
+func NewLocalStorage(directory string) *LocalStorage {
+	return &LocalStorage{directory: directory}
+}
+
+func (s *LocalStorage) Put(contentType string, content io.Reader) (string, error) {
+	data, err := io.ReadAll(content)
+	if err != nil {
+		return "", err
+	}
+	if err := os.MkdirAll(s.directory, 0755); err != nil {
+		return "", err
+	}
+	storedName := uuid.NewString() + contentTypeExtensions[contentType]
+	if err := os.WriteFile(filepath.Join(s.directory, storedName), data, 0666); err != nil {
+		return "", err
+	}
+	return storedName, nil
+}
+
+func (s *LocalStorage) Open(storedName string) (domain.Blob, error) {
+	if !isPlainName(storedName) {
+		return domain.Blob{}, domain.ErrBlobNotFound
+	}
+	file, err := os.Open(filepath.Join(s.directory, storedName))
+	if err != nil {
+		return domain.Blob{}, domain.ErrBlobNotFound
+	}
+	info, err := file.Stat()
+	if err != nil || info.IsDir() {
+		_ = file.Close()
+		return domain.Blob{}, domain.ErrBlobNotFound
+	}
+	return domain.Blob{Content: file, Size: info.Size(), ModTime: info.ModTime()}, nil
+}
+
+func (s *LocalStorage) Delete(storedName string) error {
+	if !isPlainName(storedName) {
+		return nil
+	}
+	err := os.Remove(filepath.Join(s.directory, storedName))
+	if err != nil && !errors.Is(err, os.ErrNotExist) {
+		return err
+	}
+	return nil
+}
+
+func isPlainName(name string) bool {
+	return name != "" && name != "." && name != ".." && filepath.Base(name) == name && filepath.ToSlash(name) == name
+}

internal/infrastructure/storage/storage.go

@@ -0,0 +1,22 @@
+package storage
+
+import (
+	domain "RenewCMS/internal/domain/image"
+	"fmt"
+)
+
+const DriverLocal = "local"
+
+type Config struct {
+	Driver   string
+	Location string
+}
+
+func Open(config Config) (domain.Storage, error) {
+	switch config.Driver {
+	case DriverLocal:
+		return NewLocalStorage(config.Location), nil
+	default:
+		return nil, fmt.Errorf("unsupported storage driver %q", config.Driver)
+	}
+}