feat: Add action mutations for undelegation and reward claiming

planetarium · Dec 2, 2024 · f69e1b7 · f69e1b7
1 parent 1c006a6
commit f69e1b7
Showing 1 changed file with 75 additions and 4 deletions.
diff --git a/NineChronicles.Headless/GraphTypes/ActionMutation.cs b/NineChronicles.Headless/GraphTypes/ActionMutation.cs
@@ -601,12 +601,12 @@ public ActionMutation(NineChroniclesNodeService service)
             );
 
             Field<NonNullGraphType<TxIdType>>("stake",
-                description: "Claim reward for self delegation of validator.",
+                description: "Stake the NCG.",
                 arguments: new QueryArguments(
                     new QueryArgument<NonNullGraphType<IntGraphType>>
                     {
                         Name = "amount",
-                        Description = "Amount to stake."
+                        Description = "Total amount of NCG to stake."
                     }
                 ),
                 resolve: context =>
@@ -731,12 +731,12 @@ public ActionMutation(NineChroniclesNodeService service)
                 }
             );
             Field<NonNullGraphType<TxIdType>>("delegateValidator",
-                description: "Unjail validator",
+                description: "Delegate Validator",
                 arguments: new QueryArguments(
                     new QueryArgument<NonNullGraphType<BigIntGraphType>>
                     {
                         Name = "amount",
-                        Description = "Amount of guild gold to stake."
+                        Description = "Amount of guild gold to delegate."
                     }
                 ),
                 resolve: context =>
@@ -767,6 +767,77 @@ public ActionMutation(NineChroniclesNodeService service)
                     }
                 }
             );
+            Field<NonNullGraphType<TxIdType>>("undelegateValidator",
+                description: "Undelegate validator",
+                arguments: new QueryArguments(
+                    new QueryArgument<NonNullGraphType<BigIntGraphType>>
+                    {
+                        Name = "share",
+                        Description = "Share to undelegate."
+                    }
+                ),
+                resolve: context =>
+                {
+                    try
+                    {
+                        BlockChain? blockChain = service.BlockChain;
+                        if (blockChain is null)
+                        {
+                            throw new InvalidOperationException($"{nameof(blockChain)} is null.");
+                        }
+
+                        BigInteger share = context.GetArgument<BigInteger>("amount");
+                        var actions = new[] { new UndelegateValidator(share) };
+                        Transaction tx = blockChain.MakeTransaction(
+                            service.MinerPrivateKey,
+                            actions,
+                            Currencies.Mead * 1,
+                            1L);
+                        return tx.Id;
+                    }
+                    catch (Exception e)
+                    {
+                        var msg = $"Unexpected exception occurred during {typeof(ActionMutation)}: {e}";
+                        context.Errors.Add(new ExecutionError(msg, e));
+                        throw;
+                    }
+                }
+            );
+            Field<NonNullGraphType<TxIdType>>("claimStakeReward",
+                description: "Claim Stake Reward",
+                arguments: new QueryArguments(
+                    new QueryArgument<NonNullGraphType<AddressType>>
+                    {
+                        Name = "avatarAddress",
+                        Description = "Avatar address to claim stake reward."
+                    }
+                ),
+                resolve: context =>
+                {
+                    try
+                    {
+                        BlockChain? blockChain = service.BlockChain;
+                        if (blockChain is null)
+                        {
+                            throw new InvalidOperationException($"{nameof(blockChain)} is null.");
+                        }
+
+                        var actions = new[] { new ClaimStakeReward() };
+                        Transaction tx = blockChain.MakeTransaction(
+                            service.MinerPrivateKey,
+                            actions,
+                            Currencies.Mead * 1,
+                            1L);
+                        return tx.Id;
+                    }
+                    catch (Exception e)
+                    {
+                        var msg = $"Unexpected exception occurred during {typeof(ActionMutation)}: {e}";
+                        context.Errors.Add(new ExecutionError(msg, e));
+                        throw;
+                    }
+                }
+            );
         }
     }
 }