Add Devices to account DTO

hhvrc · Jul 10, 2023 · 4bd5752 · 4bd5752
1 parent bf93aca
commit 4bd5752
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 0 deletions.
diff --git a/Common/DTOs/AccountDto.cs b/Common/DTOs/AccountDto.cs
@@ -28,6 +28,11 @@ namespace ZapMe.DTOs;
 
     public required string StatusText { get; init; }
 
+    /// <summary>
+    /// Ids of devices this account owns
+    /// </summary>
+    public required IEnumerable<Guid> OwnedDeviceIds { get; init; }
+
     /// <summary>
     /// Ids of users this account has friended
     /// </summary>

diff --git a/Common/Database/Models/Device.cs b/Common/Database/Models/Device.cs
@@ -40,5 +40,7 @@ public void Configure(EntityTypeBuilder<DeviceEntity> builder)
 
         builder.HasIndex(ud => ud.OwnerId);
         builder.HasIndex(ud => ud.AccessToken).IsUnique();
+
+        builder.HasOne(ud => ud.Owner).WithMany(u => u.OwnedDevices).OnDelete(DeleteBehavior.Cascade);
     }
 }
diff --git a/Common/Database/Models/User.cs b/Common/Database/Models/User.cs
@@ -57,6 +57,7 @@ public sealed class UserEntity
     public List<UserRelationEntity> RelationsOutgoing { get; private set; } = new List<UserRelationEntity>();
     public List<UserRelationEntity> RelationsIncoming { get; private set; } = new List<UserRelationEntity>();
     public List<SSOConnectionEntity> SSOConnections { get; private set; } = new List<SSOConnectionEntity>();
+    public List<DeviceEntity> OwnedDevices { get; private set; } = new List<DeviceEntity>();
 }
 
 public sealed class UserEntityConfiguration : IEntityTypeConfiguration<UserEntity>

diff --git a/Common/Mappers/UserMapper.cs b/Common/Mappers/UserMapper.cs
@@ -61,6 +61,7 @@ public static AccountDto MapToAccountDto(UserEntity user)
             BannerUrl = user.ProfileBanner?.PublicUrl,
             Status = user.Status,
             StatusText = user.StatusText,
+            OwnedDeviceIds = user.OwnedDevices.Select(d => d.Id),
             FriendUserIds = user.RelationsOutgoing.Where(r => r.FriendStatus == UserPartialRelationType.Accepted).Select(r => r.ToUserId),
             SSOConnections = user.SSOConnections.Select(oc => oc.ProviderName),
             CreatedAt = user.CreatedAt,