calculate cart totals afer running extensions

src/StoreApi/Routes/V1/AbstractCartRoute.php

@@ -78,7 +78,7 @@ public function __construct( SchemaController $schema_controller, AbstractSchema
 	 */
 	public function get_response( \WP_REST_Request $request ) {
 		$this->load_cart_session( $request );
-		$this->calculate_totals();
+		$this->cart_controller->calculate_totals();
 
 		if ( $this->requires_nonce( $request ) ) {
 			$nonce_check = $this->check_nonce( $request );
@@ -229,16 +229,6 @@ protected function cart_updated( \WP_REST_Request $request ) {
 		}
 	}
 
-	/**
-	 * Ensures the cart totals are calculated before an API response is generated.
-	 */
-	protected function calculate_totals() {
-		wc()->cart->get_cart();
-		wc()->cart->calculate_fees();
-		wc()->cart->calculate_shipping();
-		wc()->cart->calculate_totals();
-	}
-
 	/**
 	 * For non-GET endpoints, require and validate a nonce to prevent CSRF attacks.
 	 *

src/StoreApi/Schemas/V1/CartExtensionsSchema.php

@@ -69,12 +69,16 @@ public function get_item_response( $request = null ) {
 				400
 			);
 		}
+
+		$controller = new CartController();
+
 		if ( is_callable( $callback ) ) {
 			$callback( $request['data'] );
+			// We recalculate the cart if we had something to run.
+			$controller->calculate_totals();
 		}
 
-		$controller = new CartController();
-		$cart       = $controller->get_cart_instance();
+		$cart = $controller->get_cart_instance();
 
 		return rest_ensure_response( $this->cart_schema->get_item_response( $cart ) );
 	}

src/StoreApi/Utilities/CartController.php

@@ -32,6 +32,17 @@ public function load_cart() {
 		}
 	}
 
+	/**
+	 * Recalculates the cart totals.
+	 */
+	public function calculate_totals() {
+		$cart = $this->get_cart_instance();
+		$cart->get_cart();
+		$cart->calculate_fees();
+		$cart->calculate_shipping();
+		$cart->calculate_totals();
+	}
+
 	/**
 	 * Based on the core cart class but returns errors rather than rendering notices directly.
 	 *