[Bookings] Apply attendance status to UI (#16260)

Author: RafaelKayumov

Modules/Sources/Networking/Model/Bookings/Booking.swift

@@ -29,7 +29,6 @@ public struct Booking: Codable, GeneratedCopiable, Hashable, GeneratedFakeable {
         return BookingStatus(rawValue: statusKey) ?? .unknown
     }
 
-    /// periphery: ignore - will be used in UI in upcoming PRs
     public var attendanceStatus: BookingAttendanceStatus {
         return BookingAttendanceStatus(rawValue: attendanceStatusKey) ?? .unknown
     }
@@ -208,7 +207,6 @@ public enum BookingStatus: String, CaseIterable {
     case unknown
 }
 
-/// periphery: ignore - will be used in UI in upcoming PRs
 public enum BookingAttendanceStatus: String, CaseIterable {
     case booked
     case checkedIn = "checked-in"

Modules/Sources/Networking/Remote/BookingsRemote.swift

@@ -17,6 +17,12 @@ public protocol BookingsRemoteProtocol {
     func loadBooking(bookingID: Int64,
                      siteID: Int64) async throws -> Booking?
 
+    func updateBooking(
+        from siteID: Int64,
+        bookingID: Int64,
+        attendanceStatus: BookingAttendanceStatus
+    ) async throws -> Booking?
+
     func fetchResource(resourceID: Int64,
                        siteID: Int64) async throws -> BookingResource?
 }
@@ -88,6 +94,28 @@ public final class BookingsRemote: Remote, BookingsRemoteProtocol {
         return try await enqueue(request, mapper: mapper)
     }
 
+    public func updateBooking(
+        from siteID: Int64,
+        bookingID: Int64,
+        attendanceStatus: BookingAttendanceStatus
+    ) async throws -> Booking? {
+        let path = "\(Path.bookings)/\(bookingID)"
+        let parameters = [
+            ParameterKey.attendanceStatus: attendanceStatus.rawValue
+        ]
+        let request = JetpackRequest(
+            wooApiVersion: .wcBookings,
+            method: .put,
+            siteID: siteID,
+            path: path,
+            parameters: parameters,
+            availableAsRESTRequest: true
+        )
+
+        let mapper = BookingMapper(siteID: siteID)
+        return try await enqueue(request, mapper: mapper)
+    }
+
     public func fetchResource(
         resourceID: Int64,
         siteID: Int64
@@ -132,5 +160,6 @@ public extension BookingsRemote {
         static let startDateAfter: String  = "start_date_after"
         static let search: String          = "search"
         static let order: String           = "order"
+        static let attendanceStatus        = "attendance_status"
     }
 }

Modules/Sources/Yosemite/Actions/BookingAction.swift

@@ -52,4 +52,16 @@ public enum BookingAction: Action {
     case fetchResource(siteID: Int64,
                        resourceID: Int64,
                        onCompletion: (Result<BookingResource, Error>) -> Void)
+
+    /// Updates a booking attendance status.
+    ///
+    /// - Parameter siteID: The site ID of the booking.
+    /// - Parameter bookingID: The ID of the booking to be updated.
+    /// - Parameter status: The new attendance status.
+    /// - Parameter onCompletion: called when update completes, returns an error in case of a failure.
+    ///
+    case updateBookingAttendanceStatus(siteID: Int64,
+                                       bookingID: Int64,
+                                       status: BookingAttendanceStatus,
+                                       onCompletion: (Error?) -> Void)
 }

Modules/Sources/Yosemite/Model/Model.swift

@@ -31,6 +31,7 @@ public typealias BookingCustomerInfo = Networking.BookingCustomerInfo
 public typealias BookingPaymentInfo = Networking.BookingPaymentInfo
 public typealias BookingProductInfo = Networking.BookingProductInfo
 public typealias BookingResource = Networking.BookingResource
+public typealias BookingAttendanceStatus = Networking.BookingAttendanceStatus
 public typealias CreateBlazeCampaign = Networking.CreateBlazeCampaign
 public typealias FallibleCancelable = Hardware.FallibleCancelable
 public typealias CommentStatus = Networking.CommentStatus

Modules/Sources/Yosemite/Stores/BookingStore.swift

@@ -63,11 +63,17 @@ public class BookingStore: Store {
                            onCompletion: onCompletion)
         case let .fetchResource(siteID, resourceID, onCompletion):
             fetchResource(siteID: siteID, resourceID: resourceID, onCompletion: onCompletion)
+        case .updateBookingAttendanceStatus(let siteID, let bookingID, let status, let onCompletion):
+            performUpdateBookingAttendanceStatus(
+                siteID: siteID,
+                bookingID: bookingID,
+                status: status,
+                onCompletion: onCompletion
+            )
         }
     }
 }
 
-
 // MARK: - Services
 //
 private extension BookingStore {
@@ -246,6 +252,50 @@ private extension BookingStore {
             }
         }
     }
+
+    func performUpdateBookingAttendanceStatus(
+        siteID: Int64,
+        bookingID: Int64,
+        status: BookingAttendanceStatus,
+        onCompletion: @escaping (Error?) -> Void
+    ) {
+        updateBookingAttendanceStatusLocally(
+            siteID: siteID,
+            bookingID: bookingID,
+            statusKey: status
+        ) { _ in
+            //TODO: - booking status remote update + rollback status in case of error
+            onCompletion(nil)
+        }
+    }
+
+    /// Updates local (Storage) Booking attendance status
+    func updateBookingAttendanceStatusLocally(
+        siteID: Int64,
+        bookingID: Int64,
+        statusKey: BookingAttendanceStatus,
+        onCompletion: @escaping (BookingAttendanceStatus) -> Void
+    ) {
+        storageManager.performAndSave({ storage -> BookingAttendanceStatus in
+            guard let booking = storage.loadBooking(
+                siteID: siteID,
+                bookingID: bookingID
+            ) else {
+                return statusKey
+            }
+
+            let oldStatus = booking.attendanceStatusKey
+            booking.attendanceStatusKey = statusKey.rawValue
+            return BookingAttendanceStatus(rawValue: oldStatus ?? "") ?? .unknown
+        }, completion: { result in
+            switch result {
+            case .success(let status):
+                onCompletion(status)
+            case .failure:
+                onCompletion(statusKey)
+            }
+        }, on: .main)
+    }
 }
 
 

Modules/Tests/YosemiteTests/Mocks/MockBookingsRemote.swift

@@ -46,4 +46,8 @@ final class MockBookingsRemote: BookingsRemoteProtocol {
         }
         return try result.get()
     }
+
+    func updateBooking(from siteID: Int64, bookingID: Int64, attendanceStatus: Networking.BookingAttendanceStatus) async throws -> Networking.Booking? {
+        return nil
+    }
 }

WooCommerce/Classes/Bookings/BookingList/BookingListView.swift

@@ -136,7 +136,7 @@ private extension BookingListView {
                 HStack {
                     // TODO: update this when attendance status is available
                     // Update badge colors if design changes as statuses are not clarified now.
-                    statusBadge(text: "Booked", color: Layout.defaultBadgeColor)
+                    statusBadge(text: booking.attendanceStatus.localizedTitle, color: Layout.defaultBadgeColor)
                     statusBadge(text: booking.bookingStatus.localizedTitle, color: Layout.defaultBadgeColor)
                     Spacer()
                 }

WooCommerce/Classes/ViewModels/Booking Details/AttendanceContent.swift

@@ -1,9 +1,12 @@
 import Foundation
+import Networking
 
 extension BookingDetailsViewModel {
-    struct AttendanceContent {
-        /// Hardcoded attendance value
-        /// Will be replaced with model value or binding
-        let value = "Booked"
+    final class AttendanceContent {
+        private(set) var value = ""
+
+        func update(with booking: Booking) {
+            value = booking.attendanceStatus.localizedTitle
+        }
     }
 }

WooCommerce/Classes/ViewModels/Booking Details/BookingAttendanceStatus+Localization.swift

@@ -0,0 +1,39 @@
+import Foundation
+import Networking
+
+extension BookingAttendanceStatus {
+    var localizedTitle: String {
+        switch self {
+        case .booked:
+            return NSLocalizedString(
+                "BookingAttendanceStatus.booked",
+                value: "Booked",
+                comment: "Title for 'Booked' booking attendance status."
+            )
+        case .checkedIn:
+            return NSLocalizedString(
+                "BookingAttendanceStatus.checkedIn",
+                value: "Checked In",
+                comment: "Title for 'Checked In' booking attendance status."
+            )
+        case .cancelled:
+            return NSLocalizedString(
+                "BookingAttendanceStatus.cancelled",
+                value: "Cancelled",
+                comment: "Title for 'Cancelled' booking attendance status."
+            )
+        case .noShow:
+            return NSLocalizedString(
+                "BookingAttendanceStatus.noShow",
+                value: "No Show",
+                comment: "Title for 'No Show' booking attendance status."
+            )
+        case .unknown:
+            return NSLocalizedString(
+                "BookingAttendanceStatus.unknown",
+                value: "Unknown",
+                comment: "Title for 'Unknown' booking attendance status."
+            )
+        }
+    }
+}

WooCommerce/Classes/ViewModels/Booking Details/BookingDetailsViewModel.swift

@@ -28,6 +28,10 @@ final class BookingDetailsViewModel: ObservableObject {
     @Published private(set) var navigationTitle = ""
     @Published private(set) var sections: [Section] = []
 
+    var bookingAttendanceStatus: BookingAttendanceStatus {
+        booking.attendanceStatus
+    }
+
     init(booking: Booking,
          stores: StoresManager = ServiceLocator.stores,
          storage: StorageManagerType = ServiceLocator.storageManager) {
@@ -92,6 +96,7 @@ private extension BookingDetailsViewModel {
         }
         headerContent.update(with: booking)
         appointmentDetailsContent.update(with: booking, resource: bookingResource)
+        attendanceContent.update(with: booking)
         paymentContent.update(with: booking)
     }
 
@@ -137,6 +142,24 @@ extension BookingDetailsViewModel {
     }
 }
 
+// MARK: Attendance status
+
+extension BookingDetailsViewModel {
+    func updateAttendanceStatus(to newStatus: BookingAttendanceStatus) {
+        let action = BookingAction.updateBookingAttendanceStatus(
+            siteID: booking.siteID,
+            bookingID: booking.bookingID,
+            status: newStatus
+        ) { error in
+            if let error {
+                DDLogError("⛔️ Error updating booking attendance status: \(error)")
+                // TODO: Show an error notice to the user
+            }
+        }
+        stores.dispatch(action)
+    }
+}
+
 private extension BookingDetailsViewModel {
     @MainActor
     func fetchResource() async -> BookingResource? {