From ecaaf8c229512841d962ede5e6a10fcd01061ec0 Mon Sep 17 00:00:00 2001
From: Matt Bruce <matt.bruce@verizon.com>
Date: Tue, 14 Feb 2023 13:55:11 -0600
Subject: [PATCH] helper method for setNeedsLayout()

Signed-off-by: Matt Bruce <matt.bruce@verizon.com>
---
 VDS/Protocols/ViewProtocol.swift | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/VDS/Protocols/ViewProtocol.swift b/VDS/Protocols/ViewProtocol.swift
index 38ebcd96..d8acfb57 100644
--- a/VDS/Protocols/ViewProtocol.swift
+++ b/VDS/Protocols/ViewProtocol.swift
@@ -14,3 +14,11 @@ public protocol ViewProtocol {
     func setup()
 }
 
+extension ViewProtocol where Self: UIView {
+    public func removeFromSuperview(_ view: UIView){
+        if view.superview != nil {
+            view.removeFromSuperview()
+            setNeedsDisplay()
+        }
+    }
+}