woocommerce/shortcodes/shortcode-my_account.php

329 lines
10 KiB
PHP

<?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;
$user_id = get_current_user_id();
$load_address = woocommerce_get_address_to_edit();
$address = array(
'first_name' => get_user_meta( $user_id, $load_address . '_first_name', true ),
'last_name' => get_user_meta( $user_id, $load_address . '_last_name', true ),
'company' => get_user_meta( $user_id, $load_address . '_company', true ),
'email' => get_user_meta( $user_id, $load_address . '_email', true ),
'phone' => get_user_meta( $user_id, $load_address . '_phone', true ),
'address' => get_user_meta( $user_id, $load_address . '_address_1', true ),
'address2' => get_user_meta( $user_id, $load_address . '_address_2', true ),
'city' => get_user_meta( $user_id, $load_address . '_city', true ),
'state' => get_user_meta( $user_id, $load_address . '_state', true ),
'postcode' => get_user_meta( $user_id, $load_address . '_postcode', true ),
'country' => get_user_meta( $user_id, $load_address . '_country', true )
);
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();
$_POST = array_map( 'woocommerce_clean', $_POST );
if ( empty( $_POST[ 'address_first_name' ] ) )
$woocommerce->add_error( __( 'First name is a required field.', 'woothemes' ) );
if ( empty( $_POST[ 'address_last_name' ] ) )
$woocommerce->add_error( __( 'Last name is a required field.', 'woothemes') );
if ( empty( $_POST[ 'address_address_1' ] ) )
$woocommerce->add_error( __( 'Address is a required field.', 'woothemes') );
if ( empty( $_POST[ 'address_city' ] ) )
$woocommerce->add_error( __( 'City is a required field.', 'woothemes') );
if ( empty( $_POST[ 'address_postcode' ] ) )
$woocommerce->add_error( __( 'Postcode is a required field.', 'woothemes') );
if ( empty( $_POST[ 'address_country' ] ) )
$woocommerce->add_error( __( 'Country is a required field.', 'woothemes' ) );
if ( empty( $_POST[ 'address_state' ] ) )
$woocommerce->add_error( __( 'State is a required field.', 'woothemes' ) );
// Billing only
if ( $load_address == 'billing' ) {
if ( empty( $_POST[ 'address_email' ] ) )
$woocommerce->add_error( __( 'Email is a required field.', 'woothemes' ) );
if ( empty( $_POST[ 'address_phone' ] ) )
$woocommerce->add_error( __( 'Phone number is a required field.', 'woothemes' ) );
if ( ! $woocommerce->validation->is_email( $_POST[ 'address_email' ] ) )
$woocommerce->add_error( __( 'Please enter a valid email address.', 'woothemes' ) );
if ( ! $woocommerce->validation->is_phone( $_POST[ 'address_phone' ] ) )
$woocommerce->add_error( __( 'Please enter a valid phone number.', 'woothemes' ) );
}
if ( ! $woocommerce->validation->is_postcode( $_POST[ 'address_postcode' ], $_POST[ 'address_country' ] ) )
$woocommerce->add_error( __( 'Please enter a valid postcode/ZIP.', 'woothemes' ) );
else
$_POST[ 'address_postcode' ] = $woocommerce->validation->format_postcode( $_POST[ 'address_postcode' ], $_POST[ 'address_country' ] );
if ( $woocommerce->error_count() == 0 ) {
update_user_meta( $user_id, $load_address . '_first_name', $_POST[ 'address_first_name' ] );
update_user_meta( $user_id, $load_address . '_last_name', $_POST[ 'address_last_name' ] );
update_user_meta( $user_id, $load_address . '_company', $_POST[ 'address_company' ] );
update_user_meta( $user_id, $load_address . '_address_1', $_POST[ 'address_address_1' ] );
update_user_meta( $user_id, $load_address . '_address_2', $_POST[ 'address_address_2' ] );
update_user_meta( $user_id, $load_address . '_city', $_POST[ 'address_city' ] );
update_user_meta( $user_id, $load_address . '_postcode', $_POST[ 'address_postcode' ] );
update_user_meta( $user_id, $load_address . '_country', $_POST[ 'address_country' ] );
update_user_meta( $user_id, $load_address . '_state', $_POST[ 'address_state' ] );
if ( $load_address == 'billing' ) {
update_user_meta( $user_id, $load_address . '_email', $_POST['address_email'] );
update_user_meta( $user_id, $load_address . '_phone', $_POST['address_phone'] );
}
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 );
}