<?php
/**
 * Iconic WDS - Change ASAP timeslot fee for shipping method.
 *
 * @param array $timeslots The timeslots.
 * @return array The modified timeslots.
 */
function iconic_wds_change_asap_timeslot_fee_for_shipping_method( $timeslots ) {
	global $iconic_wds;

	// Todo modify the Shipping method ASAP rates
	$asap_rates = array(
		'free_shipping:1' => 5,
		'flat_rate:5' => 6,
	);

	$chosen_shipping_method = $iconic_wds->get_chosen_shipping_method();
	
	if ( isset( $asap_rates[ $chosen_shipping_method ] ) ) {
		$timeslots['asap']['fee'] = array(
			'value' => $asap_rates[ $chosen_shipping_method ],
			'formatted' => wc_price( $asap_rates[ $chosen_shipping_method ] ),
		);

		$timeslots['asap']['formatted_with_fee'] = $timeslots['asap']['fee']['value'] > 0 ? sprintf( '%s (+%s)', $timeslots['asap']['formatted'], wp_strip_all_tags( $timeslots['asap']['fee']['formatted'] ) ) : $timeslots['asap']['formatted'];
		$timeslots['asap']['value'] = sprintf( 'asap|%01.2f', $asap_rates[ $chosen_shipping_method ] );		
	}

	return $timeslots;
}

add_filter( 'iconic_wds_timeslots', 'iconic_wds_change_asap_timeslot_fee_for_shipping_method', 10, 2 );