From 677c352b9eae42227b3722b08477d868ba4b9014 Mon Sep 17 00:00:00 2001
From: Scott Pfeil <Scott.Pfeil3@verizonwireless.com>
Date: Mon, 2 May 2022 17:54:37 -0400
Subject: [PATCH] convenience functions

---
 .../Behaviors/Protocols/PageBehaviorHandlerProtocol.swift    | 2 +-
 MVMCoreUI/Behaviors/Protocols/PageBehaviorProtocol.swift     | 5 ++++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/MVMCoreUI/Behaviors/Protocols/PageBehaviorHandlerProtocol.swift b/MVMCoreUI/Behaviors/Protocols/PageBehaviorHandlerProtocol.swift
index cdabba87..79f3364f 100644
--- a/MVMCoreUI/Behaviors/Protocols/PageBehaviorHandlerProtocol.swift
+++ b/MVMCoreUI/Behaviors/Protocols/PageBehaviorHandlerProtocol.swift
@@ -36,7 +36,7 @@ public extension PageBehaviorHandlerProtocol {
     }
     
     /// Executes all behaviors of type.
-    public func executeBehaviors<T>(_ behaviorBlock: (_ behavior: T) -> Void) {
+    func executeBehaviors<T>(_ behaviorBlock: (_ behavior: T) -> Void) {
         behaviors?.compactMap { $0 as? T }.forEach { behaviorBlock($0) }
     }
 }
diff --git a/MVMCoreUI/Behaviors/Protocols/PageBehaviorProtocol.swift b/MVMCoreUI/Behaviors/Protocols/PageBehaviorProtocol.swift
index 3f14f3cc..76bc7f84 100644
--- a/MVMCoreUI/Behaviors/Protocols/PageBehaviorProtocol.swift
+++ b/MVMCoreUI/Behaviors/Protocols/PageBehaviorProtocol.swift
@@ -50,8 +50,11 @@ public protocol PageCustomActionHandlerBehavior: PageBehaviorProtocol {
 }
 
 public extension MVMCoreUIDelegateObject {
+    var behaviorModelDelegate: PageBehaviorHandlerModelProtocol? {
+        (moleculeDelegate as? PageProtocol)?.pageModel as? PageBehaviorHandlerModelProtocol
+    }
     
     weak var behaviorTemplateDelegate: (PageBehaviorHandlerProtocol & NSObjectProtocol)? {
-        (moleculeDelegate as? PageProtocol)?.pageModel as? (PageBehaviorHandlerProtocol & NSObjectProtocol)
+        moleculeDelegate as? (PageBehaviorHandlerProtocol & NSObjectProtocol)
     }
 }