Reload button when opted in or out

Author: hola-soy-milk

ShiftDetailInteractor.swift

@@ -24,14 +24,16 @@ class ShiftDetailInteractor {
     if let user = UserManager.currentUser {
       if shift.currentUserAssigned.boolValue {
         APIClient.optOut(user, shift: self.shift, success: { (json) in
-          print("Opted out")
+          self.shift.currentUserAssigned = false
+          self.loadShift()
         }) { (message) in
           print("No good")
         }
       }
       else {
         APIClient.optIn(user, shift: self.shift, success: { (json) in
-          print("Opted In")
+          self.shift.currentUserAssigned = true
+          self.loadShift()
         }) { (message) in
           print("No good")
         }

Where2Help/EventMapper.swift

@@ -17,7 +17,7 @@ struct Shift: JSONJoy {
     let volunteersNeeded: Int!
     let volunteersCount: Int!
     let createdAt: NSDate!
-    let currentUserAssigned: Bool!
+    var currentUserAssigned: Bool!
 
     init(_ decoder: JSONDecoder) throws {
         ID = try decoder["id"].getInt()

Where2Help/ShiftDetailViewController.swift

@@ -30,10 +30,13 @@ class ShiftDetailViewController: UIViewController, ShiftHandler {
     super.viewDidLoad()
   }
 
+  override func viewWillAppear(animated: Bool) {
+    super.viewWillAppear(animated)
+    interactor.setup(self, event: event, shift: shift)
+  }
+
   override func viewDidLayoutSubviews() {
     super.viewDidLayoutSubviews()
-    interactor.setup(self, event: event, shift: shift)
-    shift.currentUserAssigned.boolValue ? imInButton.optedIn() : imInButton.optedOut()
   }
 
   // MARK: - ShiftHandler
@@ -72,6 +75,7 @@ class ShiftDetailViewController: UIViewController, ShiftHandler {
     timeLabel.text = shift.timespanLabelText()
     dateLabel.text = shift.dateString()
     volunteerCountLabel.text = "\(shift.volunteersOutOfLabelText()) Volunteers"
+    shift.shift.currentUserAssigned.boolValue ? imInButton.optedIn() : imInButton.optedOut()
   }
 
   ///////////////////////////////////