Move servers to LBPU.PL.Servers

ProjectLighthouse.Servers.GameServer/Controllers/FriendsController.cs

@@ -0,0 +1,95 @@
+#nullable enable
+using System.Xml.Serialization;
+using LBPUnion.ProjectLighthouse.Helpers;
+using LBPUnion.ProjectLighthouse.Serialization;
+using LBPUnion.ProjectLighthouse.Types;
+using LBPUnion.ProjectLighthouse.Types.Profiles;
+using Microsoft.AspNetCore.Mvc;
+using Microsoft.EntityFrameworkCore;
+
+namespace LBPUnion.ProjectLighthouse.Servers.GameServer.Controllers;
+
+[ApiController]
+[Route("LITTLEBIGPLANETPS3_XML/")]
+public class FriendsController : ControllerBase
+{
+    private readonly Database database;
+
+    public FriendsController(Database database)
+    {
+        this.database = database;
+    }
+
+    [HttpPost("npdata")]
+    public async Task<IActionResult> NPData()
+    {
+        User? user = await this.database.UserFromGameRequest(this.Request);
+        if (user == null) return this.StatusCode(403, "");
+
+        this.Request.Body.Position = 0;
+        string bodyString = await new StreamReader(this.Request.Body).ReadToEndAsync();
+
+        XmlSerializer serializer = new(typeof(NPData));
+        NPData? npData = (NPData?)serializer.Deserialize(new StringReader(bodyString));
+        if (npData == null) return this.BadRequest();
+
+        SanitizationHelper.SanitizeStringsInClass(npData);
+
+        List<User> friends = new();
+        foreach (string friendName in npData.Friends)
+        {
+            User? friend = await this.database.Users.FirstOrDefaultAsync(u => u.Username == friendName);
+            if (friend == null) continue;
+
+            friends.Add(friend);
+        }
+
+        List<int> blockedUsers = new();
+        foreach (string blockedUserName in npData.BlockedUsers)
+        {
+            User? blockedUser = await this.database.Users.FirstOrDefaultAsync(u => u.Username == blockedUserName);
+            if (blockedUser == null) continue;
+
+            blockedUsers.Add(blockedUser.UserId);
+        }
+
+        if (FriendHelper.FriendIdsByUserId.ContainsKey(user.UserId))
+        {
+            FriendHelper.FriendIdsByUserId.Remove(user.UserId);
+            FriendHelper.BlockedIdsByUserId.Remove(user.UserId);
+        }
+
+        FriendHelper.FriendIdsByUserId.Add(user.UserId, friends.Select(u => u.UserId).ToArray());
+        FriendHelper.BlockedIdsByUserId.Add(user.UserId, blockedUsers.ToArray());
+
+        string friendsSerialized = friends.Aggregate(string.Empty, (current, user1) => current + LbpSerializer.StringElement("npHandle", user1.Username));
+
+        return this.Ok(LbpSerializer.StringElement("npdata", LbpSerializer.StringElement("friends", friendsSerialized)));
+    }
+
+    [HttpGet("myFriends")]
+    public async Task<IActionResult> MyFriends()
+    {
+        (User, GameToken)? userAndToken = await this.database.UserAndGameTokenFromRequest(this.Request);
+
+        if (userAndToken == null) return this.StatusCode(403, "");
+
+        // ReSharper disable once PossibleInvalidOperationException
+        User user = userAndToken.Value.Item1;
+        GameToken gameToken = userAndToken.Value.Item2;
+
+        if (!FriendHelper.FriendIdsByUserId.TryGetValue(user.UserId, out int[]? friendIds) || friendIds == null)
+            return this.Ok(LbpSerializer.BlankElement("myFriends"));
+
+        string friends = "";
+        foreach (int friendId in friendIds)
+        {
+            User? friend = await this.database.Users.Include(u => u.Location).FirstOrDefaultAsync(u => u.UserId == friendId);
+            if (friend == null) continue;
+
+            friends += friend.Serialize(gameToken.GameVersion);
+        }
+
+        return this.Ok(LbpSerializer.StringElement("myFriends", friends));
+    }
+}