Add data-wp-key to pagination numbers

WordPress · Jan 24, 2024 · ab761aa · ab761aa
1 parent 4c96f5e
commit ab761aa
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/packages/block-library/src/query-pagination-numbers/index.php b/packages/block-library/src/query-pagination-numbers/index.php
@@ -91,14 +91,19 @@ function render_block_core_query_pagination_numbers( $attributes, $content, $blo
 	}
 
 	if ( $enhanced_pagination ) {
-		$p = new WP_HTML_Tag_Processor( $content );
+		$p      = new WP_HTML_Tag_Processor( $content );
+		$dots_i = 0;
 		while ( $p->next_tag(
-			array(
-				'tag_name'   => 'a',
-				'class_name' => 'page-numbers',
-			)
+			array( 'class_name' => 'page-numbers' )
 		) ) {
-			$p->set_attribute( 'data-wp-on--click', 'core/query::actions.navigate' );
+			if ( 'A' === $p->get_tag() ) {
+				$p->set_attribute( 'data-wp-on--click', 'core/query::actions.navigate' );
+				$p->set_attribute( 'data-wp-key', $p->get_attribute( 'href' ) );
+			} elseif ( $p->has_class( 'current' ) ) {
+				$p->set_attribute( 'data-wp-key', 'current' );
+			} elseif ( $p->has_class( 'dots' ) ) {
+				$p->set_attribute( 'data-wp-key', 'dots-' . $dots_i++ );
+			}
 		}
 		$content = $p->get_updated_html();
 	}