feat: Variable type (e.g. Certificate) can now be specified

Swoogan · Apr 15, 2019 · 911c8c4 · 911c8c4
1 parent 85f8dba
commit 911c8c4
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/Octopus-Cmdlets/AddLibraryVariable.cs b/Octopus-Cmdlets/AddLibraryVariable.cs
@@ -81,6 +81,15 @@ public class AddLibraryVariable : PSCmdlet
             ValueFromPipelineByPropertyName = true)]
         public string Value { get; set; }
 
+        /// <summary>
+        /// <para type="description">The type of the variable to create.</para>
+        /// </summary>
+        [Parameter(
+            ParameterSetName = "ByParts",
+            Mandatory = false,
+            ValueFromPipelineByPropertyName = true)]
+        public VariableType Type { get; set; }
+
         /// <summary>
         /// <para type="description">The environments to restrict the scope to.</para>
         /// </summary>
@@ -181,7 +190,7 @@ private void ProcessByObject()
 
         private void ProcessByParts()
         {
-            var variable = new VariableResource { Name = Name, Value = Value, IsSensitive = Sensitive };
+            var variable = new VariableResource { Name = Name, Value = Value, Type = Type, IsSensitive = Sensitive };
 
             if (Environments != null)
                 AddEnvironments(variable);

diff --git a/Octopus-Cmdlets/AddVariable.cs b/Octopus-Cmdlets/AddVariable.cs
@@ -66,6 +66,15 @@ public class AddVariable : PSCmdlet
             ValueFromPipelineByPropertyName = true)]
         public string Value { get; set; }
 
+        /// <summary>
+        /// <para type="description">The type of the variable to create.</para>
+        /// </summary>
+        [Parameter(
+            ParameterSetName = "ByParts",
+            Mandatory = false,
+            ValueFromPipelineByPropertyName = true)]
+        public VariableType Type { get; set; }
+
         /// <summary>
         /// <para type="description">The environments to restrict the scope to.</para>
         /// </summary>
@@ -190,7 +199,7 @@ protected override void ProcessRecord()
 
         private void ProcessByParts()
         {
-            var variable = new VariableResource { Name = Name, Value = Value, IsSensitive = Sensitive };
+            var variable = new VariableResource { Name = Name, Value = Value, Type = Type, IsSensitive = Sensitive };
 
             if (Environments != null)
                 AddEnvironments(variable);