Check my account page is set before appending link Closes #2176.

This commit is contained in:
Mike Jolley 2013-01-10 15:00:47 +00:00
parent f239a05944
commit 411ee8d026
1 changed files with 8 additions and 2 deletions

View File

@ -171,9 +171,15 @@ function woocommerce_list_pages( $pages ){
* @return string * @return string
*/ */
function woocommerce_nav_menu_items( $items, $args ) { function woocommerce_nav_menu_items( $items, $args ) {
if ( get_option('woocommerce_menu_logout_link')=='yes' && strstr($items, get_permalink(woocommerce_get_page_id('myaccount'))) && is_user_logged_in() ) if ( is_user_logged_in() && get_option('woocommerce_menu_logout_link') == 'yes' ) {
$items .= '<li class="logout"><a href="'. wp_logout_url(home_url()) .'">'.__( 'Logout', 'woocommerce' ).'</a></li>';
$my_account_page_id = woocommerce_get_page_id( 'myaccount' );
$permalink = get_permalink( $my_account_page_id );
if ( $my_account_page_id && $permalink && $items && strstr( $items, $permalink ) ) {
$items .= '<li class="logout"><a href="'. wp_logout_url( home_url() ) .'">' . __( 'Logout', 'woocommerce' ) . '</a></li>';
}
}
return $items; return $items;
} }