Perform booking attendance status update

Author: RafaelKayumov

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/Stores/BookingStore.swift

@@ -63,10 +63,20 @@ 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
+            )
         }
     }
 }
 
+private enum UpdateBookingStatusError: Error {
+    case undefinedState
+}
 
 // MARK: - Services
 //
@@ -246,6 +256,78 @@ private extension BookingStore {
             }
         }
     }
+
+    func performUpdateBookingAttendanceStatus(
+        siteID: Int64,
+        bookingID: Int64,
+        status: BookingAttendanceStatus,
+        onCompletion: @escaping (Error?) -> Void
+    ) {
+        updateBookingAttendanceStatusLocally(
+            siteID: siteID,
+            bookingID: bookingID,
+            statusKey: status
+        ) { [weak self] previousStatusKey in
+            guard let self else {
+                return onCompletion(UpdateBookingStatusError.undefinedState)
+            }
+
+            Task {
+                do {
+                    if let remoteBooking = try await self.remote.updateBooking(
+                        from: siteID,
+                        bookingID: bookingID,
+                        attendanceStatus: status
+                    ) {
+                        await self.upsertStoredBookingsInBackground(
+                            readOnlyBookings: [remoteBooking],
+                            readOnlyOrders: [],
+                            siteID: siteID
+                        )
+                    } else {
+                        onCompletion(nil)
+                    }
+                } catch {
+                    /// Revert Optimistic Update
+                    self.updateBookingAttendanceStatusLocally(
+                        siteID: siteID,
+                        bookingID: bookingID,
+                        statusKey: previousStatusKey
+                    ) { _ in
+                        onCompletion(error)
+                    }
+                }
+            }
+        }
+    }
+
+    /// 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)
+    }
 }
 
 

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

@@ -142,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? {

WooCommerce/Classes/ViewRelated/Bookings/Booking Details/BookingDetailsView.swift

@@ -85,7 +85,7 @@ struct BookingDetailsView: View {
         }
         .sheet(isPresented: $showingStatusSheet) {
             UpdateAttendanceStatusView(selectedStatus: viewModel.bookingAttendanceStatus) { selectedStatus in
-                print("Selected status: \(selectedStatus)")
+                viewModel.updateAttendanceStatus(to: selectedStatus)
             }
             .padding(.top)
             .presentationDetents([.medium, .large])