Merge pull request #161 from heoblitz/main

Add UIKit component support
YusukeHosonuma · Sep 29, 2021 · f4d9fa5 · f4d9fa5
2 parents 67d448d + b89b9e3
commit f4d9fa5
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/Sources/Core/PrettyDescriber.swift b/Sources/Core/PrettyDescriber.swift
@@ -6,7 +6,13 @@
 // Copyright (c) 2020 Yusuke Hosonuma.
 //
 
-import Foundation
+#if canImport(UIKit)
+    import UIKit
+    typealias UIComponent = UIResponder
+#else
+    import Foundation
+    typealias UIComponent = AnyClass
+#endif
 
 struct PrettyDescriber {
     var formatter: PrettyFormatter
@@ -175,15 +181,25 @@ struct PrettyDescriber {
                     return value.debugDescription
 
                 case let value as CustomStringConvertible:
-                    return value.description
+                    if #available(iOS 10.0, tvOS 10.0, *),
+                        let _ = value as? UIComponent {
+                        return nil
+                    } else {
+                        return value.description
+                    }
 
                 default:
                     return nil
                 }
             } else {
                 switch target {
                 case let value as CustomStringConvertible:
-                    return value.description
+                    if #available(iOS 10.0, tvOS 10.0, *),
+                        let _ = value as? UIComponent {
+                        return nil
+                    } else {
+                        return value.description
+                    }
 
                 case let value as CustomDebugStringConvertible:
                     return value.debugDescription