Support the other button locations mapping

modules/ppcp-compat/src/Settings/StylingSettingsMapHelper.php

@@ -9,6 +9,8 @@ declare(strict_types=1);
 
 namespace WooCommerce\PayPalCommerce\Compat\Settings;
 
+use RuntimeException;
+use WooCommerce\PayPalCommerce\Button\Helper\ContextTrait;
 use WooCommerce\PayPalCommerce\Settings\DTO\LocationStylingDTO;
 
 /**
@@ -18,6 +20,9 @@ use WooCommerce\PayPalCommerce\Settings\DTO\LocationStylingDTO;
  * @psalm-import-type oldSettingsKey from SettingsMap
  */
 class StylingSettingsMapHelper {
+
+	use ContextTrait;
+
 	/**
 	 * Maps old setting keys to new setting style names.
 	 *
@@ -34,7 +39,12 @@ class StylingSettingsMapHelper {
 	 */
 	public function map(): array {
 
-		$mapped_settings = array();
+		$mapped_settings = array(
+			'smart_button_locations'     => '',
+			'pay_later_button_locations' => '',
+			'disable_funding'            => '',
+			'googlepay_button_enabled'   => '',
+		);
 
 		foreach ( $this->locations_map() as $old_location_name => $new_location_name ) {
 			foreach ( $this->styles() as $style ) {
@@ -50,28 +60,45 @@ class StylingSettingsMapHelper {
 	 * Retrieves the value of a mapped key from the new settings.
 	 *
 	 * @param string               $old_key The key from the legacy settings.
-	 * @param LocationStylingDTO[] $model The list of location styling models.
+	 * @param LocationStylingDTO[] $styling_models The list of location styling models.
 	 *
-	 * @return mixed|null The value of the mapped setting, or null if not found.
+	 * @return mixed The value of the mapped setting, (null if not found).
 	 */
-	public function mapped_value( string $old_key, array $model ) {
+	public function mapped_value(string $old_key, array $styling_models) {
-		foreach ( $this->locations_map() as $old_location_name => $new_location_name ) {
+		switch ($old_key) {
-			foreach ( $this->styles() as $style ) {
+			case 'smart_button_locations':
-				if ( $old_key !== $this->get_old_styling_setting_key( $old_location_name, $style ) ) {
+				return $this->mapped_smart_button_locations_value($styling_models);
-					continue;
+
+			case 'pay_later_button_locations':
+				return $this->mapped_pay_later_button_locations_value($styling_models);
+
+			case 'disable_funding':
+				return $this->mapped_disabled_funding_value($styling_models);
+
+			case 'googlepay_button_enabled':
+				return $this->mapped_google_pay_or_apple_pay_enabled_value($styling_models, 'googleplay');
+
+			case 'applepay_button_enabled':
+				return $this->mapped_google_pay_or_apple_pay_enabled_value($styling_models, 'applepay');
+
+			default:
+				foreach ($this->locations_map() as $old_location_name => $new_location_name) {
+					foreach ($this->styles() as $style) {
+						if ($old_key !== $this->get_old_styling_setting_key($old_location_name, $style)) {
+							continue;
+						}
+
+						$location_settings = $styling_models[$new_location_name] ?? false;
+
+						if (!$location_settings instanceof LocationStylingDTO) {
+							continue;
+						}
+
+						return $location_settings->$style ?? null;
+					}
 				}
-
+				return null;
-				$location_settings = $model[ $new_location_name ] ?? false;
-
-				if ( ! $location_settings instanceof LocationStylingDTO ) {
-					continue;
-				}
-
-				return $location_settings->$style ?? null;
-			}
 		}
-
-		return null;
 	}
 
 	/**
@@ -89,6 +116,23 @@ class StylingSettingsMapHelper {
 		);
 	}
 
+	/**
+	 * Returns a mapping of current context to new button location names.
+	 *
+	 * @return string[] The mapping of current context to new button location names.
+	 */
+	protected function current_context_to_new_button_location_map(): array {
+		return array(
+			'product'                => 'product',
+			'cart'                   => 'cart',
+			'cart-block'             => 'cart',
+			'checkout'               => 'classic_checkout',
+			'pay-now'                => 'classic_checkout',
+			'mini-cart'              => 'mini_cart',
+			'checkout-block'         => 'express_checkout',
+		);
+	}
+
 	/**
 	 * Returns the available style names.
 	 *
@@ -117,4 +161,97 @@ class StylingSettingsMapHelper {
 		$location_setting_name_part = $location === 'checkout' ? '' : "_{$location}";
 		return "button{$location_setting_name_part}_{$style}";
 	}
+
+	/**
+	 * Retrieves the mapped smart button locations from the new settings.
+	 *
+	 * @param LocationStylingDTO[] $styling_models The list of location styling models.
+	 * @return string[] The list of enabled smart button locations.
+	 */
+	protected function mapped_smart_button_locations_value( array $styling_models ): array {
+		$enabled_locations = array();
+		$locations         = array_flip( $this->locations_map() );
+
+		foreach ( $styling_models as $model ) {
+			if ( ! $model->enabled ) {
+				continue;
+			}
+
+			$enabled_locations[] = $locations[$model->location] ?? '';
+		}
+
+		return $enabled_locations;
+	}
+
+	/**
+	 * Retrieves the mapped pay later button locations from the new settings.
+	 *
+	 * @param LocationStylingDTO[] $styling_models The list of location styling models.
+	 * @return string[] The list of locations where the Pay Later button is enabled.
+	 */
+	protected function mapped_pay_later_button_locations_value( array $styling_models ): array {
+		$enabled_locations = array();
+		$locations         = array_flip( $this->locations_map() );
+		foreach ( $styling_models as $model ) {
+			if ( ! $model->enabled || ! in_array( 'paylater', $model->methods, true ) ) {
+				continue;
+			}
+
+			$enabled_locations[] = $locations[$model->location] ?? '';
+		}
+
+		return $enabled_locations;
+	}
+
+	/**
+	 * Retrieves the mapped disabled funding value from the new settings.
+	 *
+	 * @param LocationStylingDTO[] $styling_models The list of location styling models.
+	 * @return array|null The list of disabled funding, or null if none are disabled.
+	 */
+	protected function mapped_disabled_funding_value( array $styling_models ): ?array {
+		$disabled_funding         = array();
+		$locations_to_context_map = $this->current_context_to_new_button_location_map();
+
+		foreach ( $styling_models as $model ) {
+			if ( $model->location !== $locations_to_context_map[$this->context()] || in_array( 'venmo', $model->methods, true ) ) {
+				continue;
+			}
+
+			$disabled_funding[] = 'venmo';
+
+			return $disabled_funding;
+		}
+
+		return null;
+	}
+
+	/**
+	 * Retrieves the mapped enabled/disabled Google Pay or Apple Pay value from the new settings.
+	 *
+	 * @param LocationStylingDTO[] $styling_models The list of location styling models.
+	 * @param 'googlepay'|'applepay' $button_name The button name ('googlepay' or 'applepay').
+	 * @return int The enabled (1) or disabled (0) state.
+	 * @throws RuntimeException If an invalid button name is provided.
+	 */
+	protected function mapped_google_pay_or_apple_pay_enabled_value( array $styling_models, string $button_name ): int {
+		if ( $button_name !== 'googlepay' && $button_name !== 'applepay' ) {
+			throw new RuntimeException( 'Wrong button name is provided. Either "googlepay" or "applepay" can be used' );
+		}
+
+		$locations_to_context_map = $this->current_context_to_new_button_location_map();
+
+		foreach ( $styling_models as $model ) {
+			if ( ! $model->enabled
+				|| $model->location !== $locations_to_context_map[$this->context()]
+				|| ! in_array( $button_name, $model->methods, true )
+			) {
+				continue;
+			}
+
+			return 1;
+		}
+
+		return 0;
+	}
 }