fix: use model's RootNode instead of parameter for consistency in GetMapKeyNodeOrRoot methods

Author: TristanSpeakEasy

openapi/core/callbacks.go

@@ -25,9 +25,13 @@ func (c *Callback) GetMapKeyNodeOrRoot(key string, rootNode *yaml.Node) *yaml.No
 		return rootNode
 	}
 
-	for i := 0; i < len(rootNode.Content); i += 2 {
-		if rootNode.Content[i].Value == key {
-			return rootNode.Content[i]
+	if c.RootNode == nil {
+		return rootNode
+	}
+
+	for i := 0; i < len(c.RootNode.Content); i += 2 {
+		if c.RootNode.Content[i].Value == key {
+			return c.RootNode.Content[i]
 		}
 	}
 

openapi/core/paths.go

@@ -25,9 +25,13 @@ func (p *Paths) GetMapKeyNodeOrRoot(key string, rootNode *yaml.Node) *yaml.Node
 		return rootNode
 	}
 
-	for i := 0; i < len(rootNode.Content); i += 2 {
-		if rootNode.Content[i].Value == key {
-			return rootNode.Content[i]
+	if p.RootNode == nil {
+		return rootNode
+	}
+
+	for i := 0; i < len(p.RootNode.Content); i += 2 {
+		if p.RootNode.Content[i].Value == key {
+			return p.RootNode.Content[i]
 		}
 	}
 
@@ -66,9 +70,13 @@ func (p *PathItem) GetMapKeyNodeOrRoot(key string, rootNode *yaml.Node) *yaml.No
 		return rootNode
 	}
 
-	for i := 0; i < len(rootNode.Content); i += 2 {
-		if rootNode.Content[i].Value == key {
-			return rootNode.Content[i]
+	if p.RootNode == nil {
+		return rootNode
+	}
+
+	for i := 0; i < len(p.RootNode.Content); i += 2 {
+		if p.RootNode.Content[i].Value == key {
+			return p.RootNode.Content[i]
 		}
 	}
 

openapi/core/responses.go

@@ -26,9 +26,13 @@ func (r *Responses) GetMapKeyNodeOrRoot(key string, rootNode *yaml.Node) *yaml.N
 		return rootNode
 	}
 
-	for i := 0; i < len(rootNode.Content); i += 2 {
-		if rootNode.Content[i].Value == key {
-			return rootNode.Content[i]
+	if r.RootNode == nil {
+		return rootNode
+	}
+
+	for i := 0; i < len(r.RootNode.Content); i += 2 {
+		if r.RootNode.Content[i].Value == key {
+			return r.RootNode.Content[i]
 		}
 	}
 

openapi/core/security.go

@@ -37,9 +37,13 @@ func (s *SecurityRequirement) GetMapKeyNodeOrRoot(key string, rootNode *yaml.Nod
 		return rootNode
 	}
 
-	for i := 0; i < len(rootNode.Content); i += 2 {
-		if rootNode.Content[i].Value == key {
-			return rootNode.Content[i]
+	if s.RootNode == nil {
+		return rootNode
+	}
+
+	for i := 0; i < len(s.RootNode.Content); i += 2 {
+		if s.RootNode.Content[i].Value == key {
+			return s.RootNode.Content[i]
 		}
 	}