<?php

/**

 * Server-side rendering of the `core/query-pagination-numbers` block.

 *

 * @package WordPress

 */

/**

 * Renders the `core/query-pagination-numbers` block on the server.

 *

 * @since 5.8.0

 *

 * @global WP_Query $wp_query WordPress Query object.

 *

 * @param array    $attributes Block attributes.

 * @param string   $content    Block default content.

 * @param WP_Block $block      Block instance.

 *

 * @return string Returns the pagination numbers for the Query.

 */

function render_block_core_query_pagination_numbers( $attributes, $content, $block ) {

	$page_key            = isset( $block->context['queryId'] ) ? 'query-' . $block->context['queryId'] . '-page' : 'query-page';

	$enhanced_pagination = isset( $block->context['enhancedPagination'] ) && $block->context['enhancedPagination'];

	$page                = empty( $_GET[ $page_key ] ) ? 1 : (int) $_GET[ $page_key ];

	$max_page            = isset( $block->context['query']['pages'] ) ? (int) $block->context['query']['pages'] : 0;

	$wrapper_attributes = get_block_wrapper_attributes();

	$content            = '';

	global $wp_query;

	$mid_size = isset( $block->attributes['midSize'] ) ? (int) $block->attributes['midSize'] : null;

	if ( isset( $block->context['query']['inherit'] ) && $block->context['query']['inherit'] ) {

		// Take into account if we have set a bigger `max page`

		// than what the query has.

		$total         = ! $max_page || $max_page > $wp_query->max_num_pages ? $wp_query->max_num_pages : $max_page;

		$paginate_args = array(

			'prev_next' => false,

			'total'     => $total,

		);

		if ( null !== $mid_size ) {

			$paginate_args['mid_size'] = $mid_size;

		}

		$content = paginate_links( $paginate_args );

	} else {

		$block_query = new WP_Query( build_query_vars_from_query_block( $block, $page ) );

		// `paginate_links` works with the global $wp_query, so we have to

		// temporarily switch it with our custom query.

		$prev_wp_query = $wp_query;

		$wp_query      = $block_query;

		$total         = ! $max_page || $max_page > $wp_query->max_num_pages ? $wp_query->max_num_pages : $max_page;

		$paginate_args = array(

			'base'      => '%_%',

			'format'    => "?$page_key=%#%",

			'current'   => max( 1, $page ),

			'total'     => $total,

			'prev_next' => false,

		);

		if ( null !== $mid_size ) {

			$paginate_args['mid_size'] = $mid_size;

		}

		if ( 1 !== $page ) {

			/**

			 * `paginate_links` doesn't use the provided `format` when the page is `1`.

			 * This is great for the main query as it removes the extra query params

			 * making the URL shorter, but in the case of multiple custom queries is

			 * problematic. It results in returning an empty link which ends up with

			 * a link to the current page.

			 *

			 * A way to address this is to add a `fake` query arg with no value that

			 * is the same for all custom queries. This way the link is not empty and

			 * preserves all the other existent query args.

			 *

			 * @see https://developer.wordpress.org/reference/functions/paginate_links/

			 *

			 * The proper fix of this should be in core. Track Ticket:

			 * @see https://core.trac.wordpress.org/ticket/53868

			 *

			 * TODO: After two WP versions (starting from the WP version the core patch landed),

			 * we should remove this and call `paginate_links` with the proper new arg.

			 */

			$paginate_args['add_args'] = array( 'cst' => '' );

		}

		// We still need to preserve `paged` query param if exists, as is used

		// for Queries that inherit from global context.

		$paged = empty( $_GET['paged'] ) ? null : (int) $_GET['paged'];

		if ( $paged ) {

			$paginate_args['add_args'] = array( 'paged' => $paged );

		}

		$content = paginate_links( $paginate_args );

		wp_reset_postdata(); // Restore original Post Data.

		$wp_query = $prev_wp_query;

	}

	if ( empty( $content ) ) {

		return '';

	}

	if ( $enhanced_pagination ) {

		$p         = new WP_HTML_Tag_Processor( $content );

		$tag_index = 0;

		while ( $p->next_tag(

			array( 'class_name' => 'page-numbers' )

		) ) {

			if ( null === $p->get_attribute( 'data-wp-key' ) ) {

				$p->set_attribute( 'data-wp-key', 'index-' . $tag_index++ );

			}

			if ( 'A' === $p->get_tag() ) {

				$p->set_attribute( 'data-wp-on--click', 'core/query::actions.navigate' );

			}

		}

		$content = $p->get_updated_html();

	}

	return sprintf(

		'<div %1$s>%2$s</div>',

		$wrapper_attributes,

		$content

	);

}

/**

 * Registers the `core/query-pagination-numbers` block on the server.

 *

 * @since 5.8.0

 */

function register_block_core_query_pagination_numbers() {

	register_block_type_from_metadata(

		__DIR__ . '/query-pagination-numbers',

		array(

			'render_callback' => 'render_block_core_query_pagination_numbers',

		)

	);

}

add_action( 'init', 'register_block_core_query_pagination_numbers' );