From 48388343ef06794a1b9a494135d7d1c0f7d1bc41 Mon Sep 17 00:00:00 2001
From: Scott Pfeil <Scott.Pfeil3@verizonwireless.com>
Date: Mon, 2 May 2022 17:51:08 -0400
Subject: [PATCH] Move function for scoping purposes

---
 .../PageBehaviorHandlerModelProtocol.swift       | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/MVMCoreUI/Behaviors/Protocols/PageBehaviorHandlerModelProtocol.swift b/MVMCoreUI/Behaviors/Protocols/PageBehaviorHandlerModelProtocol.swift
index d9b8da9f..bb752694 100644
--- a/MVMCoreUI/Behaviors/Protocols/PageBehaviorHandlerModelProtocol.swift
+++ b/MVMCoreUI/Behaviors/Protocols/PageBehaviorHandlerModelProtocol.swift
@@ -23,3 +23,19 @@ public extension PageBehaviorHandlerModelProtocol {
         self.behaviors = newBehaviors
     }
 }
+
+public extension PageBehaviorHandlerModelProtocol where Self: MoleculeTreeTraversalProtocol {
+    
+    /// Traverses all models and adds any required behavior models.
+    mutating func traverseAndAddRequiredBehaviors() {
+        let behaviorModels: [PageBehaviorModelProtocol] = reduceDepthFirstTraverse(options: .childFirst, depth: 0, initialResult: []) { (accumulator, molecule, depth) in
+            if let behaviorRequirer = molecule as? PageBehaviorProtocolRequirer {
+                return accumulator + behaviorRequirer.getRequiredBehaviors()
+            }
+            return accumulator
+        }
+        for behavior in behaviorModels {
+            add(behavior: behavior)
+        }
+    }
+}