register mgmt sdk to mgmt core client

Author: chunyu3

eng/scripts/automation/GenerateAndBuildLib.ps1

@@ -177,6 +177,45 @@ function Update-CIYmlFile() {
     }
 }
 
+function RegisterMgmtSDKToMgmtCoreClient () {
+    param(
+        [string]$sdkRepoRootPath
+    )
+    $packagesPath = Join-Path $sdkRepoRootPath "sdk"
+    $track2MgmtDirs = Get-ChildItem -Path "$packagesPath" -Directory -Recurse -Depth 1 | Where-Object { $_.Name -match "(Azure.ResourceManager.)" -and $(Test-Path("$($_.FullName)/src")) }
+    Write-Host "Updating mgmt core client ci.mgmt.yml"
+    #add path for each mgmt library into Azure.ResourceManager
+    $armCiFile = "$sdkRepoRootPath/sdk/resourcemanager/ci.mgmt.yml"
+    $armLines = Get-Content $armCiFile
+    $newLines = [System.Collections.ArrayList]::new()
+    $startIndex = $track2MgmtDirs[0].FullName.IndexOf(("\sdk\")) + 1
+    $shouldRemove = $false
+    foreach($line in $armLines) {
+        if($line.StartsWith("  paths:")) {
+            $newLines.Add($line) | Out-Null
+            $newLines.Add("    include:") | Out-Null
+            $newLines.Add("    - sdk/resourcemanager") | Out-Null
+            $newLines.Add("    - common/ManagementTestShared") | Out-Null
+            $newLines.Add("    - common/ManagementCoreShared") | Out-Null
+            foreach($dir in $track2MgmtDirs) {
+                $newLine = "    - $($dir.FullName.Substring($startIndex, $dir.FullName.Length - $startIndex).Replace('\', '/'))"
+                $newLines.Add($newLine) | Out-Null
+            }
+            $shouldRemove = $true
+            Continue
+        }
+
+        if($shouldRemove) {
+            if($line.StartsWith(" ")) {
+                Continue
+            }
+            $shouldRemove = $false
+        }
+
+        $newLines.Add($line) | Out-Null
+    }
+    Set-Content -Path $armCiFile $newLines
+}
 <#
 .SYNOPSIS
 Prepare the SDK pacakge for data-plane.
@@ -334,6 +373,8 @@ function New-MgmtPackageFolder() {
       Push-Location $projectFolder
       dotnet new azuremgmt --provider $packageName --includeCI true --force
       Pop-Location
+
+      RegisterMgmtSDKToMgmtCoreClient -sdkRepoRootPath $sdkPath
     }
 
     # update the readme path.