1 files changed, 14 insertions, 15 deletions
diff --git a/internal/media/handler.go b/internal/media/handler.go
index 9c1d3227e..f2450bbc0 100644
--- a/internal/media/handler.go
+++ b/internal/media/handler.go
@@ -26,8 +26,8 @@ import (
 	"strings"
 	"time"
 
+	"git.iim.gay/grufwub/go-store/kv"
 	"github.com/sirupsen/logrus"
-	"github.com/superseriousbusiness/gotosocial/internal/blob"
 	"github.com/superseriousbusiness/gotosocial/internal/config"
 	"github.com/superseriousbusiness/gotosocial/internal/db"
 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
@@ -84,19 +84,19 @@ type Handler interface {
 }
 
 type mediaHandler struct {
-	config  *config.Config
-	db      db.DB
-	storage blob.Storage
-	log     *logrus.Logger
+	config *config.Config
+	db     db.DB
+	store  *kv.KVStore
+	log    *logrus.Logger
 }
 
 // New returns a new handler with the given config, db, storage, and logger
-func New(config *config.Config, database db.DB, storage blob.Storage, log *logrus.Logger) Handler {
+func New(config *config.Config, database db.DB, store *kv.KVStore, log *logrus.Logger) Handler {
 	return &mediaHandler{
-		config:  config,
-		db:      database,
-		storage: storage,
-		log:     log,
+		config: config,
+		db:     database,
+		store:  store,
+		log:    log,
 	}
 }
 
@@ -256,13 +256,13 @@ func (mh *mediaHandler) ProcessLocalEmoji(ctx context.Context, emojiBytes []byte
 	emojiStaticURL := fmt.Sprintf("%s/%s/%s/%s/%s.png", URLbase, instanceAccount.ID, Emoji, Static, newEmojiID)
 	emojiStaticPath := fmt.Sprintf("%s/%s/%s/%s/%s.png", mh.config.StorageConfig.BasePath, instanceAccount.ID, Emoji, Static, newEmojiID)
 
-	// store the original
-	if err := mh.storage.StoreFileAt(emojiPath, original.image); err != nil {
+	// Store the original emoji
+	if err := mh.store.Put(emojiPath, original.image); err != nil {
 		return nil, fmt.Errorf("storage error: %s", err)
 	}
 
-	// store the static
-	if err := mh.storage.StoreFileAt(emojiStaticPath, static.image); err != nil {
+	// Store the static emoji
+	if err := mh.store.Put(emojiStaticPath, static.image); err != nil {
 		return nil, fmt.Errorf("storage error: %s", err)
 	}
 
@@ -293,7 +293,6 @@ func (mh *mediaHandler) ProcessLocalEmoji(ctx context.Context, emojiBytes []byte
 }
 
 func (mh *mediaHandler) ProcessRemoteHeaderOrAvatar(ctx context.Context, t transport.Transport, currentAttachment *gtsmodel.MediaAttachment, accountID string) (*gtsmodel.MediaAttachment, error) {
-
 	if !currentAttachment.Header && !currentAttachment.Avatar {
 		return nil, errors.New("provided attachment was set to neither header nor avatar")
 	}