User defined script parameters

AddMenuItem.cs

@@ -1,5 +1,9 @@
-using SSMSObjectExplorerMenu.objects;
+using SSMSObjectExplorerMenu.enums;
+using SSMSObjectExplorerMenu.extensions;
+using SSMSObjectExplorerMenu.objects;
 using System;
+using System.Collections.Generic;
+using System.Linq;
 using System.Reflection;
 using System.Windows.Forms;
 using MenuItem = SSMSObjectExplorerMenu.objects.MenuItem;
@@ -13,16 +17,24 @@ public AddMenuItem(NodeInfo nodeInfo)
 			InitializeComponent();
 
 			labelVersion.Text = Assembly.GetExecutingAssembly().GetName().Version.ToString();
+
 			comboContext.Text = nodeInfo.UrnPath;
-			this.ActiveControl = textName;
+
+            this.comboContext.DataSource =
+                Enum.GetValues(typeof(MenuItemContext))
+                    .Cast<MenuItemContext>()
+                    .Select(ctx => new { Displayed = ctx.ToStringDescription(), Value = ctx }).ToList();
+            this.comboContext.DisplayMember = nameof(ComboBoxItem<MenuItemContext>.Displayed);
+            this.comboContext.ValueMember = nameof(ComboBoxItem<MenuItemContext>.Value);
+
+            this.ActiveControl = textName;
 			buttonOpen.Visible = false;
 		}
 
 		public MenuItem GetMenuItem()
 		{
-			return new MenuItem(true, comboContext.Text, textName.Text, textPath.Text, checkExecute.Checked, checkConfirm.Checked);
+			return new MenuItem(true, (MenuItemContext)comboContext.SelectedValue, textName.Text, textPath.Text, checkExecute.Checked, checkConfirm.Checked, listViewUserDefinedParam.GetUserDefinedParams());
 		}
-
 
 		private void textName_TextChanged(object sender, EventArgs e)
 		{
@@ -36,8 +48,7 @@ private void textPath_TextChanged(object sender, EventArgs e)
 
 		private void ValidateInputs()
 		{ 
-			if (comboContext.Text.Trim().Length > 0
-				&& textName.Text.Trim().Length > 0
+			if (textName.Text.Trim().Length > 0
 				&& textPath.Text.Trim().Length > 0)
 			{
 				buttonOK.Enabled = true;
@@ -86,5 +97,59 @@ private void checkExecute_CheckedChanged(object sender, EventArgs e)
 				checkConfirm.Checked = false;
 			}
 		}
-	}
+
+        private void buttonAddUserDefinedParam_Click(object sender, EventArgs e)
+        {
+            var addDialog = new AddUserDefinedParameter(GetArgumentNamesInUse());
+            if (addDialog.ShowDialog() == DialogResult.OK)
+            {
+                var newParam = addDialog.Parameter;
+                var newListViewItem = new ListViewItem { Text = newParam.Name, Tag = newParam };
+                newListViewItem.SubItems.Add(new ListViewItem.ListViewSubItem { Text = newParam.Type.ToStringDescription() });
+                listViewUserDefinedParam.Items.Add(newListViewItem);
+            }
+        }
+
+        private void buttonEditUserDefinedParam_Click(object sender, EventArgs e)
+        {
+			var selectedItem = this.listViewUserDefinedParam.GetSelectedItems().SingleOrDefault();
+            if (selectedItem != null)
+            {
+				var editDialog = new AddUserDefinedParameter(GetArgumentNamesInUse(), true, (UserDefinedParameter)selectedItem.Tag);
+				if (editDialog.ShowDialog() == DialogResult.OK)
+				{
+					var editedParam = editDialog.Parameter;
+					selectedItem.Text = editedParam.Name;
+					selectedItem.SubItems[1].Text = Enum.GetName(typeof(UserDefinedParameterType), editedParam.Type);
+					selectedItem.Tag = editedParam;
+                }
+            }
+        }
+
+        private void buttonRemoveUserDefinedParam_Click(object sender, EventArgs e)
+        {
+			var selectedItems = this.listViewUserDefinedParam.GetSelectedItems();
+            if (selectedItems.Any() &&
+                DialogResult.Yes == MessageBox.Show("Are you sure?", "Delete parameter(s)", MessageBoxButtons.YesNo, MessageBoxIcon.Warning))
+            {
+                foreach (var item in selectedItems)
+                {
+                    this.listViewUserDefinedParam.Items.Remove(item);
+                }
+            }
+        }
+
+        private void listViewUserDefinedParam_SelectedIndexChanged(object sender, EventArgs e)
+        {
+			// In case of removing items, a minimal delay is needed to work with the state of the listview after the item has been removed.
+            this.BeginInvoke((Action)(() =>
+			{
+				var selectedItemsCount = this.listViewUserDefinedParam.GetSelectedItems().Count();
+				this.buttonEditUserDefinedParameter.Enabled = selectedItemsCount == 1;
+				this.buttonRemoveUserDefinedParam.Enabled = selectedItemsCount > 0;
+			}));
+        }
+
+        private IEnumerable<string> GetArgumentNamesInUse() => this.listViewUserDefinedParam.Items.Cast<ListViewItem>().Select(item => item.Text).Concat(Utils.ParametersFromContext);
+    }
 }