Search-TssWorkflowTemplate - convert Type param to enum

thycotic-ps · Oct 27, 2021 · 6e98c79 · 6e98c79
1 parent 658106f
commit 6e98c79
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 7 deletions.
diff --git a/docs/commands/workflow-templates/Search-TssWorkflowTemplate.md b/docs/commands/workflow-templates/Search-TssWorkflowTemplate.md
@@ -6,8 +6,8 @@ Search Workflow Templates
 ## SYNTAX
 
 ```
-Search-TssWorkflowTemplate [-TssSession] <Session> [-IncludeInactive] [-Type <String>] [-SortBy <String>]
- [<CommonParameters>]
+Search-TssWorkflowTemplate [-TssSession] <Session> [-IncludeInactive] [-Type <WorkflowTypes>]
+ [-SortBy <String>] [<CommonParameters>]
 ```
 
 ## DESCRIPTION
@@ -59,9 +59,10 @@ Accept wildcard characters: False
 Workflow Type (AccessRequest, SecretEraseRequest)
 
 ```yaml
-Type: String
+Type: WorkflowTypes
 Parameter Sets: (All)
 Aliases:
+Accepted values: AccessRequest, SecretEraseRequest
 
 Required: False
 Position: Named

diff --git a/src/Thycotic.SecretServer/enums/workflows/WorkflowType.cs b/src/Thycotic.SecretServer/enums/workflows/WorkflowType.cs
@@ -0,0 +1,13 @@
+using System;
+using System.Threading.Tasks;
+using System.Management.Automation;
+using System.Management.Automation.Runspaces;
+
+namespace Thycotic.PowerShell.Enums
+{
+    public enum WorkflowTypes
+    {
+        AccessRequest,
+        SecretEraseRequest
+    }
+}
diff --git a/src/functions/workflow-templates/Search-TssWorkflowTemplate.ps1 b/src/functions/workflow-templates/Search-TssWorkflowTemplate.ps1
@@ -34,9 +34,7 @@ function Search-TssWorkflowTemplate {
         $IncludeInactive,
 
         # Workflow Type (AccessRequest, SecretEraseRequest)
-        [Parameter()]
-        [ValidateSet('AccessRequest','SecretEraseRequest')]
-        [string]
+        [Thycotic.PowerShell.Enums.WorkflowTypes]
         $Type,
 
         # Sort by specific property, default Name
@@ -57,7 +55,7 @@ function Search-TssWorkflowTemplate {
             $uri = $uri, "filter.includeInactive=$([boolean]$IncludeInactive)" -join '&'
 
             if ($tssParams.ContainsKey('Type')) {
-                $uri = $uri, "filter.workflowType=$Type" -join '&'
+                $uri = $uri, "filter.workflowType=$([string]$Type)" -join '&'
             }
             $invokeParams.Uri = $uri
             $invokeParams.Method = 'GET'