Migrate Astro from generation to main (Azure#26643)

* Move Astro to main * Update ChangeLog.md --------- Co-authored-by: azure-powershell-bot <[email protected]> Co-authored-by: Vincent Dai <[email protected]>
azure-sdk · Nov 8, 2024 · c2b91f8 · c2b91f8
1 parent ba10fa1
commit c2b91f8
Show file tree

Hide file tree

Showing 24 changed files with 542 additions and 411 deletions.
diff --git a/src/Astro/Astro.Autorest/README.md b/src/Astro/Astro.Autorest/README.md
@@ -48,6 +48,7 @@ subject-prefix: $(service-name)
 # Disable transform IdentityType as GET+PUT can not replace patch
 # 1. Organizations_CreateOrUpdate can not update resources
 # 2. the input schemas of PUT and PATCH are different
+flatten-userassignedidentity: false
 disable-transform-identity-type: true
 
 directive:
@@ -74,4 +75,7 @@ directive:
   - where:
       verb: Update
     hide: true
+  - from: UserAssignedIdentities.dictionary.cs
+    where: $
+    transform: $ = $.replace('null != property.Key && null != property.Value', 'null != property.Key');
 ```
diff --git a/src/Astro/Astro.Autorest/build-module.ps1 b/src/Astro/Astro.Autorest/build-module.ps1
@@ -75,6 +75,8 @@ if(-not $NotIsolated -and -not $Debugger) {
 $binFolder = Join-Path $PSScriptRoot 'bin'
 $objFolder = Join-Path $PSScriptRoot 'obj'
 
+$isAzure = [System.Convert]::ToBoolean('true')
+
 if(-not $Debugger) {
   Write-Host -ForegroundColor Green 'Cleaning build folders...'
   $null = Remove-Item -Recurse -ErrorAction SilentlyContinue -Path $binFolder, $objFolder
@@ -151,7 +153,7 @@ if($NoDocs) {
     $null = Get-ChildItem -Path $docsFolder -Recurse -Exclude 'README.md' | Remove-Item -Recurse -ErrorAction SilentlyContinue
   }
   $null = New-Item -ItemType Directory -Force -Path $docsFolder
-  $addComplexInterfaceInfo = ![System.Convert]::ToBoolean('true')
+  $addComplexInterfaceInfo = !$isAzure
   Export-ProxyCmdlet -ModuleName $moduleName -ModulePath $modulePaths -ExportsFolder $exportsFolder -InternalFolder $internalFolder -ModuleDescription $moduleDescription -DocsFolder $docsFolder -ExamplesFolder $examplesFolder -ModuleGuid $guid -AddComplexInterfaceInfo:$addComplexInterfaceInfo
 }
 
@@ -186,5 +188,4 @@ if (-not $DisableAfterBuildTasks){
   }
 }
 
-
-Write-Host -ForegroundColor Green '-------------Done-------------'
+Write-Host -ForegroundColor Green '-------------Done-------------'
diff --git a/src/Astro/Astro.Autorest/custom/New-AzAstroOrganization.ps1 b/src/Astro/Astro.Autorest/custom/New-AzAstroOrganization.ps1
@@ -100,7 +100,7 @@ function New-AzAstroOrganization {
         # Last name of the user
         ${UserLastName},
 
-        [Parameter(ParameterSetName='CreateExpanded')]
+        [Parameter(ParameterSetName = 'CreateExpanded')]
         [Microsoft.Azure.PowerShell.Cmdlets.Astro.Category('Body')]
         [System.Management.Automation.SwitchParameter]
         # Decides if enable a system assigned identity for the resource.
@@ -282,7 +282,7 @@ function New-AzAstroOrganization {
     )
 
     process {
-        if($PSBoundParameters.ContainsKey('EnableSystemAssignedIdentity') -or $PSBoundParameters.ContainsKey('UserAssignedIdentity') ){
+        if ($PSBoundParameters.ContainsKey('EnableSystemAssignedIdentity') -or $PSBoundParameters.ContainsKey('UserAssignedIdentity')) {
             $supportsSystemAssignedIdentity = $PSBoundParameters.ContainsKey('EnableSystemAssignedIdentity')
             $supportsUserAssignedIdentity = $PSBoundParameters.ContainsKey("UserAssignedIdentity") -and $UserAssignedIdentity.Length -gt 0
 
@@ -299,9 +299,20 @@ function New-AzAstroOrganization {
             else {
                 $PSBoundParameters.Add("IdentityType", "None")
             }
-
+
+            # If user input UserAssignedIdentity
+            if ($PSBoundParameters.ContainsKey('UserAssignedIdentity')) {
+                $userIdentityObject = [Microsoft.Azure.PowerShell.Cmdlets.Astro.Models.UserAssignedIdentity]::New()
+                $PSBoundParameters.IdentityUserAssignedIdentity = @{}
+                foreach ($item in $PSBoundParameters.UserAssignedIdentity) {
+                    $PSBoundParameters.IdentityUserAssignedIdentity.Add($item, $userIdentityObject )
+                }
+
+                $null = $PSBoundParameters.Remove('UserAssignedIdentity')
+            }
+
             # remove EnableSystemAssignedIdentity
-            if($PSBoundParameters.ContainsKey('EnableSystemAssignedIdentity')) {
+            if ($PSBoundParameters.ContainsKey('EnableSystemAssignedIdentity')) {
                 $null = $PSBoundParameters.Remove("EnableSystemAssignedIdentity")
             }
         }