diff options
Diffstat (limited to 'internal/media')
| -rw-r--r-- | internal/media/handler.go | 29 | ||||
| -rw-r--r-- | internal/media/processicon.go | 4 | ||||
| -rw-r--r-- | internal/media/processimage.go | 5 |
3 files changed, 18 insertions, 20 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") } diff --git a/internal/media/processicon.go b/internal/media/processicon.go index 5fae63198..9ef4d77ec 100644 --- a/internal/media/processicon.go +++ b/internal/media/processicon.go @@ -85,13 +85,13 @@ func (mh *mediaHandler) processHeaderOrAvi(imageBytes []byte, contentType string // we store the original... originalPath := fmt.Sprintf("%s/%s/%s/%s/%s.%s", mh.config.StorageConfig.BasePath, accountID, mediaType, Original, newMediaID, extension) - if err := mh.storage.StoreFileAt(originalPath, original.image); err != nil { + if err := mh.store.Put(originalPath, original.image); err != nil { return nil, fmt.Errorf("storage error: %s", err) } // and a thumbnail... smallPath := fmt.Sprintf("%s/%s/%s/%s/%s.%s", mh.config.StorageConfig.BasePath, accountID, mediaType, Small, newMediaID, extension) - if err := mh.storage.StoreFileAt(smallPath, small.image); err != nil { + if err := mh.store.Put(smallPath, small.image); err != nil { return nil, fmt.Errorf("storage error: %s", err) } diff --git a/internal/media/processimage.go b/internal/media/processimage.go index a8a6d0716..3e7e60fb7 100644 --- a/internal/media/processimage.go +++ b/internal/media/processimage.go @@ -73,13 +73,13 @@ func (mh *mediaHandler) processImageAttachment(data []byte, minAttachment *gtsmo // we store the original... originalPath := fmt.Sprintf("%s/%s/%s/%s/%s.%s", mh.config.StorageConfig.BasePath, minAttachment.AccountID, Attachment, Original, newMediaID, extension) - if err := mh.storage.StoreFileAt(originalPath, original.image); err != nil { + if err := mh.store.Put(originalPath, original.image); err != nil { return nil, fmt.Errorf("storage error: %s", err) } // and a thumbnail... smallPath := fmt.Sprintf("%s/%s/%s/%s/%s.jpeg", mh.config.StorageConfig.BasePath, minAttachment.AccountID, Attachment, Small, newMediaID) // all thumbnails/smalls are encoded as jpeg - if err := mh.storage.StoreFileAt(smallPath, small.image); err != nil { + if err := mh.store.Put(smallPath, small.image); err != nil { return nil, fmt.Errorf("storage error: %s", err) } @@ -130,5 +130,4 @@ func (mh *mediaHandler) processImageAttachment(data []byte, minAttachment *gtsmo } return attachment, nil - } |