Merge pull request #4 from bmresearch/add-request-airdrop-non-mainnet

hoakbuilds · web-flow · commit ba4ec75fa62a · 2022-05-22T19:25:26.000+01:00
feat: ability to request airdrop when non-mainnet
diff --git a/Anvil/Models/ApplicationState.cs b/Anvil/Models/ApplicationState.cs
@@ -28,13 +28,30 @@ public string RpcUrl
             set => this.RaiseAndSetIfChanged(ref _rpcUrl, value);
         }
 
+
         [DataMember]
         private Cluster _cluster = Cluster.MainNet;
+        
+        /// <summary>
+        /// A setting used in order to choose which cluster to use as an RPC Client Provider.
+        /// </summary>
         public Cluster Cluster
         {
             get => _cluster;
             set => this.RaiseAndSetIfChanged(ref _cluster, value);
         }
+        
+        [DataMember]
+        private Cluster _network = Cluster.MainNet;
+
+        /// <summary>
+        /// A flag used in order to tag the current network according to the received genesis hash.
+        /// </summary>
+        public Cluster Network
+        {
+            get => _network;
+            set => this.RaiseAndSetIfChanged(ref _network, value);
+        }
 
         [DataMember]
         private string _storePath = AppContext.BaseDirectory;
diff --git a/Anvil/ViewModels/Common/TransactionSubmissionViewModel.cs b/Anvil/ViewModels/Common/TransactionSubmissionViewModel.cs
@@ -44,9 +44,11 @@ public async void CopyTransactionHashToClipboard()
         /// Initialize the transaction submission.
         /// </summary>
         /// <param name="rpcClientProvider">The rpc client provider.</param>
-        public TransactionSubmissionViewModel(IRpcClientProvider rpcClientProvider)
+        /// <param name="transactionHash">The transaction hash.</param>
+        public TransactionSubmissionViewModel(IRpcClientProvider rpcClientProvider, string transactionHash = null)
         {
             _rpcClientProvider = rpcClientProvider;
+            TransactionHash = transactionHash;
         }
 
         /// <summary>
@@ -56,6 +58,8 @@ public TransactionSubmissionViewModel(IRpcClientProvider rpcClientProvider)
         /// <returns>A task that performs the action and may return a boolean which represents whether it was a successful submission or not.</returns>
         public async Task<bool> SubmitTransaction(byte[] transaction)
         {
+            if (TransactionHash != null) return false;
+
             Progress = "Submitting transaction..";
             var txSig = await _rpcClientProvider.Client.SendTransactionAsync(transaction,
                 commitment: Solnet.Rpc.Types.Commitment.Confirmed);
diff --git a/Anvil/ViewModels/MainWindowViewModel.cs b/Anvil/ViewModels/MainWindowViewModel.cs
@@ -320,7 +320,7 @@ private void _keyStoreService_OnStartupStateChanged(object sender, Services.Even
             if (!_keyStoreService.IsProcessing && KeyStoreServiceState == KeyStoreServiceState.Done)
             {
                 _walletViewModel ??= new WalletViewModel(_appLifetime, _internetService, _walletService,
-                    _rpcProvider, _keyStoreService, _addressBookService);
+                    _rpcProvider, _keyStoreService, _addressBookService, _appState);
                 CurrentView = _walletViewModel;
                 WalletUnlocked = true;
                 RestoreFromWalletSnapshot();
@@ -358,7 +358,7 @@ private async void OnWalletImport(WalletImport walletImport)
                 // private key import
                 await _keyStoreService.InitializeWalletWithPrivateKey(walletImport.PrivateKeyFilePath, walletImport.Alias, walletImport.Password);
 
-                _walletViewModel ??= new WalletViewModel(_appLifetime, _internetService, _walletService, _rpcProvider, _keyStoreService, _addressBookService);
+                _walletViewModel ??= new WalletViewModel(_appLifetime, _internetService, _walletService, _rpcProvider, _keyStoreService, _addressBookService, _appState);
                 CurrentView = _walletViewModel;
                 WalletUnlocked = true;
             }
@@ -367,7 +367,7 @@ private async void OnWalletImport(WalletImport walletImport)
                 // mnemonic import
                 await _keyStoreService.InitializeWallet(walletImport.Mnemonic, walletImport.Password);
 
-                _walletViewModel ??= new WalletViewModel(_appLifetime, _internetService, _walletService, _rpcProvider, _keyStoreService, _addressBookService);
+                _walletViewModel ??= new WalletViewModel(_appLifetime, _internetService, _walletService, _rpcProvider, _keyStoreService, _addressBookService, _appState);
                 CurrentView = _walletViewModel;
                 WalletUnlocked = true;
             }
@@ -393,7 +393,7 @@ public void ChangeView(string view)
             switch (view)
             {
                 case "Wallet":
-                    _walletViewModel ??= new WalletViewModel(_appLifetime, _internetService, _walletService, _rpcProvider, _keyStoreService, _addressBookService);
+                    _walletViewModel ??= new WalletViewModel(_appLifetime, _internetService, _walletService, _rpcProvider, _keyStoreService, _addressBookService, _appState);
                     CurrentView = _walletViewModel;
                     break;
                 case "WatchOnly":
diff --git a/Anvil/ViewModels/SettingsViewModel.cs b/Anvil/ViewModels/SettingsViewModel.cs
@@ -75,8 +75,29 @@ public SettingsViewModel(ApplicationState appState, IRpcClientProvider rpcClient
             GetVersionInfo();
         }
 
+        private async void GetNetworkInfo()
+        {
+            var g = await _rpcClientProvider.Client.GetGenesisHashAsync();
+
+            var cluster = GetClusterForBlockhash(g.Result);
+
+            _appState.Network = cluster;
+        }
+
+        private Cluster GetClusterForBlockhash(string blockhash)
+        {
+            return blockhash switch
+            {
+                "5eykt4UsFv8P8NJdTREpY1vzqKqZKvdpKuc147dw2N9d" => Cluster.MainNet,
+                "4uhcVJyU9pJkvQyS88uRDiswHXSCkY3zQawwpjk2NsNY" => Cluster.TestNet,
+                "EtWTRABZaYq6iMfeYKouRu166VU2xqa1wcaWoxPkrZBG" => Cluster.DevNet,
+                _ => Cluster.DevNet,
+            };
+        }
+
         private async void GetVersionInfo()
         {
+
             var v = await _rpcClientProvider.Client.GetVersionAsync();
 
             if (v.WasRequestSuccessfullyHandled)
@@ -136,6 +157,7 @@ public void ApplyChanges()
                 default:
                     break;
             }
+            GetNetworkInfo();
             GetVersionInfo();
         }
 
diff --git a/Anvil/ViewModels/Wallet/WalletViewModel.cs b/Anvil/ViewModels/Wallet/WalletViewModel.cs
@@ -1,4 +1,5 @@
 ﻿using Anvil.Core.ViewModels;
+using Anvil.Models;
 using Anvil.Services;
 using Anvil.Services.Network;
 using Anvil.Services.Wallets;
@@ -55,10 +56,13 @@ public class WalletViewModel : ViewModelBase
 
         #endregion
 
+        private ApplicationState _appState;
+
         public WalletViewModel(IClassicDesktopStyleApplicationLifetime appLifetime,
             InternetConnectionService internetConnectionService,
             IWalletService walletService, IRpcClientProvider rpcClientProvider,
-            KeyStoreService keyStoreService, AddressBookService addressBookService)
+            KeyStoreService keyStoreService, AddressBookService addressBookService,
+            ApplicationState applicationState)
         {
             _appLifetime = appLifetime;
             _internetConnectionService = internetConnectionService;
@@ -70,9 +74,13 @@ public WalletViewModel(IClassicDesktopStyleApplicationLifetime appLifetime,
             _walletService.OnWalletServiceStateChanged += OnWalletServiceStateChanged;
             _keyStoreService = keyStoreService;
             _addressBookService = addressBookService;
+            _appState = applicationState;
+
+            _appState.PropertyChanged += ApplicationStateChanged;
 
             DerivationWalletsCollection = new();
             PrivateKeyWalletsCollection = new();
+            CanRequestAirdrop = _appState.Network != Cluster.MainNet;
             NoConnection = !_internetConnectionService.IsConnected;
             TransactionSubmission = TransactionSubmissionViewModel.NoShow();
             TransactionSubmission.WhenAnyValue(x => x.SubmittingTransaction)
@@ -84,11 +92,36 @@ public WalletViewModel(IClassicDesktopStyleApplicationLifetime appLifetime,
             RestoreFromWalletSnapshot(false);
         }
 
+        private void ApplicationStateChanged(object sender, System.ComponentModel.PropertyChangedEventArgs e)
+        {
+            if (e.PropertyName == "Network")
+            {
+                CanRequestAirdrop = _appState.Network != Cluster.MainNet;
+            }
+        }
+
         private void OnNetworkConnectionChanged(object sender, Services.Network.Events.NetworkConnectionChangedEventArgs e)
         {
             NoConnection = !e.Connected;
         }
 
+        public async void RequestAirdrop()
+        {
+            var req = await _rpcClient.RequestAirdropAsync(_currentWallet.Address, SolHelper.ConvertToLamports(1m));
+            if (req.WasSuccessful)
+            {
+                if(req.Result != null)
+                {
+                    TransactionSubmission = new TransactionSubmissionViewModel(_rpcClientProvider, req.Result);
+                    TransactionSubmission.Progress = "Successfully requested airdrop.";
+                    await TransactionSubmission.PollConfirmation();
+                    await GetAccountBalance();
+                    await Task.Delay(15000);
+                    TransactionSubmission = TransactionSubmissionViewModel.NoShow();
+                }
+            }
+        }
+
         private async Task<SendSolanaDialogViewModel> GetDestinationAddressAndAmount()
         {
             var vm = new SendSolanaDialogViewModel()
@@ -330,7 +363,7 @@ private Tuple<bool, string> ValidateMnemonicDelegate(string text)
 
             return new Tuple<bool, string>(false, "Mnemonic is invalid.");
         }
-        
+
         /// <summary>
         /// Private key validation for the dialog.
         /// </summary>
@@ -784,6 +817,13 @@ private void UpdateTokenCollection()
             }
         }
 
+        private bool _canRequestAirdrop;
+        public bool CanRequestAirdrop
+        {
+            get => _canRequestAirdrop;
+            set => this.RaiseAndSetIfChanged(ref _canRequestAirdrop, value);
+        }
+
         private bool _canSubmitTransaction;
         public bool CanSubmitTransaction
         {
diff --git a/Anvil/Views/Wallet/WalletView.axaml b/Anvil/Views/Wallet/WalletView.axaml
@@ -304,6 +304,18 @@
 								<avalonia:MaterialIcon Kind="Send" VerticalAlignment="Center"/>
 							</StackPanel>
 						</Button>
+						<Button
+							Command="{Binding RequestAirdrop}"
+							VerticalAlignment="Center"
+							HorizontalAlignment="Center"
+							IsVisible="{Binding CanRequestAirdrop}"
+							Margin="5"
+							ToolTip.Tip="Request Airdrop">
+							<StackPanel
+								Orientation="Horizontal">
+								<avalonia:MaterialIcon Kind="ArrowDownBold" VerticalAlignment="Center"/>
+							</StackPanel>
+						</Button>
 					</StackPanel>
 					<common:TransactionSubmissionView
 						DockPanel.Dock="Top"
