【WordPressカスタマイズ】複数のカスタム投稿タイプの記事を1つのカレンダーに表示したい

どーも黒ブタです。

WordPressには標準でget_calendar()というテンプレートタグでカレンダーを表示する機能があるのですが、これに出力される記事はポストタイプがpost(通常投稿)のみで、カスタム投稿には対応していません。

「各カスタム投稿毎にカレンダーを別表示したい」などのカスタマイズ用の記事はいくつか見つかったのですが、僕が今回やりたかったのは

“ 複数のカスタム投稿の記事を1つのカレンダーに表示したい ”

というもので、Google先生で探してもめぼしい記事は見つからなかった為、自分でカスタマイズすることにしました。

カレンダー出力に関するコアファイルですが、wp-includeフォルダ内にある「general-template.php」の2050行目付近に記述があります。
こちらのコードを参考にして独自テーマのfunctions.phpに追加していきます。

「general-template.php」の2049行目〜2267行目までのコードの間に「post_type = 'post'」の記述が4箇所ほどあります。
※(2073行目、2117行目、2125行目、2191行目)

上記コードの書き換えをすることによって複数の投稿タイプの記事を1つのカレンダーに表示させることができます。

こちらのコードはSELECT文を使ってデータベースからデータを取得していますので、書き換え方法は下記となります。

post_type = 'post'
↓
(post_type = 'post' OR post_type = 'カスタム投稿タイプ1' OR post_type = 'カスタム投稿タイプ2')

4箇所まとめて変更したいので、変数に格納して呼び出します。
今回カスタム投稿タイプ1はnews、カスタム投稿タイプ2はblogとします。

カレンダーに追加しても、日付アーカイブにも出力されなければならないので、日付アーカイブにもカスタム投稿が反映されるように最初に追加記述します。

最後に、カレンダーを出力したい箇所に「<?php get_custom_calendar(); ?>」を埋め込みます。

//日付アーカイブにカスタム投稿追加
function add_post_date_archive( $wp_query ) {
 if ($wp_query->is_main_query() && $wp_query->is_date()) {
   $wp_query->set( 'post_type', array( 'post','news','blog'));
 }
}
add_action( 'pre_get_posts', 'add_post_date_archive' , 10 , 1);

//カレンダーにカスタム投稿を複数追加
function get_custom_calendar( $initial = true, $echo = true ) {
 //追加したいカスタム投稿タイプを変数に格納
 $custom_calendar_post_type = "(post_type = 'post' OR post_type = 'news' OR post_type = 'blog')";

 global $wpdb, $m, $monthnum, $year, $wp_locale, $posts;
 $key = md5( $m . $monthnum . $year );
 $cache = wp_cache_get( 'get_calendar', 'calendar' );

 if ( $cache && is_array( $cache ) && isset( $cache[ $key ] ) ) {
  /** This filter is documented in wp-includes/general-template.php */
  $output = apply_filters( 'get_calendar', $cache[ $key ] );

  if ( $echo ) {
   echo $output;
  return;
 }

 return $output;
 }

 if ( ! is_array( $cache ) ) {
  $cache = array();
 }

 // Quick check. If we have no posts at all, abort!
 if ( ! $posts ) {
  $gotsome = $wpdb->get_var( "SELECT 1 as test FROM $wpdb->posts WHERE $custom_calendar_post_type AND post_status = 'publish' LIMIT 1" );
  if ( ! $gotsome ) {
   $cache[ $key ] = '';
   wp_cache_set( 'get_calendar', $cache, 'calendar' );
   return;
  }
 }

 if ( isset( $_GET['w'] ) ) {
  $w = (int) $_GET['w'];
 }
 // week_begins = 0 stands for Sunday
 $week_begins = (int) get_option( 'start_of_week' );

 // Let's figure out when we are
 if ( ! empty( $monthnum ) && ! empty( $year ) ) {
  $thismonth = zeroise( intval( $monthnum ), 2 );
  $thisyear = (int) $year;
 } elseif ( ! empty( $w ) ) {
  // We need to get the month from MySQL
  $thisyear = (int) substr( $m, 0, 4 );
  //it seems MySQL's weeks disagree with PHP's
  $d = ( ( $w - 1 ) * 7 ) + 6;
  $thismonth = $wpdb->get_var( "SELECT DATE_FORMAT((DATE_ADD('{$thisyear}0101', INTERVAL $d DAY) ), '%m')" );
 } elseif ( ! empty( $m ) ) {
  $thisyear = (int) substr( $m, 0, 4 );
  if ( strlen( $m ) < 6 ) {
   $thismonth = '01';
  } else {
   $thismonth = zeroise( (int) substr( $m, 4, 2 ), 2 );
  }
 } else {
  $thisyear = current_time( 'Y' );
  $thismonth = current_time( 'm' );
 }

 $unixmonth = mktime( 0, 0, 0, $thismonth, 1, $thisyear );
 $last_day = date( 't', $unixmonth );

 // Get the next and previous month and year with at least one post
 $previous = $wpdb->get_row(
  "SELECT MONTH(post_date) AS month, YEAR(post_date) AS year
  FROM $wpdb->posts
  WHERE post_date < '$thisyear-$thismonth-01'
  AND $custom_calendar_post_type AND post_status = 'publish'
  ORDER BY post_date DESC
  LIMIT 1"
 );
 $next = $wpdb->get_row(
  "SELECT MONTH(post_date) AS month, YEAR(post_date) AS year
  FROM $wpdb->posts
  WHERE post_date > '$thisyear-$thismonth-{$last_day} 23:59:59'
  AND $custom_calendar_post_type AND post_status = 'publish'
  ORDER BY post_date ASC
  LIMIT 1"
 );

 /* translators: Calendar caption: 1: month name, 2: 4-digit year */
 $calendar_caption = _x( '%1$s %2$s', 'calendar caption' );
 $calendar_output = '<table id="wp-calendar">
 <caption>' . sprintf(
  $calendar_caption,
  $wp_locale->get_month( $thismonth ),
  date( 'Y', $unixmonth )
 ) . '</caption>
 <thead>
 <tr>';

 $myweek = array();

 for ( $wdcount = 0; $wdcount <= 6; $wdcount++ ) {
  $myweek[] = $wp_locale->get_weekday( ( $wdcount + $week_begins ) % 7 );
 }

 foreach ( $myweek as $wd ) {
  $day_name = $initial ? $wp_locale->get_weekday_initial( $wd ) : $wp_locale->get_weekday_abbrev( $wd );
  $wd = esc_attr( $wd );
  $calendar_output .= "\n\t\t<th scope=\"col\" title=\"$wd\">$day_name</th>";
 }

 $calendar_output .= '
  </tr>
  </thead>

  <tfoot>
  <tr>
 ';

 if ( $previous ) {
  $calendar_output .= "\n\t\t" . '<td colspan="3" id="prev"><a href="' . get_month_link( $previous->year, $previous->month ) . '">&laquo; ' .
  $wp_locale->get_month_abbrev( $wp_locale->get_month( $previous->month ) ) .
  '</a></td>';
 } else {
  $calendar_output .= "\n\t\t" . '<td colspan="3" id="prev" class="pad">&nbsp;</td>';
 }

 $calendar_output .= "\n\t\t" . '<td class="pad">&nbsp;</td>';

 if ( $next ) {
  $calendar_output .= "\n\t\t" . '<td colspan="3" id="next"><a href="' . get_month_link( $next->year, $next->month ) . '">' .
  $wp_locale->get_month_abbrev( $wp_locale->get_month( $next->month ) ) .
  ' &raquo;</a></td>';
 } else {
  $calendar_output .= "\n\t\t" . '<td colspan="3" id="next" class="pad">&nbsp;</td>';
 }

 $calendar_output .= '
  </tr>
  </tfoot>

  <tbody>
  <tr>
 ';

 $daywithpost = array();

 // Get days with posts
 $dayswithposts = $wpdb->get_results(
  "SELECT DISTINCT DAYOFMONTH(post_date)
  FROM $wpdb->posts WHERE post_date >= '{$thisyear}-{$thismonth}-01 00:00:00'
  AND $custom_calendar_post_type AND post_status = 'publish'
  AND post_date <= '{$thisyear}-{$thismonth}-{$last_day} 23:59:59'",
  ARRAY_N
 );
 if ( $dayswithposts ) {
  foreach ( (array) $dayswithposts as $daywith ) {
   $daywithpost[] = $daywith[0];
  }
 }

 // See how much we should pad in the beginning
 $pad = calendar_week_mod( date( 'w', $unixmonth ) - $week_begins );
 if ( 0 != $pad ) {
  $calendar_output .= "\n\t\t" . '<td colspan="' . esc_attr( $pad ) . '" class="pad">&nbsp;</td>';
 }

 $newrow = false;
 $daysinmonth = (int) date( 't', $unixmonth );

 for ( $day = 1; $day <= $daysinmonth; ++$day ) {
  if ( isset( $newrow ) && $newrow ) {
   $calendar_output .= "\n\t</tr>\n\t<tr>\n\t\t";
  }
  $newrow = false;

  if ( $day == current_time( 'j' ) &&
  $thismonth == current_time( 'm' ) &&
  $thisyear == current_time( 'Y' ) ) {
   $calendar_output .= '<td id="today">';
  } else {
   $calendar_output .= '<td>';
  }

  if ( in_array( $day, $daywithpost ) ) {
   // any posts today?
   $date_format = date( _x( 'F j, Y', 'daily archives date format' ), strtotime( "{$thisyear}-{$thismonth}-{$day}" ) );
   /* translators: Post calendar label. %s: Date */
   $label = sprintf( __( 'Posts published on %s' ), $date_format );
   $calendar_output .= sprintf(
   '<a href="%s" aria-label="%s">%s</a>',
   get_day_link( $thisyear, $thismonth, $day ),
   esc_attr( $label ),
   $day
   );
  } else {
   $calendar_output .= $day;
  }
  $calendar_output .= '</td>';

  if ( 6 == calendar_week_mod( date( 'w', mktime( 0, 0, 0, $thismonth, $day, $thisyear ) ) - $week_begins ) ) {
   $newrow = true;
  }
 }

 $pad = 7 - calendar_week_mod( date( 'w', mktime( 0, 0, 0, $thismonth, $day, $thisyear ) ) - $week_begins );
 if ( $pad != 0 && $pad != 7 ) {
  $calendar_output .= "\n\t\t" . '<td class="pad" colspan="' . esc_attr( $pad ) . '">&nbsp;</td>';
 }
 $calendar_output .= "\n\t</tr>\n\t</tbody>\n\t</table>";

 $cache[ $key ] = $calendar_output;
 wp_cache_set( 'get_calendar', $cache, 'calendar' );

 if ( $echo ) {
  /**
  * Filters the HTML calendar output.
  *
  * @since 3.0.0
  *
  * @param string $calendar_output HTML output of the calendar.
  */
  echo apply_filters( 'get_calendar', $calendar_output );
  return;
 }
 /** This filter is documented in wp-includes/general-template.php */
 return apply_filters( 'get_calendar', $calendar_output );
}