Skip to content

Commit 7661d7e

Browse files
author
BradleyBartlett
committed
make RegistrationName mandatory for Remove-AzsRegistration
1 parent 7471ab1 commit 7661d7e

File tree

1 file changed

+9
-49
lines changed

1 file changed

+9
-49
lines changed

Registration/RegisterWithAzure.psm1

Lines changed: 9 additions & 49 deletions
Original file line numberDiff line numberDiff line change
@@ -317,15 +317,15 @@ function Remove-AzsRegistration{
317317
[Parameter(Mandatory = $true)]
318318
[String] $PrivilegedEndpoint,
319319

320+
[Parameter(Mandatory = $true)]
321+
[String] $RegistrationName,
322+
320323
[Parameter(Mandatory = $false)]
321324
[String] $ResourceGroupName = 'azurestack',
322325

323326
[Parameter(Mandatory = $false)]
324327
[String] $ResourceGroupLocation = 'westcentralus',
325328

326-
[Parameter(Mandatory = $false)]
327-
[String] $RegistrationName,
328-
329329
[Parameter(Mandatory = $false)]
330330
[ValidateNotNullorEmpty()]
331331
[PSObject] $AzureContext = (Get-AzureRmContext)
@@ -351,56 +351,16 @@ function Remove-AzsRegistration{
351351
# Find registration resource in Azure
352352
Log-Output "Searching for registration resource in Azure..."
353353
$registrationResource = $null
354-
if ($RegistrationName)
354+
355+
$registrationResourceId = "/subscriptions/$($AzureContext.Subscription.SubscriptionId)/resourceGroups/$ResourceGroupName/providers/Microsoft.AzureStack/registrations/$registrationName"
356+
$registrationResource = Get-AzureRmResource -ResourceId $registrationResourceId -ErrorAction Ignore
357+
if ($registrationResource.Properties.cloudId -eq $stampInfo.CloudId)
355358
{
356-
$registrationResourceId = "/subscriptions/$($AzureContext.Subscription.SubscriptionId)/resourceGroups/$ResourceGroupName/providers/Microsoft.AzureStack/registrations/$registrationName"
357-
$registrationResource = Get-AzureRmResource -ResourceId $registrationResourceId -ErrorAction Ignore
358-
if ($registrationResource.Properties.cloudId -eq $stampInfo.CloudId)
359-
{
360-
Log-Output "Registration resource found: $($registrationResource.ResourceId)"
361-
}
362-
else
363-
{
364-
Log-Throw "The registration resource found does not correlate the current environment's Cloud-Id. `r`nEnvironment Cloud Id: $($stampinfo.CloudId) `r`nResource Cloud Id: $($registrationResource.Properties.cloudId)" -CallingFunction $($PSCmdlet.MyInvocation.MyCommand.Name)
365-
}
359+
Log-Output "Registration resource found: $($registrationResource.ResourceId)"
366360
}
367361
else
368362
{
369-
Log-Output "Parameter 'RegistrationName' not supplied. Searching through all registration resources under current context."
370-
try
371-
{
372-
Log-Output "Attempting to retrieve resources using command: 'Find-AzureRmResource -ResourceType Microsoft.AzureStack/registrations -ResourceGroupNameEquals $ResourceGroupName'"
373-
$registrationResources = Find-AzureRmResource -ResourceType Microsoft.AzureStack/registrations -ResourceGroupNameEquals $ResourceGroupName
374-
}
375-
catch
376-
{
377-
Log-Warning "Could not retrieve resources from Azure `r`n$($_)"
378-
}
379-
380-
if ($registrationResources.Count -eq 0)
381-
{
382-
try
383-
{
384-
Log-Output "Attempting to retrieve resources using command: 'Get-AzureRmResource -ResourceType microsoft.azurestack/registrations -ResourceGroupName $ResourceGroupName'"
385-
$registrationresources = Get-AzureRmResource -ResourceType microsoft.azurestack/registrations -ResourceGroupName $ResourceGroupName
386-
}
387-
catch
388-
{
389-
Log-Throw "Unable to retrieve registration resource(s) from Azure `r`n$($_)" -CallingFunction $($PSCmdlet.MyInvocation.MyCommand.Name)
390-
}
391-
}
392-
393-
Log-Output "Found $($registrationResources.Count) registration resources. Finding a matching CloudId may take some time."
394-
foreach ($resource in $registrationResources)
395-
{
396-
$resourceObject = Get-AzureRmResource -ResourceId "/subscriptions/$($AzureContext.Subscription.SubscriptionId)/resourceGroups/$ResourceGroupName/providers/Microsoft.AzureStack/registrations/$($resource.name)"
397-
$resourceCloudId = (($resourceObject.Properties.ToString()) | ConvertFrom-Json).cloudId
398-
if ($resourceCloudId -eq $stampInfo.CloudId)
399-
{
400-
$registrationResource = $resourceObject
401-
break
402-
}
403-
}
363+
Log-Throw "The registration resource found does not correlate the current environment's Cloud-Id. `r`nEnvironment Cloud Id: $($stampinfo.CloudId) `r`nResource Cloud Id: $($registrationResource.Properties.cloudId)" -CallingFunction $($PSCmdlet.MyInvocation.MyCommand.Name)
404364
}
405365

406366
if ($registrationResource)

0 commit comments

Comments
 (0)