diff --git a/LightlessSyncServer/LightlessSyncServer/Hubs/LightlessHub.Functions.cs b/LightlessSyncServer/LightlessSyncServer/Hubs/LightlessHub.Functions.cs index c5fd851..b0910f4 100644 --- a/LightlessSyncServer/LightlessSyncServer/Hubs/LightlessHub.Functions.cs +++ b/LightlessSyncServer/LightlessSyncServer/Hubs/LightlessHub.Functions.cs @@ -211,7 +211,8 @@ public partial class LightlessHub if (isOwnerResult.ReferredGroup == null) return (false, null); - var groupPairSelf = await DbContext.GroupPairs.SingleOrDefaultAsync(g => g.GroupGID == gid || g.Group.Alias == gid && g.GroupUserUID == UserUID).ConfigureAwait(false); + var groupPairSelf = await DbContext.GroupPairs.SingleOrDefaultAsync( + g => (g.GroupGID == gid || g.Group.Alias == gid) && g.GroupUserUID == UserUID).ConfigureAwait(false); if (groupPairSelf == null || !groupPairSelf.IsModerator) return (false, null); return (true, isOwnerResult.ReferredGroup); diff --git a/LightlessSyncServer/LightlessSyncServer/Hubs/LightlessHub.Groups.cs b/LightlessSyncServer/LightlessSyncServer/Hubs/LightlessHub.Groups.cs index 80f8d5f..e31e482 100644 --- a/LightlessSyncServer/LightlessSyncServer/Hubs/LightlessHub.Groups.cs +++ b/LightlessSyncServer/LightlessSyncServer/Hubs/LightlessHub.Groups.cs @@ -799,12 +799,12 @@ public partial class LightlessHub if (!hasRights) return; var groupProfileDb = await DbContext.GroupProfiles - .FirstOrDefaultAsync(g => g.Group.GID == dto.Group.GID || g.Group.Alias == dto.Group.GID, - cancellationToken) + .Include(g => g.Group) + .FirstOrDefaultAsync(g => g.GroupGID == dto.Group.GID, cancellationToken) .ConfigureAwait(false); - ImageCheckService.ImageLoadResult profileResult = null; - ImageCheckService.ImageLoadResult bannerResult = null; + ImageCheckService.ImageLoadResult profileResult = new(); + ImageCheckService.ImageLoadResult bannerResult = new(); //Avatar image validation if (!string.IsNullOrEmpty(dto.PictureBase64)) @@ -830,27 +830,33 @@ public partial class LightlessHub } } + var sanitizedProfileImage = profileResult?.Base64Image; + var sanitizedBannerImage = bannerResult?.Base64Image; + if (groupProfileDb == null) { groupProfileDb = new GroupProfile { GroupGID = dto.Group.GID, + Group = group, ProfileDisabled = false, IsNSFW = dto.IsNsfw ?? false, }; - groupProfileDb.UpdateProfileFromDto(dto, profileResult.Base64Image, bannerResult.Base64Image); + groupProfileDb.UpdateProfileFromDto(dto, sanitizedProfileImage, sanitizedBannerImage); await DbContext.GroupProfiles.AddAsync(groupProfileDb, cancellationToken).ConfigureAwait(false); } else { + groupProfileDb.Group ??= group; + if (groupProfileDb?.ProfileDisabled ?? false) { await Clients.Caller.Client_ReceiveServerMessage(MessageSeverity.Error, "Your profile was permanently disabled and cannot be edited").ConfigureAwait(false); return; } - groupProfileDb.UpdateProfileFromDto(dto, profileResult.Base64Image, bannerResult.Base64Image); + groupProfileDb.UpdateProfileFromDto(dto, sanitizedProfileImage, sanitizedBannerImage); var userIds = await DbContext.GroupPairs .Where(p => p.GroupGID == groupProfileDb.GroupGID) diff --git a/LightlessSyncServer/LightlessSyncServer/Utils/Extensions.cs b/LightlessSyncServer/LightlessSyncServer/Utils/Extensions.cs index e0d09d9..3911ff0 100644 --- a/LightlessSyncServer/LightlessSyncServer/Utils/Extensions.cs +++ b/LightlessSyncServer/LightlessSyncServer/Utils/Extensions.cs @@ -17,22 +17,22 @@ public static class Extensions if (profile == null || dto == null) return; - profile.Base64GroupProfileImage = string.IsNullOrWhiteSpace(base64PictureString) ? null : base64PictureString; - profile.Base64GroupBannerImage = string.IsNullOrWhiteSpace(base64BannerString) ? null : base64BannerString; + if (base64PictureString != null) profile.Base64GroupProfileImage = base64PictureString; + if (base64BannerString != null) profile.Base64GroupBannerImage = base64BannerString; if (dto.Tags != null) profile.Tags = dto.Tags; if (dto.Description != null) profile.Description = dto.Description; if (dto.IsNsfw.HasValue) profile.IsNSFW = dto.IsNsfw.Value; } - public static void UpdateProfileFromDto(this UserProfileData profile, UserProfileDto dto, string? base64PictureString, string? base64BannerString = null) + public static void UpdateProfileFromDto(this UserProfileData profile, UserProfileDto dto, string? base64PictureString = null, string? base64BannerString = null) { ArgumentNullException.ThrowIfNull(profile); ArgumentNullException.ThrowIfNull(dto); if (profile == null || dto == null) return; - profile.Base64ProfileImage = string.IsNullOrWhiteSpace(base64PictureString) ? null : base64PictureString; - profile.Base64BannerImage = string.IsNullOrWhiteSpace(base64BannerString) ? null : base64BannerString; + if (base64PictureString != null) profile.Base64ProfileImage = base64PictureString; + if (base64BannerString != null) profile.Base64BannerImage = base64BannerString; if (dto.Tags != null) profile.Tags = dto.Tags; if (dto.Description != null) profile.UserDescription = dto.Description; if (dto.IsNSFW.HasValue) profile.IsNSFW = dto.IsNSFW.Value; @@ -45,7 +45,8 @@ public static class Extensions return new GroupProfileDto(Group: null, Description: null, Tags: null, PictureBase64: null, BannerBase64: null, IsNsfw: false, IsDisabled: false); } - var groupData = groupProfile.Group?.ToGroupData(); + var groupData = groupProfile.Group?.ToGroupData() + ?? (!string.IsNullOrWhiteSpace(groupProfile.GroupGID) ? new GroupData(groupProfile.GroupGID) : null); return new GroupProfileDto( groupData,