diff options
Diffstat (limited to 'internal/processing/admin')
| -rw-r--r-- | internal/processing/admin/admin.go | 8 | ||||
| -rw-r--r-- | internal/processing/admin/emoji.go | 48 | 
2 files changed, 27 insertions, 29 deletions
diff --git a/internal/processing/admin/admin.go b/internal/processing/admin/admin.go index 217d10dfe..bdb586588 100644 --- a/internal/processing/admin/admin.go +++ b/internal/processing/admin/admin.go @@ -38,21 +38,21 @@ type Processor interface {  	DomainBlocksGet(ctx context.Context, account *gtsmodel.Account, export bool) ([]*apimodel.DomainBlock, gtserror.WithCode)  	DomainBlockGet(ctx context.Context, account *gtsmodel.Account, id string, export bool) (*apimodel.DomainBlock, gtserror.WithCode)  	DomainBlockDelete(ctx context.Context, account *gtsmodel.Account, id string) (*apimodel.DomainBlock, gtserror.WithCode) -	EmojiCreate(ctx context.Context, account *gtsmodel.Account, user *gtsmodel.User, form *apimodel.EmojiCreateRequest) (*apimodel.Emoji, error) +	EmojiCreate(ctx context.Context, account *gtsmodel.Account, user *gtsmodel.User, form *apimodel.EmojiCreateRequest) (*apimodel.Emoji, gtserror.WithCode)  }  type processor struct {  	tc            typeutils.TypeConverter -	mediaHandler  media.Handler +	mediaManager  media.Manager  	fromClientAPI chan messages.FromClientAPI  	db            db.DB  }  // New returns a new admin processor. -func New(db db.DB, tc typeutils.TypeConverter, mediaHandler media.Handler, fromClientAPI chan messages.FromClientAPI) Processor { +func New(db db.DB, tc typeutils.TypeConverter, mediaManager media.Manager, fromClientAPI chan messages.FromClientAPI) Processor {  	return &processor{  		tc:            tc, -		mediaHandler:  mediaHandler, +		mediaManager:  mediaManager,  		fromClientAPI: fromClientAPI,  		db:            db,  	} diff --git a/internal/processing/admin/emoji.go b/internal/processing/admin/emoji.go index 4989d8e8d..6ef78aa65 100644 --- a/internal/processing/admin/emoji.go +++ b/internal/processing/admin/emoji.go @@ -19,55 +19,53 @@  package admin  import ( -	"bytes"  	"context"  	"errors"  	"fmt"  	"io"  	apimodel "github.com/superseriousbusiness/gotosocial/internal/api/model" +	"github.com/superseriousbusiness/gotosocial/internal/db" +	"github.com/superseriousbusiness/gotosocial/internal/gtserror"  	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"  	"github.com/superseriousbusiness/gotosocial/internal/id" +	"github.com/superseriousbusiness/gotosocial/internal/uris"  ) -func (p *processor) EmojiCreate(ctx context.Context, account *gtsmodel.Account, user *gtsmodel.User, form *apimodel.EmojiCreateRequest) (*apimodel.Emoji, error) { -	if user.Admin { -		return nil, fmt.Errorf("user %s not an admin", user.ID) +func (p *processor) EmojiCreate(ctx context.Context, account *gtsmodel.Account, user *gtsmodel.User, form *apimodel.EmojiCreateRequest) (*apimodel.Emoji, gtserror.WithCode) { +	if !user.Admin { +		return nil, gtserror.NewErrorNotAuthorized(fmt.Errorf("user %s not an admin", user.ID), "user is not an admin")  	} -	// open the emoji and extract the bytes from it -	f, err := form.Image.Open() -	if err != nil { -		return nil, fmt.Errorf("error opening emoji: %s", err) +	data := func(innerCtx context.Context) (io.Reader, int, error) { +		f, err := form.Image.Open() +		return f, int(form.Image.Size), err  	} -	buf := new(bytes.Buffer) -	size, err := io.Copy(buf, f) + +	emojiID, err := id.NewRandomULID()  	if err != nil { -		return nil, fmt.Errorf("error reading emoji: %s", err) -	} -	if size == 0 { -		return nil, errors.New("could not read provided emoji: size 0 bytes") +		return nil, gtserror.NewErrorInternalError(fmt.Errorf("error creating id for new emoji: %s", err), "error creating emoji ID")  	} -	// allow the mediaHandler to work its magic of processing the emoji bytes, and putting them in whatever storage backend we're using -	emoji, err := p.mediaHandler.ProcessLocalEmoji(ctx, buf.Bytes(), form.Shortcode) +	emojiURI := uris.GenerateURIForEmoji(emojiID) + +	processingEmoji, err := p.mediaManager.ProcessEmoji(ctx, data, form.Shortcode, emojiID, emojiURI, nil)  	if err != nil { -		return nil, fmt.Errorf("error reading emoji: %s", err) +		return nil, gtserror.NewErrorInternalError(fmt.Errorf("error processing emoji: %s", err), "error processing emoji")  	} -	emojiID, err := id.NewULID() +	emoji, err := processingEmoji.LoadEmoji(ctx)  	if err != nil { -		return nil, err +		var alreadyExistsError *db.ErrAlreadyExists +		if errors.As(err, &alreadyExistsError) { +			return nil, gtserror.NewErrorConflict(fmt.Errorf("emoji with shortcode %s already exists", form.Shortcode), fmt.Sprintf("emoji with shortcode %s already exists", form.Shortcode)) +		} +		return nil, gtserror.NewErrorInternalError(fmt.Errorf("error loading emoji: %s", err), "error loading emoji")  	} -	emoji.ID = emojiID  	apiEmoji, err := p.tc.EmojiToAPIEmoji(ctx, emoji)  	if err != nil { -		return nil, fmt.Errorf("error converting emoji to apitype: %s", err) -	} - -	if err := p.db.Put(ctx, emoji); err != nil { -		return nil, fmt.Errorf("database error while processing emoji: %s", err) +		return nil, gtserror.NewErrorInternalError(fmt.Errorf("error converting emoji: %s", err), "error converting emoji to api representation")  	}  	return &apiEmoji, nil  | 
