From 21bb324156f582e918a097ea744e52fc21b2ddf4 Mon Sep 17 00:00:00 2001 From: kim <89579420+NyaaaWhatsUpDoc@users.noreply.github.com> Date: Wed, 26 Jun 2024 15:01:16 +0000 Subject: [chore] media and emoji refactoring (#3000) * start updating media manager interface ready for storing attachments / emoji right away * store emoji and media as uncached immediately, then (re-)cache on Processing{}.Load() * remove now unused media workers * fix tests and issues * fix another test! * fix emoji activitypub uri setting behaviour, fix remainder of test compilation issues * fix more tests * fix (most of) remaining tests, add debouncing to repeatedly failing media / emojis * whoops, rebase issue * remove kim's whacky experiments * do some reshuffling, ensure emoji uri gets set * ensure marked as not cached on cleanup * tweaks to media / emoji processing to handle context canceled better * ensure newly fetched emojis actually get set in returned slice * use different varnames to be a bit more obvious * move emoji refresh rate limiting to dereferencer * add exported dereferencer functions for remote media, use these for recaching in processor * add check for nil attachment in updateAttachment() * remove unused emoji and media fields + columns * see previous commit * fix old migrations expecting image_updated_at to exists (from copies of old models) * remove freshness checking code (seems to be broken...) * fix error arg causing nil ptr exception * finish documentating functions with comments, slight tweaks to media / emoji deref error logic * remove some extra unneeded boolean checking * finish writing documentation (code comments) for exported media manager methods * undo changes to migration snapshot gtsmodels, updated failing migration to have its own snapshot * move doesColumnExist() to util.go in migrations package --- internal/federation/dereferencing/emoji_test.go | 45 +++++++++++++------------ 1 file changed, 24 insertions(+), 21 deletions(-) (limited to 'internal/federation/dereferencing/emoji_test.go') diff --git a/internal/federation/dereferencing/emoji_test.go b/internal/federation/dereferencing/emoji_test.go index 08365741f..fdb815762 100644 --- a/internal/federation/dereferencing/emoji_test.go +++ b/internal/federation/dereferencing/emoji_test.go @@ -19,6 +19,7 @@ package dereferencing_test import ( "context" + "fmt" "testing" "time" @@ -32,48 +33,50 @@ type EmojiTestSuite struct { func (suite *EmojiTestSuite) TestDereferenceEmojiBlocking() { ctx := context.Background() - fetchingAccount := suite.testAccounts["local_account_1"] emojiImageRemoteURL := "http://example.org/media/emojis/1781772.gif" emojiImageStaticRemoteURL := "http://example.org/media/emojis/1781772.gif" emojiURI := "http://example.org/emojis/1781772" emojiShortcode := "peglin" - emojiID := "01GCBMGNZBKMEE1KTZ6PMJEW5D" emojiDomain := "example.org" emojiDisabled := false emojiVisibleInPicker := false - ai := &media.AdditionalEmojiInfo{ - Domain: &emojiDomain, - ImageRemoteURL: &emojiImageRemoteURL, - ImageStaticRemoteURL: &emojiImageStaticRemoteURL, - Disabled: &emojiDisabled, - VisibleInPicker: &emojiVisibleInPicker, - } - - processingEmoji, err := suite.dereferencer.GetRemoteEmoji(ctx, fetchingAccount.Username, emojiImageRemoteURL, emojiShortcode, emojiDomain, emojiID, emojiURI, ai, false) - suite.NoError(err) - - // make a blocking call to load the emoji from the in-process media - emoji, err := processingEmoji.LoadEmoji(ctx) + emoji, err := suite.dereferencer.GetEmoji( + ctx, + emojiShortcode, + emojiDomain, + emojiImageRemoteURL, + media.AdditionalEmojiInfo{ + URI: &emojiURI, + Domain: &emojiDomain, + ImageRemoteURL: &emojiImageRemoteURL, + ImageStaticRemoteURL: &emojiImageStaticRemoteURL, + Disabled: &emojiDisabled, + VisibleInPicker: &emojiVisibleInPicker, + }, + false, + ) suite.NoError(err) suite.NotNil(emoji) - suite.Equal(emojiID, emoji.ID) + expectPath := fmt.Sprintf("/emoji/original/%s.gif", emoji.ID) + expectStaticPath := fmt.Sprintf("/emoji/static/%s.png", emoji.ID) + suite.WithinDuration(time.Now(), emoji.CreatedAt, 10*time.Second) suite.WithinDuration(time.Now(), emoji.UpdatedAt, 10*time.Second) suite.Equal(emojiShortcode, emoji.Shortcode) suite.Equal(emojiDomain, emoji.Domain) suite.Equal(emojiImageRemoteURL, emoji.ImageRemoteURL) suite.Equal(emojiImageStaticRemoteURL, emoji.ImageStaticRemoteURL) - suite.Contains(emoji.ImageURL, "/emoji/original/01GCBMGNZBKMEE1KTZ6PMJEW5D.gif") - suite.Contains(emoji.ImageStaticURL, "emoji/static/01GCBMGNZBKMEE1KTZ6PMJEW5D.png") - suite.Contains(emoji.ImagePath, "/emoji/original/01GCBMGNZBKMEE1KTZ6PMJEW5D.gif") - suite.Contains(emoji.ImageStaticPath, "/emoji/static/01GCBMGNZBKMEE1KTZ6PMJEW5D.png") + suite.Contains(emoji.ImageURL, expectPath) + suite.Contains(emoji.ImageStaticURL, expectStaticPath) + suite.Contains(emoji.ImagePath, expectPath) + suite.Contains(emoji.ImageStaticPath, expectStaticPath) suite.Equal("image/gif", emoji.ImageContentType) suite.Equal("image/png", emoji.ImageStaticContentType) suite.Equal(37796, emoji.ImageFileSize) suite.Equal(7951, emoji.ImageStaticFileSize) - suite.WithinDuration(time.Now(), emoji.ImageUpdatedAt, 10*time.Second) + suite.WithinDuration(time.Now(), emoji.UpdatedAt, 10*time.Second) suite.False(*emoji.Disabled) suite.Equal(emojiURI, emoji.URI) suite.False(*emoji.VisibleInPicker) -- cgit v1.2.3