[Shipping labels] Set selected package from "Add Package" flow

Networking/Networking/Model/ShippingLabel/Packages/Carriers and Rates/ShippingLabelPackageSelected.swift

@@ -43,7 +43,7 @@ extension ShippingLabelPackageSelected: Encodable {
         var container = encoder.container(keyedBy: CodingKeys.self)
 
         try container.encode(id, forKey: .id)
-        try container.encode(boxID, forKey: .boxID)
+        try container.encode(boxID.isEmpty ? "0" : boxID, forKey: .boxID)
         try container.encode(length, forKey: .length)
         try container.encode(width, forKey: .width)
         try container.encode(height, forKey: .height)

WooCommerce/Classes/ViewRelated/Orders/Order Details/Shipping Labels/WooShipping Create Shipping Labels/WooShipping Package and Rate Selection/WooShippingAddCustomPackageViewModel.swift

@@ -53,7 +53,7 @@ final class WooShippingAddCustomPackageViewModel: ObservableObject {
     }
 
     private var packageDataFromCurrentData: WooShippingPackageDataRepresentable {
-        return WooShippingPackageData(id: UUID().uuidString,
+        return WooShippingPackageData(id: packageTemplateName,
                                       name: packageTemplateName,
                                       length: fieldValues[.length] ?? "",
                                       width: fieldValues[.width] ?? "",

WooCommerce/Classes/ViewRelated/Orders/Order Details/Shipping Labels/WooShipping Create Shipping Labels/WooShipping Package and Rate Selection/WooShippingPackageAndRatePlaceholder.swift

@@ -1,7 +1,11 @@
 import SwiftUI
 
 struct WooShippingPackageAndRatePlaceholder: View {
+    /// Action to perform when a package is selected.
+    let onSelectPackage: (WooShippingPackageDataRepresentable) -> Void
+
     @State private var showAddPackage: Bool = false
+
     var body: some View {
         VStack(spacing: .zero) {
             Button {
@@ -25,7 +29,7 @@ struct WooShippingPackageAndRatePlaceholder: View {
         .roundedBorder(cornerRadius: Layout.borderCornerRadius, lineColor: Color(.border), lineWidth: Layout.borderLineWidth, dashed: true)
         .sheet(isPresented: $showAddPackage) {
             WooShippingAddPackageView { packageData in
-                // TODO: use packageData
+                onSelectPackage(packageData)
                 showAddPackage = false
             }
         }
@@ -58,6 +62,6 @@ private extension WooShippingPackageAndRatePlaceholder {
 }
 
 #Preview {
-    WooShippingPackageAndRatePlaceholder()
+    WooShippingPackageAndRatePlaceholder(onSelectPackage: { _ in })
         .padding()
 }

WooCommerce/Classes/ViewRelated/Orders/Order Details/Shipping Labels/WooShipping Create Shipping Labels/WooShippingCreateLabelsView.swift

@@ -57,7 +57,7 @@ struct WooShippingCreateLabelsView: View {
                         WooShippingServiceView(viewModel: shippingService)
                             .padding(.horizontal, -16)
                     } else {
-                        WooShippingPackageAndRatePlaceholder()
+                        WooShippingPackageAndRatePlaceholder(onSelectPackage: viewModel.selectPackage)
                     }
                 }
                 .padding(16)

WooCommerce/Classes/ViewRelated/Orders/Order Details/Shipping Labels/WooShipping Create Shipping Labels/WooShippingCreateLabelsViewModel.swift

@@ -141,6 +141,24 @@ final class WooShippingCreateLabelsViewModel: ObservableObject {
         self.stores = stores
     }
 
+    /// Handles package selection for the shipping label.
+    /// Selecting a package also refreshes the available rates for the shipping service.
+    func selectPackage(_ packageData: WooShippingPackageDataRepresentable) {
+        // For now we support purchasing labels in a single package for a single shipment.
+        // In future milestones we can handle an array of packages with unique IDs for each shipment.
+        let package = ShippingLabelPackageSelected(id: "shipment_0",
+                                                   boxID: packageData.id,
+                                                   length: Double(packageData.length) ?? 0,
+                                                   width: Double(packageData.width) ?? 0,
+                                                   height: Double(packageData.height) ?? 0,
+                                                   weight: itemsDataSource.items.map(\.weight).reduce(0, +) + (Double(packageData.weight) ?? 0),
+                                                   isLetter: WooShippingPackageType(rawValue: packageData.packageType) == .envelope,
+                                                   hazmatCategory: nil, // Hazmat support will be added in a future milestone
+                                                   customsForm: nil) // Customs form support will be added in a future milestone
+        selectedPackage = package
+        shippingService?.loadLabelRates(for: package)
+    }
+
     /// Purchases a shipping label with the provided label details and settings.
     func purchaseLabel() {
         guard isPurchaseButtonEnabled, !isPurchasingLabel, let originSiteAddress, let destinationAddress, let selectedPackage, let selectedRate else {
@@ -175,13 +193,6 @@ final class WooShippingCreateLabelsViewModel: ObservableObject {
 
 // MARK: Utils
 private extension WooShippingCreateLabelsViewModel {
-    /// Handles changes to the selected package.
-    /// Selecting a package also refreshes the available rates for the shipping service.
-    func handleNewSelectedPackage(_ selectedPackage: ShippingLabelPackageSelected) {
-        self.selectedPackage = selectedPackage
-        shippingService?.loadLabelRates(for: selectedPackage)
-    }
-
     /// Provides the formatted label and amount for a shipping rate, based on the provided base rate.
     func formatShippingRate(name: String, rate: Double, basedOn baseRate: Double? = nil) -> (title: String, amount: String) {
         let amount = {

WooCommerce/WooCommerceTests/ViewRelated/Shipping Label/WooShipping Create Shipping Labels/WooShippingCreateLabelsViewModelTests.swift

@@ -250,6 +250,34 @@ final class WooShippingCreateLabelsViewModelTests: XCTestCase {
         // Check isPurchaseLabel is false after purchase
         XCTAssertFalse(viewModel.isPurchasingLabel)
     }
+
+    func test_selectPackage_sets_selectedPackage_with_package_data() {
+        // Given
+        let packageData = WooShippingPackageData(id: "small_flat_box",
+                                                 name: "Small Flat Rate Box",
+                                                 length: "21.91",
+                                                 width: "13.65",
+                                                 height: "4.13",
+                                                 dimensionsUnit: "cm",
+                                                 weight: "0",
+                                                 weightUnit: "kg",
+                                                 source: .predefined("usps"),
+                                                 packageType: "box")
+        let viewModel = WooShippingCreateLabelsViewModel(order: Order.fake())
+
+        // When
+        viewModel.selectPackage(packageData)
+
+        // Then
+        XCTAssertNotNil(viewModel.selectedPackage)
+        XCTAssertEqual(viewModel.selectedPackage?.id, "shipment_0")
+        XCTAssertEqual(viewModel.selectedPackage?.boxID, "small_flat_box")
+        XCTAssertEqual(viewModel.selectedPackage?.length, 21.91)
+        XCTAssertEqual(viewModel.selectedPackage?.width, 13.65)
+        XCTAssertEqual(viewModel.selectedPackage?.height, 4.13)
+        XCTAssertEqual(viewModel.selectedPackage?.weight, 0)
+        XCTAssertEqual(viewModel.selectedPackage?.isLetter, false)
+    }
 }
 
 private extension WooShippingCreateLabelsViewModelTests {