如何在wordpress中隐藏HTML元素，并且只对某些角色可见？

function hide_element_for_roles() {
    if ( is_user_logged_in() ) {
        $user = wp_get_current_user();
        $allowed_roles = array( 'editor', 'administrator' ); // 可见的角色列表
        if ( array_intersect( $allowed_roles, $user->roles ) ) {
            ?>
            <style type="text/css">
                .hidden-element {
                    display: none;
                }
            </style>
            <?php
        }
    }
}
add_action( 'wp_head', 'hide_element_for_roles' );