Loginout sur le menu

Ca marche bien et c’est pratique de mettre ce loginout dans le menu.

Voici le code et le LIEU où je l’ai trouvé

// Add login and logout to custom menu automatically
function __wps__add_nav_menu_items( $items, $args ) {
  global $current_user;
  if ( is_user_logged_in() ) {
    $items .= '<li id="menu-item-logout"><a href="'.esc_url( home_url( '/' ) ).'wp-login.php?action=logout">'.__('Logout', WPS_TEXT_DOMAIN).'</a></li>';
  } else {
    $items .= '<li id="menu-item-login"><a href="'.esc_url( home_url( '/' ) ).'wp-login.php?redirect_to='.urlencode($_SERVER['REQUEST_URI']).'">'.__('Login', WPS_TEXT_DOMAIN).'</a></li>';
  }
  return $items;
}
add_filter('wp_nav_menu_items','__wps__add_nav_menu_items', 10, 2);