mirror of
https://github.com/10h30/ultimatemember.git
synced 2026-07-11 18:56:10 +09:00
ultimatemember
111 lines
2.3 KiB
PHP
111 lines
2.3 KiB
PHP
<?php
|
|
|
|
class UM_Rewrite {
|
|
|
|
function __construct() {
|
|
|
|
add_filter('query_vars', array(&$this, 'query_vars'), 10, 1 );
|
|
|
|
add_action('init', array(&$this, 'rewrite_rules') );
|
|
|
|
add_action('template_redirect', array(&$this, 'locate_user_profile'), 9999 );
|
|
|
|
}
|
|
|
|
/***
|
|
*** @modify global query vars
|
|
***/
|
|
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[] = 'members_page';
|
|
return $public_query_vars;
|
|
}
|
|
|
|
/***
|
|
*** @setup rewrite rules
|
|
***/
|
|
function rewrite_rules(){
|
|
|
|
global $ultimatemember;
|
|
|
|
if ( isset( $ultimatemember->permalinks->core['user'] ) ) {
|
|
|
|
$user_page_id = $ultimatemember->permalinks->core['user'];
|
|
$account_page_id = $ultimatemember->permalinks->core['account'];
|
|
|
|
$user = get_post($user_page_id);
|
|
$user_slug = $user->post_name;
|
|
|
|
$account = get_post($account_page_id);
|
|
$account_slug = $account->post_name;
|
|
|
|
add_rewrite_rule(
|
|
'^'.$user_slug.'/([^/]*)$',
|
|
'index.php?page_id='.$user_page_id.'&um_user=$matches[1]',
|
|
'top'
|
|
);
|
|
|
|
add_rewrite_rule(
|
|
'^'.$account_slug.'/([^/]*)$',
|
|
'index.php?page_id='.$account_page_id.'&um_tab=$matches[1]',
|
|
'top'
|
|
);
|
|
|
|
flush_rewrite_rules();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
/***
|
|
*** @locate/display a profile
|
|
***/
|
|
function locate_user_profile() {
|
|
global $post, $ultimatemember;
|
|
|
|
if ( um_queried_user() && um_is_core_page('user') ) {
|
|
|
|
if ( um_get_option('permalink_base') == 'user_login' ) {
|
|
|
|
$user_id = username_exists( um_queried_user() );
|
|
|
|
// Try nice name
|
|
if ( !$user_id ) {
|
|
$the_user = get_user_by( 'slug', um_queried_user() );
|
|
if ( isset( $the_user->ID ) ){
|
|
$user_id = $the_user->ID;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
if ( um_get_option('permalink_base') == 'user_id' ) {
|
|
$user_id = $ultimatemember->user->user_exists_by_id( um_queried_user() );
|
|
|
|
}
|
|
|
|
if ( um_get_option('permalink_base') == 'name' ) {
|
|
$user_id = $ultimatemember->user->user_exists_by_name( um_queried_user() );
|
|
|
|
}
|
|
|
|
/** USER EXISTS SET USER AND CONTINUE **/
|
|
|
|
if ( $user_id ) {
|
|
|
|
um_set_requested_user( $user_id );
|
|
|
|
} else {
|
|
|
|
exit( wp_redirect( um_get_core_page('user') ) );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} |