mirror of
https://github.com/10h30/ultimatemember.git
synced 2026-06-05 15:09:37 +09:00
nikitasinelnikov
478 lines
14 KiB
PHP
478 lines
14 KiB
PHP
<?php if ( ! defined( 'ABSPATH' ) ) exit;
|
|
|
|
|
|
/**
|
|
* Error procesing hook for login
|
|
*
|
|
* @param $args
|
|
*/
|
|
function um_submit_form_errors_hook_login( $args ) {
|
|
$is_email = false;
|
|
|
|
$form_id = $args['form_id'];
|
|
$mode = $args['mode'];
|
|
$user_password = $args['user_password'];
|
|
|
|
|
|
if ( isset( $args['username'] ) && $args['username'] == '' ) {
|
|
UM()->form()->add_error( 'username', __( 'Please enter your username or email', 'ultimate-member' ) );
|
|
}
|
|
|
|
if ( isset( $args['user_login'] ) && $args['user_login'] == '' ) {
|
|
UM()->form()->add_error( 'user_login', __( 'Please enter your username', 'ultimate-member' ) );
|
|
}
|
|
|
|
if ( isset( $args['user_email'] ) && $args['user_email'] == '' ) {
|
|
UM()->form()->add_error( 'user_email', __( 'Please enter your email', 'ultimate-member' ) );
|
|
}
|
|
|
|
if ( isset( $args['username'] ) ) {
|
|
$authenticate = $args['username'];
|
|
$field = 'username';
|
|
if ( is_email( $args['username'] ) ) {
|
|
$is_email = true;
|
|
$data = get_user_by('email', $args['username'] );
|
|
$user_name = isset( $data->user_login ) ? $data->user_login : null;
|
|
} else {
|
|
$user_name = $args['username'];
|
|
}
|
|
} elseif ( isset( $args['user_email'] ) ) {
|
|
$authenticate = $args['user_email'];
|
|
$field = 'user_email';
|
|
$is_email = true;
|
|
$data = get_user_by('email', $args['user_email'] );
|
|
$user_name = isset( $data->user_login ) ? $data->user_login : null;
|
|
} else {
|
|
$field = 'user_login';
|
|
$user_name = $args['user_login'];
|
|
$authenticate = $args['user_login'];
|
|
}
|
|
|
|
if ( $args['user_password'] == '' ) {
|
|
UM()->form()->add_error( 'user_password', __( 'Please enter your password', 'ultimate-member' ) );
|
|
}
|
|
|
|
$user = get_user_by( 'login', $user_name );
|
|
if ( $user && wp_check_password( $args['user_password'], $user->data->user_pass, $user->ID ) ) {
|
|
UM()->login()->auth_id = username_exists( $user_name );
|
|
} else {
|
|
UM()->form()->add_error( 'user_password', __( 'Password is incorrect. Please try again.', 'ultimate-member' ) );
|
|
}
|
|
|
|
// @since 4.18 replacement for 'wp_login_failed' action hook
|
|
// see WP function wp_authenticate()
|
|
$ignore_codes = array( 'empty_username', 'empty_password' );
|
|
|
|
$user = apply_filters( 'authenticate', null, $authenticate, $args['user_password'] );
|
|
if ( is_wp_error( $user ) && ! in_array( $user->get_error_code(), $ignore_codes ) ) {
|
|
UM()->form()->add_error( $user->get_error_code(), __( $user->get_error_message(), 'ultimate-member' ) );
|
|
}
|
|
|
|
$user = apply_filters( 'wp_authenticate_user', $user, $args['user_password'] );
|
|
if ( is_wp_error( $user ) && ! in_array( $user->get_error_code(), $ignore_codes ) ) {
|
|
UM()->form()->add_error( $user->get_error_code(), __( $user->get_error_message(), 'ultimate-member' ) );
|
|
}
|
|
|
|
// if there is an error notify wp
|
|
if ( UM()->form()->has_error( $field ) || UM()->form()->has_error( $user_password ) || UM()->form()->count_errors() > 0 ) {
|
|
do_action( 'wp_login_failed', $user_name );
|
|
}
|
|
}
|
|
add_action( 'um_submit_form_errors_hook_login', 'um_submit_form_errors_hook_login', 10 );
|
|
|
|
|
|
/**
|
|
* Display the login errors from other plugins
|
|
*
|
|
* @param $args
|
|
*/
|
|
function um_display_login_errors( $args ) {
|
|
if ( UM()->form()->count_errors() > 0 ) {
|
|
$errors = UM()->form()->errors;
|
|
// hook for other plugins to display error
|
|
$error_keys = array_keys( $errors );
|
|
}
|
|
|
|
if ( isset( $args['custom_fields'] ) ) {
|
|
$custom_fields = $args['custom_fields'];
|
|
}
|
|
|
|
if ( ! empty( $error_keys ) && ! empty( $custom_fields ) ) {
|
|
foreach ( $error_keys as $error ) {
|
|
if ( trim( $error ) && ! isset( $custom_fields[ $error ] ) && ! empty( $errors[ $error ] ) ) {
|
|
$error_message = apply_filters( 'login_errors', $errors[ $error ], $error );
|
|
if ( empty( $error_message ) ) {
|
|
return;
|
|
}
|
|
echo '<p class="um-notice err um-error-code-' . esc_attr( $error ) . '"><i class="um-icon-ios-close-empty" onclick="jQuery(this).parent().fadeOut();"></i>' . $error_message . '</p>';
|
|
}
|
|
}
|
|
}
|
|
}
|
|
add_action( 'um_before_login_fields', 'um_display_login_errors' );
|
|
|
|
|
|
/**
|
|
* Login checks thru the frontend login
|
|
*
|
|
* @param $args
|
|
*/
|
|
function um_submit_form_errors_hook_logincheck( $args ) {
|
|
// Logout if logged in
|
|
if ( is_user_logged_in() ) {
|
|
wp_logout();
|
|
}
|
|
|
|
$user_id = ( isset( UM()->login()->auth_id ) ) ? UM()->login()->auth_id : '';
|
|
um_fetch_user( $user_id );
|
|
|
|
$status = um_user( 'account_status' ); // account status
|
|
switch( $status ) {
|
|
|
|
// If user can't login to site...
|
|
case 'inactive':
|
|
case 'awaiting_admin_review':
|
|
case 'awaiting_email_confirmation':
|
|
case 'rejected':
|
|
um_reset_user();
|
|
exit( wp_redirect( add_query_arg( 'err', esc_attr( $status ), UM()->permalinks()->get_current_url() ) ) );
|
|
break;
|
|
|
|
}
|
|
|
|
if ( isset( $args['form_id'] ) && $args['form_id'] == UM()->shortcodes()->core_login_form() && UM()->form()->errors && ! isset( $_POST[ UM()->honeypot ] ) ) {
|
|
exit( wp_redirect( um_get_core_page( 'login' ) ) );
|
|
}
|
|
|
|
}
|
|
add_action( 'um_submit_form_errors_hook_logincheck', 'um_submit_form_errors_hook_logincheck', 9999 );
|
|
|
|
|
|
/**
|
|
* Store last login timestamp
|
|
*
|
|
* @param $user_id
|
|
*/
|
|
function um_store_lastlogin_timestamp( $user_id ) {
|
|
update_user_meta( $user_id, '_um_last_login', current_time( 'timestamp' ) );
|
|
}
|
|
add_action( 'um_on_login_before_redirect', 'um_store_lastlogin_timestamp', 10, 1 );
|
|
|
|
|
|
/**
|
|
* @param $login
|
|
*/
|
|
function um_store_lastlogin_timestamp_( $login ) {
|
|
$user = get_user_by( 'login', $login );
|
|
um_store_lastlogin_timestamp( $user->ID );
|
|
|
|
$attempts = (int) get_user_meta( $user->ID, 'password_rst_attempts', true );
|
|
if ( $attempts ) {
|
|
//don't create meta but update if it's exists only
|
|
update_user_meta( $user->ID, 'password_rst_attempts', 0 );
|
|
}
|
|
}
|
|
add_action( 'wp_login', 'um_store_lastlogin_timestamp_' );
|
|
|
|
|
|
/**
|
|
* Login user process
|
|
*
|
|
* @param array $args
|
|
*/
|
|
function um_user_login( $args ) {
|
|
extract( $args );
|
|
|
|
$rememberme = ( isset( $args['rememberme'] ) && 1 == $args['rememberme'] && isset( $_REQUEST['rememberme'] ) ) ? 1 : 0;
|
|
|
|
if ( ( UM()->options()->get( 'deny_admin_frontend_login' ) && ! isset( $_GET['provider'] ) ) && strrpos( um_user('wp_roles' ), 'administrator' ) !== false ) {
|
|
wp_die( __( 'This action has been prevented for security measures.', 'ultimate-member' ) );
|
|
}
|
|
|
|
UM()->user()->auto_login( um_user( 'ID' ), $rememberme );
|
|
|
|
/**
|
|
* UM hook
|
|
*
|
|
* @type action
|
|
* @title um_on_login_before_redirect
|
|
* @description Hook that runs after successful login and before user is redirected
|
|
* @input_vars
|
|
* [{"var":"$user_id","type":"int","desc":"User ID"}]
|
|
* @change_log
|
|
* ["Since: 2.0"]
|
|
* @usage add_action( 'um_on_login_before_redirect', 'function_name', 10, 1 );
|
|
* @example
|
|
* <?php
|
|
* add_action( 'um_on_login_before_redirect', 'my_on_login_before_redirect', 10, 1 );
|
|
* function my_on_login_before_redirect( $user_id ) {
|
|
* // your code here
|
|
* }
|
|
* ?>
|
|
*/
|
|
do_action( 'um_on_login_before_redirect', um_user( 'ID' ) );
|
|
|
|
// Priority redirect
|
|
if ( ! empty( $args['redirect_to'] ) ) {
|
|
exit( wp_safe_redirect( $args['redirect_to'] ) );
|
|
}
|
|
|
|
// Role redirect
|
|
$after_login = um_user( 'after_login' );
|
|
if ( empty( $after_login ) ) {
|
|
exit( wp_redirect( um_user_profile_url() ) );
|
|
}
|
|
|
|
switch ( $after_login ) {
|
|
|
|
case 'redirect_admin':
|
|
exit( wp_redirect( admin_url() ) );
|
|
break;
|
|
|
|
case 'redirect_url':
|
|
/**
|
|
* UM hook
|
|
*
|
|
* @type filter
|
|
* @title um_login_redirect_url
|
|
* @description Change redirect URL after successful login
|
|
* @input_vars
|
|
* [{"var":"$url","type":"string","desc":"Redirect URL"},
|
|
* {"var":"$id","type":"int","desc":"User ID"}]
|
|
* @change_log
|
|
* ["Since: 2.0"]
|
|
* @usage
|
|
* <?php add_filter( 'um_login_redirect_url', 'function_name', 10, 2 ); ?>
|
|
* @example
|
|
* <?php
|
|
* add_filter( 'um_login_redirect_url', 'my_login_redirect_url', 10, 2 );
|
|
* function my_login_redirect_url( $url, $id ) {
|
|
* // your code here
|
|
* return $url;
|
|
* }
|
|
* ?>
|
|
*/
|
|
$redirect_url = apply_filters( 'um_login_redirect_url', um_user( 'login_redirect_url' ), um_user( 'ID' ) );
|
|
exit( wp_redirect( $redirect_url ) );
|
|
break;
|
|
|
|
case 'refresh':
|
|
exit( wp_redirect( UM()->permalinks()->get_current_url() ) );
|
|
break;
|
|
|
|
case 'redirect_profile':
|
|
default:
|
|
exit( wp_redirect( um_user_profile_url() ) );
|
|
break;
|
|
|
|
}
|
|
}
|
|
add_action( 'um_user_login', 'um_user_login', 10 );
|
|
|
|
|
|
/**
|
|
* Form processing
|
|
*
|
|
* @param $args
|
|
*/
|
|
function um_submit_form_login( $args ) {
|
|
|
|
if ( ! isset( UM()->form()->errors ) ) {
|
|
/**
|
|
* UM hook
|
|
*
|
|
* @type action
|
|
* @title um_user_login
|
|
* @description Hook that runs after successful submit login form
|
|
* @input_vars
|
|
* [{"var":"$args","type":"array","desc":"Form data"}]
|
|
* @change_log
|
|
* ["Since: 2.0"]
|
|
* @usage add_action( 'um_user_login', 'function_name', 10, 1 );
|
|
* @example
|
|
* <?php
|
|
* add_action( 'um_user_login', 'my_user_login', 10, 1 );
|
|
* function my_user_login( $args ) {
|
|
* // your code here
|
|
* }
|
|
* ?>
|
|
*/
|
|
do_action( 'um_user_login', $args );
|
|
}
|
|
|
|
/**
|
|
* UM hook
|
|
*
|
|
* @type action
|
|
* @title um_user_login_extra_hook
|
|
* @description Hook that runs after successful submit login form
|
|
* @input_vars
|
|
* [{"var":"$args","type":"array","desc":"Form data"}]
|
|
* @change_log
|
|
* ["Since: 2.0"]
|
|
* @usage add_action( 'um_user_login_extra_hook', 'function_name', 10, 1 );
|
|
* @example
|
|
* <?php
|
|
* add_action( 'um_user_login_extra_hook', 'my_user_login_extra', 10, 1 );
|
|
* function my_user_login_extra( $args ) {
|
|
* // your code here
|
|
* }
|
|
* ?>
|
|
*/
|
|
do_action( 'um_user_login_extra_hook', $args );
|
|
}
|
|
add_action( 'um_submit_form_login', 'um_submit_form_login', 10 );
|
|
|
|
|
|
/**
|
|
* Show the submit button
|
|
*
|
|
* @param $args
|
|
*/
|
|
function um_add_submit_button_to_login( $args ) {
|
|
/**
|
|
* UM hook
|
|
*
|
|
* @type filter
|
|
* @title um_login_form_button_one
|
|
* @description Change Login Form Primary button
|
|
* @input_vars
|
|
* [{"var":"$primary_btn_word","type":"string","desc":"Button text"},
|
|
* {"var":"$args","type":"array","desc":"Login Form arguments"}]
|
|
* @change_log
|
|
* ["Since: 2.0"]
|
|
* @usage
|
|
* <?php add_filter( 'um_login_form_button_one', 'function_name', 10, 2 ); ?>
|
|
* @example
|
|
* <?php
|
|
* add_filter( 'um_login_form_button_one', 'my_login_form_button_one', 10, 2 );
|
|
* function my_login_form_button_one( $primary_btn_word, $args ) {
|
|
* // your code here
|
|
* return $primary_btn_word;
|
|
* }
|
|
* ?>
|
|
*/
|
|
$primary_btn_word = apply_filters('um_login_form_button_one', $args['primary_btn_word'], $args );
|
|
|
|
if ( ! isset( $primary_btn_word ) || $primary_btn_word == '' ){
|
|
$primary_btn_word = UM()->options()->get( 'login_primary_btn_word' );
|
|
}
|
|
|
|
/**
|
|
* UM hook
|
|
*
|
|
* @type filter
|
|
* @title um_login_form_button_two
|
|
* @description Change Login Form Secondary button
|
|
* @input_vars
|
|
* [{"var":"$secondary_btn_word","type":"string","desc":"Button text"},
|
|
* {"var":"$args","type":"array","desc":"Login Form arguments"}]
|
|
* @change_log
|
|
* ["Since: 2.0"]
|
|
* @usage
|
|
* <?php add_filter( 'um_login_form_button_two', 'function_name', 10, 2 ); ?>
|
|
* @example
|
|
* <?php
|
|
* add_filter( 'um_login_form_button_two', 'my_login_form_button_two', 10, 2 );
|
|
* function my_login_form_button_two( $secondary_btn_word, $args ) {
|
|
* // your code here
|
|
* return $secondary_btn_word;
|
|
* }
|
|
* ?>
|
|
*/
|
|
$secondary_btn_word = apply_filters( 'um_login_form_button_two', $args['secondary_btn_word'], $args );
|
|
|
|
if ( ! isset( $secondary_btn_word ) || $secondary_btn_word == '' ){
|
|
$secondary_btn_word = UM()->options()->get( 'login_secondary_btn_word' );
|
|
}
|
|
|
|
$secondary_btn_url = ! empty( $args['secondary_btn_url'] ) ? $args['secondary_btn_url'] : um_get_core_page( 'register' );
|
|
/**
|
|
* UM hook
|
|
*
|
|
* @type filter
|
|
* @title um_login_form_button_two_url
|
|
* @description Change Login Form Secondary button URL
|
|
* @input_vars
|
|
* [{"var":"$secondary_btn_url","type":"string","desc":"Button URL"},
|
|
* {"var":"$args","type":"array","desc":"Login Form arguments"}]
|
|
* @change_log
|
|
* ["Since: 2.0"]
|
|
* @usage
|
|
* <?php add_filter( 'um_login_form_button_two_url', 'function_name', 10, 2 ); ?>
|
|
* @example
|
|
* <?php
|
|
* add_filter( 'um_login_form_button_two_url', 'my_login_form_button_two_url', 10, 2 );
|
|
* function my_login_form_button_two_url( $secondary_btn_url, $args ) {
|
|
* // your code here
|
|
* return $secondary_btn_url;
|
|
* }
|
|
* ?>
|
|
*/
|
|
$secondary_btn_url = apply_filters( 'um_login_form_button_two_url', $secondary_btn_url, $args ); ?>
|
|
|
|
<div class="um-col-alt">
|
|
|
|
<?php if ( isset( $args['show_rememberme'] ) && $args['show_rememberme'] ) {
|
|
UM()->fields()->checkbox( 'rememberme', __( 'Keep me signed in', 'ultimate-member' ), false ); ?>
|
|
<div class="um-clear"></div>
|
|
<?php }
|
|
|
|
if ( isset( $args['secondary_btn'] ) && $args['secondary_btn'] != 0 ) { ?>
|
|
|
|
<div class="um-left um-half">
|
|
<input type="submit" value="<?php esc_attr_e( wp_unslash( $primary_btn_word ), 'ultimate-member' ); ?>" class="um-button" id="um-submit-btn" />
|
|
</div>
|
|
<div class="um-right um-half">
|
|
<a href="<?php echo esc_url( $secondary_btn_url ); ?>" class="um-button um-alt">
|
|
<?php _e( wp_unslash( $secondary_btn_word ), 'ultimate-member' ); ?>
|
|
</a>
|
|
</div>
|
|
|
|
<?php } else { ?>
|
|
|
|
<div class="um-center">
|
|
<input type="submit" value="<?php esc_attr_e( wp_unslash( $primary_btn_word ), 'ultimate-member' ); ?>" class="um-button" id="um-submit-btn" />
|
|
</div>
|
|
|
|
<?php } ?>
|
|
|
|
<div class="um-clear"></div>
|
|
|
|
</div>
|
|
|
|
<?php
|
|
}
|
|
add_action( 'um_after_login_fields', 'um_add_submit_button_to_login', 1000 );
|
|
|
|
|
|
/**
|
|
* Display a forgot password link
|
|
*
|
|
* @param $args
|
|
*/
|
|
function um_after_login_submit( $args ) {
|
|
if ( $args['forgot_pass_link'] == 0 ) {
|
|
return;
|
|
} ?>
|
|
|
|
<div class="um-col-alt-b">
|
|
<a href="<?php echo esc_url( um_get_core_page( 'password-reset' ) ); ?>" class="um-link-alt">
|
|
<?php _e( 'Forgot your password?', 'ultimate-member' ); ?>
|
|
</a>
|
|
</div>
|
|
|
|
<?php
|
|
}
|
|
add_action( 'um_after_login_fields', 'um_after_login_submit', 1001 );
|
|
|
|
|
|
/**
|
|
* Show Fields
|
|
*
|
|
* @param $args
|
|
*/
|
|
function um_add_login_fields( $args ) {
|
|
echo UM()->fields()->display( 'login', $args );
|
|
}
|
|
add_action( 'um_main_login_fields', 'um_add_login_fields', 100 );
|