trail 2

Author: sameersyd

MotionToastView.podspec

@@ -8,7 +8,7 @@
 
 Pod::Spec.new do |s|
   s.name             = 'MotionToastView'
-  s.version          = '0.1.5'
+  s.version          = '0.1.6'
   s.summary          = 'A Beautiful Toast Library for iOS Swift'
 
 # This description is used to generate tags and improve search results.

Source/MotionToastStyles/MTPale.swift

@@ -51,32 +51,43 @@ class MTPale: UIView {
         switch toastType {
             case .success:
                 headLabel.text = "Success"
-                circleImg.image = UIImage(named: "success_icon_white", in: bundle, compatibleWith: nil)!
+                circleImg.image = loadImage(name: "success_icon_white")
                 sideBarView.backgroundColor = UIColor(hex: "6FCF97")
                 circleView.backgroundColor = UIColor(hex: "6FCF97")
                 toastView.backgroundColor = UIColor(named: "alpha_green_dark", in: bundle, compatibleWith: nil)!
                 break
             case .error:
                 headLabel.text = "Error"
-                circleImg.image = UIImage(named: "error_icon_white", in: bundle, compatibleWith: nil)!
+                circleImg.image = loadImage(name: "error_icon_white")
                 sideBarView.backgroundColor = UIColor(hex: "EB5757")
                 circleView.backgroundColor = UIColor(hex: "EB5757")
                 toastView.backgroundColor = UIColor(named: "alpha_red_dark", in: bundle, compatibleWith: nil)!
                 break
             case .warning:
                 headLabel.text = "Warning"
-                circleImg.image = UIImage(named: "warning_icon_white", in: bundle, compatibleWith: nil)!
+                circleImg.image = loadImage(name: "warning_icon_white")
                 sideBarView.backgroundColor = UIColor(hex: "F2C94C")
                 circleView.backgroundColor = UIColor(hex: "F2C94C")
                 toastView.backgroundColor = UIColor(named: "alpha_yellow_dark", in: bundle, compatibleWith: nil)!
                 break
             case .info:
                 headLabel.text = "Info"
-                circleImg.image = UIImage(named: "info_icon_white", in: bundle, compatibleWith: nil)!
+                circleImg.image = loadImage(name: "info_icon_white")
                 sideBarView.backgroundColor = UIColor(hex: "2F80ED")
                 circleView.backgroundColor = UIColor(hex: "2F80ED")
                 toastView.backgroundColor = UIColor(named: "alpha_blue_dark", in: bundle, compatibleWith: nil)!
                 break
         }
     }
+    
+    func loadImage(name: String) -> UIImage? {
+        let podBundle = Bundle(for: MTPale.self)
+        if let url = podBundle.url(forResource: "MotionToastView", withExtension: "bundle") {
+            let bundle = Bundle(url: url)
+            return UIImage(named: name, in: bundle, compatibleWith: nil)
+        }
+        print("MotionToastView")
+        return nil
+    }
+    
 }

Source/MotionToastStyles/MTVibrant.swift

@@ -49,27 +49,37 @@ class MTVibrant: UIView {
             case .success:
                 headLabel.text = "Success"
                 headLabel.textColor = UIColor(named: "white_green", in: bundle, compatibleWith: nil)!
-                circleImg.image = UIImage(named: "success_icon", in: bundle, compatibleWith: nil)!
+                circleImg.image = loadImage(name: "success_icon")
                 toastView.backgroundColor = UIColor(named: "green_dark", in: bundle, compatibleWith: nil)!
                 break
             case .error:
                 headLabel.text = "Error"
                 headLabel.textColor = UIColor(named: "white_red", in: bundle, compatibleWith: nil)!
-                circleImg.image = UIImage(named: "error_icon", in: bundle, compatibleWith: nil)!
+                circleImg.image = loadImage(name: "error_icon")
                 toastView.backgroundColor = UIColor(named: "red_dark", in: bundle, compatibleWith: nil)!
                 break
             case .warning:
                 headLabel.text = "Warning"
                 headLabel.textColor = UIColor(named: "white_yellow", in: bundle, compatibleWith: nil)!
-                circleImg.image = UIImage(named: "warning_icon", in: bundle, compatibleWith: nil)!
+                circleImg.image = loadImage(name: "warning_icon")
                 toastView.backgroundColor = UIColor(named: "yellow_dark", in: bundle, compatibleWith: nil)!
                 break
             case .info:
                 headLabel.text = "Info"
                 headLabel.textColor = UIColor(named: "white_blue", in: bundle, compatibleWith: nil)!
-                circleImg.image = UIImage(named: "info_icon", in: bundle, compatibleWith: nil)!
+                circleImg.image = loadImage(name: "info_icon")
                 toastView.backgroundColor = UIColor(named: "blue_dark", in: bundle, compatibleWith: nil)!
                 break
         }
     }
+    
+    func loadImage(name: String) -> UIImage? {
+        let podBundle = Bundle(for: MTVibrant.self)
+        if let url = podBundle.url(forResource: "MotionToastView", withExtension: "bundle") {
+            let bundle = Bundle(url: url)
+            return UIImage(named: name, in: bundle, compatibleWith: nil)
+        }
+        print("MotionToastView")
+        return nil
+    }
 }