/**
 * WordPress 禁止某些用户评论 - 龙笑天下
 * https://www.ilxtx.com/wp-ban-user-comment.html
 */
//在资料页面添加选项
function lxtx_usercmt_admin_init(){ 
	// 编辑用户资料
	add_action( 'edit_user_profile', 'lxtx_usercmt_edit_user_profile' );
	add_action( 'edit_user_profile_update', 'lxtx_usercmt_user_profile_update' ); 
}
add_action('admin_init', 'lxtx_usercmt_admin_init' );
//在个人资料页面添加一个复选框
function lxtx_usercmt_edit_user_profile() {
	if ( !current_user_can( 'edit_users' ) ) {
		return;
	} 
	global $user_id; 
	// 用户不能禁止自己
	$current_user = wp_get_current_user();
	$current_user_id = $current_user->ID;
	if ( $current_user_id == $user_id ) {
		return;
	}
	?>
	<!-- <h3>权限设置</h3> -->
	<table class="form-table">
	<tr>
		<th scope="row">禁止用户评论</th>
		<td><label for="lxtx_usercmt_ban"><input name="lxtx_usercmt_ban" type="checkbox" id="lxtx_usercmt_ban" 
		<?php if (lxtx_usercmt_is_user_banned( $user_id )){echo 'checked="checked"';} ?> /> 请谨慎操作，选中则禁止！</label></td>
	</tr>
	</table>
	<?php
}
//添加一个函数来将这个选项的值保存到数据库中
function lxtx_usercmt_user_profile_update() { 
	if ( !current_user_can( 'edit_users' ) ) {
		return;
	} 
	global $user_id; 
	// 用户不能禁止自己
	$current_user    = wp_get_current_user();
	$current_user_id = $current_user->ID;
	if ( $current_user_id == $user_id ) {
		return;
	} 
	// 锁定
	if( isset( $_POST['lxtx_usercmt_ban'] ) && $_POST['lxtx_usercmt_ban'] = 'on' ) {
		lxtx_usercmt_ban_user( $user_id );
	} else { // 解锁
		lxtx_usercmt_unban_user( $user_id );
	} 
}
//禁止用户
function lxtx_usercmt_ban_user( $user_id ) { 
	$old_status = lxtx_usercmt_is_user_banned( $user_id ); 
	// 更新状态
	if ( !$old_status ) {
		update_user_option( $user_id, 'lxtx_usercmt_banned', true, false );
	}
}
//解禁用户
function lxtx_usercmt_unban_user( $user_id ) { 
	$old_status = lxtx_usercmt_is_user_banned( $user_id ); 
	// 更新状态
	if ( $old_status ) {
		update_user_option( $user_id, 'lxtx_usercmt_banned', false, false );
	}
}
//判断用户是否被禁止
function lxtx_usercmt_is_user_banned( $user_id ) {
	return get_user_option( 'lxtx_usercmt_banned', $user_id, false );
}