PSAtera.psm1
|
function New-AteraGetRequest { [CmdletBinding()] param ( [Parameter(Mandatory)] [string] $Endpoint, [Parameter()] [bool] $Paginate=$true ) $Headers = @{ "accept" = "application/json" "X-API-KEY" = Get-AteraAPIKey } $ItemsInPage = 50 $Uri = "https://app.atera.com/api/v3$($endpoint)?itemsInPage=$ItemsInPage" $items = @() $index = 0 do { Write-Debug "[PSAtera] Request for $Uri" $data = Invoke-RestMethod -Uri $Uri -Method "GET" -Headers $Headers if (!$Paginate) { return $data } $items += $data.items $index += 1 $Uri = $data.nextLink } while ($Uri -ne "" -and $index -lt [math]::ceiling($RecordLimit / $ItemsInPage)) return $items } function New-AteraPostRequest { [CmdletBinding()] param ( [Parameter(Mandatory)] [string] $Endpoint, [Parameter(Mandatory, ValueFromPipeline)] [Hashtable] $Body ) $Headers = @{ "accept" = "application/json" "X-API-KEY" = Get-AteraAPIKey } $Uri = "https://app.atera.com/api/v3$($endpoint)" Write-Debug "[PSAtera] Request for $Uri" $data = Invoke-RestMethod -Uri $Uri -Method "POST" -Headers $Headers -Body $body return $data } $AteraAPIKey = $env:ATERAAPIKEY <# .SYNOPSIS Set the Atera API Key used by the module. If none set, the ATERAAPIKEY environment variable will be used instead. #> function Set-AteraAPIKey { param( # Atera API Key [string]$APIKey ) $script:AteraAPIKey = $APIKey } <# .SYNOPSIS Get the Atera API Key in use by the module. #> function Get-AteraAPIKey { if (!$AteraAPIKey) { Write-Error "`$AteraAPIKey not set. Set it with either Set-AteraAPIKey or `$env:ATERAAPIKEY"; exit 1 } return $AteraAPIKey } $RecordLimit = 1000 <# .SYNOPSIS Set the maximum number of records returned by API calls. Default is set at 1,000. #> function Set-AteraRecordLimit { param( # Maximum records returned [Parameter(Mandatory=$true)] [int]$Limit ) $script:RecordLimit = $Limit } <# .SYNOPSIS Get the maximum number of records returned by API calls. #> function Get-AteraRecordLimit { $RecordLimit } Get-ChildItem -Path $PSScriptRoot/endpoints | ForEach-Object { . $_.PSPath } Export-ModuleMember -Function Get-Atera*,Set-Atera*,New-Atera* |