Send favourite users and levels to PSP clients

LittleBigRefresh · Oct 3, 2023 · 66b18bb · 66b18bb
1 parent 758b590
commit 66b18bb
Show file tree

Hide file tree

Showing 4 changed files with 56 additions and 10 deletions.
diff --git a/Refresh.GameServer/Endpoints/Game/DataTypes/Response/GameUserResponse.cs b/Refresh.GameServer/Endpoints/Game/DataTypes/Response/GameUserResponse.cs
@@ -1,7 +1,10 @@
 using System.Xml.Serialization;
 using Refresh.GameServer.Authentication;
+using Refresh.GameServer.Database;
 using Refresh.GameServer.Endpoints.ApiV3.DataTypes;
 using Refresh.GameServer.Types;
+using Refresh.GameServer.Types.Levels;
+using Refresh.GameServer.Types.Lists;
 using Refresh.GameServer.Types.UserData;
 
 namespace Refresh.GameServer.Endpoints.Game.DataTypes.Response;
@@ -39,12 +42,21 @@ public class GameUserResponse : IDataConvertableFrom<GameUserResponse, GameUser>
     [XmlElement("lbp2PurchasedSlots")] public int PurchasedSlotsLBP2 { get; set; }
     [XmlElement("lbp3PurchasedSlots")] public int PurchasedSlotsLBP3 { get; set; }
 
-    public static GameUserResponse? FromOldWithExtraData(GameUser? old, TokenGame gameVersion)
+    /// <summary>
+    /// The levels the user has favourited, only used by LBP PSP
+    /// </summary>
+    [XmlElement("favouriteSlots")] public SerializedMinimalFavouriteLevelList? FavouriteLevels { get; set; }
+    /// <summary>
+    /// The users the user has favourited, only used by LBP PSP
+    /// </summary>
+    [XmlElement("favouriteUsers")] public SerializedMinimalFavouriteUserList? FavouriteUsers { get; set; }
+
+    public static GameUserResponse? FromOldWithExtraData(GameUser? old, TokenGame gameVersion, GameDatabaseContext database)
     {
         if (old == null) return null;
 
         GameUserResponse response = FromOld(old)!;
-        response.FillInExtraData(old, gameVersion);
+        response.FillInExtraData(old, gameVersion, database);
 
         return response;
     }
@@ -85,10 +97,10 @@ public class GameUserResponse : IDataConvertableFrom<GameUserResponse, GameUser>
 
     public static IEnumerable<GameUserResponse> FromOldList(IEnumerable<GameUser> oldList) => oldList.Select(FromOld)!;
 
-    public static IEnumerable<GameUserResponse> FromOldListWithExtraData(IEnumerable<GameUser> oldList, TokenGame gameVersion) 
-        => oldList.Select(old => FromOldWithExtraData(old, gameVersion))!;
+    public static IEnumerable<GameUserResponse> FromOldListWithExtraData(IEnumerable<GameUser> oldList, TokenGame gameVersion, GameDatabaseContext database) 
+        => oldList.Select(old => FromOldWithExtraData(old, gameVersion, database))!;
 
-    private void FillInExtraData(GameUser old, TokenGame gameVersion)
+    private void FillInExtraData(GameUser old, TokenGame gameVersion, GameDatabaseContext database)
     {
         this.PlanetsHash = gameVersion switch
         {
@@ -141,10 +153,18 @@ private void FillInExtraData(GameUser old, TokenGame gameVersion)
                 throw new ArgumentOutOfRangeException(nameof(gameVersion), gameVersion, null);
         }
 
-        // Apply PSP-specific icon hashes
         if (gameVersion == TokenGame.LittleBigPlanetPSP)
         {
+            // Apply PSP-specific icon hashes
             this.IconHash = old.PspIconHash;
+
+            //Fill out PSP favourite users
+            List<GameUser> users = database.GetUsersFavouritedByUser(old, 20, 0).ToList();
+            this.FavouriteUsers = new SerializedMinimalFavouriteUserList(users.Select(SerializedUserHandle.FromUser).ToList(), users.Count);
+
+            //Fill out PSP favourite levels
+            List<GameMinimalLevelResponse> favouriteLevels = old.FavouriteLevelRelations.Select(f => GameMinimalLevelResponse.FromOld(f.Level)).ToList()!;
+            this.FavouriteLevels = new SerializedMinimalFavouriteLevelList(new SerializedMinimalLevelList(favouriteLevels, favouriteLevels.Count));
         }
     }
 }
diff --git a/Refresh.GameServer/Endpoints/Game/RelationEndpoints.cs b/Refresh.GameServer/Endpoints/Game/RelationEndpoints.cs
@@ -88,7 +88,7 @@ public Response UnfavouriteUser(RequestContext context, GameDatabaseContext data
         List<GameUser> users = database.GetUsersFavouritedByUser(user, count, skip)
             .ToList();
 
-        return new SerializedFavouriteUserList(GameUserResponse.FromOldListWithExtraData(users, token.TokenGame).ToList(), users.Count);
+        return new SerializedFavouriteUserList(GameUserResponse.FromOldListWithExtraData(users, token.TokenGame, database).ToList(), users.Count);
     }
 
     [GameEndpoint("lolcatftw/add/user/{id}", HttpMethods.Post)]

diff --git a/Refresh.GameServer/Endpoints/Game/UserEndpoints.cs b/Refresh.GameServer/Endpoints/Game/UserEndpoints.cs
@@ -19,7 +19,7 @@ public class UserEndpoints : EndpointGroup
     [GameEndpoint("user/{name}", HttpMethods.Get, ContentType.Xml)]
     [MinimumRole(GameUserRole.Restricted)]
     public GameUserResponse? GetUser(RequestContext context, GameDatabaseContext database, string name, Token token) 
-        => GameUserResponse.FromOldWithExtraData(database.GetUserByUsername(name), token.TokenGame);
+        => GameUserResponse.FromOldWithExtraData(database.GetUserByUsername(name), token.TokenGame, database);
 
     [GameEndpoint("users", HttpMethods.Get, ContentType.Xml)]
     [MinimumRole(GameUserRole.Restricted)]
@@ -35,7 +35,7 @@ public SerializedUserList GetMultipleUsers(RequestContext context, GameDatabaseC
             GameUser? user = database.GetUserByUsername(username);
             if (user == null) continue;
 
-            users.Add(GameUserResponse.FromOldWithExtraData(user, token.TokenGame)!);
+            users.Add(GameUserResponse.FromOldWithExtraData(user, token.TokenGame, database)!);
         }
 
         return new SerializedUserList
@@ -53,7 +53,7 @@ public SerializedUserList GetMultipleUsers(RequestContext context, GameDatabaseC
         List<GameUser>? friends = friendService.GetUsersFriends(user, database)?.ToList();
         if (friends == null) return null;
 
-        return new SerializedFriendsList(GameUserResponse.FromOldListWithExtraData(friends, token.TokenGame).ToList());
+        return new SerializedFriendsList(GameUserResponse.FromOldListWithExtraData(friends, token.TokenGame, database).ToList());
     }
 
     [GameEndpoint("updateUser", HttpMethods.Post, ContentType.Xml)]

diff --git a/Refresh.GameServer/Types/Lists/SerializedMinimalFavouriteUserList.cs b/Refresh.GameServer/Types/Lists/SerializedMinimalFavouriteUserList.cs
@@ -0,0 +1,26 @@
+using System.Xml.Serialization;
+using Refresh.GameServer.Endpoints.Game.DataTypes.Response;
+using Refresh.GameServer.Types.UserData;
+
+namespace Refresh.GameServer.Types.Lists;
+
+#nullable disable
+
+/// <summary>
+/// Minimal favourite user list, used only by LBP PSP, since it structures things a bit differently
+/// </summary>
+[XmlRoot("favouriteUsers")]
+[XmlType("favouriteUsers")]
+public class SerializedMinimalFavouriteUserList : SerializedList<SerializedUserHandle>
+{
+    public SerializedMinimalFavouriteUserList() {}
+
+    public SerializedMinimalFavouriteUserList(List<SerializedUserHandle> list, int count)
+    {
+        this.Total = count;
+        this.Items = list;
+    }
+
+    [XmlElement("user")]
+    public override List<SerializedUserHandle> Items { get; set; }
+}