Private/get-kubedata.ps1
|
function Get-KubeData { param ( [string]$SubscriptionId, [string]$ResourceGroup, [string]$ClusterName, [switch]$ExcludeNamespaces, [switch]$AKS, [switch]$UseAksRestApi ) # Check for kubectl if (-not (Get-Command kubectl -ErrorAction SilentlyContinue)) { Write-Host "❌ kubectl is not installed or not in PATH." -ForegroundColor Red return $false } # Check kubectl client version $kubectlVersionOutput = kubectl version --client 2>&1 if ($LASTEXITCODE -ne 0) { Write-Host "❌ kubectl is not functioning." -ForegroundColor Red Write-Host "🧾 Error: $kubectlVersionOutput" -ForegroundColor DarkGray return $false } # Check cluster connectivity with retry for AzureCLICredential error $kubectlClusterCheck = kubectl get namespaces --no-headers -o custom-columns=NAME:.metadata.name 2>&1 if ($LASTEXITCODE -ne 0 -and $kubectlClusterCheck -match "failed to get token: AzureCLICredential") { Write-Host "❗ kubectl failed with AzureCLICredential error." -ForegroundColor Yellow if (-not (Get-Command kubelogin -ErrorAction SilentlyContinue)) { Write-Host "❌ kubelogin is not installed. Cannot fix auth issue." -ForegroundColor Red return $false } try { Write-Host "🔄 Attempting to reconfigure kubeconfig with kubelogin..." -ForegroundColor Yellow $isContainer = Test-IsContainer $spnProvided = $env:AZURE_CLIENT_ID -and $env:AZURE_CLIENT_SECRET -and $env:AZURE_TENANT_ID if ($spnProvided) { # Use SPN credentials & kubelogin convert-kubeconfig -l spn Write-Host "✅ Kubeconfig reconfigured for SPN login." -ForegroundColor Green } elseif (-not $isContainer) { # Local run: Try Azure CLI credentials & kubelogin convert-kubeconfig -l azurecli Write-Host "✅ Kubeconfig reconfigured for Azure CLI login." -ForegroundColor Green } else { throw "No SPN credentials provided in container" } # Retry kubectl $kubectlClusterCheck = kubectl get namespaces --no-headers -o custom-columns=NAME:.metadata.name 2>&1 if ($LASTEXITCODE -ne 0 -or -not $kubectlClusterCheck) { throw "Retry failed: $kubectlClusterCheck" } } catch { Write-Host "❌ Failed to reconfigure kubeconfig: $_" -ForegroundColor Red return $false } } if ($LASTEXITCODE -ne 0 -or -not $kubectlClusterCheck) { Write-Host "❌ kubectl cannot connect to the cluster." -ForegroundColor Red Write-Host "🧾 Error: $kubectlClusterCheck" -ForegroundColor DarkGray return $false } Write-Host "✅ kubectl is available and connected to the cluster." -ForegroundColor Green $data = @{} $resources = @( @{ Name = "Pods"; Cmd = { kubectl get pods --all-namespaces -o json }; Key = "Pods"; Items = $false }, @{ Name = "Nodes"; Cmd = { kubectl get nodes -o json }; Key = "Nodes"; Items = $false }, @{ Name = "Top Nodes"; Cmd = { kubectl top nodes --no-headers }; Key = "TopNodes"; Raw = $true }, @{ Name = "Namespaces"; Cmd = { kubectl get namespaces -o json }; Key = "Namespaces"; Items = $true }, @{ Name = "Events"; Cmd = { kubectl get events --all-namespaces -o json }; Key = "Events"; Items = $true }, @{ Name = "Jobs"; Cmd = { kubectl get jobs --all-namespaces -o json }; Key = "Jobs"; Items = $true }, @{ Name = "DaemonSets"; Cmd = { kubectl get daemonsets --all-namespaces -o json }; Key = "DaemonSets"; Items = $true }, @{ Name = "StatefulSets"; Cmd = { kubectl get statefulsets --all-namespaces -o json }; Key = "StatefulSets"; Items = $true }, @{ Name = "Deployments"; Cmd = { kubectl get deployments --all-namespaces -o json }; Key = "Deployments"; Items = $true }, @{ Name = "PodDisruptionBudgets"; Cmd = { kubectl get pdb --all-namespaces -o json }; Key = "PodDisruptionBudgets"; Items = $true }, @{ Name = "HorizontalPodAutoscalers"; Cmd = { kubectl get hpa --all-namespaces -o json }; Key = "HorizontalPodAutoscalers"; Items = $true }, @{ Name = "Services"; Cmd = { kubectl get svc --all-namespaces -o json }; Key = "Services"; Items = $false }, @{ Name = "Ingresses"; Cmd = { kubectl get ingress --all-namespaces -o json }; Key = "Ingresses"; Items = $true }, @{ Name = "Endpoints"; Cmd = { kubectl get endpoints --all-namespaces -o json }; Key = "Endpoints"; Items = $false }, @{ Name = "ConfigMaps"; Cmd = { kubectl get configmaps --all-namespaces -o json }; Key = "ConfigMaps"; Items = $true }, @{ Name = "Secrets"; Cmd = { kubectl get secrets --all-namespaces -o json }; Key = "Secrets"; Items = $true }, @{ Name = "PersistentVolumes"; Cmd = { kubectl get pv -o json }; Key = "PersistentVolumes"; Items = $true }, @{ Name = "PersistentVolumeClaims"; Cmd = { kubectl get pvc --all-namespaces -o json }; Key = "PersistentVolumeClaims"; Items = $true }, @{ Name = "NetworkPolicies"; Cmd = { kubectl get networkpolicies --all-namespaces -o json }; Key = "NetworkPolicies"; Items = $true }, @{ Name = "Roles"; Cmd = { kubectl get roles --all-namespaces -o json }; Key = "Roles"; Items = $true }, @{ Name = "RoleBindings"; Cmd = { kubectl get rolebindings --all-namespaces -o json }; Key = "RoleBindings"; Items = $true }, @{ Name = "ClusterRoles"; Cmd = { kubectl get clusterroles -o json }; Key = "ClusterRoles"; Items = $true }, @{ Name = "ClusterRoleBindings"; Cmd = { kubectl get clusterrolebindings -o json }; Key = "ClusterRoleBindings"; Items = $true }, @{ Name = "ServiceAccounts"; Cmd = { kubectl get serviceaccounts --all-namespaces -o json }; Key = "ServiceAccounts"; Items = $true } ) Write-Host "`n[📦 Gathering Kubernetes Resource Data]" -ForegroundColor Cyan $totalResources = $resources.Count $results = $resources | ForEach-Object -Parallel { Import-Module Microsoft.PowerShell.Utility -Cmdlet Write-Host, ConvertFrom-Json $res = $_ $output = [PSCustomObject]@{ Key = $res.Key Label = $res.Name Raw = $res.Raw Value = $null Success = $true Error = $null } Write-Host "▶️ Starting $($res.Name)" -ForegroundColor Yellow try { $result = & $res.Cmd if ($null -ne $result) { if ($res.Raw) { $output.Value = @($result -split "`n" | Where-Object { $_ }) } else { $jsonResult = $result | ConvertFrom-Json $output.Value = if ($res.Items) { $jsonResult.items } else { $jsonResult } } } else { $output.Value = @() } } catch { $output.Success = $false $output.Error = $_.Exception.Message } Write-Host "✔️ Finished $($res.Name)" -ForegroundColor Cyan return $output } -ThrottleLimit 8 # Show progress based on completed results $completed = $results.Count $percentComplete = ([int]($completed / $totalResources * 100)) Write-Progress -Activity "Gathering Kubernetes Resources" ` -Status "$completed/$totalResources ($percentComplete%)" ` -PercentComplete $percentComplete Write-Progress -Activity "Gathering Kubernetes Resources" -Completed Write-Host "`n[📋 Results]" -ForegroundColor Cyan foreach ($r in $results) { if ($r.Success) { Write-Host "✅ $($r.Label)" -ForegroundColor Green $data[$r.Key] = $r.Value } else { Write-Host "❌ $($r.Label): $($r.Error)" -ForegroundColor Red Write-Host "Critical error: Stopping execution due to failure in $($r.Label) - $($r.Error)" -ForegroundColor DarkGray return $false # return error flag return } } # Custom Resources Write-Host -NoNewline "`n🤖 Fetching Custom Resource Instances..." -ForegroundColor Yellow $data.CustomResourcesByKind = @{} try { $crds = kubectl get crds -o json | ConvertFrom-Json foreach ($crd in $crds.items) { $kind = $crd.spec.names.kind $plural = $crd.spec.names.plural $group = $crd.spec.group $version = ($crd.spec.versions | Where-Object { $_.served -and $_.storage } | Select-Object -First 1).name if (-not $version) { $version = $crd.spec.versions[0].name } $apiVersion = "$group/$version" try { $items = kubectl get $plural --all-namespaces -o json --api-version=$apiVersion 2>$null | ConvertFrom-Json $data.CustomResourcesByKind[$kind] = $items.items } catch {} } Write-Host "`r✅ Custom Resource Instances fetched. " -ForegroundColor Green } catch { Write-Host "`r❌ Failed to fetch CRDs: $($_.Exception.Message)" -ForegroundColor Red return $false } # AKS Metadata if ($AKS -and $ResourceGroup -and $ClusterName -and $SubscriptionId) { Write-Host -NoNewline "`n🤖 Fetching AKS Metadata..." -ForegroundColor Yellow try { if (-not $ClusterName -or $ClusterName -match "^\s*$" -or $ClusterName -match "[^\w-]" -or $ClusterName.Length -gt 63) { Write-Host "`r❌ ClusterName is missing, empty, or invalid (alphanumeric, hyphens only, max 63 chars)." -ForegroundColor Red return $false } $isContainer = Test-IsContainer $spnProvided = $env:AZURE_CLIENT_ID -and $env:AZURE_CLIENT_SECRET -and $env:AZURE_TENANT_ID $apiVersion = "2025-01-01" $uri = "https://management.azure.com/subscriptions/$SubscriptionId/resourceGroups/$ResourceGroup/providers/Microsoft.ContainerService/managedClusters/${ClusterName}?api-version=${apiVersion}" try { if ($spnProvided) { Write-Host "`r🔐 Using SPN credentials to acquire token..." -ForegroundColor Yellow $tokenUrl = "https://login.microsoftonline.com/$($env:AZURE_TENANT_ID)/oauth2/token" $headers = @{ "Content-Type" = "application/x-www-form-urlencoded" } $body = @{ grant_type = "client_credentials" client_id = $env:AZURE_CLIENT_ID client_secret = $env:AZURE_CLIENT_SECRET resource = "https://management.azure.com/" } $tokenResponse = Invoke-RestMethod -Uri $tokenUrl -Method POST -Headers $headers -Body $body -UseBasicParsing $accessToken = $tokenResponse.access_token } else { if (-not (Get-Command az -ErrorAction SilentlyContinue)) { Write-Host "`r❌ Azure CLI not found, and SPN not provided." -ForegroundColor Red return $false } Write-Host "`r🔐 Using Azure CLI to get user token..." -ForegroundColor Yellow $accessToken = az account get-access-token --resource https://management.azure.com/ --query accessToken -o tsv } $headers = @{ Authorization = "Bearer $accessToken" } $data.AksCluster = Invoke-RestMethod -Uri $uri -Headers $headers -UseBasicParsing Write-Host "`r✅ AKS Metadata fetched via REST API." -ForegroundColor Green } catch { Write-Host "`r❌ Failed to fetch AKS Metadata: $($_.Exception.Message)" -ForegroundColor Red return $false } Write-Host -NoNewline "`n🤖 Fetching Constraints..." -ForegroundColor Yellow try { $data.ConstraintTemplates = @( (kubectl get constrainttemplates -o json --ignore-not-found | ConvertFrom-Json -ErrorAction SilentlyContinue).items ?? @() ) $data.Constraints = @( (kubectl get constraints -A -o json --ignore-not-found | ConvertFrom-Json -ErrorAction SilentlyContinue).items ?? @() ) Write-Host "`r✅ Constraints fetched. " -ForegroundColor Green } catch { Write-Host "`r⚠️ Constraints not found, continuing with empty data: $($_.Exception.Message)" -ForegroundColor Yellow $data.ConstraintTemplates = @() $data.Constraints = @() } } catch { Write-Host "`r❌ Unexpected error during AKS metadata fetch: $($_.Exception.Message)" -ForegroundColor Red return $false } } else { Write-Host "`r❌ Required parameters missing. Set AKS, ResourceGroup, ClusterName, and SubscriptionId." -ForegroundColor Red return $false } # Namespace filtering if ($ExcludeNamespaces) { Write-Host "`n🤖 🚫 Excluding selected namespaces..." -ForegroundColor Yellow foreach ($key in @( 'Pods', 'Jobs', 'Deployments', 'Services', 'DaemonSets', 'StatefulSets', 'PersistentVolumeClaims', 'Events', 'NetworkPolicies', 'Roles', 'RoleBindings', 'Constraints' )) { if ($data.ContainsKey($key)) { $data[$key] = Exclude-Namespaces -items $data[$key] } } } return $data } |