diff --git a/GMERemittance/Model/SuccessMessage.swift b/GMERemittance/Model/SuccessMessage.swift index 7c3653cf..935594f6 100644 --- a/GMERemittance/Model/SuccessMessage.swift +++ b/GMERemittance/Model/SuccessMessage.swift @@ -11,6 +11,7 @@ import ObjectMapper class SuccessMessage: Mappable { var message: String? + var extra: String? required init?(map: Map) { @@ -18,6 +19,7 @@ class SuccessMessage: Mappable { func mapping(map: Map) { message <- map["Message"] + extra <- map["Extra"] } } diff --git a/GMERemittance/Module/PennyTestSubmit/User Interface/View/PennyTestSubmitViewController.swift b/GMERemittance/Module/PennyTestSubmit/User Interface/View/PennyTestSubmitViewController.swift index 26d0974b..d4ac1530 100644 --- a/GMERemittance/Module/PennyTestSubmit/User Interface/View/PennyTestSubmitViewController.swift +++ b/GMERemittance/Module/PennyTestSubmit/User Interface/View/PennyTestSubmitViewController.swift @@ -128,6 +128,13 @@ class PennyTestSubmitViewController: UIViewController { ] self.submit(params: params, success: { (response) in + let balance = response.extra ?? "" + let userInfo = [SideMenuNavigationNotifications.availableBalance : balance] + + UserDefaults.standard.set(balance, forKey: UserKeys.availableBalance) + NotificationCenter.default.post(name: self.getAvailableBalanceNotificationName(), object: nil, userInfo: userInfo) + // show alert and dismiss + self.alertWithOk(message: response.message ?? "", title: "Success", okTitle: "Ok", style: UIAlertControllerStyle.alert , OkStyle: .default, okAction: { UserDefaults.standard.set(PennyTestStatusCode.completed.rawValue, forKey: AppConstants.pennyTestStatusCode) @@ -138,6 +145,10 @@ class PennyTestSubmitViewController: UIViewController { } } + func getAvailableBalanceNotificationName() -> Notification.Name { + return Notification.Name.init(rawValue: SideMenuNavigationNotifications.availableBalance) + } + private func getFcmToken() -> String? { let token = InstanceID.instanceID().token() return token