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() + } + } +}