@@ -43,22 +43,20 @@ $currentIdentityOid = $null
4343$currentIdentityDisplay = $null
4444$principalType = " User"
4545
46- # Try signed-in user first (single API call to get both id and userPrincipalName)
47- $userInfo = az ad signed- in- user show 2> $null | ConvertFrom-Json
48- if ($userInfo ) {
46+ # Try signed-in user first
47+ $currentIdentityOid = az ad signed- in- user show -- query " id " - o tsv 2> $null
48+ if ($currentIdentityOid ) {
4949 $identityType = " user"
50- $currentIdentityOid = $userInfo.id
51- $currentIdentityDisplay = $userInfo.userPrincipalName
50+ $currentIdentityDisplay = az ad signed- in- user show -- query " userPrincipalName" - o tsv 2> $null
5251 $principalType = " User"
5352 Write-Host " ✓ Detected identity type: User ($currentIdentityDisplay )"
5453} else {
55- # Fallback to service principal (single API call per resource)
54+ # Fallback to service principal
5655 $spAppId = az account show -- query " user.name" - o tsv 2> $null
5756 if ($spAppId -and $spAppId -ne " null" ) {
58- $spInfo = az ad sp show -- id $spAppId 2> $null | ConvertFrom-Json
59- if ($spInfo ) {
60- $currentIdentityOid = $spInfo.id
61- $currentIdentityDisplay = $spInfo.displayName
57+ $currentIdentityOid = az ad sp show -- id $spAppId -- query " id" - o tsv 2> $null
58+ $currentIdentityDisplay = az ad sp show -- id $spAppId -- query " displayName" - o tsv 2> $null
59+ if ($currentIdentityOid ) {
6260 $identityType = " servicePrincipal"
6361 $principalType = " ServicePrincipal"
6462 Write-Host " ✓ Detected identity type: Service Principal ($currentIdentityDisplay , OID: $currentIdentityOid )"
@@ -127,21 +125,21 @@ Write-Host ""
127125Write-Host " --- Step 1: Set Function App Client Key ---"
128126
129127# Discover function app
130- $functionAppName = az functionapp list -- resource- group $ResourceGroupName -- query " [0 ].name" - o tsv 2> $null
131- if (-not $functionAppName ) {
128+ $functionApps = az functionapp list -- resource- group $ResourceGroupName -- query " [].name" - o tsv 2> $null
129+ if (-not $functionApps ) {
132130 Write-Warning " ⚠ No function apps found in resource group '$ResourceGroupName '. Skipping function key setup."
133131}
134132else {
135- $functionAppName = $functionAppName .Trim ()
133+ $functionAppName = ( $functionApps -split " `n " )[ 0 ] .Trim()
136134 Write-Host " ✓ Discovered function app: $functionAppName "
137135
138136 # Discover key vault
139- $keyVaultName = az keyvault list -- resource- group $ResourceGroupName -- query " [0 ].name" - o tsv 2> $null
140- if (-not $keyVaultName ) {
137+ $keyVaults = az keyvault list -- resource- group $ResourceGroupName -- query " [].name" - o tsv 2> $null
138+ if (-not $keyVaults ) {
141139 Write-Warning " ⚠ No Key Vault found. Skipping function key setup."
142140 }
143141 else {
144- $keyVaultName = $keyVaultName .Trim ()
142+ $keyVaultName = ( $keyVaults -split " `n " )[ 0 ] .Trim()
145143 Write-Host " ✓ Discovered Key Vault: $keyVaultName "
146144
147145 # Ensure the current identity has 'Key Vault Secrets User' role on the Key Vault
@@ -228,12 +226,12 @@ else {
228226Write-Host " "
229227Write-Host " --- Step 2: Create PostgreSQL Tables ---"
230228
231- $serverFqdn = az postgres flexible- server list -- resource- group $ResourceGroupName -- query " [0 ].fullyQualifiedDomainName" - o tsv 2> $null
232- if (-not $serverFqdn ) {
229+ $pgServers = az postgres flexible- server list -- resource- group $ResourceGroupName -- query " [].fullyQualifiedDomainName" - o tsv 2> $null
230+ if (-not $pgServers ) {
233231 Write-Host " No PostgreSQL Flexible Server found in resource group. Skipping table creation."
234232}
235233else {
236- $serverFqdn = $serverFqdn .Trim ()
234+ $serverFqdn = ( $pgServers -split " `n " )[ 0 ] .Trim()
237235 $serverName = $serverFqdn.Split (' .' )[0 ]
238236 Write-Host " ✓ Discovered PostgreSQL server: $serverName ($serverFqdn )"
239237
0 commit comments