Merge pull request #32 from planetarium/implment-more-api

Implement `avatars/{address}/inventory` API

NineChroniclesUtilBackend/Controllers/AgentController.cs

@@ -22,40 +22,39 @@ public class AccountController : ControllerBase
 {
     [HttpGet("{agentAddress}/avatars")]
     public async Task<AvatarsResponse> GetAvatars(string agentAddress, IStateService stateService)
+    { 
+        var avatarStates = await GetAvatarStates(new Address(agentAddress), stateService);
+        return new AvatarsResponse(avatarStates
+            .Select(a => new Avatar(a.address.ToString(), a.name, a.level))
+            .ToList());
+    }
+
+    private static async Task<List<AvatarState>> GetAvatarStates(Address agentAddress, IStateService stateService)
     {
-        async Task<List<AvatarState>> GetAvatarsState(Address agentAddress)
+        var rawState = await stateService.GetState(agentAddress, Addresses.Agent) ??
+                       await stateService.GetState(agentAddress);
+        var agentState = rawState switch
         {
-            var rawState = await stateService.GetState(agentAddress, Addresses.Agent) ??
-                           await stateService.GetState(agentAddress);
-            var agentState = rawState switch
-            {
-                Dictionary agentStateDictionary => new AgentState(agentStateDictionary),
-                List agentStateList => new AgentState(agentStateList),
-                _ => throw new ArgumentException(nameof(agentAddress)),
-            };
-
-            List<AvatarState> avatars = new List<AvatarState>();
+            Dictionary agentStateDictionary => new AgentState(agentStateDictionary),
+            List agentStateList => new AgentState(agentStateList),
+            _ => throw new ArgumentException(nameof(agentAddress)),
+        };
 
-            foreach(var avatarAddressKey in agentState.avatarAddresses.Keys)
+        var avatars = new List<AvatarState>();
+        foreach(var avatarAddress in agentState.avatarAddresses.Values)
+        {
+            var rawAvatarState =
+                await stateService.GetState(avatarAddress, Addresses.Avatar) ??
+                await stateService.GetState(avatarAddress);
+            var avatarState = rawAvatarState switch
             {
-                var avatarAddress = agentState.avatarAddresses[avatarAddressKey];
-                var rawAvatarState =
-                    await stateService.GetState(avatarAddress, Addresses.Avatar) ??
-                    await stateService.GetState(avatarAddress);
-
-                var avatarState = rawAvatarState switch
-                {
-                    Dictionary avatarStateDictionary => new AvatarState(avatarStateDictionary),
-                    List avatarStateList => new AvatarState(avatarStateList),
-                    _ => throw new ArgumentException(nameof(avatarAddress))
-                };
-                avatars.Add(avatarState);
-            }
-
-            return avatars;
+                Dictionary avatarStateDictionary => new AvatarState(avatarStateDictionary),
+                List avatarStateList => new AvatarState(avatarStateList),
+                _ => throw new ArgumentException(nameof(avatarAddress))
+            };
+            avatars.Add(avatarState);
         }
-        var avatars = await GetAvatarsState(new Address(agentAddress));
 
-        return new AvatarsResponse(avatars.Select(a => new Avatar(a.address.ToString(), a.name, a.level)).ToList());
+        return avatars;
     }
 }

NineChroniclesUtilBackend/Controllers/AvatarController.cs

@@ -0,0 +1,14 @@
+using Microsoft.AspNetCore.Mvc;
+using NineChroniclesUtilBackend.Models.Avatar;
+using NineChroniclesUtilBackend.Repositories;
+
+namespace NineChroniclesUtilBackend.Controllers;
+
+[ApiController]
+[Route("avatars")]
+public class AvatarController(AvatarRepository avatarRepository) : ControllerBase
+{
+    [HttpGet("{avatarAddress}/inventory")]
+    public Inventory GetInventory(string avatarAddress) =>
+        avatarRepository.GetInventory(avatarAddress);
+}

NineChroniclesUtilBackend/Models/Arena/ArenaRanking.cs

@@ -1,5 +1,4 @@
 using System.Text.Json.Serialization;
-using NineChroniclesUtilBackend.Models.Agent;
 
 namespace NineChroniclesUtilBackend.Models.Arena;
 
@@ -28,5 +27,5 @@ int Score
     public int Score { get; set; } = Score;
 
     [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
-    public Avatar? Avatar { get; set; }
+    public Agent.Avatar? Avatar { get; set; }
 }

NineChroniclesUtilBackend/Models/Avatar/Inventory.cs

@@ -0,0 +1,11 @@
+using MongoDB.Bson;
+using NineChroniclesUtilBackend.Models.Items;
+
+namespace NineChroniclesUtilBackend.Models.Avatar;
+
+public class Inventory(BsonValue inventory)
+{
+    public List<Equipment> Equipments { get; set; } = inventory["Equipments"].AsBsonArray
+        .Select(e => new Equipment(e.AsBsonDocument))
+        .ToList();
+}

NineChroniclesUtilBackend/Models/Items/Equipment.cs

@@ -0,0 +1,8 @@
+using MongoDB.Bson;
+
+namespace NineChroniclesUtilBackend.Models.Items;
+
+public class Equipment(BsonValue equipment) : NonFungibleItem(equipment)
+{
+    public int Level { get; set; } = equipment["level"].AsInt32;
+}

NineChroniclesUtilBackend/Models/Items/Item.cs

@@ -0,0 +1,9 @@
+using MongoDB.Bson;
+using Nekoyume.Model.Item;
+
+namespace NineChroniclesUtilBackend.Models.Items;
+
+public class Item(BsonValue item)
+{
+    public ItemSubType ItemSubType { get; set; } = (ItemSubType)item["ItemSubType"].AsInt32;
+}

NineChroniclesUtilBackend/Models/Items/NonFungibleItem.cs

@@ -0,0 +1,11 @@
+using MongoDB.Bson;
+
+namespace NineChroniclesUtilBackend.Models.Items;
+
+public class NonFungibleItem(BsonValue nonFungibleItem) : Item(nonFungibleItem)
+{
+    public Guid NonFungibleId { get; set; } =
+        Guid.TryParse(nonFungibleItem["NonFungibleId"].AsString, out var nonFungibleId)
+            ? nonFungibleId
+            : Guid.Empty;
+}

NineChroniclesUtilBackend/Program.cs

@@ -25,6 +25,7 @@
 builder.Services.AddSingleton<IStateService, HeadlessStateService>();
 builder.Services.AddSingleton<MongoDBCollectionService>();
 builder.Services.AddSingleton<ArenaRankingRepository>();
+builder.Services.AddSingleton<AvatarRepository>();
 builder.Services.AddControllers();
 builder.Services.AddHeadlessGQLClient()
     .ConfigureHttpClient((provider, client) =>

NineChroniclesUtilBackend/Repositories/AvatarRepository.cs

@@ -0,0 +1,20 @@
+using MongoDB.Bson;
+using MongoDB.Driver;
+using NineChroniclesUtilBackend.Models.Avatar;
+using NineChroniclesUtilBackend.Services;
+
+namespace NineChroniclesUtilBackend.Repositories;
+
+public class AvatarRepository(MongoDBCollectionService mongoDBCollectionService)
+{
+    private readonly IMongoCollection<BsonDocument> _avatarsCollection =
+        mongoDBCollectionService.GetCollection<BsonDocument>("avatars");
+
+    public Inventory GetInventory(string avatarAddress)
+    {
+        var filter = Builders<BsonDocument>.Filter.Eq(f => f["Avatar"]["address"], avatarAddress);
+        var projection = Builders<BsonDocument>.Projection.Include(f => f["Avatar"]["inventory"]["Equipments"]);
+        var document = _avatarsCollection.Find(filter).Project(projection).FirstOrDefault();
+        return new Inventory(document["Avatar"]["inventory"]);
+    }
+}

README.md

@@ -6,5 +6,6 @@ A backend service that provides 9c-related utility APIs.
 ```
 dotnet tool restore
 dotnet graphql generate NineChroniclesUtilBackend
+dotnet graphql generate NineChroniclesUtilBackend.Store
 dotnet build
-```
+```