Fill out add deploy profile.

allisterb · allisterb · commit 3d9cb6d33e41 · 2025-04-29T20:22:20.000-04:00
diff --git a/tests/Stratis.DevEx.Wpf.Test/BlockchainExplorer/BlockchainExplorerToolWindowControl.xaml.cs b/tests/Stratis.DevEx.Wpf.Test/BlockchainExplorer/BlockchainExplorerToolWindowControl.xaml.cs
@@ -1,4 +1,6 @@
 ﻿using System;
+using System.IO;
+using System.Security.Cryptography;
 using System.Linq;
 using System.Numerics;
 using System.Threading.Tasks;
@@ -19,6 +21,8 @@
 using static Stratis.DevEx.Result;
 using System.IO;
 using Stratis.DevEx;
+using System.Diagnostics.Eventing.Reader;
+using System.Text;
 
 namespace Stratis.VS.StratisEVM.UI
 {
@@ -419,39 +423,7 @@ private void DeleteNetworkCmd_CanExecute(object sender, CanExecuteRoutedEventArg
                 e.CanExecute = true;
             }
         }
-        #endregion
-
-        #region Methods
-        private BlockchainInfo GetSelectedItem(object sender)
-        {
-            var window = (BlockchainExplorerToolWindowControl)sender;
-            var tree = window.BlockchainExplorerTree;
-            return tree.SelectedItem;
-        }
-        private void ShowValidationErrors(Wpc.TextBlock textBlock, string message)
-        {
-            textBlock.Visibility = Visibility.Visible;
-            textBlock.Text = message;
-        }
 
-        private void ShowProgressRing(ProgressRing progressRing)
-        {
-            progressRing.IsEnabled = true;
-            progressRing.Visibility = Visibility.Visible;
-        }
-
-        private void HideProgressRing(ProgressRing progressRing)
-        {
-            progressRing.IsEnabled = false;
-            progressRing.Visibility = Visibility.Hidden;
-        }
-
-        #endregion
-
-        #region Fields
-        internal BlockchainExplorerToolWindow window;
-
-        #endregion
 
         private async void EditAccountCmd_Executed(object sender, ExecutedRoutedEventArgs e)
         {
@@ -472,7 +444,7 @@ private async void EditAccountCmd_Executed(object sender, ExecutedRoutedEventArg
                 var acctpubkey = (Wpc.TextBlock)((StackPanel)sp.Children[0]).Children[1];
                 acctpubkey.Text = item.Name;
                 var acctlabel = (Wpc.TextBox)((StackPanel)sp.Children[0]).Children[3];
-                acctlabel.Text = (string)item.Data["Label"]; 
+                acctlabel.Text = (string)item.Data["Label"];
                 var validForClose = false;
                 dw.ButtonClicked += (cd, args) =>
                 {
@@ -485,7 +457,7 @@ private async void EditAccountCmd_Executed(object sender, ExecutedRoutedEventArg
                 var r = await dw.ShowAsync();
                 if (r != ContentDialogResult.Primary)
                 {
-                    acctlabel.Text = (string) item.Data["Label"];
+                    acctlabel.Text = (string)item.Data["Label"];
                     return;
                 }
                 item.Data["Label"] = acctlabel.Text;
@@ -514,35 +486,155 @@ private async void EditAccountCmd_Executed(object sender, ExecutedRoutedEventArg
 
         private async void NewDeployProfileCmd_Executed(object sender, ExecutedRoutedEventArgs e)
         {
-            var window = (BlockchainExplorerToolWindowControl)sender;
-            var tree = window.BlockchainExplorerTree;
-            var item = GetSelectedItem(sender);
-            var dw = new BlockchainExplorerDialog(RootContentDialog)
+            int roundUp(int numToRound, int multiple)
             {
-                Title = "Add Deploy Profile",
-                PrimaryButtonIcon = new SymbolIcon(SymbolRegular.Save20),
-                Content = (StackPanel)TryFindResource("AddDeployProfileDialog"),
-                PrimaryButtonText = "Save",
-                CloseButtonText = "Cancel",
-            };
-            var sp = (StackPanel)dw.Content;
-            var name = (Wpc.TextBox)((StackPanel)sp.Children[0]).Children[1];
-            var endpoint = (ComboBox)((StackPanel)sp.Children[1]).Children[1];
-            var accounts = (ComboBox)((StackPanel)sp.Children[2]).Children[1];
-            endpoint.ItemsSource = item.GetNetworkEndPoints();
-            endpoint.SelectedIndex = 0; 
-            accounts.ItemsSource = item.GetNetworkAccounts();
-            var validForClose = false;
-
-            dw.ButtonClicked += (cd, args) =>
+                if (multiple == 0)
+                    return numToRound;
+
+                int remainder = numToRound % multiple;
+                if (remainder == 0)
+                    return numToRound;
+
+                return numToRound + multiple - remainder;
+            }
+
+            try
             {
-                validForClose = true;
-            };
-            dw.Closing += (d, args) =>
+                var window = (BlockchainExplorerToolWindowControl)sender;
+                var tree = window.BlockchainExplorerTree;
+                var item = GetSelectedItem(sender);
+                var dw = new BlockchainExplorerDialog(RootContentDialog)
+                {
+                    Title = "Add Deploy Profile",
+                    PrimaryButtonIcon = new SymbolIcon(SymbolRegular.Save20),
+                    Content = (StackPanel)TryFindResource("AddDeployProfileDialog"),
+                    PrimaryButtonText = "Save",
+                    CloseButtonText = "Cancel",
+                };
+                var sp = (StackPanel)dw.Content;
+                var name = (Wpc.TextBox)((StackPanel)sp.Children[0]).Children[1];
+                var endpoint = (ComboBox)((StackPanel)sp.Children[1]).Children[1];
+                var accounts = (ComboBox)((StackPanel)sp.Children[2]).Children[1];
+                var pkey = (Wpc.TextBox)((StackPanel)sp.Children[3]).Children[1];
+                var errors = (Wpc.TextBlock)((StackPanel)sp.Children[4]).Children[0];
+                endpoint.ItemsSource = item.GetNetworkEndPoints();
+                endpoint.SelectedIndex = 0;
+                accounts.ItemsSource = item.GetNetworkAccounts();
+                var validForClose = false;
+
+                dw.ButtonClicked += (cd, args) =>
+                {
+                    validForClose = false;
+                    errors.Visibility = Visibility.Hidden;
+
+                    if (args.Button == ContentDialogButton.Primary)
+                    {
+                        if (!string.IsNullOrEmpty(name.Text) && accounts.SelectedValue != null && endpoint.SelectedValue != null)
+                        {
+                            var dp = item.GetNetworkDeployProfiles();
+                            if (dp.Contains(name.Text))
+                            {
+                                ShowValidationErrors(errors, "The " + name.Text + " deploy profile already exists.");
+                            }
+                            else
+                            {
+                                validForClose = true;
+                            }
+                        }
+                        else
+                        {
+                            ShowValidationErrors(errors, "Enter a deploy profile name and select a valid endpoint and account");
+                        }
+                    }
+                    else
+                    {
+                        validForClose = true;
+                    }
+                };
+
+                dw.Closing += (d, args) =>
+                {
+                    args.Cancel = !validForClose;
+                };
+
+                var r = await dw.ShowAsync();
+                if (r != ContentDialogResult.Primary)
+                {
+                    name.Text = "";
+                    endpoint.ItemsSource = null;
+                    accounts.ItemsSource = null;
+                    return;
+                }
+                else if (!string.IsNullOrEmpty(pkey.Text))
+                {
+                    byte[] pkeydata = UnicodeEncoding.ASCII.GetBytes(pkey.Text);
+                    var len = pkeydata.Length; 
+                    var lb = BitConverter.GetBytes(len);
+                    var round = roundUp(pkeydata.Length, 16);
+                    Array.Resize(ref pkeydata, round);  
+                    ProtectedMemory.Protect(pkeydata, MemoryProtectionScope.SameLogon);
+                    var encodeddata = lb.Concat(pkeydata).ToArray();  
+                    item.AddDeployProfile(name.Text, (string)endpoint.SelectedValue, (string)accounts.SelectedValue, encodeddata);
+                }
+                else
+                {
+                    item.AddDeployProfile(name.Text, (string)endpoint.SelectedValue, (string)accounts.SelectedValue);
+                }
+                if (!tree.RootItem.Save("BlockchainExplorerTree", out var ex))
+                {
+#if IS_VSIX
+                VSUtil.ShowModalErrorDialogBox("Error saving tree data: " + ex?.Message);
+#else
+                    System.Windows.MessageBox.Show("Error saving tree data: " + ex?.Message);
+#endif
+                }
+                else
+                {
+                    tree.Refresh();
+                }
+            }
+            catch (Exception ex)
             {
-                args.Cancel = !validForClose;
-            };
-            var r = await dw.ShowAsync();
+#if IS_VSIX
+                VSUtil.ShowModalErrorDialogBox(ex?.Message);
+#else
+                System.Windows.MessageBox.Show(ex?.Message);
+#endif
+            }
+        }
+        #endregion
+
+        #region Methods
+        private BlockchainInfo GetSelectedItem(object sender)
+        {
+            var window = (BlockchainExplorerToolWindowControl)sender;
+            var tree = window.BlockchainExplorerTree;
+            return tree.SelectedItem;
+        }
+        private void ShowValidationErrors(Wpc.TextBlock textBlock, string message)
+        {
+            textBlock.Visibility = Visibility.Visible;
+            textBlock.Text = message;
         }
+
+        private void ShowProgressRing(ProgressRing progressRing)
+        {
+            progressRing.IsEnabled = true;
+            progressRing.Visibility = Visibility.Visible;
+        }
+
+        private void HideProgressRing(ProgressRing progressRing)
+        {
+            progressRing.IsEnabled = false;
+            progressRing.Visibility = Visibility.Hidden;
+        }
+
+        #endregion
+
+        #region Fields
+        internal BlockchainExplorerToolWindow window;
+
+        #endregion
+
     }
 }
diff --git a/tests/Stratis.DevEx.Wpf.Test/BlockchainExplorer/BlockchainExplorerTreeResources.xaml b/tests/Stratis.DevEx.Wpf.Test/BlockchainExplorer/BlockchainExplorerTreeResources.xaml
@@ -300,6 +300,9 @@
             <DataTrigger Binding="{Binding Path=Kind}" Value="Account">
                 <Setter TargetName="NodeImage" Property="Source" Value="{StaticResource AccountDrawingImage}" />
             </DataTrigger>
+            <DataTrigger Binding="{Binding Path=Kind}" Value="DeployProfile">
+                <Setter TargetName="NodeImage" Property="Source" Value="{StaticResource DeployProfileDrawingImage}" />
+            </DataTrigger>
             <MultiDataTrigger>
                 <MultiDataTrigger.Conditions>
                     <Condition Binding="{Binding Path=Kind}" Value="Network" />
@@ -450,16 +453,10 @@
     <!-- Dialog style -->
     <Style x:Key="DialogStyle" TargetType="{x:Type StackPanel}">
         <Style.Resources>
-            <Style TargetType="{x:Type ui:TextBox}">
-                <Setter Property="FontSize" Value="12" />
-            </Style>
             <Style TargetType="{x:Type ui:NumberBox}">
                 <Setter Property="FontSize" Value="12" />
                 <Setter Property="HorizontalContentAlignment" Value="Right" />
             </Style>
-            <Style TargetType="{x:Type ui:TextBlock}">
-                <Setter Property="FontSize" Value="12" />
-            </Style>
         </Style.Resources>
     </Style>
 
@@ -517,26 +514,25 @@
 
     <StackPanel x:Key="AddDeployProfileDialog" Orientation="Vertical" VerticalAlignment="Top" Style="{DynamicResource DialogStyle}">
         <StackPanel Orientation="Vertical" >
-            <ui:TextBlock Text="Deploy Profile Name:" VerticalAlignment="Center" Margin="1,0,0,2" />
+            <ui:TextBlock Text="Name:" VerticalAlignment="Center" Margin="1,0,0,2" />
             <ui:TextBox x:Name="DeployProfileNameTextBox"  Width="150" Margin="0,2,10,4" VerticalAlignment="Center" />
         </StackPanel>
         <StackPanel Orientation="Vertical">
-            <ui:TextBlock Text="Select the deployment endpoint:" VerticalAlignment="Center"  Margin="1,0,0,2" />
+            <ui:TextBlock Text="Endpoint:" VerticalAlignment="Center"  Margin="1,0,0,2" />
             <ComboBox x:Name="DeployProfileEndpointListView" Width="150" Margin="0,2,10,4" VerticalAlignment="Center"/>
         </StackPanel>
         <StackPanel Orientation="Vertical">
-            <ui:TextBlock Text="Select the account to use:" VerticalAlignment="Center"  Margin="1,0,0,2" />
+            <ui:TextBlock Text="Account:" VerticalAlignment="Center"  Margin="1,0,0,2" />
             <ComboBox x:Name="DeployProfileAccountListView" Width="150" Margin="0,2,10,4" VerticalAlignment="Center"/>
         </StackPanel>
         <StackPanel Orientation="Vertical">
-            <ui:TextBlock Text="Account private key (optional):" VerticalAlignment="Center" Margin="1,0,0,2" />
+            <ui:TextBlock Text="Private Key (optional):" VerticalAlignment="Center" Margin="1,0,0,2" />
             <ui:PasswordBox Name="DeployProfileAccountPrivateKeyPasswordBox" Width="150" Margin="0,2,10,4" VerticalAlignment="Center"/>
         </StackPanel>
         <StackPanel Orientation="Vertical">
-            <Grid>
-                <ui:TextBlock Name = "AddDeployProfileDialogValidationErrors" Grid.Row="0" Foreground="Red" Visibility="Hidden" VerticalAlignment="Center" TextWrapping="WrapWithOverflow" 
+            <ui:TextBlock Name = "AddDeployProfileDialogValidationErrors" Grid.Row="0" Foreground="Red" Visibility="Hidden" VerticalAlignment="Center" TextWrapping="WrapWithOverflow" 
                         ScrollViewer.VerticalScrollBarVisibility="Auto" ScrollViewer.CanContentScroll="True"/>
-            </Grid>
+ 
         </StackPanel>
     </StackPanel>
 </ResourceDictionary>
diff --git a/tests/Stratis.DevEx.Wpf.Test/BlockchainExplorer/BlockchainViewModel.cs b/tests/Stratis.DevEx.Wpf.Test/BlockchainExplorer/BlockchainViewModel.cs
@@ -4,6 +4,8 @@
 using System.Collections.Specialized;
 using System.ComponentModel;
 using System.IO;
+using System.Text;
+using System.Security.Cryptography;
 using System.Linq;
 using System.Numerics;
 using System.Net.Http;
@@ -115,6 +117,21 @@ public BlockchainInfo AddAccount(string pubkey, string label = null)
             };
             return AddChild(BlockchainInfoKind.Account, pubkey, data);  
         }
+
+        public BlockchainInfo AddDeployProfile(string name, string endpoint, string account, byte[] pkey = null)
+        {
+            var data = new Dictionary<string, object>()
+            {
+                {"EndPoint",  endpoint},
+                {"Account",  account},
+            };
+            if (pkey != null)
+            {
+                data["PrivateKey"] = pkey;
+            }
+            return AddChild(BlockchainInfoKind.DeployProfile, name, data);
+        }
+
         public override int GetHashCode() => Key.GetHashCode();
 
         public override bool Equals(object obj) => obj is BlockchainInfo bi ? Key == bi.Key : false; 
@@ -133,6 +150,8 @@ public BlockchainInfo AddAccount(string pubkey, string label = null)
 
         public IEnumerable<string> GetNetworkAccounts() => GetChild("Accounts", BlockchainInfoKind.Folder).GetChildren(BlockchainInfoKind.Account).Select(bi => bi.Name);
 
+        public IEnumerable<string> GetNetworkDeployProfiles() => GetChild("Deploy Profiles", BlockchainInfoKind.Folder).GetChildren(BlockchainInfoKind.DeployProfile).Select(bi => bi.Name);
+
         public bool Save(string path, out Exception e)
         {
             try
diff --git a/tests/Stratis.DevEx.Wpf.Test/Stratis.DevEx.Wpf.Test.csproj b/tests/Stratis.DevEx.Wpf.Test/Stratis.DevEx.Wpf.Test.csproj
@@ -44,6 +44,7 @@
     <Reference Include="System" />
     <Reference Include="System.Data" />
     <Reference Include="System.Numerics" />
+    <Reference Include="System.Security" />
     <Reference Include="System.Xml" />
     <Reference Include="Microsoft.CSharp" />
     <Reference Include="System.Core" />
