禁止指定用户登录（小黑屋） - wordpress 开发

1
//在资料页面添加选项
2
function dj_rc_admin_init(){
3
    // 编辑用户资料
4
    add_action( 'edit_user_profile', 'dj_rc_edit_user_profile' );
5
    add_action( 'edit_user_profile_update', 'dj_rc_edit_user_profile_update' );
6
}
7
add_action('admin_init', 'dj_rc_admin_init' );
8

9
//在个人资料页面添加一个复选框
10
function dj_rc_edit_user_profile() {
11
    if ( !current_user_can( 'edit_users' ) ) {
12
        return;
13
    }
14
    global $user_id;
15
    // 用户不能禁止自己
16
    $current_user = wp_get_current_user();
17
    $current_user_id = $current_user->ID;
18
    if ( $current_user_id == $user_id ) {
19
        return;
20
    }
21
    ?>
22
    <h3>权限设置</h3>
23
    <table class="form-table">
24
    <tr>
25
        <th scope="row">禁止用户登录</th>
26
        <td><label for="dj_rc_ban"><input name="dj_rc_ban" type="checkbox" id="dj_rc_ban"
27
        <?php if (dj_rc_is_user_banned( $user_id )){echo 'checked="checked"';} ?> /> 勾起来，盘Ta，什么玩意儿……</label></td>
28
    </tr>
29
    </table>
30
    <?php
31
}
32

33
//添加一个函数来将这个选项的值保存到数据库中
34
function dj_rc_edit_user_profile_update() {
35
    if ( !current_user_can( 'edit_users' ) ) {
36
        return;
37
    }
38
    global $user_id;
39
    // 用户不能禁止自己
40
    $current_user    = wp_get_current_user();
41
    $current_user_id = $current_user->ID;
42
    if ( $current_user_id == $user_id ) {
43
        return;
44
    }
45
    // 锁定
46
    if( isset( $_POST['dj_rc_ban'] ) && $_POST['dj_rc_ban'] = 'on' ) {
47
        dj_rc_ban_user( $user_id );
48
    } else { // 解锁
49
        dj_rc_unban_user( $user_id );
50
    }
51
}
52

53
//禁止用户
54
function dj_rc_ban_user( $user_id ) {
55
    $old_status = dj_rc_is_user_banned( $user_id );
56
    // 更新状态
57
    if ( !$old_status ) {
58
        update_user_option( $user_id, 'dj_rc_banned', true, false );
59
    }
60
}
61

62
//解禁用户
63
function dj_rc_unban_user( $user_id ) {
64
    $old_status = dj_rc_is_user_banned( $user_id );
65
    // 更新状态
66
    if ( $old_status ) {
67
        update_user_option( $user_id, 'dj_rc_banned', false, false );
68
    }
69
}
70

71
//判断用户是否被禁止
72
function dj_rc_is_user_banned( $user_id ) {
73
    return get_user_option( 'dj_rc_banned', $user_id, false );
74
}
75

76
//阻止已禁止的用户登录
77
function dj_rc_authenticate_user( $user ) {
78
    if ( is_wp_error( $user ) ) {
79
        return $user;
80
    }
81
    // 如果用户被禁止，则返回错误提示，这里是显示给登录用户看的，错误提示可根据自己的喜好修改。
82
    $banned = get_user_option( 'dj_rc_banned', $user->ID, false );
83
    if ( $banned ) {
84
        return new WP_Error( 'dj_rc_banned', __('抱歉，您已被禁止登录！请联系站长解禁。', 'rc') );
85
    }
86
    return $user;
87
}
88
//将该函数挂载到 wp_authenticate_user 钩子
89
add_filter( 'wp_authenticate_user', 'dj_rc_authenticate_user', 1 );

1
add_filter( 'rest_api_init', function($wp_rest_server){
2
  $array = array(3,9);//关入小黑屋用户id，英文,隔开
3
    $user_id = get_current_user_id();
4
  if(in_array($user_id,$array)){
5
        $allow_cookie = apply_filters('b2_login_cookie', b2_get_option('normal_login','allow_cookie'));
6
        if((string)$allow_cookie === '1'){
7
            wp_logout();
8
        }
9
    wp_die('您已被关入小黑屋，请联系管理员解封');
10
  }
11
}, 1 );