<?php
/**
 * My Account Shortcodes
 * 
 * Shows the 'my account' section where the customer can view past orders and update their information.
 *
 * @package		WooCommerce
 * @category	Shortcode
 * @author		WooThemes
 */
 
/**
 * Shortcode wrappers
 */
function get_woocommerce_my_account( $atts ) {
	global $woocommerce;
	return $woocommerce->shortcode_wrapper('woocommerce_my_account', $atts); 
}
function get_woocommerce_edit_address() {
	global $woocommerce;
	return $woocommerce->shortcode_wrapper('woocommerce_edit_address'); 
}
function get_woocommerce_change_password() {
	global $woocommerce;
	return $woocommerce->shortcode_wrapper('woocommerce_change_password'); 
}
function get_woocommerce_view_order() {
	global $woocommerce;
	return $woocommerce->shortcode_wrapper('woocommerce_view_order'); 
}

/**
 * My Account Shortcode.
 *
 * @package WooCommerce
 * @since 1.4
 */
function woocommerce_my_account( $atts ) {
	global $woocommerce, $current_user, $recent_orders;
	
	if ( ! is_user_logged_in() ) :
		
		woocommerce_get_template( 'myaccount/login.php' );
	
	else :

		extract(shortcode_atts(array(
	    	'recent_orders' => 5
		), $atts));
		
	  	$recent_orders = ('all' == $recent_orders) ? -1 : $recent_orders;
		
		get_currentuserinfo();
		
		woocommerce_get_template( 'myaccount/my-account.php' );

	endif;
		
}

/**
 * Edit Address Shortcode.
 *
 * @todo Address fields should be loaded using the array defined in 
 * the checkout class, and the fields should be built off of that.
 *
 * Adapted from spencerfinnell's pull request
 *
 * @package WooCommerce
 * @since 1.4
 */
function woocommerce_edit_address() {
	global $woocommerce, $load_address, $address;
	
	if ( ! is_user_logged_in() ) :
		wp_safe_redirect( get_permalink( get_option( 'woocommerce_myaccount_page_id' ) ) );
		exit;
	endif;

	$load_address = woocommerce_get_address_to_edit();
	
	$address = $woocommerce->countries->get_address_fields( get_user_meta( get_current_user_id(), $load_address . '_country', true ), $load_address . '_' );
	
	woocommerce_get_template( 'myaccount/edit-address.php' );
}

/**
 * Save and and update a billing or shipping address if the
 * form was submitted through the user account page.
 *
 * @todo Address fields should be loaded using the array defined in 
 * the checkout class.
 *
 * @package WooCommerce
 * @since 1.4
 */
function woocommerce_save_address() {
	global $woocommerce;

	if ( 'POST' !== strtoupper( $_SERVER[ 'REQUEST_METHOD' ] ) )
		return;

	if ( empty( $_POST[ 'action' ] ) || ( 'edit_address' !== $_POST[ 'action' ] ) )
		return;

	$woocommerce->verify_nonce( 'edit_address' );

	$user_id = get_current_user_id();

	if ( $user_id <= 0 ) return;

	$load_address = woocommerce_get_address_to_edit();
	
	$address = $woocommerce->countries->get_address_fields( esc_attr($_POST[ $load_address . '_country' ]), $load_address . '_' );

	foreach ($address as $key => $field) :
		
		if (!isset($field['type'])) $field['type'] = 'text';
		
		// Get Value
		switch ($field['type']) :
			case "checkbox" :
				$_POST[$key] = isset($_POST[$key]) ? 1 : 0;
			break;
			default :
				$_POST[$key] = isset($_POST[$key]) ? woocommerce_clean($_POST[$key]) : '';
			break;
		endswitch;
		
		// Hook to allow modification of value
		$_POST[$key] = apply_filters('woocommerce_process_myaccount_field_' . $key, $_POST[$key]);
		
		// Validation: Required fields
		if ( isset($field['required']) && $field['required'] && empty($_POST[$key]) ) $woocommerce->add_error( $field['label'] . ' ' . __('is a required field.', 'woothemes') );
		
		// Postcode
		if ($key=='billing_postcode' || $key=='shipping_postcode') :
			if ( ! $woocommerce->validation->is_postcode( $_POST[$key], $_POST[ $load_address . '_country' ] ) ) :
				$woocommerce->add_error( __( 'Please enter a valid postcode/ZIP.', 'woothemes' ) ); 
			else :
				$_POST[$key] = $woocommerce->validation->format_postcode( $_POST[$key], $_POST[ $load_address . '_country' ] );
			endif;
		endif;
		
	endforeach;

	if ( $woocommerce->error_count() == 0 ) {
		
		foreach ($address as $key => $field) :
			update_user_meta( $user_id, $key, $_POST[$key] );
		endforeach;

		do_action( 'woocommerce_customer_save_address', $user_id );

		wp_safe_redirect( get_permalink( get_option( 'woocommerce_myaccount_page_id' ) ) );
		exit;
	}
}
add_action( 'template_redirect', 'woocommerce_save_address' );

/**
 * Figure out which address is being viewed/edited.
 *
 * @package WooCommerce
 * @since 1.4
 */
function woocommerce_get_address_to_edit() {
	$load_address = 'billing';

	if ( isset( $_GET[ 'address' ] ) )
		$load_address = esc_attr( $_GET[ 'address' ] );

	if ( $load_address == 'billing' )
		$load_address = 'billing'; 
	else 
		$load_address = 'shipping';

	return $load_address;
}

/**
 * Change Password Shortcode
 *
 * @package WooCommerce
 * @since 1.4
 */
function woocommerce_change_password() {
	global $woocommerce;
	
	if ( ! is_user_logged_in() ) :
		wp_safe_redirect( get_permalink( get_option( 'woocommerce_myaccount_page_id' ) ) );
		exit;
	endif;

	woocommerce_get_template( 'myaccount/change-password.php' );
}

/**
 * Save the password and redirect back to the my account page.
 *
 * @package WooCommerce
 * @since 1.4
 */
function woocommerce_save_password() {
	global $woocommerce;

	if ( 'POST' !== strtoupper( $_SERVER[ 'REQUEST_METHOD' ] ) )
		return;

	if ( empty( $_POST[ 'action' ] ) || ( 'change_password' !== $_POST[ 'action' ] ) )
		return;

	$woocommerce->verify_nonce( 'change_password' );

	$user_id = get_current_user_id();

	if ( $user_id <= 0 )
		return;
		
	$_POST = array_map( 'woocommerce_clean', $_POST );

	if ( empty( $_POST[ 'password_1' ] ) || empty( $_POST[ 'password_2' ] ) )
		$woocommerce->add_error( __( 'Please enter your password.', 'woothemes' ) );
	
	if ( $_POST[ 'password_1' ] !== $_POST[ 'password_2' ] )
		$woocommerce->add_error( __('Passwords do not match.', 'woothemes') );
		
	if ( $woocommerce->error_count() == 0 ) {
		
		wp_update_user( array ('ID' => $user_id, 'user_pass' => esc_attr( $_POST['password_1'] ) ) ) ;

		do_action( 'woocommerce_customer_change_password', $user_id );

		wp_safe_redirect( get_permalink( get_option( 'woocommerce_myaccount_page_id' ) ) );
		exit;
	}
}
add_action( 'template_redirect', 'woocommerce_save_password' );

/**
 * View Order Shortcode
 *
 * @package WooCommerce
 * @since 1.4
 */
function woocommerce_view_order() {
	global $woocommerce;
	
	if ( ! is_user_logged_in() ) :
		wp_safe_redirect( get_permalink( get_option( 'woocommerce_myaccount_page_id' ) ) );
		exit;
	endif;
	
	$user_id      	= get_current_user_id();
	$order_id		= (isset($_GET['order'])) ? $_GET['order'] : 0;
	$order 			= &new woocommerce_order( $order_id );
	
	if ( $order_id==0 || $order->user_id != $user_id ) :
		wp_safe_redirect( get_permalink( get_option( 'woocommerce_myaccount_page_id' ) ) );
		exit;
	endif;
	
	$status = get_term_by('slug', $order->status, 'shop_order_status');
	
	echo '<p>'
	. sprintf( __('Order <mark>#%s</mark> made on <mark>%s</mark>', 'woothemes'), $order->id, date(get_option('date_format'), strtotime($order->order_date)) )
	. sprintf( __('. Order status: <mark>%s</mark>', 'woothemes'), __($status->name, 'woothemes') )
	. '.</p>';

	$notes = $order->get_customer_order_notes();
	if ($notes) :
		?>
		<h2><?php _e('Order Updates', 'woothemes'); ?></h2>
		<ol class="commentlist notes">	
			<?php foreach ($notes as $note) : ?>
			<li class="comment note">
				<div class="comment_container">			
					<div class="comment-text">
						<p class="meta"><?php echo date_i18n('l jS \of F Y, h:ia', strtotime($note->comment_date)); ?></p>
						<div class="description">
							<?php echo wpautop(wptexturize($note->comment_content)); ?>
						</div>
		  				<div class="clear"></div>
		  			</div>
					<div class="clear"></div>			
				</div>
			</li>
			<?php endforeach; ?>
		</ol>
		<?php
	endif;
	
	do_action( 'woocommerce_view_order', $order_id );
}