320 lines
9.7 KiB
PHP
320 lines
9.7 KiB
PHP
<?php
|
|
namespace um\core;
|
|
|
|
if ( ! defined( 'ABSPATH' ) ) {
|
|
exit;
|
|
}
|
|
|
|
if ( ! class_exists( 'um\core\Rewrite' ) ) {
|
|
|
|
/**
|
|
* Class Rewrite
|
|
* @package um\core
|
|
*/
|
|
class Rewrite {
|
|
|
|
/**
|
|
* Rewrite constructor.
|
|
*/
|
|
public function __construct() {
|
|
if ( ! defined( 'DOING_AJAX' ) ) {
|
|
add_action( 'wp_loaded', array( $this, 'maybe_flush_rewrite_rules' ) );
|
|
}
|
|
|
|
//add rewrite rules
|
|
add_filter( 'query_vars', array( &$this, 'query_vars' ) );
|
|
add_filter( 'rewrite_rules_array', array( &$this, 'add_rewrite_rules' ) );
|
|
|
|
add_action( 'template_redirect', array( &$this, 'redirect_author_page' ), 9999 );
|
|
add_action( 'template_redirect', array( &$this, 'locate_user_profile' ), 9999 );
|
|
}
|
|
|
|
/**
|
|
* Update "flush" option for reset rules on wp_loaded hook.
|
|
*/
|
|
public function reset_rules() {
|
|
update_option( 'um_flush_rewrite_rules', 1 );
|
|
}
|
|
|
|
/**
|
|
* Reset Rewrite rules if need it.
|
|
*/
|
|
public function maybe_flush_rewrite_rules() {
|
|
if ( get_option( 'um_flush_rewrite_rules' ) ) {
|
|
flush_rewrite_rules( false );
|
|
delete_option( 'um_flush_rewrite_rules' );
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Modify global query vars.
|
|
*
|
|
* @param array $public_query_vars
|
|
*
|
|
* @return array
|
|
*/
|
|
public function query_vars( $public_query_vars ) {
|
|
$public_query_vars[] = 'um_user';
|
|
$public_query_vars[] = 'um_tab';
|
|
$public_query_vars[] = 'profiletab';
|
|
$public_query_vars[] = 'subnav';
|
|
|
|
$public_query_vars[] = 'um_page';
|
|
$public_query_vars[] = 'um_action';
|
|
$public_query_vars[] = 'um_field';
|
|
$public_query_vars[] = 'um_form';
|
|
$public_query_vars[] = 'um_verify';
|
|
|
|
return $public_query_vars;
|
|
}
|
|
|
|
/**
|
|
* Add UM rewrite rules.
|
|
*
|
|
* @param array $rules
|
|
*
|
|
* @return array
|
|
*/
|
|
public function add_rewrite_rules( $rules ) {
|
|
$newrules = array();
|
|
|
|
// NGINX-config `rewrite ^/um-download/([^/]+)/([^/]+)/([^/]+)/([^/]+)/?$ /index.php?um_action=download&um_form=$1&um_field=$2&um_user=$3&um_verify=$4 last;`
|
|
$newrules['um-download/([^/]+)/([^/]+)/([^/]+)/([^/]+)/?$'] = 'index.php?um_action=download&um_form=$matches[1]&um_field=$matches[2]&um_user=$matches[3]&um_verify=$matches[4]';
|
|
|
|
if ( isset( UM()->config()->permalinks['user'] ) ) {
|
|
|
|
$user_page_id = UM()->config()->permalinks['user'];
|
|
$user = get_post( $user_page_id );
|
|
|
|
if ( isset( $user->post_name ) ) {
|
|
$user_slug = $user->post_name;
|
|
$newrules[ $user_slug . '/([^/]+)/?$' ] = 'index.php?page_id=' . $user_page_id . '&um_user=$matches[1]';
|
|
}
|
|
|
|
if ( UM()->external_integrations()->is_wpml_active() ) {
|
|
global $sitepress;
|
|
|
|
$active_languages = $sitepress->get_active_languages();
|
|
|
|
foreach ( $active_languages as $language_code => $language ) {
|
|
$lang_post_id = wpml_object_id_filter( $user_page_id, 'post', false, $language_code );
|
|
$lang_post_obj = get_post( $lang_post_id );
|
|
|
|
if ( isset( $lang_post_obj->post_name ) && $lang_post_obj->post_name !== $user->post_name ) {
|
|
$user_slug = $lang_post_obj->post_name;
|
|
$newrules[ $user_slug . '/([^/]+)/?$' ] = 'index.php?page_id=' . $lang_post_id . '&um_user=$matches[1]&lang=' . $language_code;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if ( isset( UM()->config()->permalinks['account'] ) ) {
|
|
$account_page_id = UM()->config()->permalinks['account'];
|
|
$account = get_post( $account_page_id );
|
|
|
|
if ( isset( $account->post_name ) ) {
|
|
$account_slug = $account->post_name;
|
|
$newrules[ $account_slug . '/([^/]+)?$' ] = 'index.php?page_id=' . $account_page_id . '&um_tab=$matches[1]';
|
|
}
|
|
|
|
if ( UM()->external_integrations()->is_wpml_active() ) {
|
|
global $sitepress;
|
|
|
|
$active_languages = $sitepress->get_active_languages();
|
|
|
|
foreach ( $active_languages as $language_code => $language ) {
|
|
$lang_post_id = wpml_object_id_filter( $account_page_id, 'post', false, $language_code );
|
|
$lang_post_obj = get_post( $lang_post_id );
|
|
|
|
if ( isset( $lang_post_obj->post_name ) && $lang_post_obj->post_name !== $account->post_name ) {
|
|
$account_slug = $lang_post_obj->post_name;
|
|
$newrules[ $account_slug . '/([^/]+)/?$' ] = 'index.php?page_id=' . $lang_post_id . '&um_user=$matches[1]&lang=' . $language_code;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return $newrules + $rules;
|
|
}
|
|
|
|
/**
|
|
* Author page to user profile redirect.
|
|
*/
|
|
public function redirect_author_page() {
|
|
if ( is_author() && UM()->options()->get( 'author_redirect' ) ) {
|
|
$id = get_query_var( 'author' );
|
|
um_fetch_user( $id );
|
|
wp_safe_redirect( um_user_profile_url() );
|
|
exit;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Getting the user_id based on the User Profile slug like when Base Permalink setting equals 'user_login'.
|
|
*
|
|
* @since 2.7.0
|
|
*
|
|
* @return bool|int|mixed
|
|
*/
|
|
private function get_user_id_by_user_login_slug() {
|
|
$permalink_base = UM()->options()->get( 'permalink_base' );
|
|
if ( 'custom_meta' === $permalink_base ) {
|
|
$custom_meta = UM()->options()->get( 'permalink_base_custom_meta' );
|
|
if ( empty( $custom_meta ) ) {
|
|
// Set default permalink base if custom meta is empty.
|
|
$permalink_base = 'user_login';
|
|
} else {
|
|
// Ignore username slug if custom meta slug exists.
|
|
$user_id = username_exists( um_queried_user() );
|
|
$custom_permalink = get_user_meta( $user_id, 'um_user_profile_url_slug_' . $permalink_base, true );
|
|
if ( ! empty( $custom_permalink ) && um_queried_user() !== $custom_permalink ) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
$user_id = username_exists( um_queried_user() );
|
|
//Try
|
|
if ( ! $user_id ) {
|
|
// Search by Profile Slug
|
|
$args = array(
|
|
'fields' => 'ids',
|
|
'meta_query' => array(
|
|
array(
|
|
'key' => 'um_user_profile_url_slug_' . $permalink_base,
|
|
'value' => strtolower( um_queried_user() ),
|
|
'compare' => '=',
|
|
),
|
|
),
|
|
'number' => 1,
|
|
);
|
|
|
|
$ids = new \WP_User_Query( $args );
|
|
if ( $ids->total_users > 0 ) {
|
|
$user_id = current( $ids->get_results() );
|
|
}
|
|
}
|
|
|
|
// Try nice name
|
|
if ( ! $user_id ) {
|
|
$slug = um_queried_user();
|
|
$slug = str_replace( '.', '-', $slug );
|
|
$the_user = get_user_by( 'slug', $slug );
|
|
if ( isset( $the_user->ID ) ) {
|
|
$user_id = $the_user->ID;
|
|
}
|
|
|
|
if ( ! $user_id ) {
|
|
$user_id = UM()->user()->user_exists_by_email_as_username( um_queried_user() );
|
|
}
|
|
|
|
if ( ! $user_id ) {
|
|
$user_id = UM()->user()->user_exists_by_email_as_username( $slug );
|
|
}
|
|
}
|
|
|
|
return $user_id;
|
|
}
|
|
|
|
/**
|
|
* Locate/display a profile.
|
|
*/
|
|
public function locate_user_profile() {
|
|
$permalink_base = UM()->options()->get( 'permalink_base' );
|
|
if ( 'custom_meta' === $permalink_base ) {
|
|
$custom_meta = UM()->options()->get( 'permalink_base_custom_meta' );
|
|
if ( empty( $custom_meta ) ) {
|
|
// Set default permalink base if custom meta is empty.
|
|
$permalink_base = 'user_login';
|
|
}
|
|
}
|
|
|
|
if ( um_queried_user() && um_is_core_page( 'user' ) ) {
|
|
if ( 'user_login' === $permalink_base ) {
|
|
$user_id = $this->get_user_id_by_user_login_slug();
|
|
}
|
|
|
|
if ( 'user_id' === $permalink_base && UM()->common()->users()::user_exists( um_queried_user() ) ) {
|
|
$user_id = um_queried_user();
|
|
}
|
|
|
|
if ( 'hash' === $permalink_base ) {
|
|
$user_id = UM()->user()->user_exists_by_hash( um_queried_user() );
|
|
}
|
|
|
|
if ( 'custom_meta' === $permalink_base ) {
|
|
$user_id = UM()->user()->user_exists_by_custom_meta( um_queried_user() );
|
|
if ( ! $user_id ) {
|
|
// Try user_login by default.
|
|
$user_id = $this->get_user_id_by_user_login_slug();
|
|
}
|
|
}
|
|
|
|
if ( in_array( $permalink_base, array( 'name', 'name_dash', 'name_dot', 'name_plus' ), true ) ) {
|
|
$user_id = UM()->user()->user_exists_by_name( um_queried_user() );
|
|
}
|
|
|
|
/** USER EXISTS SET USER AND CONTINUE **/
|
|
|
|
if ( ! empty( $user_id ) ) {
|
|
um_set_requested_user( $user_id );
|
|
/**
|
|
* Fires after setting requested user.
|
|
*
|
|
* @param {int} $user_id Requested User ID.
|
|
*
|
|
* @since 1.3.x
|
|
* @hook um_access_profile
|
|
*
|
|
* @example <caption>Some action on user access profile and requested user isset.</caption>
|
|
* add_action( 'um_access_profile', 'my_access_profile', 10, 1 );
|
|
* function my_access_profile( $user_id ) {
|
|
* // your code here
|
|
* }
|
|
*/
|
|
do_action( 'um_access_profile', $user_id );
|
|
} else {
|
|
wp_safe_redirect( um_get_core_page( 'user' ) );
|
|
exit;
|
|
}
|
|
} elseif ( um_is_core_page( 'user' ) ) {
|
|
if ( is_user_logged_in() ) { // just redirect to their profile
|
|
$query = UM()->permalinks()->get_query_array();
|
|
|
|
$url = um_user_profile_url( um_user( 'ID' ) );
|
|
|
|
if ( $query ) {
|
|
foreach ( $query as $key => $val ) {
|
|
$url = add_query_arg( $key, $val, $url );
|
|
}
|
|
}
|
|
wp_safe_redirect( $url );
|
|
exit;
|
|
}
|
|
|
|
/**
|
|
* Filters the redirect URL from user profile for not logged-in user.
|
|
*
|
|
* @param {string} $url Redirect URL. By default, it's a home page.
|
|
*
|
|
* @return {string} Redirect URL.
|
|
*
|
|
* @since 1.3.x
|
|
* @hook um_locate_user_profile_not_loggedin__redirect
|
|
*
|
|
* @example <caption>Change redirect URL from user profile for not logged-in user to WordPress native login.</caption>
|
|
* function my_user_profile_not_loggedin__redirect( $url ) {
|
|
* // your code here
|
|
* $url = wp_login_url();
|
|
* return $url;
|
|
* }
|
|
* add_filter( 'um_locate_user_profile_not_loggedin__redirect', 'my_user_profile_not_loggedin__redirect' );
|
|
*/
|
|
$redirect_to = apply_filters( 'um_locate_user_profile_not_loggedin__redirect', home_url() );
|
|
if ( ! empty( $redirect_to ) ) {
|
|
um_safe_redirect( $redirect_to );
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|