More plumbo

Author: amy-at-kickstarter

Kickstarter-iOS/Features/PledgeShippingLocation/PledgeShippingLocationViewController.swift

@@ -135,7 +135,7 @@ final class PledgeShippingLocationViewController: UIViewController {
       .observeForUI()
       .observeValues { [weak self] project, shippingRules, location in
         self?.presentShippingRules(
-          project, shippingRules: shippingRules, selectedLocation: location
+          project: project, shippingRules: shippingRules, selectedLocation: location
         )
       }
 
@@ -169,10 +169,22 @@ final class PledgeShippingLocationViewController: UIViewController {
   // MARK: - Functions
 
   private func presentShippingRules(
-    _: Project, shippingRules: [ShippingRule], selectedLocation: Location
+    project: Project, shippingRules _: [ShippingRule], selectedLocation: Location
   ) {
+    let locations: [Location] = project.rewards.map { $0.shippingRulesExpanded }
+      .map { rules in
+        rules?.map { $0.location }
+      }
+      .reduce([]) { allLocations, locations in
+        guard let locations = locations else {
+          return allLocations
+        }
+
+        return allLocations + locations
+      }
+
     let viewController = ShippingLocationsViewController(
-      withLocations: shippingRules.map { $0.location },
+      withLocations: locations,
       selectedLocation: selectedLocation
     ) { location in