introduce several sub-fields in "avatar" field

planetarium · May 30, 2024 · e400fe7 · e400fe7
1 parent 857fafe
commit e400fe7
Showing 1 changed file with 102 additions and 16 deletions.
diff --git a/Mimir/GraphQL/Types/AvatarType.cs b/Mimir/GraphQL/Types/AvatarType.cs
@@ -1,6 +1,9 @@
+using HotChocolate.Resolvers;
 using Lib9c.GraphQL.Enums;
+using Lib9c.GraphQL.Types;
 using Libplanet.Crypto;
 using Mimir.GraphQL.Objects;
+using Mimir.Models.Agent;
 using Mimir.Repositories;
 
 namespace Mimir.GraphQL.Types;
@@ -10,40 +13,123 @@ public class AvatarType : ObjectType<AvatarObject>
     protected override void Configure(IObjectTypeDescriptor<AvatarObject> descriptor)
     {
         descriptor
-            .Field("inventory")
-            .Type<InventoryType>()
+            .Field("agentAddress")
+            .Type<AddressType>()
             .Resolve(context =>
             {
-                var avatarRepo = context.Services.GetService<AvatarRepository>();
-                if (avatarRepo is null)
+                var avatar = GetAvatar(context);
+                if (avatar is null)
                 {
                     return null;
                 }
 
-                var planetName = context.ScopedContextData.TryGetValue("planetName", out var pn)
-                    ? (PlanetName?)pn
-                    : null;
-                if (planetName is null)
+                context.ScopedContextData = context.ScopedContextData.Add("avatar", avatar);
+                return new Address(avatar.AgentAddress);
+            });
+        descriptor
+            .Field("name")
+            .Type<StringType>()
+            .Resolve(context =>
+            {
+                var avatar = GetAvatar(context);
+                if (avatar is null)
                 {
                     return null;
                 }
-
-                var avatarAddress = context.ScopedContextData.TryGetValue("avatarAddress", out var aa)
-                    ? (Address?)aa
-                    : null;
-                if (avatarAddress is null)
+
+                context.ScopedContextData = context.ScopedContextData.Add("avatar", avatar);
+                return avatar.AvatarName;
+            });
+        descriptor
+            .Field("level")
+            .Type<IntType>()
+            .Resolve(context =>
+            {
+                var avatar = GetAvatar(context);
+                if (avatar is null)
+                {
+                    return null;
+                }
+
+                context.ScopedContextData = context.ScopedContextData.Add("avatar", avatar);
+                return avatar.Level;
+            });
+        descriptor
+            .Field("inventory")
+            .Type<InventoryType>()
+            .Resolve(context =>
+            {
+                var tuple = GetSource(context);
+                if (tuple is null)
                 {
                     return null;
                 }
-
-                var inventory = avatarRepo.GetInventory(planetName.Value.ToString(), avatarAddress.Value);
+
+                var (avatarRepo, planetName, avatarAddress) = tuple.Value;
+                var inventory = avatarRepo.GetInventory(planetName.ToString(), avatarAddress);
                 if (inventory is null)
                 {
                     return null;
                 }
-                
+
                 context.ScopedContextData = context.ScopedContextData.Add("inventory", inventory);
                 return new InventoryObject();
             });
+        descriptor
+            .Field("actionPoint")
+            .Type<IntType>()
+            .Resolve(context =>
+            {
+                var avatar = GetAvatar(context);
+                if (avatar is null)
+                {
+                    return null;
+                }
+
+                context.ScopedContextData = context.ScopedContextData.Add("avatar", avatar);
+                return avatar.ActionPoint;
+            });
+    }
+
+    private static (
+        AvatarRepository avatarRepo,
+        PlanetName planetName,
+        Address avatarAddress)? GetSource(IResolverContext context)
+    {
+        var avatarRepo = context.Services.GetService<AvatarRepository>();
+        if (avatarRepo is null)
+        {
+            return null;
+        }
+
+        var planetName = context.ScopedContextData.TryGetValue("planetName", out var pn)
+            ? (PlanetName?)pn
+            : null;
+        if (planetName is null)
+        {
+            return null;
+        }
+
+        var avatarAddress = context.ScopedContextData.TryGetValue("avatarAddress", out var aa)
+            ? (Address?)aa
+            : null;
+        if (avatarAddress is null)
+        {
+            return null;
+        }
+
+        return (avatarRepo, planetName.Value, avatarAddress.Value);
+    }
+
+    public Avatar? GetAvatar(IResolverContext context)
+    {
+        var tuple = GetSource(context);
+        if (tuple is null)
+        {
+            return null;
+        }
+
+        var (avatarRepo, planetName, avatarAddress) = tuple.Value;
+        return avatarRepo.GetAvatar(planetName.ToString(), avatarAddress);
     }
 }