BuildCustomFunctions.ps1

#Requires -Modules powershell-yaml
Param(
    [Parameter(Mandatory = $true, ValueFromPipelineByPropertyName = $true, HelpMessage = 'Path to SDK config file.')][ValidateNotNullOrEmpty()][System.String[]]$ConfigPath
    , [Parameter(Mandatory = $true, ValueFromPipelineByPropertyName = $true, HelpMessage = 'Path to SDK module manifest.')][ValidateNotNullOrEmpty()][System.String[]]$psd1Path
    , [Parameter(Mandatory = $true, ValueFromPipelineByPropertyName = $true, HelpMessage = 'Path to custom files.')][ValidateNotNullOrEmpty()][System.String[]]$CustomFolderPath
    , [Parameter(Mandatory = $true, ValueFromPipelineByPropertyName = $true, HelpMessage = 'Path to examples files.')][ValidateNotNullOrEmpty()][System.String[]]$ExamplesFolderPath
    , [Parameter(Mandatory = $true, ValueFromPipelineByPropertyName = $true, HelpMessage = 'Path to test files.')][ValidateNotNullOrEmpty()][System.String[]]$TestFolderPath
)
Try {
    If (!(Test-Path -Path:($CustomFolderPath))) {
        New-Item -Path:($CustomFolderPath) -ItemType:('Directory')
    }
    $IndentChar = '    '
    # Get AutoRest Versions
    $coreRegex = [regex]'autorest_core@([0-9]+\.[0-9]+\.[0-9]+)'
    $PSRegex = [regex]'autorest_powershell@([0-9]+\.[0-9]+\.[0-9]+)'
    $autorestRaw = npx autorest --version
    $autorestCore = (Select-String -InputObject $autorestRaw -Pattern $coreRegex).Matches.Groups[1].value
    $autorestPS = (Select-String -InputObject $autorestRaw -Pattern $PSRegex).Matches.Groups[1].value
    $MSCopyrightHeader = "`n# ----------------------------------------------------------------------------------`n# Code generated by Microsoft (R) AutoRest Code Generator (autorest: $autorestCore, generator: @autorest/powershell@$autorestPS)`n# Changes may cause incorrect behavior and will be lost if the code is regenerated.`n# ----------------------------------------------------------------------------------`n"
    $Divider = '|#|#|#|#|#|#|#|#|#|#|#|#|#|#|#|'
    $FunctionTemplate = "{0}`nFunction {1}`n{{`n$($IndentChar){2}`n$($IndentChar)Param(`n{3}`n$($IndentChar))`n$($IndentChar)Begin`n$($IndentChar){{`n{4}`n$($IndentChar)}}`n$($IndentChar)Process`n$($IndentChar){{`n{5}`n$($IndentChar)}}`n$($IndentChar)End`n$($IndentChar){{`n{6}`n$($IndentChar)}}`n}}"
    # Get config values
    $Config = Get-Content -Path:($ConfigPath) | ConvertFrom-Yaml
    $ConfigPrefix = $Config.prefix | Select-Object -First 1
    $ConfigCustomFunctionPrefix = $Config.customFunctionPrefix
    # Pagination Mapping
    $CustomPaginationMap = @{
        'Get-JCSDKGsuiteUsersToImport' = @{
            skip = 'pageToken'
            limit = 'maxResults'
            nextToken = '$result.nextPageToken'
        }
        'Get-JcSdkGsuiteUsersToImportFormatted' = @{
            skip = 'pageToken'
            limit = 'maxResults'
            nextToken = '$result.nextPageToken'
        }
        'Get-JCSDKOffice365UsersToImport'   = @{
            skip  = 'skipToken'
            limit = 'top'
            nextToken = '$result.skipToken'
        }
    }
    # Misc Functions
    Function Convert-GeneratedToCustom ([System.String]$InputString, [System.String]$ConfigPrefix, [System.String]$ConfigCustomFunctionPrefix) {
        # Swap out SDK prefix for customFunction prefix
        $InputString = $InputString.Replace($ConfigPrefix, $ConfigCustomFunctionPrefix)
        # Remove weird output conversion for the customFunctions
        $OutputMatches = $InputString | Select-String -Pattern:([regex]'(?<=\()(.*?)(?=\)\.ToJsonString\(\) \| ConvertFrom-Json)') -AllMatches
        $OutputMatches.Matches | ForEach-Object {
            $OutputMatchesFind = '({0}).ToJsonString() | ConvertFrom-Json' -f ($_.Value)
            $InputString = $InputString.Replace($OutputMatchesFind, $_.Value)
        }
        Return $InputString
    }
    # Load the module
    $ImportedModule = Import-Module $psd1Path -Force -PassThru
    # Start generation
    If (Get-Module -Name($ImportedModule.Name)) {
        # Get list of commands from module
        $Commands = Get-Command -Module:($ImportedModule.Name) # -Verb:('Get') -Noun:('JcSdkInternalEvent') # Use to troubleshoot single command
        ForEach ($Command In $Commands) {
            # Get module name
            $ModuleName = If ($Command.Module.Name -like '*.internal') {
                $Command.Module.Name.Replace('.internal', '')
            } Else {
                $Command.Module.Name
            }
            # Create new function name
            $CommandName = $Command.Name
            $NewCommandName = $CommandName.Replace($ConfigPrefix, $ConfigCustomFunctionPrefix)
            # Get content from sdk function
            $CommandFilePath = $Command.ScriptBlock.File
            Write-Host ("[STATUS] Command Source: $CommandFilePath") -BackgroundColor:('Black') -ForegroundColor:('Magenta')
            $CommandFilePathContent = Get-Content -Path:($CommandFilePath) -Raw
            $MSCopyrightHeader = "`n<#`n.Synopsis`n"
            $Divider = "`n[SPLIT]`n<#`n.Synopsis`n"
            $FunctionContent = If ($CommandFilePath -like '*ProxyCmdletDefinitions.ps1') {
                <# When the autorest generated module has been installed and imported from the PSGallery all the
            cmdlets will exist in a single ProxyCmdletDefinitions.ps1 file. We need to parse
            out the specific function in order to gather the parts we need to copy over. #>
                # Write-Host "hitting condition"

                $ProxyContent = $CommandFilePathContent.Replace($MSCopyrightHeader, $Divider)
                $ProxyContentSplit = $ProxyContent.Split("[SPLIT]")
                foreach ($functionSplit in $ProxyContentSplit) {
                    if ($functionSplit -match "function $CommandName {") {
                        # return the matched content
                        $functionSplit
                    }
                }
            } Else {
                <# When the autorest generated module has been imported from a local psd1 module the function will
            remain in their individual files. #>
                $CommandFilePathContent
            }
            # Extract the sections we want to copy over to our new function.
            $Params = $FunctionContent | Select-String -Pattern:([regex]'(?s)(    \[Parameter)(.*?)(\})') -AllMatches
            $OutputType = (($FunctionContent | Select-String -Pattern:([regex]'(\[OutputType)(.*?)(\]\s+)')).Matches.Value).TrimEnd()
            $CmdletBinding = (($FunctionContent | Select-String -Pattern:([regex]'(\[CmdletBinding)(.*?)(\]\s+)')).Matches.Value).TrimEnd()
            $DefaultParameterSetName = ($CmdletBinding | Select-String -Pattern:([regex]"(?<=DefaultParameterSetName=')(.*?)(?=')")).Matches.Value
            # Strip out parameters that match "DontShow"
            if ($NewCommandName -in $CustomPaginationMap.Keys){
                $ParameterContent = ($Params.Matches.Value | Where-Object { $_ -notlike "*`${$($CustomPaginationMap.$NewCommandName.limit)}*" -and $_ -notlike "*`${$($CustomPaginationMap.$NewCommandName.skip)}*" })
                $ContainsLimit = $Params.Matches.Value | Where-Object { $_ -like "*`${$($CustomPaginationMap.$NewCommandName.limit)}*" }
                $ContainsSkip = $Params.Matches.Value | Where-Object { $_ -like "*`${$($CustomPaginationMap.$NewCommandName.skip)}*" }
                $skip = $($CustomPaginationMap.$NewCommandName.skip)
                $limit = $($CustomPaginationMap.$NewCommandName.limit)
                $tokenizedPaginate = $true
                $skipToken = $($CustomPaginationMap.$NewCommandName.nextToken)

            } else {
                if ($NewCommandName -eq 'Get-JcSdkEvent') {
                    $ParameterContent = ($Params.Matches.Value)
                } else {
                    $ParameterContent = ($Params.Matches.Value | Where-Object { $_ -notlike '*${Limit}*' -and $_ -notlike '*${Skip}*' })
                }
                $ContainsLimit = $Params.Matches.Value | Where-Object { $_ -like '*Limit*' }
                $ContainsSkip = $Params.Matches.Value | Where-Object { $_ -like '*Skip*' }
                $skip = 'Skip'
                $limit = 'Limit'
                $tokenizedPaginate = $false
                $skipToken = '$ResultCount'

            }
            $ParameterSetLimit = ($ContainsLimit | Select-String -Pattern:([regex]"(?<=ParameterSetName=')(.*?)(?=')")).Matches.Value
            $ParameterSetSkip = ($ContainsSkip | Select-String -Pattern:([regex]"(?<=ParameterSetName=')(.*?)(?=')")).Matches.Value
            # Build CmdletBinding
            If (-not [System.String]::IsNullOrEmpty($OutputType)) {
                $CmdletBinding = "$($OutputType)`n$($IndentChar)$($CmdletBinding)"
            }
            # Build $BeginContent, $ProcessContent, and $EndContent
            $BeginContent = @()
            $ProcessContent = @()
            $EndContent = @()
            # Results logic - If the output model is undefined in the swagger spec
            $ResultsLogic = If ($Command.OutputType -like "$ModuleName.Models.*ApplicationJson*") {
                @"
($($ImportedModule.Name)\$($CommandName) -ErrorAction SilentlyContinue -errorVariable sdkError @PSBoundParameters); if (-not [System.String]::IsNullOrEmpty(`$Result)) { `$Result = `$Result.ToJsonString() | ConvertFrom-Json };
"@
            } Else {
                "$($ImportedModule.Name)\$($CommandName) @PSBoundParameters -errorAction SilentlyContinue -errorVariable sdkError"
            }
            If ($Command.Verb -in ('Get', 'Search')) {
                # Add paginate parameter if function contains Limit or Skip parameters
                If (-not [System.String]::IsNullOrEmpty($ContainsSkip) -or -not [System.String]::IsNullOrEmpty($ContainsLimit)) {
                    $ParameterSetLimitSkip = If (-not [System.String]::IsNullOrEmpty($ParameterSetLimit) -or -not [System.String]::IsNullOrEmpty($ParameterSetSkip)) {
                        $ParameterSetLimitSkipArray = @()
                        If (-not [System.String]::IsNullOrEmpty($ParameterSetLimit)) {
                            $ParameterSetLimit | ForEach-Object {
                                $ParameterSetLimitSkipArray += $_.Trim()
                            }
                        }
                        If (-not [System.String]::IsNullOrEmpty($ParameterSetSkip)) {
                            $ParameterSetSkip | ForEach-Object {
                                $ParameterSetLimitSkipArray += $_.Trim()
                            }
                        }
                        "'$(($ParameterSetLimitSkipArray | Select-Object -Unique ) -join "','")'"

                    } Else {
                        "'$(($DefaultParameterSetName | Select-Object -Unique) -join "','")'"
                    }
                    If (-not [System.String]::IsNullOrEmpty($ParameterContent)) {
                        If ($ParameterContent.Count -eq 1) {
                            # Add paginate parameter
                            $ParameterContent += ",$($IndentChar)[Parameter(DontShow)]`n$($IndentChar)[System.Boolean]`n$($IndentChar)# Set to `$true to return all results. This will overwrite any skip and limit parameter.`n$($IndentChar)`$Paginate = `$true"
                        } Else {
                            $ParameterContent += "$($IndentChar)[Parameter(DontShow)]`n$($IndentChar)[System.Boolean]`n$($IndentChar)# Set to `$true to return all results. This will overwrite any skip and limit parameter.`n$($IndentChar)`$Paginate = `$true"
                        }
                    }
                }
                # Build script content
                If ($ModuleName -eq 'JumpCloud.SDK.DirectoryInsights') {
                    # Build "Begin" block
                    $BeginContent += @"
        `$Results = @()
        `$PSBoundParameters.Add('HttpPipelineAppend', {
                param(`$req, `$callback, `$next)
                # call the next step in the Pipeline
                `$ResponseTask = `$next.SendAsync(`$req, `$callback)
                `$global:JCHttpRequest = `$req
                `$global:JCHttpRequestContent = If (-not [System.String]::IsNullOrEmpty(`$req.Content)) { `$req.Content.ReadAsStringAsync() }
                `$global:JCHttpResponse = `$ResponseTask
                # `$global:JCHttpResponseContent = If (-not [System.String]::IsNullOrEmpty(`$ResponseTask.Result.Content)) { `$ResponseTask.Result.Content.ReadAsStringAsync() }
                Return `$ResponseTask
            }
        )
"@
                    # Build "Process" block
                    # Add paginate logic if function contains Limit and Skip parameters
                    If (-not [System.String]::IsNullOrEmpty($ContainsSkip) -or -not [System.String]::IsNullOrEmpty($ContainsLimit)) {
                        $ProcessContent += @"
        If (`$Paginate -and `$PSCmdlet.ParameterSetName -in ($ParameterSetLimitSkip)) {
            `$PSBoundParameters.Remove('Paginate') | Out-Null
            Do {
                `$maxRetries = 4
                `$resultCounter = 0
                :retryLoop do {
                    `$resultCounter++
                    `$Result = $ResultsLogic
                    If (`$sdkError){
                        If (`$resultCounter -eq `$maxRetries){
                            throw `$sdkError
                        }
                        If (`$JCHttpResponse.Result.StatusCode -eq "503") {
                            Write-Warning ("503: Service Unavailable - retrying in " + (`$resultCounter * 5) + " seconds.")
                        } else {
                            throw `$sdkError
                        }
                    } else {
                        break retryLoop
                    }
                    Start-Sleep -Seconds (`$resultCounter * 5)
                } while (`$resultCounter -lt `$maxRetries)
                If (`$limit) {
                    `$Results += `$Result
                    Break
                } elseif (`$JCHttpResponse.Result.Headers.Contains('X-Search_after')) {
                    If (-not [System.String]::IsNullOrEmpty(`$Result)) {
                        `$XResultSearchAfter = (`$JCHttpResponse.Result.Headers.GetValues('X-Search_after') | ConvertFrom-Json);
                        If ([System.String]::IsNullOrEmpty(`$PSBoundParameters.SearchAfter)) {
                            If ([System.String]::IsNullOrEmpty(`$PSBoundParameters.Body)) {
                                `$PSBoundParameters.Add('SearchAfter', `$XResultSearchAfter)
                            } Else {
                                `$PSBoundParameters.Body.SearchAfter = `$XResultSearchAfter
                            }
                        } Else {
                            `$PSBoundParameters.SearchAfter = `$XResultSearchAfter
                        }
                        `$XResultCount = `$JCHttpResponse.Result.Headers.GetValues('X-Result-Count')
                        `$XLimit = `$JCHttpResponse.Result.Headers.GetValues('X-Limit')
                        `$Results += `$Result
                        Write-Debug ("ResultCount: `$(`$XResultCount); Limit: `$(`$XLimit); XResultSearchAfter: `$(`$XResultSearchAfter); ");
                        Write-Debug ('HttpRequest: ' + `$JCHttpRequest);
                        Write-Debug ('HttpRequestContent: ' + `$JCHttpRequestContent.Result);
                        Write-Debug ('HttpResponse: ' + `$JCHttpResponse.Result);
                        # Write-Debug ('HttpResponseContent: ' + `$JCHttpResponseContent.Result);
                    }
                } Else {
                    `$Results += `$Result
                    Break
                }
            }
            While (`$XResultCount -eq `$XLimit -and -not [System.String]::IsNullOrEmpty(`$Result))
        } Else {
            `$PSBoundParameters.Remove('Paginate') | Out-Null
            `$maxRetries = 4
            `$resultCounter = 0
            :retryLoop do {
                `$resultCounter++
                `$Result = $ResultsLogic
                If (`$sdkError){
                    If (`$resultCounter -eq `$maxRetries){
                        throw `$sdkError
                    }
                    If (`$JCHttpResponse.Result.StatusCode -eq "503") {
                        Write-Warning ("503: Service Unavailable - retrying in " + (`$resultCounter * 5) + " seconds.")
                    } else {
                        throw `$sdkError
                    }
                } else {
                    break retryLoop
                }
                Start-Sleep -Seconds (`$resultCounter * 5)
            } while (`$resultCounter -lt `$maxRetries)
            Write-Debug ('HttpRequest: ' + `$JCHttpRequest);
            Write-Debug ('HttpRequestContent: ' + `$JCHttpRequestContent.Result);
            Write-Debug ('HttpResponse: ' + `$JCHttpResponse.Result);
            # Write-Debug ('HttpResponseContent: ' + `$JCHttpResponseContent.Result);
            If (-not [System.String]::IsNullOrEmpty(`$Result)) {
                `$Results += `$Result
            }
        }
"@
                    } Else {
                        $ProcessContent += @"
        `$maxRetries = 4
        `$resultCounter = 0
        :retryLoop do {
            `$resultCounter++
            `$Result = $ResultsLogic
            If (`$sdkError){
                If (`$resultCounter -eq `$maxRetries){
                    throw `$sdkError
                }
                If (`$JCHttpResponse.Result.StatusCode -eq "503") {
                    Write-Warning ("503: Service Unavailable - retrying in " + (`$resultCounter * 5) + " seconds.")
                } else {
                    throw `$sdkError
                }
            } else {
                break retryLoop
            }
            Start-Sleep -Seconds (`$resultCounter * 5)
        } while (`$resultCounter -lt `$maxRetries)
        Write-Debug ('HttpRequest: ' + `$JCHttpRequest);
        Write-Debug ('HttpRequestContent: ' + `$JCHttpRequestContent.Result);
        Write-Debug ('HttpResponse: ' + `$JCHttpResponse.Result);
        # Write-Debug ('HttpResponseContent: ' + `$JCHttpResponseContent.Result);
        If (-not [System.String]::IsNullOrEmpty(`$Result))
        {
            `$Results += `$Result
        }
"@
                    }
                    # Build "End" block
                    $EndContent += @"
        # Clean up global variables
        `$GlobalVars = @('JCHttpRequest', 'JCHttpRequestContent', 'JCHttpResponse', 'JCHttpResponseContent')
        `$GlobalVars | ForEach-Object {
            If ((Get-Variable -Scope:('Global')).Where( { `$_.Name -eq `$_ })) { Remove-Variable -Name:(`$_) -Scope:('Global') }
        }
        Return `$Results
"@
                } ElseIf ($ModuleName -In ('JumpCloud.SDK.V1', 'JumpCloud.SDK.V2')) {
                    # Build "Begin" block
                    $BeginContent += @"
        `$Results = @()
        `$PSBoundParameters.Add('HttpPipelineAppend', {
                param(`$req, `$callback, `$next)
                # call the next step in the Pipeline
                `$ResponseTask = `$next.SendAsync(`$req, `$callback)
                `$global:JCHttpRequest = `$req
                `$global:JCHttpRequestContent = If (-not [System.String]::IsNullOrEmpty(`$req.Content)) { `$req.Content.ReadAsStringAsync() }
                `$global:JCHttpResponse = `$ResponseTask
                # `$global:JCHttpResponseContent = If (-not [System.String]::IsNullOrEmpty(`$ResponseTask.Result.Content)) { `$ResponseTask.Result.Content.ReadAsStringAsync() }
                Return `$ResponseTask
            }
        )
"@
                    # Build "Process" block
                    # Add paginate logic if function contains Limit and Skip parameters
                    If (-not [System.String]::IsNullOrEmpty($ContainsSkip) -or -not [System.String]::IsNullOrEmpty($ContainsLimit)) {
                        $ProcessContent += @"
        If (`$Paginate -and `$PSCmdlet.ParameterSetName -in ($ParameterSetLimitSkip)) {
            `$PSBoundParameters.Remove('Paginate') | Out-Null
"@
                        If (-not [System.String]::IsNullOrEmpty($ContainsLimit)) {
                            $ProcessContent += @"
            If ([System.String]::IsNullOrEmpty(`$PSBoundParameters.$limit)) {
                `$PSBoundParameters.Add('$limit', 100)
            }
"@
                        }
                        If ( (-not [System.String]::IsNullOrEmpty($ContainsSkip)) -and ($tokenizedPaginate -eq $false) ) {
                            $ProcessContent += @"
            If ([System.String]::IsNullOrEmpty(`$PSBoundParameters.$skip)) {
                `$PSBoundParameters.Add('$skip', 0)
            }
"@
                        }
                        $ProcessContent += @"
            Do {
"@
                        If (-not [System.String]::IsNullOrEmpty($ContainsLimit)) {
                            $ProcessContent += @"
                Write-Debug (`"Limit: `$(`$PSBoundParameters.$limit); `");
"@
                        }
                        If (-not [System.String]::IsNullOrEmpty($ContainsSkip)) {
                            $ProcessContent += @"
                Write-Debug (`"Skip: `$(`$PSBoundParameters.$skip); `");
"@
                        }
                $ProcessContent += @"
                `$maxRetries = 4
                `$resultCounter = 0
                :retryLoop do {
                    `$resultCounter++
                    `$Result = $ResultsLogic
                    If (`$sdkError){
                        If (`$resultCounter -eq `$maxRetries){
                            throw `$sdkError
                        }
                        If (`$JCHttpResponse.Result.StatusCode -eq "503") {
                            Write-Warning ("503: Service Unavailable - retrying in " + (`$resultCounter * 5) + " seconds.")
                        } else {
                            throw `$sdkError
                        }
                    } else {
                        break retryLoop
                    }
                    Start-Sleep -Seconds (`$resultCounter * 5)
                } while (`$resultCounter -lt `$maxRetries)
                Write-Debug ('HttpRequest: ' + `$JCHttpRequest);
                Write-Debug ('HttpRequestContent: ' + `$JCHttpRequestContent.Result);
                Write-Debug ('HttpResponse: ' + `$JCHttpResponse.Result);
                # Write-Debug ('HttpResponseContent: ' + `$JCHttpResponseContent.Result);
                `$Result = If ('Results' -in `$Result.PSObject.Properties.Name) {
                    `$Result.results
                } Else {
                    `$Result
                }
                If (-not [System.String]::IsNullOrEmpty(`$Result))
                {
"@
                        If ((-not [System.String]::IsNullOrEmpty($ContainsLimit) -or -not [System.String]::IsNullOrEmpty($ContainsSkip)) -And ($tokenizedPaginate -eq $false)) {
                            $ProcessContent += @"
                    `$ResultCount = (`$Result | Measure-Object).Count;
"@
                            $ProcessContent += @"
                    `$Results += `$Result;
"@
                        }
                        If ((-not [System.String]::IsNullOrEmpty($ContainsLimit) -or -not [System.String]::IsNullOrEmpty($ContainsSkip)) -And ($tokenizedPaginate -eq $true)) {
                            $ProcessContent += @"
                    `$ResultCount = (`$Result.users | Measure-Object).Count;
"@
                            $ProcessContent += @"
                    `$Results += `$Result.users;
"@
                        }
                        If ((-not [System.String]::IsNullOrEmpty($ContainsSkip)) -And ($tokenizedPaginate -eq $false)) {
                            $ProcessContent += @"
                    `$PSBoundParameters.$skip += $skipToken
"@
                        }
                        If ((-not [System.String]::IsNullOrEmpty($ContainsSkip)) -And ($tokenizedPaginate -eq $true)) {
                            $ProcessContent += @"
                    `$PSBoundParameters.$skip = $skipToken
"@
                        }
                        $ProcessContent += @"
                }
        }
"@
                        $ProcessContent += If (-not [System.String]::IsNullOrEmpty($ContainsLimit)) {
                            @"
            While (`$ResultCount -eq `$PSBoundParameters.$limit -and -not [System.String]::IsNullOrEmpty(`$Result))
"@
                        } Else {
                            @"
            While (-not [System.String]::IsNullOrEmpty(`$Result))
"@
                        }
                        if ($tokenizedPaginate -eq $false){
                        $ProcessContent += @"
        } Else {
            `$PSBoundParameters.Remove('Paginate') | Out-Null
            `$maxRetries = 4
            `$resultCounter = 0
            :retryLoop do {
                `$resultCounter++
                `$Result = $ResultsLogic
                If (`$sdkError){
                    If (`$resultCounter -eq `$maxRetries){
                        throw $`sdkError
                    }
                    If (`$JCHttpResponse.Result.StatusCode -eq "503") {
                        Write-Warning ("503: Service Unavailable - retrying in " + (`$resultCounter * 5) + " seconds.")
                    } else {
                        throw $`sdkError
                    }
                } else {
                    break retryLoop
                }
                Start-Sleep -Seconds (`$resultCounter * 5)
            } while (`$resultCounter -lt `$maxRetries)
            Write-Debug ('HttpRequest: ' + `$JCHttpRequest);
            Write-Debug ('HttpRequestContent: ' + `$JCHttpRequestContent.Result);
            Write-Debug ('HttpResponse: ' + `$JCHttpResponse.Result);
            # Write-Debug ('HttpResponseContent: ' + `$JCHttpResponseContent.Result);
            `$Result = If ('Results' -in `$Result.PSObject.Properties.Name) {
                `$Result.results
            } Else {
                `$Result
            }
            If (-not [System.String]::IsNullOrEmpty(`$Result)) {
                `$Results += `$Result;
            }
        }
"@
                        } elseif ($tokenizedPaginate -eq $true){
                        $ProcessContent += @"
        } Else {
            `$PSBoundParameters.Remove('Paginate') | Out-Null
            `$maxRetries = 4
            `$resultCounter = 0
            :retryLoop do {
                `$resultCounter++
                `$Result = $ResultsLogic
                If (`$sdkError){
                    If (`$resultCounter -eq `$maxRetries){
                        throw `$sdkError
                    }
                    If (`$JCHttpResponse.Result.StatusCode -eq "503") {
                        Write-Warning ("503: Service Unavailable - retrying in " + (`$resultCounter * 5) + " seconds.")
                    } else {
                        throw `$sdkError
                    }
                } else {
                    break retryLoop
                }
                Start-Sleep -Seconds (`$resultCounter * 5)
            } while (`$resultCounter -lt `$maxRetries)
            Write-Debug ('HttpRequest: ' + `$JCHttpRequest);
            Write-Debug ('HttpRequestContent: ' + `$JCHttpRequestContent.Result);
            Write-Debug ('HttpResponse: ' + `$JCHttpResponse.Result);
            # Write-Debug ('HttpResponseContent: ' + `$JCHttpResponseContent.Result);
            `$Result = If ('Results' -in `$Result.PSObject.Properties.Name) {
                `$Result.results
            } Else {
                `$Result
            }
            If (-not [System.String]::IsNullOrEmpty(`$Result)) {
                `$Results += `$Result.users;
            }
        }
"@
                        }
                    } Else {
                        $ProcessContent += @"
        `$maxRetries = 4
        `$resultCounter = 0
        :retryLoop do {
            `$resultCounter++
            `$Result = $ResultsLogic
            If (`$sdkError){
                If (`$resultCounter -eq `$maxRetries){
                    throw $sdkError
                }
                If (`$JCHttpResponse.Result.StatusCode -eq "503") {
                    Write-Warning ("503: Service Unavailable - retrying in " + (`$resultCounter * 5) + " seconds.")
                } else {
                    throw `$sdkError
                }
            } else {
                break retryLoop
            }
            Start-Sleep -Seconds (`$resultCounter * 5)
        } while (`$resultCounter -lt `$maxRetries)
        Write-Debug ('HttpRequest: ' + `$JCHttpRequest);
        Write-Debug ('HttpRequestContent: ' + `$JCHttpRequestContent.Result);
        Write-Debug ('HttpResponse: ' + `$JCHttpResponse.Result);
        # Write-Debug ('HttpResponseContent: ' + `$JCHttpResponseContent.Result);
        `$Result = If ('Results' -in `$Result.PSObject.Properties.Name) {
            `$Result.results
        } Else {
            `$Result
        }
        If (-not [System.String]::IsNullOrEmpty(`$Result)) {
            `$Results += `$Result;
        }
"@
                    }
                    # Build "End" block
                    $EndContent += @"
        # Clean up global variables
        `$GlobalVars = @('JCHttpRequest', 'JCHttpRequestContent', 'JCHttpResponse','JCHttpResponseContent')
        `$GlobalVars | ForEach-Object {
            If ((Get-Variable -Scope:('Global')).Where( { `$_.Name -eq `$_ })) { Remove-Variable -Name:(`$_) -Scope:('Global') }
        }
        Return `$Results
"@
                } Else {
                    Write-Error ('Unknown module $($ModuleName)')
                }
            } ElseIf ($Command.Verb -in ('Restart', 'Invoke', 'New', 'Set', 'Remove', 'Start', 'Unlock', 'Update', 'Reset', 'Grant', 'Import', 'Clear', 'Lock', 'Stop', 'Sync', 'Initialize')) {
                # Build "Begin" block
                $BeginContent += @"
        `$Results = @()
        `$PSBoundParameters.Add('HttpPipelineAppend', {
                param(`$req, `$callback, `$next)
                # call the next step in the Pipeline
                `$ResponseTask = `$next.SendAsync(`$req, `$callback)
                `$global:JCHttpRequest = `$req
                # `$global:JCHttpRequestContent = If (-not [System.String]::IsNullOrEmpty(`$req.Content)) { `$req.Content.ReadAsStringAsync() }
                `$global:JCHttpResponse = `$ResponseTask
                # `$global:JCHttpResponseContent = If (-not [System.String]::IsNullOrEmpty(`$ResponseTask.Result.Content)) { `$ResponseTask.Result.Content.ReadAsStringAsync() }
                Return `$ResponseTask
            }
        )
"@
                # Build "Process" block
                # temp fix for Set-JcSdkUserAssociation TODO: remove when DE-2630 is done
                if ($NewCommandName -eq ("Set-JcSdkUserAssociation" -or "Set-JcSdkSystemAssociation")) {
                    $ProcessContent += @"
        if ((-not `$PSBoundParameters['attributes']) -and (`$PSBoundParameters['op'] -eq 'remove')) {
            `$PSBoundParameters.Add('attributes', @{ '' = '' })
        } elseif ((`$PSBoundParameters['body']) -And (`$Body.Op -eq 'remove')) {
            `$Body.Attributes = @{ '' = '' }
        }
"@
                }
                $ProcessContent += @"
        `$maxRetries = 4
        `$resultCounter = 0
        :retryLoop do {
            `$resultCounter++
            `$Results = $ResultsLogic
            If (`$sdkError){
                If (`$resultCounter -eq `$maxRetries){
                    throw `$sdkError
                }
                If (`$JCHttpResponse.Result.StatusCode -eq "503") {
                    Write-Warning ("503: Service Unavailable - retrying in " + (`$resultCounter * 5) + " seconds.")
                } else {
                    throw `$sdkError
                }
            } else {
                break retryLoop
            }
            Start-Sleep -Seconds (`$resultCounter * 5)
        } while (`$resultCounter -lt `$maxRetries)
"@
                # Build "End" block
                $EndContent += @"
        Write-Debug ('HttpRequest: ' + `$JCHttpRequest);
        # Write-Debug ('HttpRequestContent: ' + `$JCHttpRequestContent.Result);
        Write-Debug ('HttpResponse: ' + `$JCHttpResponse.Result);
        # Write-Debug ('HttpResponseContent: ' + `$JCHttpResponseContent.Result);
        # Clean up global variables
        `$GlobalVars = @('JCHttpRequest', 'JCHttpRequestContent', 'JCHttpResponse', 'JCHttpResponseContent')
        `$GlobalVars | ForEach-Object {
            If ((Get-Variable -Scope:('Global')).Where( { `$_.Name -eq `$_ })) {
                Remove-Variable -Name:(`$_) -Scope:('Global')
            }
        }
        Return `$Results
"@
            } Else {
                Write-Error ("Unmapped command: $CommandName")
                $null
            }
            If (-not [System.String]::IsNullOrEmpty($BeginContent) -and -not [System.String]::IsNullOrEmpty($ProcessContent) -and -not [System.String]::IsNullOrEmpty($EndContent)) {
                # Build "Function"
                $NewScript = $FunctionTemplate -f '', $NewCommandName, $CmdletBinding, ($ParameterContent -join ", `n`n"), ($BeginContent -join "`n"), ($ProcessContent -join "`n"), ($EndContent -join "`n")
                # Fix line endings
                $NewScript = $NewScript.Replace("`r`n", "`n").Trim()
                # Export the function
                Write-Host ("[STATUS] Building: $NewCommandName") -BackgroundColor:('Black') -ForegroundColor:('Magenta') # | Tee-Object -FilePath:($LogFilePath) -Append
                $OutputFilePath = "$CustomFolderPath/$NewCommandName.ps1"
                [System.IO.File]::WriteAllLines($OutputFilePath, $NewScript, (New-Object System.Text.UTF8Encoding $true))
            }
        }
    } Else {
        Write-Error ('No modules found.')
    }
} Catch {
    Get-Error
    Write-Error ($_)
}