fix: sherweb tenant filtering

Author: JohnDuprey

Modules/CIPPCore/Public/Entrypoints/Invoke-ListCSPsku.ps1

@@ -14,13 +14,19 @@ Function Invoke-ListCSPsku {
     Write-LogMessage -headers $Request.Headers -API $APINAME -message 'Accessed this API' -Sev 'Debug'
     $TenantFilter = $Request.Query.tenantFilter
 
-    if ($Request.Query.currentSkuOnly) {
-        $GraphRequest = Get-SherwebCurrentSubscription -TenantFilter $TenantFilter
-    } else {
-        $GraphRequest = Get-SherwebCatalog -TenantFilter $TenantFilter
+    try {
+        if ($Request.Query.currentSkuOnly) {
+            $GraphRequest = Get-SherwebCurrentSubscription -TenantFilter $TenantFilter
+        } else {
+            $GraphRequest = Get-SherwebCatalog -TenantFilter $TenantFilter
+        }
+    } catch {
+        $GraphRequest = [PSCustomObject]@{
+            name = @(@{value = 'Error getting catalog' })
+            sku  = $_.Exception.Message
+        }
     }
 
-
     Push-OutputBinding -Name Response -Value ([HttpResponseContext]@{
             StatusCode = [HttpStatusCode]::OK
             Body       = @($GraphRequest)

Modules/CippExtensions/Public/Sherweb/Get-SherwebCatalog.ps1

@@ -4,13 +4,18 @@ function Get-SherwebCatalog {
         [string]$CustomerId,
         [string]$TenantFilter
     )
+
     if ($TenantFilter) {
-        Get-ExtensionMapping -Extension 'Sherweb' | Where-Object { $_.RowKey -eq $TenantFilter } | ForEach-Object {
-            Write-Host "Extracted customer id from tenant filter - It's $($_.IntegrationId)"
-            $CustomerId = $_.IntegrationId
-        }
+        $TenantFilter = (Get-Tenants -TenantFilter $TenantFilter).customerId
+        $CustomerId = Get-ExtensionMapping -Extension 'Sherweb' | Where-Object { $_.RowKey -eq $TenantFilter } | Select-Object -ExpandProperty IntegrationId
+    }
+
+    if (![string]::IsNullOrEmpty($CustomerId)) {
+        Write-Information "Getting catalog for $CustomerId"
+        $AuthHeader = Get-SherwebAuthentication
+        $SubscriptionsList = Invoke-RestMethod -Uri "https://api.sherweb.com/service-provider/v1/customer-catalogs/$CustomerId" -Method GET -Headers $AuthHeader
+        return $SubscriptionsList.catalogItems
+    } else {
+        throw 'No Sherweb mapping found'
     }
-    $AuthHeader = Get-SherwebAuthentication
-    $SubscriptionsList = Invoke-RestMethod -Uri "https://api.sherweb.com/service-provider/v1/customer-catalogs/$CustomerId" -Method GET -Headers $AuthHeader
-    return $SubscriptionsList.catalogItems
 }

Modules/CippExtensions/Public/Sherweb/Get-SherwebCurrentSubscription.ps1

@@ -11,6 +11,7 @@ function Get-SherwebCurrentSubscription {
         $CustomerId = Get-ExtensionMapping -Extension 'Sherweb' | Where-Object { $_.RowKey -eq $TenantFilter } | Select-Object -ExpandProperty IntegrationId
     }
 
+    Write-Information "Getting current subscription for $CustomerId"
     $AuthHeader = Get-SherwebAuthentication
     $Uri = "https://api.sherweb.com/service-provider/v1/billing/subscriptions/details?customerId=$CustomerId"
     $SubscriptionDetails = Invoke-RestMethod -Uri $Uri -Method GET -Headers $AuthHeader

Modules/CippExtensions/Public/Sherweb/Get-SherwebCustomerConfiguration.ps1

@@ -5,10 +5,8 @@ function Get-SherwebCustomerConfiguration {
         [string]$TenantFilter
     )
     if ($TenantFilter) {
-        Get-ExtensionMapping -Extension 'Sherweb' | Where-Object { $_.RowKey -eq $TenantFilter } | ForEach-Object {
-            Write-Host "Extracted customer id from tenant filter - It's $($_.IntegrationId)"
-            $CustomerId = $_.IntegrationId
-        }
+        $TenantFilter = (Get-Tenants -TenantFilter $TenantFilter).customerId
+        $CustomerId = Get-ExtensionMapping -Extension 'Sherweb' | Where-Object { $_.RowKey -eq $TenantFilter } | Select-Object -ExpandProperty IntegrationId
     }
     $AuthHeader = Get-SherwebAuthentication
     $Uri = "https://api.sherweb.com/service-provider/v1/customers/$($CustomerId)/platforms-configurations/"

Modules/CippExtensions/Public/Sherweb/Remove-SherwebSubscription.ps1

@@ -7,10 +7,8 @@ function Remove-SherwebSubscription {
         [string]$TenantFilter
     )
     if ($TenantFilter) {
-        Get-ExtensionMapping -Extension 'Sherweb' | Where-Object { $_.RowKey -eq $TenantFilter } | ForEach-Object {
-            Write-Host "Extracted customer id from tenant filter - It's $($_.IntegrationId)"
-            $CustomerId = $_.IntegrationId
-        }
+        $TenantFilter = (Get-Tenants -TenantFilter $TenantFilter).customerId
+        $CustomerId = Get-ExtensionMapping -Extension 'Sherweb' | Where-Object { $_.RowKey -eq $TenantFilter } | Select-Object -ExpandProperty IntegrationId
     }
     $AuthHeader = Get-SherwebAuthentication
     $Body = ConvertTo-Json -Depth 10 -InputObject @{

Modules/CippExtensions/Public/Sherweb/Set-SherwebSubscription.ps1

@@ -10,10 +10,8 @@ function Set-SherwebSubscription {
         [string]$TenantFilter
     )
     if ($TenantFilter) {
-        Get-ExtensionMapping -Extension 'Sherweb' | Where-Object { $_.RowKey -eq $TenantFilter } | ForEach-Object {
-            Write-Host "Extracted customer id from tenant filter - It's $($_.IntegrationId)"
-            $CustomerId = $_.IntegrationId
-        }
+        $TenantFilter = (Get-Tenants -TenantFilter $TenantFilter).customerId
+        $CustomerId = Get-ExtensionMapping -Extension 'Sherweb' | Where-Object { $_.RowKey -eq $TenantFilter } | Select-Object -ExpandProperty IntegrationId
     }
     $AuthHeader = Get-SherwebAuthentication
     $ExistingSubscription = Get-SherwebCurrentSubscription -CustomerId $CustomerId -SKU $SKU