Добавьте план членства пользователя в качестве основного класса в WooCommerce ⇐ CSS

[Anonymous]

Мне нужно добавить в качестве основного класса идентификатор членства пользователя в WooCommerce.
Для этого я использую следующий код:

Код: Выделить всё

add_filter( 'body_class', 'am_add_slug_body_class' );

function am_add_slug_body_class( $classes ) {
global $post;
if ( isset( $post ) ) {
$classes[] = $post->post_type . '-' . $post->post_name;
}
return $classes;
}

/* Add Logged-Out Body Class */

add_filter('body_class','am_logged_in_filter');

function am_logged_in_filter($classes) {
if( is_user_logged_in() ) {
$classes[] = '';
} else {
$classes[] = 'logged-out';
}
return $classes;
}

/* Add Gold Membership to Body Class */

add_filter('body_class','am_membership_class_filter');

function am_membership_class_filter($classes) {
if( wc_memberships_is_user_member( null, 'prime' ) ) {
$classes[] = 'prime';
}

return $classes;

if( wc_memberships_is_user_member( null, 'platinum' ) ) {
$classes[] = 'platinum';
}
return $classes;

if( wc_memberships_is_user_member( null, 'pro' ) ) {
$classes[] = 'pro';
}
return $classes;

}

Но это работает только с первым планом членства «Прайм» (Премиум), так как планы Platinum и Pro не работают.
Есть ли какая-нибудь проблема в этом коде? Почему мой код работает некорректно?

Подробнее здесь: https://stackoverflow.com/questions/784 ... oocommerce