Fixed Swift ObjC Wrapper of layout (#212)

There could be an issue if you inherit from one of the provided Objective C layouts from LayoutKit. This fix makes it so that if you do that, it will call the correct overridden methods.

Author: drumnkyle

Sources/ObjCSupport/LOKLayout.swift

@@ -20,6 +20,35 @@ import CoreGraphics
 
 extension LOKLayout {
     var unwrapped: Layout {
-        return (self as? WrappedLayout)?.layout ?? (self as? LOKBaseLayout)?.layout ?? ReverseWrappedLayout(layout: self)
+        /*
+         Need to check to see if `self` is one of the LayoutKit provided layouts.
+         If so, we want to cast it as `LOKBaseLayout` and return the wrapped layout
+         object directly.
+
+         If `self` is not one of the LayoutKit provided layouts, we want to wrap it
+         so that the methods of the class are called. We do this to make sure that if
+         someone were to subclass one of the LayoutKit provided layouts, we would want
+         to call their overriden methods instead of just the underlying layout object directly.
+         */
+        let allLayoutClasses = [
+            LOKButtonLayout.self,
+            LOKInsetLayout.self,
+            LOKLabelLayout.self,
+            LOKOverlayLayout.self,
+            LOKSizeLayout.self,
+            LOKStackLayout.self,
+            LOKTextViewLayout.self
+        ]
+        if let object = self as? NSObject {
+            if allLayoutClasses.contains(where: { object.isMember(of: $0) }) {
+                // Executes if `self` is one of the LayoutKit provided classes; not if it's a subclass
+                guard let layout = (self as? LOKBaseLayout)?.layout else {
+                    assertionFailure("LayoutKit provided layout does not inherit from LOKBaseLayout")
+                    return ReverseWrappedLayout(layout: self)
+                }
+                return layout
+            }
+        }
+        return (self as? WrappedLayout)?.layout ?? ReverseWrappedLayout(layout: self)
     }
 }