fix load img

takecian · May 29, 2017 · 072a62a · 072a62a
1 parent 79b7f83
commit 072a62a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/SwiftShareBubbles/ShareAttirbute.swift b/SwiftShareBubbles/ShareAttirbute.swift
@@ -68,7 +68,12 @@ extension ShareAttirbute {
     private static func createButton(bubbleId: Int, imageName: String, color: UIColor) -> ShareAttirbute? {
         let classBundle = Bundle(for: SwiftShareBubbles.self)
         guard let bundle = Bundle(path: "\(classBundle.bundlePath)/SwiftShareBubbles.bundle") else { return nil }
-        guard let icon = UIImage(named: "SwiftShareBubbles.bundle/\(imageName)", in: bundle, compatibleWith: nil) else { return nil }
-        return ShareAttirbute(bubbleId: bubbleId, icon: icon, backgroundColor: color)
+        if let icon = UIImage(named: "SwiftShareBubbles.bundle/\(imageName)", in: bundle, compatibleWith: nil) {
+            return ShareAttirbute(bubbleId: bubbleId, icon: icon, backgroundColor: color)
+        } else if let icon = UIImage(named: imageName, in: bundle, compatibleWith: nil) {
+            return ShareAttirbute(bubbleId: bubbleId, icon: icon, backgroundColor: color)
+        } else{
+            return nil
+        }
     }
 }