编辑文件

编辑文件：class-wc-shortcode-my-account.php

<?php
/**
 * My Account Shortcodes
 *
 * Shows the 'my account' section where the customer can view past orders and update their information.
 *
 * @package WooCommerce\Shortcodes\My_Account
 * @version 2.0.0
 */

defined( 'ABSPATH' ) || exit;

/**
 * Shortcode my account class.
 */
class WC_Shortcode_My_Account {

/**
	 * Get the shortcode content.
	 *
	 * @param array $atts Shortcode attributes.
	 *
	 * @return string
	 */
	public static function get( $atts ) {
		return WC_Shortcodes::shortcode_wrapper( array( __CLASS__, 'output' ), $atts );
	}

/**
	 * Output the shortcode.
	 *
	 * @param array $atts Shortcode attributes.
	 */
	public static function output( $atts ) {
		global $wp;

// Check cart class is loaded or abort.
		if ( is_null( WC()->cart ) ) {
			return;
		}

self::my_account_add_notices();

// Show the lost password page. This can still be accessed directly by logged in accounts which is important for the initial create password links sent via email.
		if ( isset( $wp->query_vars['lost-password'] ) ) {
			self::lost_password();
			return;
		}

// Show login form if not logged in.
		if ( ! is_user_logged_in() ) {
			wc_get_template( 'myaccount/form-login.php' );
			return;
		}

// Output the my account page.
		self::my_account( $atts );
	}

/**
	 * Add notices to the my account page.
	 *
	 * Historically a filter has existed to render a message above the my account page content while the user is
	 * logged out. See `woocommerce_my_account_message`.
	 */
	private static function my_account_add_notices() {
		global $wp;

if ( ! is_user_logged_in() ) {
			/**
			 * Filters the message shown on the 'my account' page when the user is not logged in.
			 *
			 * @since 2.6.0
			 */
			$message = apply_filters( 'woocommerce_my_account_message', '' );

if ( ! empty( $message ) ) {
				wc_add_notice( $message );
			}
		}

// After password reset, add confirmation message.
		if ( ! empty( $_GET['password-reset'] ) ) { // phpcs:ignore WordPress.Security.NonceVerification.Recommended
			wc_add_notice( __( 'Your password has been reset successfully.', 'woocommerce' ) );
		}

// After logging out without a nonce, add confirmation message.
		if ( isset( $wp->query_vars['customer-logout'] ) && is_user_logged_in() ) {
			/* translators: %s: logout url */
			wc_add_notice( sprintf( __( 'Are you sure you want to log out? <a href="%s">Confirm and log out</a>', 'woocommerce' ), wc_logout_url() ) );
		}

if ( get_user_option( 'default_password_nag' ) && ( wc_is_current_account_menu_item( 'dashboard' ) || wc_is_current_account_menu_item( 'edit-account' ) ) ) {
			wc_add_notice(
				sprintf(
					// translators: %s: site name.
					__( 'Your account with %s is using a temporary password. We emailed you a link to change your password.', 'woocommerce' ),
					esc_html( wp_specialchars_decode( get_option( 'blogname' ), ENT_QUOTES ) )
				),
				'notice',
				array(),
				true
			);
		}
	}

/**
	 * My account page.
	 *
	 * @param array $atts Shortcode attributes.
	 */
	private static function my_account( $atts ) {
		$args = shortcode_atts(
			array(
				'order_count' => 15, // @deprecated 2.6.0. Keep for backward compatibility.
			),
			$atts,
			'woocommerce_my_account'
		);

wc_get_template(
			'myaccount/my-account.php',
			array(
				'current_user' => get_user_by( 'id', get_current_user_id() ),
				'order_count'  => 'all' === $args['order_count'] ? -1 : $args['order_count'],
			)
		);
	}

/**
	 * View order page.
	 *
	 * @param int $order_id Order ID.
	 */
	public static function view_order( $order_id ) {
		$order = wc_get_order( $order_id );

if ( ! $order || ! current_user_can( 'view_order', $order_id ) ) {
			wc_print_notice(
				esc_html__( 'Invalid order.', 'woocommerce' ) . ' <a href="' . esc_url( wc_get_page_permalink( 'myaccount' ) ) . '" class="wc-forward">' . esc_html__( 'My account', 'woocommerce' ) . '</a>',
				'error'
			);
			return;
		}

// Backwards compatibility.
		$status       = new stdClass();
		$status->name = wc_get_order_status_name( $order->get_status() );

wc_get_template(
			'myaccount/view-order.php',
			array(
				'status'   => $status, // @deprecated 2.2.
				'order'    => $order,
				'order_id' => $order->get_id(),
			)
		);
	}

/**
	 * Edit account details page.
	 */
	public static function edit_account() {
		wc_get_template( 'myaccount/form-edit-account.php', array( 'user' => get_user_by( 'id', get_current_user_id() ) ) );
	}

/**
	 * Edit address page.
	 *
	 * @param string $load_address Type of address; 'billing' or 'shipping'.
	 */
	public static function edit_address( $load_address = 'billing' ) {
		$current_user = wp_get_current_user();
		$load_address = sanitize_key( $load_address );
		$country      = get_user_meta( get_current_user_id(), $load_address . '_country', true );

if ( ! $country ) {
			$country = WC()->countries->get_base_country();
		}

if ( 'billing' === $load_address ) {
			$allowed_countries = WC()->countries->get_allowed_countries();

if ( ! array_key_exists( $country, $allowed_countries ) ) {
				$country = current( array_keys( $allowed_countries ) );
			}
		}

if ( 'shipping' === $load_address ) {
			$allowed_countries = WC()->countries->get_shipping_countries();

if ( ! array_key_exists( $country, $allowed_countries ) ) {
				$country = current( array_keys( $allowed_countries ) );
			}
		}

$address = WC()->countries->get_address_fields( $country, $load_address . '_' );

// Enqueue scripts.
		wp_enqueue_script( 'wc-country-select' );
		wp_enqueue_script( 'wc-address-i18n' );

// Prepare values.
		foreach ( $address as $key => $field ) {

$value = get_user_meta( get_current_user_id(), $key, true );

if ( ! $value ) {
				switch ( $key ) {
					case 'billing_email':
					case 'shipping_email':
						$value = $current_user->user_email;
						break;
				}
			}

$address[ $key ]['value'] = apply_filters( 'woocommerce_my_account_edit_address_field_value', $value, $key, $load_address );
		}

wc_get_template(
			'myaccount/form-edit-address.php',
			array(
				'load_address' => $load_address,
				'address'      => apply_filters( 'woocommerce_address_to_edit', $address, $load_address ),
			)
		);
	}

/**
	 * Lost password page handling.
	 */
	public static function lost_password() {
		/**
		 * After sending the reset link, don't show the form again.
		 */
		if ( ! empty( $_GET['reset-link-sent'] ) ) { // WPCS: input var ok, CSRF ok.
			wc_get_template( 'myaccount/lost-password-confirmation.php' );
			return;

/**
			 * Process reset key / login from email confirmation link
			 */
		} elseif ( ! empty( $_GET['show-reset-form'] ) ) { // WPCS: input var ok, CSRF ok.
			if ( isset( $_COOKIE[ 'wp-resetpass-' . COOKIEHASH ] ) && 0 < strpos( $_COOKIE[ 'wp-resetpass-' . COOKIEHASH ], ':' ) ) {  // @codingStandardsIgnoreLine
				list( $rp_id, $rp_key ) = array_map( 'wc_clean', explode( ':', wp_unslash( $_COOKIE[ 'wp-resetpass-' . COOKIEHASH ] ), 2 ) ); // @codingStandardsIgnoreLine
				$userdata               = get_userdata( absint( $rp_id ) );
				$rp_login               = $userdata ? $userdata->user_login : '';
				$user                   = self::check_password_reset_key( $rp_key, $rp_login );

// Reset key / login is correct, display reset password form with hidden key / login values.
				if ( is_object( $user ) ) {
					wc_get_template(
						'myaccount/form-reset-password.php',
						array(
							'key'   => $rp_key,
							'login' => $rp_login,
						)
					);
					return;
				}
			}
		}

// Show lost password form by default.
		wc_get_template(
			'myaccount/form-lost-password.php',
			array(
				'form' => 'lost_password',
			)
		);
	}

/**
	 * Handles sending password retrieval email to customer.
	 *
	 * Based on retrieve_password() in core wp-login.php.
	 *
	 * @uses $wpdb WordPress Database object
	 * @return bool True: when finish. False: on error
	 */
	public static function retrieve_password() {
		$login = isset( $_POST['user_login'] ) ? sanitize_user( wp_unslash( $_POST['user_login'] ) ) : ''; // WPCS: input var ok, CSRF ok.

if ( empty( $login ) ) {

wc_add_notice( __( 'Enter a username or email address.', 'woocommerce' ), 'error' );

return false;

} else {
			// Check on username first, as customers can use emails as usernames.
			$user_data = get_user_by( 'login', $login );
		}

// If no user found, check if it login is email and lookup user based on email.
		if ( ! $user_data && is_email( $login ) && apply_filters( 'woocommerce_get_username_from_email', true ) ) {
			$user_data = get_user_by( 'email', $login );
		}

$errors = new WP_Error();

do_action( 'lostpassword_post', $errors, $user_data );

if ( $errors->get_error_code() ) {
			wc_add_notice( $errors->get_error_message(), 'error' );

return false;
		}

if ( ! $user_data ) {
			wc_add_notice( __( 'Invalid username or email.', 'woocommerce' ), 'error' );

return false;
		}

// Redefining user_login ensures we return the right case in the email.
		$user_login = $user_data->user_login;

do_action( 'retrieve_password', $user_login );

$allow = apply_filters( 'allow_password_reset', true, $user_data->ID );

if ( ! $allow ) {

wc_add_notice( __( 'Password reset is not allowed for this user', 'woocommerce' ), 'error' );

return false;

} elseif ( is_wp_error( $allow ) ) {

wc_add_notice( $allow->get_error_message(), 'error' );

return false;
		}

// Get password reset key (function introduced in WordPress 4.4).
		$key = get_password_reset_key( $user_data );

// Send email notification.
		WC()->mailer(); // Load email classes.
		do_action( 'woocommerce_reset_password_notification', $user_login, $key );

return true;
	}

/**
	 * Retrieves a user row based on password reset key and login.
	 *
	 * @uses $wpdb WordPress Database object.
	 * @param string $key   Hash to validate sending user's password.
	 * @param string $login The user login.
	 * @return WP_User|bool User's database row on success, false for invalid keys
	 */
	public static function check_password_reset_key( $key, $login ) {
		// Check for the password reset key.
		// Get user data or an error message in case of invalid or expired key.
		$user = check_password_reset_key( $key, $login );

if ( is_wp_error( $user ) ) {
			wc_add_notice( __( 'This key is invalid or has already been used. Please reset your password again if needed.', 'woocommerce' ), 'error' );
			return false;
		}

return $user;
	}

/**
	 * Handles resetting the user's password.
	 *
	 * @since 9.4.0 This will log the user in after resetting the password/session.
	 *
	 * @param WP_User $user     The user.
	 * @param string  $new_pass New password for the user in plaintext.
	 */
	public static function reset_password( $user, $new_pass ) {
		// phpcs:ignore WooCommerce.Commenting.CommentHooks.MissingHookComment
		do_action( 'password_reset', $user, $new_pass );

wp_set_password( $new_pass, $user->ID );
		update_user_meta( $user->ID, 'default_password_nag', false );

/**
		 * Fires after the user's password has been reset via WooCommerce.
		 *
		 * This provides parity with WordPress core's reset_password() function.
		 *
		 * @since 10.9.0
		 * @param WP_User $user     The user.
		 * @param string  $new_pass New user password in plaintext.
		 */
		do_action( 'after_password_reset', $user, $new_pass );
		self::set_reset_password_cookie();
		wc_set_customer_auth_cookie( $user->ID );

// phpcs:ignore WooCommerce.Commenting.CommentHooks.MissingHookComment
		if ( ! apply_filters( 'woocommerce_disable_password_change_notification', false ) ) {
			wp_password_change_notification( $user );
		}
	}

/**
	 * Set or unset the cookie.
	 *
	 * @param string $value Cookie value.
	 */
	public static function set_reset_password_cookie( $value = '' ) {
		$rp_cookie = 'wp-resetpass-' . COOKIEHASH;
		$rp_path   = isset( $_SERVER['REQUEST_URI'] ) ? current( explode( '?', wp_unslash( $_SERVER['REQUEST_URI'] ) ) ) : ''; // WPCS: input var ok, sanitization ok.

if ( $value ) {
			setcookie( $rp_cookie, $value, 0, $rp_path, COOKIE_DOMAIN, is_ssl(), true );
		} else {
			setcookie( $rp_cookie, ' ', time() - YEAR_IN_SECONDS, $rp_path, COOKIE_DOMAIN, is_ssl(), true );
		}
	}

/**
	 * Show the add payment method page.
	 */
	public static function add_payment_method() {
		if ( ! is_user_logged_in() ) {
			wp_safe_redirect( wc_get_page_permalink( 'myaccount' ) );
			exit();
		} else {
			do_action( 'before_woocommerce_add_payment_method' );

wc_get_template( 'myaccount/form-add-payment-method.php' );

do_action( 'after_woocommerce_add_payment_method' );
		}
	}
}