feat: Add state query for guild

planetarium · Dec 17, 2024 · 856a271 · 856a271
1 parent 6270340
commit 856a271
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 3 deletions.
diff --git a/NineChronicles.Headless/GraphTypes/GuildType.cs b/NineChronicles.Headless/GraphTypes/GuildType.cs
@@ -0,0 +1,38 @@
+using GraphQL.Types;
+using Libplanet.Crypto;
+using Libplanet.Explorer.GraphTypes;
+using Nekoyume.Model.Guild;
+
+namespace NineChronicles.Headless.GraphTypes;
+
+public class GuildType : ObjectGraphType<GuildType>
+{
+    public Address Address { get; set; }
+
+    public Address ValidatorAddress { get; set; }
+
+    public Address GuildMasterAddress { get; set; }
+
+    public GuildType()
+    {
+        Field<NonNullGraphType<AddressType>>(
+            nameof(Address),
+            description: "Address of the guild",
+            resolve: context => context.Source.Address);
+        Field<NonNullGraphType<AddressType>>(
+            nameof(ValidatorAddress),
+            description: "Validator address of the guild",
+            resolve: context => context.Source.ValidatorAddress);
+        Field<NonNullGraphType<AddressType>>(
+            nameof(GuildMasterAddress),
+            description: "Guild master address of the guild",
+            resolve: context => context.Source.GuildMasterAddress);
+    }
+
+    public static GuildType FromDelegatee(Guild guild) => new GuildType
+    {
+        Address = guild.Address,
+        ValidatorAddress = guild.ValidatorAddress,
+        GuildMasterAddress = guild.GuildMasterAddress,
+    };
+}
diff --git a/NineChronicles.Headless/GraphTypes/StateQuery.cs b/NineChronicles.Headless/GraphTypes/StateQuery.cs
@@ -721,7 +721,7 @@ public StateQuery()
                 }
             );
 
-            Field<AddressType>(
+            Field<GuildType>(
                 name: "guild",
                 description: "State for guild.",
                 arguments: new QueryArguments(
@@ -740,9 +740,13 @@ public StateQuery()
                     }
 
                     var repository = new GuildRepository(new World(context.Source.WorldState), new HallowActionContext { });
-                    var joinedGuild = (Address?)repository.GetJoinedGuild(agentAddress);
+                    if (repository.GetJoinedGuild(agentAddress) is { } guildAddress)
+                    {
+                        var guild = repository.GetGuild(guildAddress);
+                        return GuildType.FromDelegatee(guild);
+                    }
 
-                    return joinedGuild;
+                    return null;
                 }
             );