php WooCommerce：添加登录和注销链接到菜单

Posted 2021-05-04

// Add login and logout links to menu locations
add_filter( 'wp_nav_menu_items', 'login_logout_links', 10, 2 );
function login_logout_links( $items, $args ) {
  if (is_user_logged_in() && $args->theme_location == 'primary') {
	  $items .= '<li><a href="'. wp_logout_url( get_permalink( wc_get_page_id( 'myaccount' ) ) ) .'">Logout</a></li>';
  }
  elseif (!is_user_logged_in() && $args->theme_location == 'primary') {
	  $items .= '<li><a href="' . get_permalink( wc_get_page_id( 'myaccount' ) ) . '">Login</a></li>';
  }
  return $items;
}