如何在wordpress中重定向到登录页面并返回到请求的链接

<?php
/*
Template Name: Custom Login
*/

// 获取当前页面的URL
$current_url = esc_url( home_url( $_SERVER['REQUEST_URI'] ) );

// 显示登录表单
wp_login_form( array(
    'redirect' => $current_url, // 设置登录成功后的重定向URL
    'form_id' => 'custom-login-form',
    'label_username' => '用户名',
    'label_password' => '密码',
    'label_remember' => '记住我',
    'label_log_in' => '登录',
    'id_username' => 'user_login',
    'id_password' => 'user_pass',
    'id_remember' => 'rememberme',
    'id_submit' => 'wp-submit',
    'remember' => true
) );
?>

<?php
/*
Template Name: Custom Page
*/

// 检查用户是否已登录
if ( is_user_logged_in() ) {
    // 获取当前用户的ID
    $user_id = get_current_user_id();

    // 获取当前用户的角色
    $user_roles = wp_get_current_user()->roles;

    // 根据用户角色进行重定向
    if ( in_array( 'administrator', $user_roles ) ) {
        // 如果是管理员，重定向到后台管理页面
        wp_redirect( admin_url() );
        exit;
    } else {
        // 如果是其他用户，重定向到之前访问的页面
        $redirect_url = isset( $_SERVER['HTTP_REFERER'] ) ? $_SERVER['HTTP_REFERER'] : home_url();
        wp_redirect( $redirect_url );
        exit;
    }
} else {
    // 如果用户未登录，重定向到登录页面
    wp_redirect( wp_login_url() );
    exit;
}
?>