cs-codex-dist-tests/Nethereum/NethereumInteraction.cs

160 lines
5.0 KiB
C#
Raw Normal View History

2023-04-25 09:31:15 +00:00
using Logging;
using Nethereum.ABI.FunctionEncoding.Attributes;
2023-04-18 11:22:41 +00:00
using Nethereum.Contracts;
2023-04-14 10:37:05 +00:00
using Nethereum.Hex.HexTypes;
2023-04-14 08:51:35 +00:00
using Nethereum.Web3;
using System.Numerics;
using Utils;
namespace NethereumWorkflow
{
2023-04-14 10:37:05 +00:00
public class NethereumInteraction
2023-04-14 08:51:35 +00:00
{
private readonly List<Task> openTasks = new List<Task>();
2023-04-25 09:31:15 +00:00
private readonly BaseLog log;
2023-04-14 08:51:35 +00:00
private readonly Web3 web3;
private readonly string rootAccount;
2023-04-25 09:31:15 +00:00
internal NethereumInteraction(BaseLog log, Web3 web3, string rootAccount)
2023-04-14 08:51:35 +00:00
{
2023-04-25 09:31:15 +00:00
this.log = log;
2023-04-14 08:51:35 +00:00
this.web3 = web3;
this.rootAccount = rootAccount;
}
2023-04-18 11:22:41 +00:00
public string GetTokenAddress(string marketplaceAddress)
{
2023-04-25 09:31:15 +00:00
log.Debug(marketplaceAddress);
2023-04-18 11:22:41 +00:00
var function = new GetTokenFunction();
var handler = web3.Eth.GetContractQueryHandler<GetTokenFunction>();
return Time.Wait(handler.QueryAsync<string>(marketplaceAddress, function));
}
2023-04-18 11:22:41 +00:00
public void TransferWeiTo(string account, decimal amount)
{
2023-04-25 09:31:15 +00:00
log.Debug($"{amount} --> {account}");
if (amount < 1 || string.IsNullOrEmpty(account)) throw new ArgumentException("Invalid arguments for AddToBalance");
var value = ToHexBig(amount);
var transactionId = Time.Wait(web3.Eth.TransactionManager.SendTransactionAsync(rootAccount, account, value));
openTasks.Add(web3.Eth.TransactionManager.TransactionReceiptService.PollForReceiptAsync(transactionId));
2023-04-18 11:22:41 +00:00
}
public void MintTestTokens(string account, decimal amount, string tokenAddress)
{
2023-04-25 09:31:15 +00:00
log.Debug($"({tokenAddress}) {amount} --> {account}");
if (amount < 1 || string.IsNullOrEmpty(account)) throw new ArgumentException("Invalid arguments for MintTestTokens");
2023-04-18 11:22:41 +00:00
var function = new MintTokensFunction
{
Holder = account,
Amount = ToBig(amount)
};
var handler = web3.Eth.GetContractTransactionHandler<MintTokensFunction>();
openTasks.Add(handler.SendRequestAndWaitForReceiptAsync(tokenAddress, function));
}
public decimal GetBalance(string tokenAddress, string account)
2023-04-14 08:51:35 +00:00
{
2023-04-25 09:31:15 +00:00
log.Debug($"({tokenAddress}) {account}");
var function = new GetTokenBalanceFunction
{
Owner = account
};
var handler = web3.Eth.GetContractQueryHandler<GetTokenBalanceFunction>();
2023-04-19 07:57:37 +00:00
return ToDecimal(Time.Wait(handler.QueryAsync<BigInteger>(tokenAddress, function)));
2023-04-14 08:51:35 +00:00
}
public void WaitForAllTransactions()
{
var tasks = openTasks.ToArray();
openTasks.Clear();
Task.WaitAll(tasks);
}
public void EnsureSynced(string marketplaceAddress, string marketplaceAbi)
{
WaitUntilSynced();
WaitForContract(marketplaceAddress, marketplaceAbi);
}
private void WaitUntilSynced()
{
2023-04-25 09:31:15 +00:00
log.Debug();
Time.WaitUntil(() =>
{
var sync = Time.Wait(web3.Eth.Syncing.SendRequestAsync());
var number = Time.Wait(web3.Eth.Blocks.GetBlockNumber.SendRequestAsync());
var numberOfBlocks = ToDecimal(number);
return !sync.IsSyncing && numberOfBlocks > 256;
}, TimeSpan.FromMinutes(1), TimeSpan.FromSeconds(3));
}
private void WaitForContract(string marketplaceAddress, string marketplaceAbi)
{
2023-04-25 09:31:15 +00:00
log.Debug();
Time.WaitUntil(() =>
{
try
{
var contract = web3.Eth.GetContract(marketplaceAbi, marketplaceAddress);
return contract != null;
}
catch
{
return false;
}
}, TimeSpan.FromMinutes(1), TimeSpan.FromSeconds(3));
}
2023-04-14 08:51:35 +00:00
private HexBigInteger ToHexBig(decimal amount)
{
2023-04-18 11:22:41 +00:00
var bigint = ToBig(amount);
2023-04-14 08:51:35 +00:00
var str = bigint.ToString("X");
return new HexBigInteger(str);
}
2023-04-14 10:37:05 +00:00
2023-04-18 11:22:41 +00:00
private BigInteger ToBig(decimal amount)
{
return new BigInteger(amount);
}
private decimal ToDecimal(HexBigInteger hexBigInteger)
{
return ToDecimal(hexBigInteger.Value);
}
private decimal ToDecimal(BigInteger bigInteger)
{
return (decimal)bigInteger;
}
2023-04-14 08:51:35 +00:00
}
2023-04-18 11:22:41 +00:00
[Function("token", "address")]
public class GetTokenFunction : FunctionMessage
{
}
[Function("mint")]
public class MintTokensFunction : FunctionMessage
{
[Parameter("address", "holder", 1)]
public string Holder { get; set; }
[Parameter("uint256", "amount", 2)]
public BigInteger Amount { get; set; }
}
[Function("balanceOf", "uint256")]
public class GetTokenBalanceFunction : FunctionMessage
{
[Parameter("address", "owner", 1)]
public string Owner { get; set; }
}
2023-04-14 08:51:35 +00:00
}