Add descendantNodeRepresentingObject to Node.swift.

Frameworks/RSTree/RSTree/Node.swift

@@ -114,17 +114,13 @@ public final class Node: Hashable {
 	}
 	
 	public func childNodeRepresentingObject(_ obj: AnyObject) -> Node? {
-		
+
-		guard let childNodes = childNodes else {
+		return findNodeRepresentingObject(obj, recursively: false)
-			return nil
+	}
-		}
+
-		
+	public func descendantNodeRepresentingObject(_ obj: AnyObject) -> Node? {
-		for oneNode in childNodes {
+
-			if oneNode.representedObject === obj {
+		return findNodeRepresentingObject(obj, recursively: true)
-				return oneNode
-			}
-		}
-		return nil
 	}
 
 	public func hasAncestor(in nodes: [Node]) -> Bool {
@@ -198,3 +194,23 @@ public extension Array where Element == Node {
 		return self.map{ $0.representedObject }
 	}
 }
+
+private extension Node {
+
+	func findNodeRepresentingObject(_ obj: AnyObject, recursively: Bool = false) -> Node? {
+
+		guard let childNodes = childNodes else {
+			return nil
+		}
+		for childNode in childNodes {
+			if childNode.representedObject === obj {
+				return childNode
+			}
+			if recursively, let foundNode = childNode.descendantNodeRepresentingObject(obj) {
+				return foundNode
+			}
+		}
+
+		return nil
+	}
+}