notfire/Iceshrimp.NET
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[backend/api] Fix edge case where GetAvatar and GetBanner can be both IsAvatar and IsBanner

Browse source
This commit is contained in:
pancakes 2025-01-30 22:26:39 +10:00 • committed by Laura Hausmann
parent 6322d5fb91
commit 16a0a76528
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
Iceshrimp.Backend/Controllers/Web/ProfileController.cs
View file

@ -117,6 +117,8 @@ public class ProfileController(
var user = HttpContext.GetUserOrFail();
var file = await db.DriveFiles
.Include(p => p.UserAvatar)
.Include(p => p.UserBanner)
.FirstOrDefaultAsync(p => p.UserId == user.Id && p.UserAvatar != null)
?? throw GracefulException.RecordNotFound();
@ -129,8 +131,8 @@ public class ProfileController(
ContentType = file.Type,
Sensitive = file.IsSensitive,
Description = file.Comment,
IsAvatar = true,
IsBanner = false
IsAvatar = file.UserAvatar != null,
IsBanner = file.UserBanner != null
};
}
@ -189,6 +191,8 @@ public class ProfileController(
var user = HttpContext.GetUserOrFail();
var file = await db.DriveFiles
.Include(p => p.UserAvatar)
.Include(p => p.UserBanner)
.FirstOrDefaultAsync(p => p.UserId == user.Id && p.UserBanner != null)
?? throw GracefulException.RecordNotFound();
@ -201,8 +205,8 @@ public class ProfileController(
ContentType = file.Type,
Sensitive = file.IsSensitive,
Description = file.Comment,
IsAvatar = false,
IsBanner = true
IsAvatar = file.UserAvatar != null,
IsBanner = file.UserBanner != null
};
}