Connect actions from shipping line row to NewOrderViewModel

Author: ealeksandrov

WooCommerce/Classes/ViewRelated/Orders/Order Creation/NewOrder.swift

@@ -40,7 +40,7 @@ struct NewOrder: View {
                         Spacer(minLength: Layout.sectionSpacing)
 
                         if viewModel.shouldShowPaymentSection {
-                            OrderPaymentSection(viewModel: viewModel.paymentDataViewModel)
+                            OrderPaymentSection(viewModel: viewModel.paymentDataViewModel, saveShippingLineClosure: viewModel.saveShippingLine)
 
                             Spacer(minLength: Layout.sectionSpacing)
                         }

WooCommerce/Classes/ViewRelated/Orders/Order Creation/NewOrderViewModel.swift

@@ -138,6 +138,13 @@ final class NewOrderViewModel: ObservableObject {
     ///
     @Published private(set) var paymentDataViewModel = PaymentDataViewModel()
 
+    /// Saves a shipping line.
+    ///
+    /// - Parameter shippingLine: Optional shipping line object to save. `nil` will remove existing shipping line.
+    func saveShippingLine(_ shippingLine: ShippingLine?) {
+        orderDetails.shippingLine = shippingLine
+    }
+
     /// Analytics engine.
     ///
     private let analytics: Analytics

WooCommerce/Classes/ViewRelated/Orders/Order Creation/PaymentSection/OrderPaymentSection.swift

@@ -1,11 +1,15 @@
 import SwiftUI
+import Yosemite
 
 /// Represents the Payment section in an order
 ///
 struct OrderPaymentSection: View {
     /// View model to drive the view content
     let viewModel: NewOrderViewModel.PaymentDataViewModel
 
+    /// Closure to create/update the shipping line object
+    let saveShippingLineClosure: (ShippingLine?) -> Void
+
     ///   Environment safe areas
     ///
     @Environment(\.safeAreaInsets) var safeAreaInsets: EdgeInsets
@@ -22,9 +26,17 @@ struct OrderPaymentSection: View {
 
             if viewModel.shouldShowShippingTotal {
                 TitleAndValueRow(title: Localization.shippingTotal, value: .content(viewModel.shippingTotal), selectable: true) {
+                    saveShippingLineClosure(nil)
                 }
             } else {
                 Button(Localization.addShipping) {
+                    let testShippingLine = ShippingLine(shippingID: 0,
+                                                        methodTitle: "Flat Rate",
+                                                        methodID: "other",
+                                                        total: "10",
+                                                        totalTax: "",
+                                                        taxes: [])
+                    saveShippingLineClosure(testShippingLine)
                 }
                 .buttonStyle(PlusButtonStyle())
                 .padding()
@@ -60,7 +72,7 @@ struct OrderPaymentSection_Previews: PreviewProvider {
     static var previews: some View {
         let viewModel = NewOrderViewModel.PaymentDataViewModel(itemsTotal: "20.00", orderTotal: "20.00")
 
-        OrderPaymentSection(viewModel: viewModel)
+        OrderPaymentSection(viewModel: viewModel, saveShippingLineClosure: { _ in })
             .previewLayout(.sizeThatFits)
     }
 }