@@ -19,6 +19,7 @@ package tree
19
19
import (
20
20
"context"
21
21
22
+ "github.com/pkg/errors"
22
23
corev1 "k8s.io/api/core/v1"
23
24
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
24
25
unstructured "k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
@@ -83,9 +84,11 @@ func Discovery(ctx context.Context, c client.Client, namespace, name string, opt
83
84
tree := NewObjectTree (cluster , options .toObjectTreeOptions ())
84
85
85
86
// Adds cluster infra
86
- if clusterInfra , err := external .Get (ctx , c , cluster .Spec .InfrastructureRef , cluster .Namespace ); err == nil {
87
- tree .Add (cluster , clusterInfra , ObjectMetaName ("ClusterInfrastructure" ))
87
+ clusterInfra , err := external .Get (ctx , c , cluster .Spec .InfrastructureRef , cluster .Namespace )
88
+ if err != nil {
89
+ return nil , errors .Wrap (err , "get InfraCluster reference from Cluster" )
88
90
}
91
+ tree .Add (cluster , clusterInfra , ObjectMetaName ("ClusterInfrastructure" ))
89
92
90
93
if options .ShowClusterResourceSets {
91
94
addClusterResourceSetsToObjectTree (ctx , c , cluster , tree )
@@ -119,9 +122,11 @@ func Discovery(ctx context.Context, c client.Client, namespace, name string, opt
119
122
}
120
123
121
124
controlPlaneMachines := selectControlPlaneMachines (machinesList )
122
- for i := range controlPlaneMachines {
123
- cp := controlPlaneMachines [i ]
124
- addMachineFunc (controlPlane , cp )
125
+ if controlPlane != nil {
126
+ for i := range controlPlaneMachines {
127
+ cp := controlPlaneMachines [i ]
128
+ addMachineFunc (controlPlane , cp )
129
+ }
125
130
}
126
131
127
132
machinePoolList , err := getMachinePoolsInCluster (ctx , c , cluster .Namespace , cluster .Name )
0 commit comments