WordPressのログインの有無でコンテンツ切り替え

概要

WordPressで会員サイトを作る際に
ログインの有無で表示を変える必要があったので
ショートコードを作って対応したのでその記録です。

実際のコード

ログインしているユーザーは見られる

functions.php

function display_login($atts, $content = null) {
  // ユーザーがログインしている場合は、コンテンツを表示
  if (is_user_logged_in()) {
    return do_shortcode($content);
  } else {
  }
}
add_shortcode('login', 'display_login');

ログインしていないユーザーは見られる

functions.php

function display_no_login($atts, $content = null) {
  // ユーザーがログインしていない場合は、コンテンツを表示
  if (!is_user_logged_in()) {
    return do_shortcode($content);
  } else {
  }
}
add_shortcode('no_login', 'display_no_login');