Subscribing Gateway and Firewall Offers to Private Azure Marketplace for Aviatrix Deployments

# Provide Azure Private Marketplace Collection Name. By default, a collection is scoped to all subscriptions
$CollectionName="Default Collection"
if ($CollectionName -eq "Default Collection") {$CollectionName="DefaultCollection"}

# Install the Az.Marketplace module
Install-Module -Name Az.Marketplace -Confirm:$false -Force

# Get the Private Store ID
$PrivateStore = Get-AzMarketplacePrivateStore
if ($PrivateStore.Availability -ne "enabled") {
    throw "Azure Private Marketplace is not enabled. Follow https://learn.microsoft.com/en-us/marketplace/create-manage-private-azure-marketplace-new#enabledisable-private-azure-marketplace to enable it"
}
$PrivateStoreId = $PrivateStore.PrivateStoreId

# Get the Collection ID
$Collection = Get-AzMarketplacePrivateStoreCollection -PrivateStoreId $PrivateStoreId | Where-Object {$_.CollectionName -eq $CollectionName}
if (!($Collection.Enabled)) {
    throw "Azure Private Marketplace Collection: $CollectionName is not enabled. Follow https://learn.microsoft.com/en-us/marketplace/create-manage-private-azure-marketplace-new#enabledisable-a-collection to enable it"
}
$CollectionId = $Collection.CollectionId

# Add the offer to the private Azure Marketplace
$Params = @{
    privateStoreId = $PrivateStoreId
    collectionId = $CollectionId
    offerId = "aviatrix-systems.aviatrix-gateway"
    SpecificPlanIdLimitation = @("aviatrix-gateway-g3")
}

# Set the offer to the private Marketplace
Set-AzMarketplacePrivateStoreCollectionOffer @Params

# Replace <offer_id> and <specific_plan_id_limitation> (or sku) with appropriate values
$OfferId = "aviatrix-systems.aviatrix-gateway"
$SpecificPlanIdLimitation = "aviatrix-gateway-g3" # Example for 8.0 G3 Gateway

$CollectionName="Default Collection"
if ($CollectionName -eq "Default Collection") {$CollectionName="DefaultCollection"}

Install-Module -Name Az.Marketplace -Confirm:$false -Force

$PrivateStore = Get-AzMarketplacePrivateStore
if ($PrivateStore.Availability -ne "enabled") {
    throw "Azure Private Marketplace is not enabled. Follow https://learn.microsoft.com/en-us/marketplace/create-manage-private-azure-marketplace-new#enabledisable-private-azure-marketplace to enable it"
}
$PrivateStoreId = $PrivateStore.PrivateStoreId

$Collection = Get-AzMarketplacePrivateStoreCollection -PrivateStoreId $PrivateStoreId | Where-Object {$_.CollectionName -eq $CollectionName}
if (!($Collection.Enabled)) {
    throw "Azure Private Marketplace Collection: $CollectionName is not enabled. Follow https://learn.microsoft.com/en-us/marketplace/create-manage-private-azure-marketplace-new#enabledisable-a-collection to enable it"
}
$CollectionId = $Collection.CollectionId

$Params = @{
    privateStoreId = $PrivateStoreId
    collectionId = $CollectionId
    offerId = $OfferId
    SpecificPlanIdLimitation = @($SpecificPlanIdLimitation)
}

Set-AzMarketplacePrivateStoreCollectionOffer @Params

$PublisherId = "aviatrix-systems"
$OfferId = "aviatrix-gateway"
$PlanId = "aviatrix-gateway-g3"
Set-AzMarketplaceTerms -Publisher $PublisherId -Product $OfferId -Name $PlanId -Accept

az vm image terms accept --publisher aviatrix-systems --offer aviatrix-gateway --plan aviatrix-gateway-g3