<?php

add_action( 'parse_query', 'bogo_parse_query', 10, 1 );

function bogo_parse_query( $query ) {
	$qv = &$query->query_vars;

	if ( ! empty( $qv['bogo_suppress_locale_query'] ) ) {
		return;
	}

	if ( $query->is_preview() and ( $qv['page_id'] or $qv['p'] ) ) {
		$qv['bogo_suppress_locale_query'] = true;
		return;
	}

	if ( isset( $qv['post_type'] ) and 'any' !== $qv['post_type'] ) {
		$localizable = array_filter(
			(array) $qv['post_type'],
			'bogo_is_localizable_post_type'
		);

		if ( empty( $localizable ) ) {
			$qv['bogo_suppress_locale_query'] = true;
			return;
		}
	}

	$lang = $qv['lang'] ?? '';

	if ( is_admin() ) {
		$locale = $lang;
	} else {
		if ( $lang ) {
			$locale = bogo_get_closest_locale( $lang );
		} else {
			$locale = get_locale();
		}

		if ( empty( $locale ) ) {
			$locale = bogo_get_default_locale();
		}
	}

	if ( empty( $locale ) or ! bogo_is_available_locale( $locale ) ) {
		$qv['bogo_suppress_locale_query'] = true;
		return;
	}

	$qv['lang'] = $locale;

	if ( is_admin() ) {
		return;
	}

	if (
		$query->is_home and
		'page' === get_option( 'show_on_front' ) and
		get_option( 'page_on_front' )
	) {
		$query_keys = array_keys( wp_parse_args( $query->query ) );
		$query_keys = array_diff(
			$query_keys,
			array( 'preview', 'page', 'paged', 'cpage', 'lang' )
		);

		if ( empty( $query_keys ) ) {
			$query->is_page = true;
			$query->is_home = false;
			$qv['page_id'] = get_option( 'page_on_front' );

			if ( ! empty( $qv['paged'] ) ) {
				$qv['page'] = $qv['paged'];
				unset( $qv['paged'] );
			}
		}
	}

	if ( '' != $qv['pagename'] ) {
		$query->queried_object = bogo_get_page_by_path( $qv['pagename'], $locale );

		if ( ! empty( $query->queried_object ) ) {
			$query->queried_object_id = (int) $query->queried_object->ID;
		} else {
			unset( $query->queried_object );
			unset( $query->queried_object_id );
		}

		if (
			'page' === get_option( 'show_on_front' ) and
			isset( $query->queried_object_id ) and
			$query->queried_object_id == get_option( 'page_for_posts' )
		) {
			$query->is_page = false;
			$query->is_home = true;
			$query->is_posts_page = true;
		}
	}

	if (
		isset( $qv['post_type'] ) and
		'any' !== $qv['post_type'] and
		! is_array( $qv['post_type'] ) and
		! empty( $qv['name'] )
	) {
		$query->queried_object = bogo_get_page_by_path(
			$qv['name'], $locale, $qv['post_type']
		);

		if ( ! empty( $query->queried_object ) ) {
			$query->queried_object_id = (int) $query->queried_object->ID;
		} else {
			unset( $query->queried_object );
			unset( $query->queried_object_id );
		}
	}

	if (
		$query->is_posts_page and
		( ! isset( $qv['withcomments'] ) or ! $qv['withcomments'] )
	) {
		$query->is_comment_feed = false;
	}

	$query->is_singular =
		( $query->is_single || $query->is_page || $query->is_attachment );

	$query->is_embed =
		$query->is_embed && ( $query->is_singular || $query->is_404 );
}


add_filter( 'posts_join', 'bogo_posts_join', 10, 2 );

function bogo_posts_join( $join, $query ) {
	global $wpdb;

	$qv = &$query->query_vars;

	if ( ! empty( $qv['bogo_suppress_locale_query'] ) ) {
		return $join;
	}

	$locale = $qv['lang'] ?? '';

	if ( ! bogo_is_available_locale( $locale ) ) {
		return $join;
	}

	if ( ! $meta_table = _get_meta_table( 'post' ) ) {
		return $join;
	}

	$join .= " LEFT JOIN $meta_table AS postmeta_bogo ON ($wpdb->posts.ID = postmeta_bogo.post_id AND postmeta_bogo.meta_key = '_locale')";

	return $join;
}


add_filter( 'posts_where', 'bogo_posts_where', 10, 2 );

function bogo_posts_where( $where, $query ) {
	global $wpdb;

	$qv = &$query->query_vars;

	if ( ! empty( $qv['bogo_suppress_locale_query'] ) ) {
		return $where;
	}

	$locale = $qv['lang'] ?? '';

	if ( ! bogo_is_available_locale( $locale ) ) {
		return $where;
	}

	if ( ! $meta_table = _get_meta_table( 'post' ) ) {
		return $where;
	}

	$where .= " AND (1=0";

	$where .= $wpdb->prepare( " OR postmeta_bogo.meta_value LIKE %s", $locale );

	if ( bogo_is_default_locale( $locale ) ) {
		$where .= " OR postmeta_bogo.meta_id IS NULL";
	}

	$where .= ")";

	return $where;
}


add_filter( 'option_sticky_posts', 'bogo_option_sticky_posts', 10, 1 );

function bogo_option_sticky_posts( $posts ) {
	if ( is_home() ) {
		$locale = get_locale();

		$posts = array_filter( $posts,
			static function ( $post_id ) use ( $locale ) {
				return bogo_get_post_locale( $post_id ) === $locale;
			}
		);
	}

	return $posts;
}


add_filter( 'option_page_on_front', 'bogo_get_local_post', 10, 1 );
add_filter( 'option_page_for_posts', 'bogo_get_local_post', 10, 1 );

function bogo_get_local_post( $post_id ) {
	global $wpdb;

	if ( is_admin() or empty( $post_id ) ) {
		return $post_id;
	}

	$post_type = get_post_type( $post_id );

	if (
		! post_type_exists( $post_type ) or
		! bogo_is_localizable_post_type( $post_type )
	) {
		return $post_id;
	}

	$locale = get_locale();

	if ( bogo_get_post_locale( $post_id ) === $locale ) {
		return $post_id;
	}

	$locale_query = $wpdb->prepare(
		"pm2.meta_value LIKE %s",
		$locale
	);

	if ( bogo_is_default_locale( $locale ) ) {
		$locale_query = $wpdb->prepare(
			"(pm2.meta_value LIKE %s OR pm2.meta_id IS NULL)",
			$locale
		);
	}

	$original_post = get_post_meta( $post_id, '_original_post', true );

	// For back-compat
	if ( empty( $original_post ) ) {
		$original_post = $post_id;
	}

	$original_post_query = $wpdb->prepare(
		"pm1.meta_value = %s",
		$original_post
	);

	// For back-compat
	if ( is_int( $original_post ) ) {
		$original_post_query = $wpdb->prepare(
			"(ID = %d OR pm1.meta_value = %d)",
			$original_post,
			$original_post
		);
	}

	$translation = $wpdb->get_var( $wpdb->prepare(
		"SELECT ID FROM %i AS posts LEFT JOIN %i AS pm1 ON posts.ID = pm1.post_id AND pm1.meta_key = '_original_post' LEFT JOIN %i AS pm2 ON posts.ID = pm2.post_id AND pm2.meta_key = '_locale' WHERE post_status = 'publish' AND post_type = %s AND $locale_query AND $original_post_query",
		$wpdb->posts,
		$wpdb->postmeta,
		$wpdb->postmeta,
		$post_type,
	) );

	$translation = absint( $translation );

	if ( $translation ) {
		return $translation;
	}

	return $post_id;
}