get_calendar()の週末(土日)にクラスを追加
WP標準のカレンダーにクラスでスタイルを振り分けようとした時にハッと気づいた。
標準のカレンダーウィジェットやテンプレート関数get_calendar()は、土日のハイライトに対応していない。
お客さんのオフィシャルブログで、この対応を行わなければならなかったため、改造を試みた。
作成前の構想はこう。
- get_calendar()の基本的な機能は残しておく。
- なので、素体コードはgeneral-templateからコピペ。
- ウィジェットも対応しなければいけないのでget_calendarでフック。
add_filter()でget_calendar()を別コードへ上書きする。 - また、get_calendar()は内部でapply_filtersで自身を呼び出しているので、その部分を新たに作る関数名に置換。
- <th>と<td>のタグの中には、必ず曜日ないし日付が出力されている。
＝それ用の変数があるということなので、そいつを見てsunないしsatを出力する条件を判定する。
以上、単純明快な方法。
で、実際に作成した結果。
/*****************************************************************/
/* get_calendarの週末対応版 */
/*****************************************************************/
function get_calendar_add_weekend($initial = true, $echo = true) {
global $wpdb, $m, $monthnum, $year, $wp_locale, $posts;
$key = md5( $m . $monthnum . $year );
if ( $cache = wp_cache_get( 'get_calendar_add_weekend', 'calendar' ) ) {
if ( is_array($cache) && isset( $cache[ $key ] ) ) {
if ( $echo ) {
echo apply_filters( 'get_calendar_add_weekend', $cache[$key] );
return;
} else {
return apply_filters( 'get_calendar_add_weekend', $cache[$key] );
}
}
}
if ( !is_array($cache) )
$cache = array();
if ( !$posts ) {
$gotsome = $wpdb->get_var("SELECT 1 as test FROM $wpdb->posts WHERE post_type = 'post' AND post_status = 'publish' LIMIT 1");
if ( !$gotsome ) {
$cache[ $key ] = '';
wp_cache_set( 'get_calendar_add_weekend', $cache, 'calendar' );
return;
}
}
if ( isset($_GET['w']) )
$w = ''.intval($_GET['w']);
$week_begins = intval(get_option('start_of_week'));
if ( !empty($monthnum) && !empty($year) ) {
$thismonth = ''.zeroise(intval($monthnum), 2);
$thisyear = ''.intval($year);
} elseif ( !empty($w) ) {
$thisyear = ''.intval(substr($m, 0, 4));
$d = (($w - 1) * 7) + 6;
$thismonth = $wpdb->get_var("SELECT DATE_FORMAT((DATE_ADD('{$thisyear}0101', INTERVAL $d DAY) ), '%m')");
} elseif ( !empty($m) ) {
$thisyear = ''.intval(substr($m, 0, 4));
if ( strlen($m) < 6 )
$thismonth = '01';
else
$thismonth = ''.zeroise(intval(substr($m, 4, 2)), 2);
} else {
$thisyear = gmdate('Y', current_time('timestamp'));
$thismonth = gmdate('m', current_time('timestamp'));
}
$unixmonth = mktime(0, 0 , 0, $thismonth, 1, $thisyear);
$last_day = date('t', $unixmonth);
$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 post_type = 'post' 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 post_type = 'post' AND post_status = 'publish'
ORDER BY post_date ASC
LIMIT 1");
$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 = (true == $initial) ? $wp_locale->get_weekday_initial($wd) : $wp_locale->get_weekday_abbrev($wd);
$wd = esc_attr($wd);
//曜日からクラスにする
$day_class = "";
if($wd == "日曜日"){
$day_class = " class=\"sun\"";
}else if($wd == "土曜日"){
$day_class = " class=\"sat\"";
}else{
$day_class = "";
}
//クラスの変数を追加
//$calendar_output .= "\n\t\t<th scope=\"col\" title=\"$wd\">$day_name</th>";
$calendar_output .= "\n\t\t<th scope=\"col\" title=\"$wd\"$day_class>$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) . '">« ' . $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"> </td>';
}
$calendar_output .= "\n\t\t".'<td class="pad"> </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)) . ' »</a></td>';
} else {
$calendar_output .= "\n\t\t".'<td colspan="3" id="next" class="pad"> </td>';
}
$calendar_output .= '
</tr>
</tfoot>
<tbody>
<tr>';
$daywithpost = array();
$dayswithposts = $wpdb->get_results("SELECT DISTINCT DAYOFMONTH(post_date)
FROM $wpdb->posts WHERE post_date >= '{$thisyear}-{$thismonth}-01 00:00:00'
AND post_type = 'post' 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];
}
}
if (strpos($_SERVER['HTTP_USER_AGENT'], 'MSIE') !== false || stripos($_SERVER['HTTP_USER_AGENT'], 'camino') !== false || stripos($_SERVER['HTTP_USER_AGENT'], 'safari') !== false)
$ak_title_separator = "\n";
else
$ak_title_separator = ', ';
$ak_titles_for_day = array();
$ak_post_titles = $wpdb->get_results("SELECT ID, post_title, DAYOFMONTH(post_date) as dom "
."FROM $wpdb->posts "
."WHERE post_date >= '{$thisyear}-{$thismonth}-01 00:00:00' "
."AND post_date <= '{$thisyear}-{$thismonth}-{$last_day} 23:59:59' "
."AND post_type = 'post' AND post_status = 'publish'"
);
if ( $ak_post_titles ) {
foreach ( (array) $ak_post_titles as $ak_post_title ) {
$post_title = esc_attr( apply_filters( 'the_title', $ak_post_title->post_title, $ak_post_title->ID ) );
if ( empty($ak_titles_for_day['day_'.$ak_post_title->dom]) )
$ak_titles_for_day['day_'.$ak_post_title->dom] = '';
if ( empty($ak_titles_for_day["$ak_post_title->dom"]) )
$ak_titles_for_day["$ak_post_title->dom"] = $post_title;
else
$ak_titles_for_day["$ak_post_title->dom"] .= $ak_title_separator . $post_title;
}
}
$pad = calendar_week_mod(date('w', $unixmonth)-$week_begins);
if ( 0 != $pad )
$calendar_output .= "\n\t\t".'<td colspan="'. esc_attr($pad) .'" class="pad"> </td>';
$daysinmonth = intval(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;
//日付から曜日を割り出してクラスにする
$day_class = "";
$in_year = date('Y', $unixmonth);
$in_month = zeroise(intval($wp_locale->get_month($thismonth)), 2);
$in_day = zeroise(intval($day), 2);
if(date('w', strtotime($in_year.'-'.$in_month.'-'.$in_day)) == 0){
$day_class = " class=\"sun\"";
}else if(date('w', strtotime($in_year.'-'.$in_month.'-'.$in_day)) == 6){
$day_class = " class=\"sat\"";
}else{
$day_class = "";
}
if ( $day == gmdate('j', current_time('timestamp')) && $thismonth == gmdate('m', current_time('timestamp')) && $thisyear == gmdate('Y', current_time('timestamp')) )
//クラスの変数を追加
//$calendar_output .= '<td id="today">';
$calendar_output .= '<td id="today"'.$day_class.'>';
else
//クラスの変数を追加
//$calendar_output .= '<td>';
$calendar_output .= '<td'.$day_class.'>';
if ( in_array($day, $daywithpost) )
$calendar_output .= '<a href="' . get_day_link( $thisyear, $thismonth, $day ) . '" title="' . esc_attr( $ak_titles_for_day[ $day ] ) . "\">$day</a>";
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) .'"> </td>';
$calendar_output .= "\n\t</tr>\n\t</tbody>\n\t</table>";
$cache[ $key ] = $calendar_output;
wp_cache_set( 'get_calendar_add_weekend', $cache, 'calendar' );
if ( $echo ) {
echo apply_filters( 'get_calendar_add_weekend', $calendar_output );
} else {
return apply_filters( 'get_calendar_add_weekend', $calendar_output );
}
}
add_filter('get_calendar', 'get_calendar_add_weekend', 10, 2);
カレンダー一つに元のコードなげえよ・・・。
WordPressの細々としたグローバル変数が何を表してるのかなんて熟知してるわけでもないから、$wp_locale->get_month()の辺りが「？？？」って状態だった。
表記を見た感じ、オブジェクトかなんかなんだろうか。
・・・馬鹿丸出しですみませんｗｗｗｗ
とりあえずこれで、WP標準のカレンダーウィジェットの設置の際や、テンプレート側でget_calendar関数を使用した際に問答無用で土日に対応したこっちのコードが実行できる。
日本語しか対応してないが、普通に使う分にはこれで問題はない・・・だろう。
恐らく多分思うに。