iter: Better reading order

Author: earthboundkid

html/iter.go

@@ -8,9 +8,22 @@ package html
 
 import "iter"
 
+// Parents returns an sequence yielding the node and its parents.
+//
+// Mutating a Node or its parents while iterating may have unexpected results.
+func (n *Node) Parents() iter.Seq[*Node] {
+	return func(yield func(*Node) bool) {
+		for p := n; p != nil; p = p.Parent {
+			if !yield(p) {
+				return
+			}
+		}
+	}
+}
+
 // ChildNodes returns a sequence yielding the immediate children of n.
 //
-// Mutating a Node while iterating through its ChildNodes may have unexpected results.
+// Mutating a Node or its ChildNodes while iterating may have unexpected results.
 func (n *Node) ChildNodes() iter.Seq[*Node] {
 	return func(yield func(*Node) bool) {
 		if n == nil {
@@ -25,22 +38,9 @@ func (n *Node) ChildNodes() iter.Seq[*Node] {
 
 }
 
-func (n *Node) all(yield func(*Node) bool) bool {
-	if !yield(n) {
-		return false
-	}
-
-	for c := range n.ChildNodes() {
-		if !c.all(yield) {
-			return false
-		}
-	}
-	return true
-}
-
 // All returns a sequence yielding all descendents of n in depth-first pre-order.
 //
-// Mutating a Node while iterating through it or its descendents may have unexpected results.
+// Mutating a Node or its descendents while iterating may have unexpected results.
 func (n *Node) All() iter.Seq[*Node] {
 	return func(yield func(*Node) bool) {
 		if n == nil {
@@ -50,15 +50,15 @@ func (n *Node) All() iter.Seq[*Node] {
 	}
 }
 
-// Parents returns an sequence yielding the node and its parents.
-//
-// Mutating a Node while iterating through it or its parents may have unexpected results.
-func (n *Node) Parents() iter.Seq[*Node] {
-	return func(yield func(*Node) bool) {
-		for p := n; p != nil; p = p.Parent {
-			if !yield(p) {
-				return
-			}
+func (n *Node) all(yield func(*Node) bool) bool {
+	if !yield(n) {
+		return false
+	}
+
+	for c := range n.ChildNodes() {
+		if !c.all(yield) {
+			return false
 		}
 	}
+	return true
 }