diff --git a/MBRateApp/Classes/MBRateUs.swift b/MBRateApp/Classes/MBRateUs.swift index a24a43c..9040ded 100644 --- a/MBRateApp/Classes/MBRateUs.swift +++ b/MBRateApp/Classes/MBRateUs.swift @@ -15,7 +15,7 @@ public class MBRateUs{ public var rateUsInfo = MBRateUsInfo() - public func showRateUs(base:UIViewController, positiveBlock:()->Void, negativeBlock:()->Void, dismissBlock:()->Void){ + public func showRateUs(base:UIViewController, positiveBlock:()->Void, negativeBlock:()->Void, dismissBlock:(()->Void)? = nil, iconAnimationBlock:((iconView: UIImageView)->Void)? = nil ){ let podBundle = NSBundle(forClass: self.dynamicType) let storyboard = UIStoryboard(name: "RateUs", bundle: podBundle) let vc = storyboard.instantiateInitialViewController() as! MBRateUsViewController @@ -23,6 +23,7 @@ public class MBRateUs{ vc.positiveBlock = positiveBlock vc.negativeBlock = negativeBlock vc.dismissBlock = dismissBlock + vc.iconAnimationBlock = iconAnimationBlock vc.rateUsInfo = self.rateUsInfo diff --git a/MBRateApp/Classes/MBRateUsViewController.swift b/MBRateApp/Classes/MBRateUsViewController.swift index b9f905b..098c67e 100644 --- a/MBRateApp/Classes/MBRateUsViewController.swift +++ b/MBRateApp/Classes/MBRateUsViewController.swift @@ -55,6 +55,7 @@ class MBRateUsViewController : UIViewController { var positiveBlock : (()->Void)? var negativeBlock : (()->Void)? var dismissBlock : (()->Void)? + var iconAnimationBlock:((iconView: UIImageView)->Void)? var shouldRate : Bool var starImageOn : UIImage @@ -85,10 +86,14 @@ class MBRateUsViewController : UIViewController { } self.imageView.image = self.rateUsInfo?.titleImage - self.dismissButton.tintColor = self.rateUsInfo?.dismissButtonColor } + override func viewWillAppear(animated: Bool) { + super.viewWillAppear(animated) + self.iconAnimationBlock?(iconView: self.imageView) + } + @IBAction func dismiss(sender: UIButton) { self.dismissViewControllerAnimated(true, completion: { _ in self.dismissBlock?()})