rework data function to provide filesize

2 files changed, 3 insertions, 3 deletions

diff --git a/internal/federation/dereferencing/account.go b/internal/federation/dereferencing/account.go
index 6ea8256d5..581c95de2 100644
--- a/internal/federation/dereferencing/account.go
+++ b/internal/federation/dereferencing/account.go
@@ -252,7 +252,7 @@ func (d *deref) fetchHeaderAndAviForAccount(ctx context.Context, targetAccount *
 			return err
 		}
 
-		data := func(innerCtx context.Context) (io.Reader, error) {
+		data := func(innerCtx context.Context) (io.Reader, int, error) {
 			return t.DereferenceMedia(innerCtx, avatarIRI)
 		}
 
@@ -274,7 +274,7 @@ func (d *deref) fetchHeaderAndAviForAccount(ctx context.Context, targetAccount *
 			return err
 		}
 
-		data := func(innerCtx context.Context) (io.Reader, error) {
+		data := func(innerCtx context.Context) (io.Reader, int, error) {
 			return t.DereferenceMedia(innerCtx, headerIRI)
 		}
 
diff --git a/internal/federation/dereferencing/media.go b/internal/federation/dereferencing/media.go
index c427f2507..0b19570f2 100644
--- a/internal/federation/dereferencing/media.go
+++ b/internal/federation/dereferencing/media.go
@@ -42,7 +42,7 @@ func (d *deref) GetRemoteMedia(ctx context.Context, requestingUsername string, a
 		return nil, fmt.Errorf("GetRemoteMedia: error parsing url: %s", err)
 	}
 
-	dataFunc := func(innerCtx context.Context) (io.Reader, error) {
+	dataFunc := func(innerCtx context.Context) (io.Reader, int, error) {
 		return t.DereferenceMedia(innerCtx, derefURI)
 	}