281 lines
9.3 KiB
PHP
281 lines
9.3 KiB
PHP
|
<?php
|
||
|
|
/**
|
||
|
|
* Lost Password Functions
|
||
|
|
*
|
||
|
|
* @package EDD
|
||
|
|
* @subpackage Functions/Login
|
||
|
|
* @copyright Copyright (c) 2022, Easy Digital Downloads, LLC
|
||
|
|
* @license https://opensource.org/licenses/gpl-2.0.php GNU Public License
|
||
|
|
* @since 1.0
|
||
|
|
*/
|
||
|
|
|
||
|
|
// Exit if accessed directly
|
||
|
|
defined( 'ABSPATH' ) || exit;
|
||
|
|
|
||
|
|
add_filter( 'wp_login_errors', 'edd_login_register_error_message', 10, 2 );
|
||
|
|
/**
|
||
|
|
* Changes the WordPress login confirmation message when using EDD's reset password link.
|
||
|
|
*
|
||
|
|
* @since 2.10
|
||
|
|
* @param object \WP_Error $errors
|
||
|
|
* @param string $redirect
|
||
|
|
* @return void
|
||
|
|
*/
|
||
|
|
function edd_login_register_error_message( $errors, $redirect ) {
|
||
|
|
$redirect_url = EDD()->session->get( 'edd_forgot_password_redirect' );
|
||
|
|
if ( empty( $redirect_url ) ) {
|
||
|
|
return $errors;
|
||
|
|
}
|
||
|
|
$message = sprintf(
|
||
|
|
/* translators: %s: Link to the referring page. */
|
||
|
|
__( 'Follow the instructions in the confirmation email you just received, then <a href="%s">return to what you were doing</a>.', 'easy-digital-downloads' ),
|
||
|
|
esc_url( $redirect_url )
|
||
|
|
);
|
||
|
|
$errors->remove( 'confirm' );
|
||
|
|
$errors->add(
|
||
|
|
'confirm',
|
||
|
|
apply_filters(
|
||
|
|
'edd_login_register_error_message',
|
||
|
|
$message,
|
||
|
|
$redirect_url
|
||
|
|
),
|
||
|
|
'message'
|
||
|
|
);
|
||
|
|
EDD()->session->set( 'edd_forgot_password_redirect', null );
|
||
|
|
|
||
|
|
return $errors;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Gets the lost password URL, customized for EDD. Using this allows the password
|
||
|
|
* reset form to redirect to the login screen with the EDD custom confirmation message.
|
||
|
|
*
|
||
|
|
* @since 2.10
|
||
|
|
* @return string
|
||
|
|
*/
|
||
|
|
function edd_get_lostpassword_url() {
|
||
|
|
|
||
|
|
$login_page_uri = edd_get_login_page_uri();
|
||
|
|
|
||
|
|
if ( empty( $login_page_uri ) ) {
|
||
|
|
return add_query_arg(
|
||
|
|
array(
|
||
|
|
'edd_forgot_password' => 'confirm',
|
||
|
|
),
|
||
|
|
wp_lostpassword_url()
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
return add_query_arg( 'action', 'lostpassword', $login_page_uri );
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Gets the password reset link for a user.
|
||
|
|
*
|
||
|
|
* @param WP_User $user
|
||
|
|
* @return false|string
|
||
|
|
*/
|
||
|
|
function edd_get_password_reset_link( $user ) {
|
||
|
|
$key = get_password_reset_key( $user );
|
||
|
|
if ( is_wp_error( $key ) ) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
return add_query_arg(
|
||
|
|
array(
|
||
|
|
'action' => 'rp',
|
||
|
|
'key' => rawurlencode( $key ),
|
||
|
|
'login' => rawurlencode( $user->user_login ),
|
||
|
|
),
|
||
|
|
wp_login_url()
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
add_action( 'lostpassword_form', 'edd_set_lostpassword_session' );
|
||
|
|
/**
|
||
|
|
* Sets a session value for the lost password redirect URI.
|
||
|
|
*
|
||
|
|
* @since 3.0.2
|
||
|
|
* @return void
|
||
|
|
*/
|
||
|
|
function edd_set_lostpassword_session() {
|
||
|
|
if ( ! empty( $_GET['edd_forgot_password'] ) && 'confirm' === $_GET['edd_forgot_password'] ) {
|
||
|
|
$url = wp_validate_redirect(
|
||
|
|
wp_get_referer(),
|
||
|
|
edd_get_checkout_uri()
|
||
|
|
);
|
||
|
|
EDD()->session->set( 'edd_forgot_password_redirect', $url );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
add_action( 'edd_user_lost_password', 'edd_handle_lost_password_request' );
|
||
|
|
/**
|
||
|
|
* Handles the lost password request from the EDD lost password block.
|
||
|
|
*
|
||
|
|
* @since 3.1
|
||
|
|
* @param array $data
|
||
|
|
* @return void
|
||
|
|
*/
|
||
|
|
function edd_handle_lost_password_request( $data ) {
|
||
|
|
if ( 'POST' === $_SERVER['REQUEST_METHOD'] ) {
|
||
|
|
$errors = retrieve_password();
|
||
|
|
if ( ! is_wp_error( $errors ) ) {
|
||
|
|
edd_set_success( 'checkemail', __( 'You did it! Check your email for instructions on resetting your password.', 'easy-digital-downloads' ) );
|
||
|
|
} else {
|
||
|
|
$error_code = $errors->get_error_code();
|
||
|
|
$message = $errors->get_error_message( $error_code );
|
||
|
|
if ( $message ) {
|
||
|
|
// WP_Error messages include "Error:" so we remove that here to prevent duplication.
|
||
|
|
$message = explode( ':', $message );
|
||
|
|
$message = ! empty( $message[1] ) ? trim( $message[1] ) : trim( $message[0] );
|
||
|
|
edd_set_error( $id, $message );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
edd_redirect( remove_query_arg( 'action', wp_get_referer() ) );
|
||
|
|
}
|
||
|
|
|
||
|
|
add_filter( 'retrieve_password_message', 'edd_retrieve_password_message', 10, 4 );
|
||
|
|
/**
|
||
|
|
* Filters the email message sent when a password reset has been requested.
|
||
|
|
*
|
||
|
|
* @since 3.1
|
||
|
|
* @param string $message The email message.
|
||
|
|
* @param string $key The activation key.
|
||
|
|
* @param string $user_login The username for the user.
|
||
|
|
* @param WP_User $user_data WP_User object.
|
||
|
|
* @return string
|
||
|
|
*/
|
||
|
|
function edd_retrieve_password_message( $message, $key, $user_login, $user_data ) {
|
||
|
|
if ( empty( $_POST['edd_action'] ) || 'user_lost_password' !== $_POST['edd_action'] ) {
|
||
|
|
return $message;
|
||
|
|
}
|
||
|
|
if ( empty( $_POST['edd_lost-password_nonce'] ) || ! wp_verify_nonce( $_POST['edd_lost-password_nonce'], 'edd-lost-password-nonce' ) ) {
|
||
|
|
return $message;
|
||
|
|
}
|
||
|
|
if ( is_multisite() ) {
|
||
|
|
$site_name = get_network()->site_name;
|
||
|
|
} else {
|
||
|
|
/*
|
||
|
|
* The blogname option is escaped with esc_html on the way into the database
|
||
|
|
* in sanitize_option. We want to reverse this for the plain text arena of emails.
|
||
|
|
*/
|
||
|
|
$site_name = wp_specialchars_decode( get_option( 'blogname' ), ENT_QUOTES );
|
||
|
|
}
|
||
|
|
$message = __( 'Someone has requested a password reset for the following account:', 'easy-digital-downloads' ) . "\r\n\r\n";
|
||
|
|
/* translators: %s: Site name. */
|
||
|
|
$message .= sprintf( __( 'Site Name: %s', 'easy-digital-downloads' ), $site_name ) . "\r\n\r\n";
|
||
|
|
/* translators: %s: User login. */
|
||
|
|
$message .= sprintf( __( 'Username: %s', 'easy-digital-downloads' ), $user_login ) . "\r\n\r\n";
|
||
|
|
$message .= __( 'If this was a mistake, ignore this email and nothing will happen.', 'easy-digital-downloads' ) . "\r\n\r\n";
|
||
|
|
$message .= __( 'To reset your password, visit the following address:', 'easy-digital-downloads' ) . "\r\n\r\n";
|
||
|
|
$message .= add_query_arg(
|
||
|
|
array(
|
||
|
|
'edd_action' => 'password_reset_requested',
|
||
|
|
'key' => $key,
|
||
|
|
'login' => rawurlencode( $user_login ),
|
||
|
|
),
|
||
|
|
esc_url_raw( $_POST['edd_redirect'] )
|
||
|
|
);
|
||
|
|
$message .= "\r\n\r\n";
|
||
|
|
|
||
|
|
if ( ! is_user_logged_in() ) {
|
||
|
|
$requester_ip = $_SERVER['REMOTE_ADDR'];
|
||
|
|
if ( $requester_ip ) {
|
||
|
|
$message .= sprintf(
|
||
|
|
/* translators: %s: IP address of password reset requester. */
|
||
|
|
__( 'This password reset request originated from the IP address %s.', 'easy-digital-downloads' ),
|
||
|
|
$requester_ip
|
||
|
|
) . "\r\n";
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return $message;
|
||
|
|
}
|
||
|
|
|
||
|
|
add_action( 'edd_password_reset_requested', 'edd_validate_password_reset_link' );
|
||
|
|
/**
|
||
|
|
* Validates the email link and sends the user to the password reset form upon success.
|
||
|
|
*
|
||
|
|
* @since 3.1
|
||
|
|
* @return void
|
||
|
|
*/
|
||
|
|
function edd_validate_password_reset_link() {
|
||
|
|
list( $rp_path ) = explode( '?', wp_unslash( $_SERVER['REQUEST_URI'] ) );
|
||
|
|
$rp_cookie = 'wp-resetpass-' . COOKIEHASH;
|
||
|
|
$redirect = remove_query_arg( array( 'key', 'login', 'edd_action' ), wp_get_referer() );
|
||
|
|
|
||
|
|
// Everything is good; move forward with the password reset.
|
||
|
|
if ( isset( $_GET['key'] ) && isset( $_GET['login'] ) ) {
|
||
|
|
$value = sprintf( '%s:%s', wp_unslash( $_GET['login'] ), wp_unslash( $_GET['key'] ) );
|
||
|
|
setcookie( $rp_cookie, $value, 0, $rp_path, COOKIE_DOMAIN, is_ssl(), true );
|
||
|
|
|
||
|
|
edd_redirect( add_query_arg( 'action', 'rp', $redirect ) );
|
||
|
|
}
|
||
|
|
|
||
|
|
$user = false;
|
||
|
|
if ( isset( $_COOKIE[ $rp_cookie ] ) && 0 < strpos( $_COOKIE[ $rp_cookie ], ':' ) ) {
|
||
|
|
list( $rp_login, $rp_key ) = explode( ':', wp_unslash( $_COOKIE[ $rp_cookie ] ), 2 );
|
||
|
|
|
||
|
|
$user = check_password_reset_key( $rp_key, $rp_login );
|
||
|
|
|
||
|
|
if ( isset( $_POST['pass1'] ) && ! hash_equals( $rp_key, $_POST['rp_key'] ) ) {
|
||
|
|
$user = false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( ! $user || is_wp_error( $user ) ) {
|
||
|
|
setcookie( $rp_cookie, ' ', time() - YEAR_IN_SECONDS, $rp_path, COOKIE_DOMAIN, is_ssl(), true );
|
||
|
|
|
||
|
|
if ( $user && $user->get_error_code() === 'expired_key' ) {
|
||
|
|
edd_set_error( 'expiredkey', __( 'Your password reset link has expired. Please request a new link below.', 'easy-digital-downloads' ) );
|
||
|
|
} else {
|
||
|
|
edd_set_error( 'invalidkey', __( 'Your password reset link appears to be invalid. Please request a new link below.', 'easy-digital-downloads' ) );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Redirect back to the lost password form instead of the password reset.
|
||
|
|
edd_redirect( add_query_arg( 'action', 'lostpassword', $redirect ) );
|
||
|
|
}
|
||
|
|
|
||
|
|
add_action( 'edd_user_reset_password', 'edd_validate_password_reset' );
|
||
|
|
/**
|
||
|
|
* Validates the password reset and redirects to the login form on success.
|
||
|
|
*
|
||
|
|
* @since 3.1
|
||
|
|
* @param array $data
|
||
|
|
* @return void
|
||
|
|
*/
|
||
|
|
function edd_validate_password_reset( $data ) {
|
||
|
|
// Check if password is one or all empty spaces.
|
||
|
|
if ( ! empty( $data['pass1'] ) ) {
|
||
|
|
$_POST['pass1'] = trim( $data['pass1'] );
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( empty( $data['pass1'] ) ) {
|
||
|
|
edd_set_error( 'empty_password', __( 'The password cannot be a space or all spaces.', 'easy-digital-downloads' ) );
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check if password fields do not match.
|
||
|
|
if ( ! empty( $data['pass1'] ) && trim( $data['pass2'] ) !== $data['pass1'] ) {
|
||
|
|
edd_set_error( 'password_reset_mismatch', __( 'The passwords do not match.', 'easy-digital-downloads' ) );
|
||
|
|
}
|
||
|
|
|
||
|
|
$user = get_user_by( 'login', $data['user_login'] );
|
||
|
|
if ( ! $user || is_wp_error( $user ) ) {
|
||
|
|
edd_set_error( 'password_reset_unsuccessful', __( 'Your password could not be reset.', 'easy-digital-downloads' ) );
|
||
|
|
}
|
||
|
|
|
||
|
|
$redirect = remove_query_arg( 'action', $data['edd_redirect'] );
|
||
|
|
// If no errors were registered then reset the password.
|
||
|
|
if ( ! edd_get_errors() ) {
|
||
|
|
reset_password( $user, $data['pass1'] );
|
||
|
|
edd_set_success( 'password_reset_successful', __( 'Your password was successfully reset.', 'easy-digital-downloads' ) );
|
||
|
|
// todo: check if this is correct
|
||
|
|
setcookie( 'wp-resetpass-' . COOKIEHASH, ' ', time() - YEAR_IN_SECONDS, wp_make_link_relative( wp_get_referer() ), COOKIE_DOMAIN, is_ssl(), true );
|
||
|
|
edd_redirect( $redirect );
|
||
|
|
}
|
||
|
|
|
||
|
|
edd_redirect( add_query_arg( 'action', 'password_reset_unsuccessful', $redirect ) );
|
||
|
|
}
|