Merge pull request #42 from JohnDuprey/dev

Add pipeline support
JohnDuprey · Aug 30, 2024 · f989e8e · f989e8e
2 parents 34c92bd + 4a1758d
commit f989e8e
Showing 1 changed file with 18 additions and 16 deletions.
diff --git a/DuoSecurity/Public/Admin API/Users/Register-DuoUser.ps1 b/DuoSecurity/Public/Admin API/Users/Register-DuoUser.ps1
@@ -26,32 +26,34 @@ function Register-DuoUser {
     #>
     [CmdletBinding(SupportsShouldProcess)]
     Param(
-        [Parameter(Mandatory = $true)]
+        [Parameter(ValueFromPipelineByPropertyName = $true, Mandatory = $true)]
         [string]$Username,
 
-        [Parameter(Mandatory = $true)]
+        [Parameter(ValueFromPipelineByPropertyName = $true, Mandatory = $true)]
         [string]$Email,
 
         [Parameter()]
         [int]$ValidSecs = 2592000
     )
 
-    $DuoRequest = @{
-        Method = 'POST'
-        Path   = '/admin/v1/users/enroll'
-        Params = @{
-            username   = $Username
-            email      = $Email
-            valid_secs = $ValidSecs
+    process {
+        $DuoRequest = @{
+            Method = 'POST'
+            Path   = '/admin/v1/users/enroll'
+            Params = @{
+                username   = $Username
+                email      = $Email
+                valid_secs = $ValidSecs
+            }
         }
-    }
 
-    if ($PSCmdlet.ShouldProcess($Email)) {
-        $Request = Invoke-DuoRequest @DuoRequest
-        if ($Request.stat -ne 'OK') {
-            $Request
-        } else {
-            $Request.response
+        if ($PSCmdlet.ShouldProcess($Email)) {
+            $Request = Invoke-DuoRequest @DuoRequest
+            if ($Request.stat -ne 'OK') {
+                $Request
+            } else {
+                $Request.response
+            }
         }
     }
 }