<?php
/**
 * Public News Sitemap Functions
 *
 * @package XML Sitemap & Google News
 */

/**
 * Response headers filter
 * Does not check if we are really in a sitemap feed.
 *
 * @param array $headers The headers array.
 *
 * @return array
 */
function xmlsf_news_nocache_headers( $headers ) {
	// Prevent proxy caches serving a cached news sitemap.
	$headers['Cache-Control'] .= ', no-store';

	return $headers;
}

add_filter( 'nocache_headers', 'xmlsf_news_nocache_headers' );

/**
 * Filter news WHERE
 * only posts from the last 48 hours
 *
 * @param string $where DB Query where clause.
 *
 * @return string
 */
function xmlsf_news_filter_where( $where = '' ) {
	return $where . ' AND post_date_gmt > \'' . gmdate( 'Y-m-d H:i:s', strtotime( '-48 hours' ) ) . '\'';
}

/**
 * Parse language string into two or three letter ISO 639 code.
 *
 * @param string $lang Unformatted language string.
 *
 * @return string
 */
function xmlsf_parse_language_string( $lang ) {
	// Lower case, no tags.
	$lang = convert_chars( strtolower( wp_strip_all_tags( $lang ) ) );

	// Convert underscores.
	$lang = str_replace( '_', '-', $lang );

	// No hyphens except...
	if ( strpos( $lang, '-' ) ) :
		if ( 0 === strpos( $lang, 'zh' ) ) {
			$lang = strpos( $lang, 'hk' ) || strpos( $lang, 'tw' ) || strpos( $lang, 'hant' ) ? 'zh-tw' : 'zh-cn';
		} else {
			// Explode on hyphen and use only first part.
			$expl = explode( '-', $lang );
			$lang = $expl[0];
		}
	endif;

	// Make sure it's max 3 letters.
	$lang = substr( $lang, 0, 2 );

	return $lang;
}

add_filter( 'xmlsf_news_language', 'xmlsf_parse_language_string', 99 );

/**
 * COMPATIBILITY
 */

/**
 * Post language filter for Polylang.
 *
 * @param string $locale Locale.
 * @param int    $post_id Post ID.
 *
 * @return string
 */
function xmlsf_polylang_post_language_filter( $locale, $post_id ) {
	return function_exists( 'pll_get_post_language' ) ? pll_get_post_language( $post_id, 'locale' ) : $locale;
}

add_filter( 'xmlsf_news_language', 'xmlsf_polylang_post_language_filter', 10, 2 );

/**
 * Post language filter for WPML.
 *
 * @param string $locale    Locale.
 * @param int    $post_id   Post ID.
 * @param string $post_type Post type.
 *
 * @return string
 */
function xmlsf_wpml_post_language_filter( $locale, $post_id, $post_type = 'post' ) {
	global $sitepress;
	return $sitepress ? apply_filters(
		'wpml_element_language_code',
		$locale,
		array(
			'element_id'   => $post_id,
			'element_type' => $post_type,
		)
	) : $locale;
}

add_filter( 'xmlsf_news_language', 'xmlsf_wpml_post_language_filter', 10, 3 );

/**
 * Google News Publisher filter for backward compat with XMLSF_GOOGLE_NEWS_NAME constant.
 *
 * @param string $name Google News Publisher name.
 *
 * @return string
 */
function xmlsf_google_news_name( $name ) {
	defined( 'XMLSF_GOOGLE_NEWS_NAME' ) || define( 'XMLSF_GOOGLE_NEWS_NAME', false );

	return XMLSF_GOOGLE_NEWS_NAME ? XMLSF_GOOGLE_NEWS_NAME : $name;
}

add_filter( 'xmlsf_news_publication_name', 'xmlsf_google_news_name' );