Add userState query

not-blond-beard · Feb 12, 2024 · dc7c264 · dc7c264
1 parent ac2f0ba
commit dc7c264
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/backend/app/Savor22b/GraphTypes/Query/Query.cs b/backend/app/Savor22b/GraphTypes/Query/Query.cs
@@ -27,6 +27,27 @@ public Query(BlockChain blockChain, Swarm? swarm = null)
         _blockChain = blockChain;
         _swarm = swarm;
 
+        Field<UserStateType>(
+            "userState",
+            description: "The specified address's user state",
+            arguments: new QueryArguments(
+                new QueryArgument<NonNullGraphType<StringGraphType>>
+                {
+                    Name = "address",
+                    Description = "The account holder's 40-hex address",
+                }
+            ),
+            resolve: context =>
+            {
+                var accountAddress = new Address(context.GetArgument<string>("address"));
+                var rootStateEncoded = blockChain.GetState(accountAddress);
+                RootState rootState = rootStateEncoded is Bencodex.Types.Dictionary bdict
+                    ? new RootState(bdict)
+                    : new RootState();
+                return rootState;
+            }
+        );
+
         Field<StringGraphType>(
             "asset",
             description: "The specified address's balance in MNT.",